diff --git a/tests/basic/test_voice.py b/tests/basic/test_voice.py index 4bb443e0a..dbfcb4987 100644 --- a/tests/basic/test_voice.py +++ b/tests/basic/test_voice.py @@ -71,8 +71,15 @@ def test_record_and_transcribe(mock_litellm, mock_soundfile): voice = Voice() # Mock the recording process - with patch("prompt_toolkit.shortcuts.prompt", return_value=""): - with patch("sounddevice.InputStream"): + with patch("sounddevice.InputStream") as mock_stream: + # Set up the mock stream to simulate some audio data + mock_data = np.zeros((1000, 1)) + mock_stream.return_value.__enter__.return_value = mock_data + + # Mock prompt_toolkit's prompt function more completely + with patch("prompt_toolkit.shortcuts.prompt") as mock_prompt: + mock_prompt.return_value = "" # Simulate pressing Enter + # Mock the transcription response mock_litellm.transcription.return_value = Mock(text="Hello, world!")