Ariver
2026-06-24 e533a8cd34995af9aaeeb321b41969a260fef882
privatevoice.src/internal/engine/engine_darwin_test.go
@@ -6,6 +6,7 @@
   "os"
   "path/filepath"
   "testing"
   "time"
   "voicesnap/internal/model"
)
@@ -48,6 +49,47 @@
   }
   if config.ModelConfig.ModelType != model.BackendNemoTransducer {
      t.Fatalf("model type = %q, want %q", config.ModelConfig.ModelType, model.BackendNemoTransducer)
   }
}
func TestOfflineConfigForZipformerUsesPlainTransducer(t *testing.T) {
   root := t.TempDir()
   files := map[string]string{
      "encoder": filepath.Join(root, "encoder-epoch-99-avg-1.int8.onnx"),
      "decoder": filepath.Join(root, "decoder-epoch-99-avg-1.onnx"),
      "joiner":  filepath.Join(root, "joiner-epoch-99-avg-1.int8.onnx"),
      "tokens":  filepath.Join(root, "tokens.txt"),
   }
   resolved := model.ResolvedModel{
      ModelID:     model.JapaneseModelID,
      BackendKind: model.BackendTransducer,
      Profile: model.ModelProfile{
         DisplayName:   "Zipformer Japanese",
         BackendKind:   model.BackendTransducer,
         ProviderOrder: []string{"cpu"},
         NumThreads:    1,
      },
      Files: files,
   }
   config, err := offlineConfigForResolvedModel(resolved, "cpu")
   if err != nil {
      t.Fatal(err)
   }
   if config.ModelConfig.Transducer.Encoder != files["encoder"] {
      t.Fatalf("encoder = %q, want %q", config.ModelConfig.Transducer.Encoder, files["encoder"])
   }
   if config.ModelConfig.Transducer.Decoder != files["decoder"] {
      t.Fatalf("decoder = %q, want %q", config.ModelConfig.Transducer.Decoder, files["decoder"])
   }
   if config.ModelConfig.Transducer.Joiner != files["joiner"] {
      t.Fatalf("joiner = %q, want %q", config.ModelConfig.Transducer.Joiner, files["joiner"])
   }
   if config.ModelConfig.Tokens != files["tokens"] {
      t.Fatalf("tokens = %q, want %q", config.ModelConfig.Tokens, files["tokens"])
   }
   if config.ModelConfig.ModelType != "" {
      t.Fatalf("model type = %q, want empty for plain transducer", config.ModelConfig.ModelType)
   }
}
@@ -142,6 +184,74 @@
   }
}
func TestRememberNonEmptyTextKeepsLastPartial(t *testing.T) {
   got := rememberNonEmptyText("最后一句", "   ")
   if got != "最后一句" {
      t.Fatalf("rememberNonEmptyText() = %q, want last non-empty text", got)
   }
   got = rememberNonEmptyText("旧内容", "新内容")
   if got != "新内容" {
      t.Fatalf("rememberNonEmptyText() = %q, want new non-empty text", got)
   }
}
func TestXASRTailPaddingMatchesFlushWindow(t *testing.T) {
   if xasrTailPaddingSamples != 24000 {
      t.Fatalf("tail padding samples = %d, want 24000", xasrTailPaddingSamples)
   }
   if len(xasrTailPadding) != xasrTailPaddingSamples {
      t.Fatalf("tail padding len = %d, want %d", len(xasrTailPadding), xasrTailPaddingSamples)
   }
}
func TestXASRHeadPaddingPreparesInitialContext(t *testing.T) {
   if xasrHeadPaddingSamples != 4000 {
      t.Fatalf("head padding samples = %d, want 4000", xasrHeadPaddingSamples)
   }
   input := []float32{0.1, -0.2, 0.3}
   got := xasrSamplesWithHeadPadding(input)
   if len(got) != xasrHeadPaddingSamples+len(input) {
      t.Fatalf("padded len = %d, want %d", len(got), xasrHeadPaddingSamples+len(input))
   }
   for i := 0; i < xasrHeadPaddingSamples; i++ {
      if got[i] != 0 {
         t.Fatalf("head padding sample %d = %v, want 0", i, got[i])
      }
   }
   for i, want := range input {
      if got[xasrHeadPaddingSamples+i] != want {
         t.Fatalf("payload sample %d = %v, want %v", i, got[xasrHeadPaddingSamples+i], want)
      }
   }
}
func TestXASRRequestsReleaseTailCaptureDelay(t *testing.T) {
   got := (&xasrStreamingEngine{}).ReleaseTailCaptureDelay()
   if got != 300*time.Millisecond {
      t.Fatalf("ReleaseTailCaptureDelay() = %v, want 300ms", got)
   }
}
func TestXASREnablesHoldPreCapture(t *testing.T) {
   if !(&xasrStreamingEngine{}).HoldPreCaptureEnabled() {
      t.Fatal("X-ASR should enable hold pre-capture")
   }
}
func TestOfflineSherpaDoesNotRequestReleaseTailCapture(t *testing.T) {
   if _, ok := any(&sherpaEngine{}).(ReleaseTailCaptureEngine); ok {
      t.Fatal("offline sherpa engine should not request release tail capture")
   }
}
func TestOfflineSherpaDoesNotEnableHoldPreCapture(t *testing.T) {
   if _, ok := any(&sherpaEngine{}).(HoldPreCaptureEngine); ok {
      t.Fatal("offline sherpa engine should not enable hold pre-capture")
   }
}
func TestXASRRealModelSmoke(t *testing.T) {
   if os.Getenv("PRIVATEVOICE_XASR_SMOKE") != "1" {
      t.Skip("set PRIVATEVOICE_XASR_SMOKE=1 to run the real X-ASR model smoke test")