From 57868a8279e371e71c1600f6bc12c40e18d8bb04 Mon Sep 17 00:00:00 2001
From: Ariver <shanghai3168@gmail.com>
Date: Tue, 23 Jun 2026 14:37:01 +0800
Subject: [PATCH] Polish input settings layout
---
privatevoice.src/internal/model/model_test.go | 197 ++++++++++++++++++++++++++++++++++++++++++++++++-
1 files changed, 193 insertions(+), 4 deletions(-)
diff --git a/privatevoice.src/internal/model/model_test.go b/privatevoice.src/internal/model/model_test.go
index c334d19..b163161 100644
--- a/privatevoice.src/internal/model/model_test.go
+++ b/privatevoice.src/internal/model/model_test.go
@@ -61,6 +61,54 @@
}
}
+func TestRegistryReturnsParakeetV3ForEuropeanLanguages(t *testing.T) {
+ profile, err := GetModelProfile(ParakeetV3ModelID)
+ if err != nil {
+ t.Fatal(err)
+ }
+ if profile.BackendKind != BackendNemoTransducer {
+ t.Fatalf("backend kind = %q, want %q", profile.BackendKind, BackendNemoTransducer)
+ }
+ for _, languageID := range []string{FrenchLanguageID, GermanLanguageID, SpanishLanguageID, ItalianLanguageID, PortugueseLanguageID} {
+ if !stringSliceContains(profile.SupportedLanguageIDs, languageID) {
+ t.Fatalf("parakeet v3 languages = %v, want %s", profile.SupportedLanguageIDs, languageID)
+ }
+ languageProfile, err := GetLanguageProfile(languageID)
+ if err != nil {
+ t.Fatal(err)
+ }
+ if languageProfile.DefaultModelID != ParakeetV3ModelID {
+ t.Fatalf("%s default model = %q, want %q", languageID, languageProfile.DefaultModelID, ParakeetV3ModelID)
+ }
+ }
+}
+
+func TestRegistryReturnsJapaneseAndKoreanDefaults(t *testing.T) {
+ tests := []struct {
+ languageID string
+ modelID string
+ }{
+ {JapaneseLanguageID, JapaneseModelID},
+ {KoreanLanguageID, KoreanModelID},
+ }
+ for _, tt := range tests {
+ languageProfile, err := GetLanguageProfile(tt.languageID)
+ if err != nil {
+ t.Fatal(err)
+ }
+ if languageProfile.DefaultModelID != tt.modelID {
+ t.Fatalf("%s default model = %q, want %q", tt.languageID, languageProfile.DefaultModelID, tt.modelID)
+ }
+ modelProfile, err := GetModelProfile(tt.modelID)
+ if err != nil {
+ t.Fatal(err)
+ }
+ if modelProfile.BackendKind != BackendTransducer {
+ t.Fatalf("%s backend = %q, want %q", tt.modelID, modelProfile.BackendKind, BackendTransducer)
+ }
+ }
+}
+
func TestRegistryReturnsQwen3ASRChinese(t *testing.T) {
profile, err := GetModelProfile(Qwen3ASRModelID)
if err != nil {
@@ -80,6 +128,25 @@
}
}
+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 {
@@ -88,8 +155,8 @@
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)
}
}
@@ -101,8 +168,8 @@
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)
}
}
@@ -228,6 +295,61 @@
}
}
+func TestValidateJapaneseZipformerRequiredFiles(t *testing.T) {
+ profile, err := GetModelProfile(JapaneseModelID)
+ if err != nil {
+ t.Fatal(err)
+ }
+ dir := t.TempDir()
+ createValidZipformerSpecialized(t, dir)
+
+ result := ValidateModelDir(profile, dir)
+ if !result.Valid {
+ t.Fatalf("expected valid japanese zipformer 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 TestValidateKoreanZipformerRequiredFiles(t *testing.T) {
+ profile, err := GetModelProfile(KoreanModelID)
+ if err != nil {
+ t.Fatal(err)
+ }
+ dir := t.TempDir()
+ createValidZipformerSpecialized(t, dir)
+
+ result := ValidateModelDir(profile, dir)
+ if !result.Valid {
+ t.Fatalf("expected valid korean zipformer 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 TestValidateZipformerSpecializedRejectsMissingJoiner(t *testing.T) {
+ profile, err := GetModelProfile(JapaneseModelID)
+ if err != nil {
+ t.Fatal(err)
+ }
+ dir := t.TempDir()
+ createValidZipformerSpecialized(t, dir)
+ if err := os.Remove(filepath.Join(dir, "joiner-epoch-99-avg-1.int8.onnx")); err != nil {
+ t.Fatal(err)
+ }
+
+ 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 {
@@ -282,6 +404,42 @@
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")
}
}
@@ -669,6 +827,17 @@
writeTestFile(t, filepath.Join(dir, "tokens.txt"), "tokens")
}
+func createValidZipformerSpecialized(t *testing.T, dir string) {
+ t.Helper()
+ if err := os.MkdirAll(dir, 0755); err != nil {
+ t.Fatal(err)
+ }
+ writeTestFile(t, filepath.Join(dir, "encoder-epoch-99-avg-1.int8.onnx"), "encoder")
+ writeTestFile(t, filepath.Join(dir, "decoder-epoch-99-avg-1.onnx"), "decoder")
+ writeTestFile(t, filepath.Join(dir, "joiner-epoch-99-avg-1.int8.onnx"), "joiner")
+ 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 {
@@ -681,6 +850,26 @@
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
--
Gitblit v1.9.3