wx44wx/pipecat
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Apply ruff formatting to test file

Browse Source
This commit is contained in:
gui217
2025-12-30 13:39:36 +02:00
parent ec5fb392c4
commit 32c3298eff
1 changed files with 12 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
tests/test_rnnoise_filter.py
View File

@@ -104,23 +104,23 @@ class TestRNNoiseFilter(unittest.IsolatedAsyncioTestCase):
async def test_rnnoise_filter_handles_empty_resampler_output(self): async def test_rnnoise_filter_handles_empty_resampler_output(self):
"""Test that RNNoise filter handles empty bytes from resampler gracefully. """Test that RNNoise filter handles empty bytes from resampler gracefully.
This reproduces the issue where mute/silence input (like "bx000") causes This reproduces the issue where mute/silence input (like "bx000") causes
the resampler to return empty bytes. The filter should handle this gracefully the resampler to return empty bytes. The filter should handle this gracefully
without raising a memory error. without raising a memory error.
""" """
filter = RNNoiseFilter() filter = RNNoiseFilter()
# Initialize with a sample rate that requires resampling (not 48kHz) # Initialize with a sample rate that requires resampling (not 48kHz)
await filter.start(sample_rate=16000) await filter.start(sample_rate=16000)
# Enable filtering # Enable filtering
await filter.process_frame(FilterEnableFrame(enable=True)) await filter.process_frame(FilterEnableFrame(enable=True))
# Create mute/silence audio chunk (like "bx000" - all zeros) # Create mute/silence audio chunk (like "bx000" - all zeros)
# This simulates a mute sound that causes resampler to return empty bytes # This simulates a mute sound that causes resampler to return empty bytes
mute_audio = b"\x00\x00" * 160 # 160 samples of silence at 16kHz (10ms) mute_audio = b"\x00\x00" * 160 # 160 samples of silence at 16kHz (10ms)
# Mock the resampler to return empty bytes when given mute input # Mock the resampler to return empty bytes when given mute input
# This simulates the behavior where resampler returns b"" for mute sounds # This simulates the behavior where resampler returns b"" for mute sounds
async def mock_resample(audio, in_rate, out_rate): async def mock_resample(audio, in_rate, out_rate):
@@ -131,13 +131,15 @@ class TestRNNoiseFilter(unittest.IsolatedAsyncioTestCase):
return b"" # Return empty bytes - this triggers the bug return b"" # Return empty bytes - this triggers the bug
# Otherwise, do normal resampling # Otherwise, do normal resampling
return audio return audio
filter._resampler_in.resample = AsyncMock(side_effect=mock_resample) filter._resampler_in.resample = AsyncMock(side_effect=mock_resample)
result = await filter.filter(mute_audio) result = await filter.filter(mute_audio)
# When resampler returns empty bytes, filter should return empty bytes # When resampler returns empty bytes, filter should return empty bytes
# (or handle it gracefully without crashing) # (or handle it gracefully without crashing)
self.assertEqual(result, b"", "Filter should return empty bytes when resampler returns empty bytes") self.assertEqual(
result, b"", "Filter should return empty bytes when resampler returns empty bytes"
)
await filter.stop() await filter.stop()