| | |
| | | } |
| | | } |
| | | |
| | | func TestRegistryReturnsMoonshineEnglish(t *testing.T) { |
| | | profile, err := GetModelProfile(MoonshineModelID) |
| | | if err != nil { |
| | | t.Fatal(err) |
| | | } |
| | | if profile.BackendKind != BackendMoonshine { |
| | | t.Fatalf("backend kind = %q, want %q", profile.BackendKind, BackendMoonshine) |
| | | } |
| | | if profile.InstallDirName != "moonshine-en" { |
| | | t.Fatalf("install dir = %q, want moonshine-en", profile.InstallDirName) |
| | | } |
| | | if len(profile.DownloadURLs) == 0 { |
| | | t.Fatal("expected moonshine download URL") |
| | | } |
| | | } |
| | | |
| | | func TestEnglishLanguageDefaultsToMoonshine(t *testing.T) { |
| | | profile, err := GetLanguageProfile(EnglishLanguageID) |
| | | if err != nil { |
| | | t.Fatal(err) |
| | | } |
| | | if profile.DefaultModelID != MoonshineModelID { |
| | | t.Fatalf("English default model = %q, want %q", profile.DefaultModelID, MoonshineModelID) |
| | | } |
| | | } |
| | | |
| | | func TestValidateSenseVoiceAnyOfAndAllOf(t *testing.T) { |
| | | dir := t.TempDir() |
| | | writeTestFile(t, filepath.Join(dir, "tokens.txt"), "tokens") |
| | |
| | | result := ValidateModelDir(DefaultModelProfile(), dir) |
| | | if result.Valid { |
| | | t.Fatal("expected zero-byte model file to fail validation") |
| | | } |
| | | } |
| | | |
| | | func TestValidateMoonshineRequiredFiles(t *testing.T) { |
| | | profile, err := GetModelProfile(MoonshineModelID) |
| | | if err != nil { |
| | | t.Fatal(err) |
| | | } |
| | | dir := t.TempDir() |
| | | createValidMoonshine(t, dir) |
| | | |
| | | result := ValidateModelDir(profile, dir) |
| | | if !result.Valid { |
| | | t.Fatalf("expected valid moonshine dir, missing=%v problems=%v", result.Missing, result.Problems) |
| | | } |
| | | for _, role := range []string{"preprocessor", "encoder", "uncached_decoder", "cached_decoder", "tokens"} { |
| | | if result.Files[role] == "" { |
| | | t.Fatalf("missing resolved file role %q", role) |
| | | } |
| | | } |
| | | } |
| | | |
| | | func TestValidateMoonshineRejectsMissingFile(t *testing.T) { |
| | | profile, err := GetModelProfile(MoonshineModelID) |
| | | if err != nil { |
| | | t.Fatal(err) |
| | | } |
| | | dir := t.TempDir() |
| | | createValidMoonshine(t, dir) |
| | | if err := os.Remove(filepath.Join(dir, "cached_decode.int8.onnx")); err != nil { |
| | | t.Fatal(err) |
| | | } |
| | | |
| | | result := ValidateModelDir(profile, dir) |
| | | if result.Valid { |
| | | t.Fatal("expected missing cached decoder to fail validation") |
| | | } |
| | | } |
| | | |
| | |
| | | writeTestFile(t, filepath.Join(dir, "model.int8.onnx"), "model") |
| | | } |
| | | |
| | | func createValidMoonshine(t *testing.T, dir string) { |
| | | t.Helper() |
| | | if err := os.MkdirAll(dir, 0755); err != nil { |
| | | t.Fatal(err) |
| | | } |
| | | writeTestFile(t, filepath.Join(dir, "preprocess.onnx"), "preprocessor") |
| | | writeTestFile(t, filepath.Join(dir, "encode.int8.onnx"), "encoder") |
| | | writeTestFile(t, filepath.Join(dir, "uncached_decode.int8.onnx"), "uncached") |
| | | writeTestFile(t, filepath.Join(dir, "cached_decode.int8.onnx"), "cached") |
| | | writeTestFile(t, filepath.Join(dir, "tokens.txt"), "tokens") |
| | | } |
| | | |
| | | func tarArchive(t *testing.T, files map[string]string) []byte { |
| | | t.Helper() |
| | | var buf bytes.Buffer |