| | |
| | | } |
| | | } |
| | | |
| | | func TestRegistryReturnsXASRZhEn960(t *testing.T) { |
| | | profile, err := GetModelProfile(XASRZhEn960ModelID) |
| | | if err != nil { |
| | | t.Fatal(err) |
| | | } |
| | | if profile.BackendKind != BackendXASRStreaming { |
| | | t.Fatalf("backend kind = %q, want %q", profile.BackendKind, BackendXASRStreaming) |
| | | } |
| | | if profile.InstallDirName != "x-asr-zh-en-960ms" { |
| | | t.Fatalf("install dir = %q, want x-asr-zh-en-960ms", profile.InstallDirName) |
| | | } |
| | | if profile.Tier != "experimental" { |
| | | t.Fatalf("tier = %q, want experimental", profile.Tier) |
| | | } |
| | | if len(profile.DownloadURLs) != 0 { |
| | | t.Fatal("expected x-asr experiment to use manual/scripted install, not in-app tar download") |
| | | } |
| | | } |
| | | |
| | | func TestChineseLanguageOffersQwen3ASRUpgrade(t *testing.T) { |
| | | profile, err := GetLanguageProfile(DefaultLanguageID) |
| | | if err != nil { |
| | |
| | | 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) |
| | | if !stringSliceContains(profile.UpgradeModelIDs, Qwen3ASRModelID) || !stringSliceContains(profile.UpgradeModelIDs, XASRZhEn960ModelID) { |
| | | t.Fatalf("Chinese upgrade models = %v, want qwen3 and x-asr", profile.UpgradeModelIDs) |
| | | } |
| | | } |
| | | |
| | |
| | | if profile.DefaultModelID != MoonshineModelID { |
| | | t.Fatalf("English default model = %q, want %q", profile.DefaultModelID, MoonshineModelID) |
| | | } |
| | | if len(profile.UpgradeModelIDs) != 1 || profile.UpgradeModelIDs[0] != ParakeetModelID { |
| | | t.Fatalf("English upgrade models = %v, want [%s]", profile.UpgradeModelIDs, ParakeetModelID) |
| | | if !stringSliceContains(profile.UpgradeModelIDs, ParakeetModelID) || !stringSliceContains(profile.UpgradeModelIDs, XASRZhEn960ModelID) { |
| | | t.Fatalf("English upgrade models = %v, want parakeet and x-asr", profile.UpgradeModelIDs) |
| | | } |
| | | } |
| | | |
| | |
| | | result := ValidateModelDir(profile, dir) |
| | | if result.Valid { |
| | | t.Fatal("expected tokenizer file to fail directory validation") |
| | | } |
| | | } |
| | | |
| | | func TestValidateXASRZhEn960RequiredFiles(t *testing.T) { |
| | | profile, err := GetModelProfile(XASRZhEn960ModelID) |
| | | if err != nil { |
| | | t.Fatal(err) |
| | | } |
| | | dir := t.TempDir() |
| | | createValidXASRZhEn960(t, dir) |
| | | |
| | | result := ValidateModelDir(profile, dir) |
| | | if !result.Valid { |
| | | t.Fatalf("expected valid x-asr dir, missing=%v problems=%v", result.Missing, result.Problems) |
| | | } |
| | | for _, role := range []string{"encoder", "decoder", "joiner", "tokens"} { |
| | | if result.Files[role] == "" { |
| | | t.Fatalf("missing resolved file role %q", role) |
| | | } |
| | | } |
| | | } |
| | | |
| | | func TestValidateXASRZhEn960RejectsMissingJoiner(t *testing.T) { |
| | | profile, err := GetModelProfile(XASRZhEn960ModelID) |
| | | if err != nil { |
| | | t.Fatal(err) |
| | | } |
| | | dir := t.TempDir() |
| | | createValidXASRZhEn960(t, dir) |
| | | if err := os.Remove(filepath.Join(dir, "joiner-960ms.onnx")); err != nil { |
| | | t.Fatal(err) |
| | | } |
| | | |
| | | result := ValidateModelDir(profile, dir) |
| | | if result.Valid { |
| | | t.Fatal("expected missing joiner to fail validation") |
| | | } |
| | | } |
| | | |
| | |
| | | writeTestFile(t, filepath.Join(dir, "tokenizer", "vocab.json"), "vocab") |
| | | } |
| | | |
| | | func createValidXASRZhEn960(t *testing.T, dir string) { |
| | | t.Helper() |
| | | if err := os.MkdirAll(dir, 0755); err != nil { |
| | | t.Fatal(err) |
| | | } |
| | | writeTestFile(t, filepath.Join(dir, "encoder-960ms.onnx"), "encoder") |
| | | writeTestFile(t, filepath.Join(dir, "decoder-960ms.onnx"), "decoder") |
| | | writeTestFile(t, filepath.Join(dir, "joiner-960ms.onnx"), "joiner") |
| | | writeTestFile(t, filepath.Join(dir, "tokens.txt"), "tokens") |
| | | } |
| | | |
| | | func stringSliceContains(values []string, target string) bool { |
| | | for _, value := range values { |
| | | if value == target { |
| | | return true |
| | | } |
| | | } |
| | | return false |
| | | } |
| | | |
| | | func tarArchive(t *testing.T, files map[string]string) []byte { |
| | | t.Helper() |
| | | var buf bytes.Buffer |