Ariver
2026-06-07 58c2c4ae5eaca3dff127f2ffb15f9dd2b160f994
privatevoice.src/internal/engine/engine_darwin_test.go
@@ -3,6 +3,7 @@
package engine
import (
   "os"
   "path/filepath"
   "testing"
@@ -96,3 +97,73 @@
      t.Fatalf("tokens = %q, want empty for qwen3-asr", config.ModelConfig.Tokens)
   }
}
func TestOnlineConfigForXASRUsesStreamingTransducer(t *testing.T) {
   root := t.TempDir()
   files := map[string]string{
      "encoder": filepath.Join(root, "encoder-960ms.onnx"),
      "decoder": filepath.Join(root, "decoder-960ms.onnx"),
      "joiner":  filepath.Join(root, "joiner-960ms.onnx"),
      "tokens":  filepath.Join(root, "tokens.txt"),
   }
   resolved := model.ResolvedModel{
      ModelID:     model.XASRZhEn960ModelID,
      BackendKind: model.BackendXASRStreaming,
      Profile: model.ModelProfile{
         DisplayName:   "X-ASR zh-en 960ms",
         BackendKind:   model.BackendXASRStreaming,
         ProviderOrder: []string{"cpu"},
         NumThreads:    1,
      },
      Files: files,
   }
   config, err := onlineConfigForResolvedModel(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 != "zipformer2" {
      t.Fatalf("model type = %q, want zipformer2", config.ModelConfig.ModelType)
   }
   if config.DecodingMethod != "greedy_search" {
      t.Fatalf("decoding method = %q, want greedy_search", config.DecodingMethod)
   }
}
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")
   }
   resolved, err := model.ResolveModel(model.XASRZhEn960ModelID)
   if err != nil {
      t.Fatal(err)
   }
   if !resolved.IsUsable() {
      t.Fatalf("x-asr model is not usable: status=%s missing=%v problems=%v", resolved.Status, resolved.Missing, resolved.Problems)
   }
   eng, err := NewWithResolvedModel(resolved)
   if err != nil {
      t.Fatal(err)
   }
   defer eng.Close()
   text, err := eng.Recognize(make([]float32, 16000))
   if err != nil {
      t.Fatal(err)
   }
   t.Logf("hardware=%s silence_text=%q", eng.HardwareInfo(), text)
}