helpers.py

tests/helpers.py

1. import math
2. import struct
3. import wave
4. from pathlib import Path
7. def write_tone(path: Path, seconds: float = 1.5, rate: int = 16000, freq: float = 440.0) -> Path:
8. with wave.open(str(path), "wb") as w:
9. w.setnchannels(1)
10. w.setsampwidth(2)
11. w.setframerate(rate)
12. n = int(seconds * rate)
13. w.writeframes(b"".join(
14. struct.pack("<h", int(8000 * math.sin(2 * math.pi * freq * i / rate)))
15. for i in range(n)))
16. return path
19. def fake_result(speakers=True):
20. """What alignment plus diarization hand to shape.py for a short two-person call."""
21. def w(word, s, e, spk):
22. d = {"word": word, "start": s, "end": e}
23. if speakers:
24. d["speaker"] = spk
25. return d
26. return {
27. "language": "en", "model": "fake",
28. "segments": [
29. {"start": 0.5, "end": 3.1, "text": "Hi, thanks for calling back. No problem.",
30. "words": [w("Hi,", 0.5, 0.7, "SPEAKER_01"), w("thanks", 0.8, 1.0, "SPEAKER_01"),
31. w("for", 1.0, 1.1, "SPEAKER_01"), w("calling", 1.1, 1.4, "SPEAKER_01"),
32. w("back.", 1.4, 1.7, "SPEAKER_01"), w("No", 2.2, 2.5, "SPEAKER_00"),
33. w("problem.", 2.5, 3.1, "SPEAKER_00")]},
34. {"start": 3.5, "end": 5.0, "text": "Did the draft come through?",
35. "words": [w("Did", 3.5, 3.7, "SPEAKER_00"), w("the", 3.7, 3.8, "SPEAKER_00"),
36. w("draft", 3.8, 4.2, "SPEAKER_00"), w("come", 4.2, 4.5, "SPEAKER_00"),
37. w("through?", 4.5, 5.0, "SPEAKER_00")]},
38. ],
39. }