mkbk
71 lines
2.7 KiB
Python
71 lines
2.7 KiB
Python
from __future__ import annotations
|
|
|
|
import tempfile
|
|
import unittest
|
|
from pathlib import Path
|
|
|
|
from owner_voice_pet.models import AudioFrame, AudioSegment, ErrorCode
|
|
from owner_voice_pet.transport import (
|
|
AudioRingBuffer,
|
|
FileReplayTransport,
|
|
MemoryAudioTransport,
|
|
SoundDeviceAudioTransport,
|
|
)
|
|
|
|
|
|
def frame(idx: int, timestamp_ms: int, metadata: dict[str, object] | None = None) -> AudioFrame:
|
|
return AudioFrame(
|
|
pcm=idx.to_bytes(2, "little", signed=False),
|
|
sample_rate=16000,
|
|
channels=1,
|
|
timestamp_ms=timestamp_ms,
|
|
frame_id=idx,
|
|
metadata=metadata or {},
|
|
)
|
|
|
|
|
|
class TransportTests(unittest.TestCase):
|
|
def test_memory_transport_replays_frames_and_captures_playback(self) -> None:
|
|
transport = MemoryAudioTransport([frame(1, 0), frame(2, 20)])
|
|
transport.start_input()
|
|
self.assertEqual([f.frame_id for f in transport.read_frames(10)], [1])
|
|
self.assertEqual([f.frame_id for f in transport.read_frames(10)], [2])
|
|
self.assertEqual(transport.read_frames(10), [])
|
|
|
|
result = transport.play_pcm(AudioSegment(b"\x00\x00", 16000, 1, 0, 100))
|
|
self.assertTrue(result.played)
|
|
self.assertEqual(len(transport.played_segments), 1)
|
|
|
|
def test_memory_transport_reports_missing_output(self) -> None:
|
|
transport = MemoryAudioTransport(output_available=False)
|
|
result = transport.play_pcm(AudioSegment(b"\x00\x00", 16000, 1, 0, 100))
|
|
self.assertFalse(result.played)
|
|
self.assertEqual(result.error.code, ErrorCode.AUDIO_OUTPUT_DEVICE_MISSING)
|
|
|
|
def test_file_replay_roundtrip(self) -> None:
|
|
frames = [frame(1, 0, {"wake": True}), frame(2, 20, {"speech": True})]
|
|
with tempfile.TemporaryDirectory() as tmp:
|
|
path = Path(tmp) / "fixture.jsonl"
|
|
FileReplayTransport.write_jsonl(path, frames)
|
|
transport = FileReplayTransport.from_jsonl(path)
|
|
transport.start_input()
|
|
self.assertTrue(transport.read_frames(10)[0].metadata["wake"])
|
|
self.assertTrue(transport.read_frames(10)[0].metadata["speech"])
|
|
|
|
def test_ring_buffer_trims_old_frames_and_requires_order(self) -> None:
|
|
buffer = AudioRingBuffer(max_duration_ms=40)
|
|
buffer.extend([frame(1, 0), frame(2, 20), frame(3, 50)])
|
|
self.assertEqual([f.frame_id for f in buffer.frames()], [2, 3])
|
|
with self.assertRaises(ValueError):
|
|
buffer.append(frame(4, 10))
|
|
|
|
def test_sounddevice_transport_is_safe_without_optional_dependency(self) -> None:
|
|
transport = SoundDeviceAudioTransport()
|
|
health = transport.health()
|
|
self.assertIsInstance(health.input_available, bool)
|
|
self.assertIsInstance(health.output_available, bool)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|