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

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 os
import queue import queue
from unittest.mock import patch, MagicMock from unittest.mock import MagicMock, patch
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 # Mock the entire sounddevice module
@pytest.fixture @pytest.fixture
def mock_sounddevice(): 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 = [ 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_sd yield mock_sd
@pytest.fixture @pytest.fixture
def mock_soundfile(): def mock_soundfile():
with patch('aider.voice.sf') 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 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 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:
Voice(device_name="nonexistent_device") Voice(device_name="nonexistent_device")
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 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 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()
@ -63,8 +70,9 @@ 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 with patch("aider.voice.sf", MagicMock()): # Need to mock sf to avoid SoundDeviceError
voice = Voice() voice = Voice()
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
@ -75,16 +83,20 @@ 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(): def test_record_and_transcribe_keyboard_interrupt():
with patch('aider.voice.sf', MagicMock()): with patch("aider.voice.sf", MagicMock()):
voice = Voice() 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() result = voice.record_and_transcribe()
assert result is None assert result is None
def test_record_and_transcribe_device_error(): def test_record_and_transcribe_device_error():
with patch('aider.voice.sf', MagicMock()): with patch("aider.voice.sf", MagicMock()):
voice = Voice() 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() result = voice.record_and_transcribe()
assert result is None assert result is None