b5f82fb48c
Some checks are pending
CI / ruff (push) Waiting to run
CI / mypy (push) Waiting to run
CI / pytest (push) Waiting to run
CI / package (push) Waiting to run
CI / Install smoke (${{ matrix.label }}) (linux, ubuntu-latest) (push) Blocked by required conditions
CI / Install smoke (${{ matrix.label }}) (macos-arm64, macos-14) (push) Blocked by required conditions
63 lines
1.9 KiB
Python
63 lines
1.9 KiB
Python
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
|