Spaces:
Sleeping
Sleeping
| """Tests for MelodyPracticeService.""" | |
| import time | |
| import numpy as np | |
| import pytest | |
| from improvisation_lab.config import Config | |
| from improvisation_lab.service.melody_practice_service import ( | |
| MelodyPracticeService, PitchResult) | |
| class TestMelodyService: | |
| def init_module(self): | |
| """Create MelodyService instance for testing.""" | |
| config = Config() | |
| self.service = MelodyPracticeService(config) | |
| def test_generate_melody(self): | |
| """Test melody generation.""" | |
| phrases = self.service.generate_melody() | |
| assert len(phrases) > 0 | |
| assert all(hasattr(phrase, "notes") for phrase in phrases) | |
| assert all(hasattr(phrase, "chord_name") for phrase in phrases) | |
| assert all(hasattr(phrase, "scale_info") for phrase in phrases) | |
| assert all(hasattr(phrase, "length") for phrase in phrases) | |
| def test_process_audio_no_voice(self): | |
| """Test processing audio with no voice detected.""" | |
| audio_data = np.zeros(1024, dtype=np.float32) | |
| result = self.service.process_audio(audio_data, target_note="A") | |
| assert isinstance(result, PitchResult) | |
| assert result.current_base_note is None | |
| assert not result.is_correct | |
| def test_process_audio_with_voice(self): | |
| """Test processing audio with voice detected.""" | |
| sample_rate = 44100 | |
| duration = 0.1 | |
| t = np.linspace(0, duration, int(sample_rate * duration)) | |
| audio_data = np.sin(2 * np.pi * 440 * t) | |
| result = self.service.process_audio(audio_data, target_note="A") | |
| assert isinstance(result, PitchResult) | |
| assert result.current_base_note == "A" | |
| assert result.is_correct | |
| def test_process_audio_incorrect_pitch(self): | |
| """Test processing audio with incorrect pitch.""" | |
| sample_rate = 44100 | |
| duration = 0.1 | |
| t = np.linspace(0, duration, int(sample_rate * duration)) | |
| # Generate 440Hz (A4) when target is C4 | |
| audio_data = np.sin(2 * np.pi * 440 * t) | |
| result = self.service.process_audio(audio_data, target_note="C") | |
| assert isinstance(result, PitchResult) | |
| assert result.current_base_note == "A" | |
| assert not result.is_correct | |
| assert result.remaining_time == self.service.config.audio.note_duration | |
| def test_correct_pitch_timing(self): | |
| """Test timing behavior with correct pitch.""" | |
| sample_rate = 44100 | |
| duration = 0.1 | |
| t = np.linspace(0, duration, int(sample_rate * duration)) | |
| audio_data = np.sin(2 * np.pi * 440 * t) | |
| # First detection | |
| result1 = self.service.process_audio(audio_data, target_note="A") | |
| initial_time = self.service.correct_pitch_start_time | |
| assert result1.is_correct | |
| assert result1.remaining_time == self.service.config.audio.note_duration | |
| # Wait a bit | |
| time.sleep(0.5) | |
| # Second detection | |
| result2 = self.service.process_audio(audio_data, target_note="A") | |
| assert result2.is_correct | |
| assert result2.remaining_time < self.service.config.audio.note_duration | |
| assert initial_time == self.service.correct_pitch_start_time | |
| def test_correct_pitch_completion(self): | |
| """Test completion of correct pitch duration.""" | |
| sample_rate = 44100 | |
| duration = 0.1 | |
| t = np.linspace(0, duration, int(sample_rate * duration)) | |
| audio_data = np.sin(2 * np.pi * 440 * t) | |
| # First detection | |
| result1 = self.service.process_audio(audio_data, target_note="A") | |
| assert result1.remaining_time == self.service.config.audio.note_duration | |
| # Wait for full duration | |
| time.sleep(self.service.config.audio.note_duration + 0.1) | |
| # Final detection | |
| result2 = self.service.process_audio(audio_data, target_note="A") | |
| assert result2.remaining_time == 0 | |