feat: add optional device parameter to Recorder
This commit is contained in:
@@ -63,3 +63,16 @@ class TestRecorder:
|
||||
assert result.shape == (16000,)
|
||||
assert result[0] == 1.0
|
||||
assert result[8000] == 0.5
|
||||
|
||||
def test_device_passed_to_inputstream(self):
|
||||
recorder = Recorder(sample_rate=16000, channels=1, min_duration=0.0, device="USB Mic")
|
||||
with patch("sounddevice.InputStream") as mock_cls:
|
||||
mock_cls.return_value = MagicMock()
|
||||
recorder.start()
|
||||
call_kwargs = mock_cls.call_args.kwargs
|
||||
assert call_kwargs["device"] == "USB Mic"
|
||||
recorder.stop()
|
||||
|
||||
def test_default_device_is_none(self):
|
||||
recorder = Recorder()
|
||||
assert recorder.device is None
|
||||
|
||||
Reference in New Issue
Block a user