| | |
| | | } |
| | | } |
| | | |
| | | func TestRegistryReturnsQwen3ASRChinese(t *testing.T) { |
| | | profile, err := GetModelProfile(Qwen3ASRModelID) |
| | | if err != nil { |
| | | t.Fatal(err) |
| | | } |
| | | if profile.BackendKind != BackendQwen3ASR { |
| | | t.Fatalf("backend kind = %q, want %q", profile.BackendKind, BackendQwen3ASR) |
| | | } |
| | | if profile.InstallDirName != "qwen3-asr-0.6b" { |
| | | t.Fatalf("install dir = %q, want qwen3-asr-0.6b", profile.InstallDirName) |
| | | } |
| | | if profile.Tier != "advanced" { |
| | | t.Fatalf("tier = %q, want advanced", profile.Tier) |
| | | } |
| | | if len(profile.DownloadURLs) == 0 { |
| | | t.Fatal("expected qwen3-asr download URL") |
| | | } |
| | | } |
| | | |
| | | func TestChineseLanguageOffersQwen3ASRUpgrade(t *testing.T) { |
| | | profile, err := GetLanguageProfile(DefaultLanguageID) |
| | | if err != nil { |
| | | t.Fatal(err) |
| | | } |
| | | if profile.DefaultModelID != DefaultModelID { |
| | | t.Fatalf("Chinese default model = %q, want %q", profile.DefaultModelID, DefaultModelID) |
| | | } |
| | | if len(profile.UpgradeModelIDs) != 1 || profile.UpgradeModelIDs[0] != Qwen3ASRModelID { |
| | | t.Fatalf("Chinese upgrade models = %v, want [%s]", profile.UpgradeModelIDs, Qwen3ASRModelID) |
| | | } |
| | | } |
| | | |
| | | func TestEnglishLanguageDefaultsToMoonshine(t *testing.T) { |
| | | profile, err := GetLanguageProfile(EnglishLanguageID) |
| | | if err != nil { |
| | |
| | | result := ValidateModelDir(profile, dir) |
| | | if result.Valid { |
| | | t.Fatal("expected missing joiner to fail validation") |
| | | } |
| | | } |
| | | |
| | | func TestValidateQwen3ASRRequiredFiles(t *testing.T) { |
| | | profile, err := GetModelProfile(Qwen3ASRModelID) |
| | | if err != nil { |
| | | t.Fatal(err) |
| | | } |
| | | dir := t.TempDir() |
| | | createValidQwen3ASR(t, dir) |
| | | |
| | | result := ValidateModelDir(profile, dir) |
| | | if !result.Valid { |
| | | t.Fatalf("expected valid qwen3-asr dir, missing=%v problems=%v", result.Missing, result.Problems) |
| | | } |
| | | for _, role := range []string{"conv_frontend", "encoder", "decoder", "tokenizer", "tokenizer_merges", "tokenizer_vocab"} { |
| | | if result.Files[role] == "" { |
| | | t.Fatalf("missing resolved file role %q", role) |
| | | } |
| | | } |
| | | if info, err := os.Stat(result.Files["tokenizer"]); err != nil || !info.IsDir() { |
| | | t.Fatalf("tokenizer path = %q, want directory, statErr=%v", result.Files["tokenizer"], err) |
| | | } |
| | | } |
| | | |
| | | func TestValidateQwen3ASRRejectsMissingTokenizerFile(t *testing.T) { |
| | | profile, err := GetModelProfile(Qwen3ASRModelID) |
| | | if err != nil { |
| | | t.Fatal(err) |
| | | } |
| | | dir := t.TempDir() |
| | | createValidQwen3ASR(t, dir) |
| | | if err := os.Remove(filepath.Join(dir, "tokenizer", "vocab.json")); err != nil { |
| | | t.Fatal(err) |
| | | } |
| | | |
| | | result := ValidateModelDir(profile, dir) |
| | | if result.Valid { |
| | | t.Fatal("expected missing tokenizer vocab to fail validation") |
| | | } |
| | | } |
| | | |
| | | func TestValidateQwen3ASRRejectsTokenizerFileInsteadOfDirectory(t *testing.T) { |
| | | profile, err := GetModelProfile(Qwen3ASRModelID) |
| | | if err != nil { |
| | | t.Fatal(err) |
| | | } |
| | | dir := t.TempDir() |
| | | createValidQwen3ASR(t, dir) |
| | | if err := os.RemoveAll(filepath.Join(dir, "tokenizer")); err != nil { |
| | | t.Fatal(err) |
| | | } |
| | | writeTestFile(t, filepath.Join(dir, "tokenizer"), "not a directory") |
| | | |
| | | result := ValidateModelDir(profile, dir) |
| | | if result.Valid { |
| | | t.Fatal("expected tokenizer file to fail directory validation") |
| | | } |
| | | } |
| | | |
| | |
| | | } |
| | | } |
| | | |
| | | func TestDownloadQwen3ASRFailureDoesNotRemoveSenseVoice(t *testing.T) { |
| | | root := t.TempDir() |
| | | senseVoiceDir := filepath.Join(root, "sensevoice-zh") |
| | | createValidSenseVoice(t, senseVoiceDir) |
| | | |
| | | profile, err := GetModelProfile(Qwen3ASRModelID) |
| | | if err != nil { |
| | | t.Fatal(err) |
| | | } |
| | | err = DownloadProfile(profile, []string{"://bad-url"}, root, nil) |
| | | if err == nil { |
| | | t.Fatal("expected download failure") |
| | | } |
| | | |
| | | result := ValidateModelDir(DefaultModelProfile(), senseVoiceDir) |
| | | if !result.Valid { |
| | | t.Fatalf("sensevoice model was damaged, missing=%v problems=%v", result.Missing, result.Problems) |
| | | } |
| | | } |
| | | |
| | | func TestDownloadProfileRemovesStalePartialDownload(t *testing.T) { |
| | | root := t.TempDir() |
| | | stalePath := filepath.Join(root, ".downloads", DefaultModelID, "old-run", "model_package") |
| | |
| | | writeTestFile(t, filepath.Join(dir, "tokens.txt"), "tokens") |
| | | } |
| | | |
| | | func createValidQwen3ASR(t *testing.T, dir string) { |
| | | t.Helper() |
| | | if err := os.MkdirAll(filepath.Join(dir, "tokenizer"), 0755); err != nil { |
| | | t.Fatal(err) |
| | | } |
| | | writeTestFile(t, filepath.Join(dir, "conv_frontend.onnx"), "conv") |
| | | writeTestFile(t, filepath.Join(dir, "encoder.int8.onnx"), "encoder") |
| | | writeTestFile(t, filepath.Join(dir, "decoder.int8.onnx"), "decoder") |
| | | writeTestFile(t, filepath.Join(dir, "tokenizer", "merges.txt"), "merges") |
| | | writeTestFile(t, filepath.Join(dir, "tokenizer", "vocab.json"), "vocab") |
| | | } |
| | | |
| | | func tarArchive(t *testing.T, files map[string]string) []byte { |
| | | t.Helper() |
| | | var buf bytes.Buffer |