mirror of
https://github.com/QuentinFuxa/WhisperLiveKit.git
synced 2026-03-07 06:14:05 +00:00
5a12c627b4
Test suite covering: - metrics.py: WER computation, timestamp accuracy, text normalization - config.py: defaults, .en model detection, policy aliases, from_namespace - timed_objects.py: ASRToken, Silence, Transcript, Segment, FrontData - hypothesis_buffer.py: insert, flush, LCP matching, pop_committed - silence_handling.py: state machine, double-counting regression test - audio_processor.py: async pipeline with MockOnlineProcessor All tests run in ~1.3s without downloading any ASR models. Add pytest and pytest-asyncio as optional test dependencies. Update .gitignore to allow tests/ directory.
59 lines
1.4 KiB
Python
59 lines
1.4 KiB
Python
"""Shared pytest fixtures for WhisperLiveKit tests."""
|
|
|
|
import json
|
|
from pathlib import Path
|
|
from types import SimpleNamespace
|
|
|
|
import pytest
|
|
|
|
from whisperlivekit.timed_objects import ASRToken, Silence, Transcript
|
|
|
|
|
|
AUDIO_TESTS_DIR = Path(__file__).parent.parent / "audio_tests"
|
|
|
|
|
|
@pytest.fixture
|
|
def sample_tokens():
|
|
"""A short sequence of ASRToken objects."""
|
|
return [
|
|
ASRToken(start=0.0, end=0.5, text="Hello"),
|
|
ASRToken(start=0.5, end=1.0, text=" world"),
|
|
ASRToken(start=1.0, end=1.5, text=" test."),
|
|
]
|
|
|
|
|
|
@pytest.fixture
|
|
def sample_silence():
|
|
"""A completed silence event."""
|
|
s = Silence(start=1.5, end=3.0, is_starting=False, has_ended=True)
|
|
s.compute_duration()
|
|
return s
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_args():
|
|
"""Minimal args namespace for AudioProcessor tests."""
|
|
return SimpleNamespace(
|
|
diarization=False,
|
|
transcription=True,
|
|
target_language="",
|
|
vac=False,
|
|
vac_chunk_size=0.04,
|
|
min_chunk_size=0.1,
|
|
pcm_input=True,
|
|
punctuation_split=False,
|
|
backend="faster-whisper",
|
|
backend_policy="localagreement",
|
|
vad=True,
|
|
)
|
|
|
|
|
|
@pytest.fixture
|
|
def ground_truth_en():
|
|
"""Ground truth transcript for the 7s English audio (if available)."""
|
|
path = AUDIO_TESTS_DIR / "00_00_07_english_1_speaker.transcript.json"
|
|
if path.exists():
|
|
with open(path) as f:
|
|
return json.load(f)
|
|
return None
|