first git

tests/test_paraformer_transcription_events.py

@@ -0,0 +1,62 @@
+import numpy as np
+
+from speech_to_speech.pipeline.messages import PartialTranscription, Transcription, VADAudio
+from speech_to_speech.STT import paraformer_handler
+from speech_to_speech.STT.paraformer_handler import ParaformerSTTHandler
+
+
+class _FakeParaformerModel:
+    def generate(self, audio):
+        return [{"text": " 今 天 天 气 不 错 "}]
+
+
+def _handler():
+    handler = object.__new__(ParaformerSTTHandler)
+    handler.model = _FakeParaformerModel()
+    return handler
+
+
+def test_progressive_paraformer_transcription_is_partial(monkeypatch):
+    monkeypatch.setattr(paraformer_handler.console, "print", lambda *args, **kwargs: None)
+    monkeypatch.setattr(paraformer_handler.torch.mps, "empty_cache", lambda: None)
+
+    result = list(
+        _handler().process(
+            VADAudio(
+                audio=np.zeros(16000, dtype=np.float32),
+                mode="progressive",
+                turn_id="turn_1",
+                turn_revision=2,
+            )
+        )
+    )
+
+    assert len(result) == 1
+    assert isinstance(result[0], PartialTranscription)
+    assert result[0].text == "今天天气不错"
+    assert result[0].turn_id == "turn_1"
+    assert result[0].turn_revision == 2
+
+
+def test_final_paraformer_transcription_is_final(monkeypatch):
+    monkeypatch.setattr(paraformer_handler.console, "print", lambda *args, **kwargs: None)
+    monkeypatch.setattr(paraformer_handler.torch.mps, "empty_cache", lambda: None)
+
+    result = list(
+        _handler().process(
+            VADAudio(
+                audio=np.zeros(16000, dtype=np.float32),
+                mode="final",
+                turn_id="turn_1",
+                turn_revision=2,
+                created_at_s=123.0,
+            )
+        )
+    )
+
+    assert len(result) == 1
+    assert isinstance(result[0], Transcription)
+    assert result[0].text == "今天天气不错"
+    assert result[0].turn_id == "turn_1"
+    assert result[0].turn_revision == 2
+    assert result[0].speech_stopped_at_s == 123.0