dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-28 16:25:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

style: Format code with black and sort imports

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

32
tests/basic/test_voice.py
View file

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