| Index: webrtc/modules/audio_processing/test/py_quality_assessment/quality_assessment/signal_processing_unittest.py
|
| diff --git a/webrtc/modules/audio_processing/test/py_quality_assessment/quality_assessment/signal_processing_unittest.py b/webrtc/modules/audio_processing/test/py_quality_assessment/quality_assessment/signal_processing_unittest.py
|
| index 7cb088c0d40722de27e3e7cc77e37e95cf49723e..3edd5387d5a4a6d7a45d19764a556cc8342fd29d 100644
|
| --- a/webrtc/modules/audio_processing/test/py_quality_assessment/quality_assessment/signal_processing_unittest.py
|
| +++ b/webrtc/modules/audio_processing/test/py_quality_assessment/quality_assessment/signal_processing_unittest.py
|
| @@ -19,15 +19,17 @@ from . import signal_processing
|
|
|
|
|
| class TestSignalProcessing(unittest.TestCase):
|
| + """Unit tests for the signal_processing module.
|
| + """
|
|
|
| def testMixSignals(self):
|
| # Generate a template signal with which white noise can be generated.
|
| silence = pydub.AudioSegment.silent(duration=1000, frame_rate=48000)
|
|
|
| # Generate two distinct AudioSegment instances with 1 second of white noise.
|
| - signal = signal_processing.SignalProcessingUtils.generate_white_noise(
|
| + signal = signal_processing.SignalProcessingUtils.GenerateWhiteNoise(
|
| silence)
|
| - noise = signal_processing.SignalProcessingUtils.generate_white_noise(
|
| + noise = signal_processing.SignalProcessingUtils.GenerateWhiteNoise(
|
| silence)
|
|
|
| # Extract samples.
|
| @@ -35,7 +37,7 @@ class TestSignalProcessing(unittest.TestCase):
|
| noise_samples = noise.get_array_of_samples()
|
|
|
| # Test target SNR -Inf (noise expected).
|
| - mix_neg_inf = signal_processing.SignalProcessingUtils.mix_signals(
|
| + mix_neg_inf = signal_processing.SignalProcessingUtils.MixSignals(
|
| signal, noise, -np.Inf)
|
| self.assertTrue(len(noise), len(mix_neg_inf)) # Check duration.
|
| mix_neg_inf_samples = mix_neg_inf.get_array_of_samples()
|
| @@ -43,7 +45,7 @@ class TestSignalProcessing(unittest.TestCase):
|
| all([x == y for x, y in zip(noise_samples, mix_neg_inf_samples)]))
|
|
|
| # Test target SNR 0.0 (different data expected).
|
| - mix_0 = signal_processing.SignalProcessingUtils.mix_signals(
|
| + mix_0 = signal_processing.SignalProcessingUtils.MixSignals(
|
| signal, noise, 0.0)
|
| self.assertTrue(len(signal), len(mix_0)) # Check duration.
|
| self.assertTrue(len(noise), len(mix_0))
|
| @@ -54,7 +56,7 @@ class TestSignalProcessing(unittest.TestCase):
|
| any([x != y for x, y in zip(noise_samples, mix_0_samples)]))
|
|
|
| # Test target SNR +Inf (signal expected).
|
| - mix_pos_inf = signal_processing.SignalProcessingUtils.mix_signals(
|
| + mix_pos_inf = signal_processing.SignalProcessingUtils.MixSignals(
|
| signal, noise, np.Inf)
|
| self.assertTrue(len(signal), len(mix_pos_inf)) # Check duration.
|
| mix_pos_inf_samples = mix_pos_inf.get_array_of_samples()
|
| @@ -63,13 +65,13 @@ class TestSignalProcessing(unittest.TestCase):
|
|
|
| def testMixSignalsMinInfPower(self):
|
| silence = pydub.AudioSegment.silent(duration=1000, frame_rate=48000)
|
| - signal = signal_processing.SignalProcessingUtils.generate_white_noise(
|
| + signal = signal_processing.SignalProcessingUtils.GenerateWhiteNoise(
|
| silence)
|
|
|
| with self.assertRaises(exceptions.SignalProcessingException):
|
| - _ = signal_processing.SignalProcessingUtils.mix_signals(
|
| + _ = signal_processing.SignalProcessingUtils.MixSignals(
|
| signal, silence, 0.0)
|
|
|
| with self.assertRaises(exceptions.SignalProcessingException):
|
| - _ = signal_processing.SignalProcessingUtils.mix_signals(
|
| + _ = signal_processing.SignalProcessingUtils.MixSignals(
|
| silence, signal, 0.0)
|
|
|