dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-30 01:04:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

test: update voice tests to mock audio dependencies properly

Browse source
This commit is contained in:
Paul Gauthier (aider) 2024-11-30 19:15:16 -08:00
parent 1a8d112055
commit 49c78f2797
1 changed files with 46 additions and 52 deletions
Download patch file Download diff file Expand all files Collapse all files

42
tests/basic/test_voice.py
View file

@ -1,39 +1,37 @@
import os import os
import queue import queue
from unittest.mock import patch from unittest.mock import patch, MagicMock
import numpy as np import numpy as np
import pytest import pytest
from aider.voice import SoundDeviceError, Voice from aider.voice import SoundDeviceError, Voice
# Mock the entire sounddevice module
@pytest.fixture @pytest.fixture
def mock_sounddevice(): def mock_sounddevice():
with patch("sounddevice.query_devices") as mock_query: with patch('aider.voice.sounddevice') as mock_sd:
mock_query.return_value = [ mock_sd.query_devices.return_value = [
{"name": "test_device", "max_input_channels": 2}, {"name": "test_device", "max_input_channels": 2},
{"name": "another_device", "max_input_channels": 1}, {"name": "another_device", "max_input_channels": 1},
] ]
yield mock_query yield mock_sd
@pytest.fixture @pytest.fixture
def mock_soundfile(): def mock_soundfile():
with patch("soundfile.SoundFile") as mock_sf: with patch('aider.voice.sf') as mock_sf:
yield mock_sf yield mock_sf
def test_voice_init_default_device(mock_sounddevice): def test_voice_init_default_device(mock_sounddevice):
voice = Voice() voice = Voice()
assert voice.device_id is None assert voice.device_id is None
assert voice.audio_format == "wav" assert voice.audio_format == "wav"
assert voice.sd == mock_sounddevice
def test_voice_init_specific_device(mock_sounddevice): def test_voice_init_specific_device(mock_sounddevice):
voice = Voice(device_name="test_device") voice = Voice(device_name="test_device")
assert voice.device_id == 0 assert voice.device_id == 0
assert voice.sd == mock_sounddevice
def test_voice_init_invalid_device(mock_sounddevice): def test_voice_init_invalid_device(mock_sounddevice):
with pytest.raises(ValueError) as exc: with pytest.raises(ValueError) as exc:
@ -41,14 +39,14 @@ def test_voice_init_invalid_device(mock_sounddevice):
assert "Device" in str(exc.value) assert "Device" in str(exc.value)
assert "not found" in str(exc.value) assert "not found" in str(exc.value)
def test_voice_init_invalid_format(): def test_voice_init_invalid_format():
with patch('aider.voice.sf', MagicMock()): # Need to mock sf to avoid SoundDeviceError
with pytest.raises(ValueError) as exc: with pytest.raises(ValueError) as exc:
Voice(audio_format="invalid") Voice(audio_format="invalid")
assert "Unsupported audio format" in str(exc.value) assert "Unsupported audio format" in str(exc.value)
def test_callback_processing(): def test_callback_processing():
with patch('aider.voice.sf', MagicMock()): # Need to mock sf to avoid SoundDeviceError
voice = Voice() voice = Voice()
voice.q = queue.Queue() voice.q = queue.Queue()
@ -65,10 +63,10 @@ def test_callback_processing():
# Verify data is queued # Verify data is queued
assert not voice.q.empty() assert not voice.q.empty()
def test_get_prompt(): def test_get_prompt():
with patch('aider.voice.sf', MagicMock()): # Need to mock sf to avoid SoundDeviceError
voice = Voice() voice = Voice()
voice.start_time = voice.start_time = os.times().elapsed voice.start_time = os.times().elapsed
voice.pct = 0.5 # 50% volume level voice.pct = 0.5 # 50% volume level
prompt = voice.get_prompt() prompt = voice.get_prompt()
@ -77,20 +75,16 @@ def test_get_prompt():
assert "█" in prompt # Should contain some filled blocks assert "█" in prompt # Should contain some filled blocks
assert "░" in prompt # Should contain some empty blocks assert "░" in prompt # Should contain some empty blocks
def test_record_and_transcribe_keyboard_interrupt():
@patch("sounddevice.InputStream") with patch('aider.voice.sf', MagicMock()):
def test_record_and_transcribe_keyboard_interrupt(mock_stream):
voice = Voice() voice = Voice()
mock_stream.side_effect = KeyboardInterrupt() with patch.object(voice, 'raw_record_and_transcribe', side_effect=KeyboardInterrupt()):
result = voice.record_and_transcribe() result = voice.record_and_transcribe()
assert result is None assert result is None
def test_record_and_transcribe_device_error():
@patch("sounddevice.InputStream") with patch('aider.voice.sf', MagicMock()):
def test_record_and_transcribe_device_error(mock_stream):
voice = Voice() voice = Voice()
mock_stream.side_effect = SoundDeviceError("Test error") with patch.object(voice, 'raw_record_and_transcribe', side_effect=SoundDeviceError("Test error")):
result = voice.record_and_transcribe() result = voice.record_and_transcribe()
assert result is None assert result is None