Index: webrtc/modules/audio_processing/test/py_quality_assessment/quality_assessment/noise_generation_unittest.py |
diff --git a/webrtc/modules/audio_processing/test/py_quality_assessment/quality_assessment/noise_generation_unittest.py b/webrtc/modules/audio_processing/test/py_quality_assessment/quality_assessment/noise_generation_unittest.py |
index 2b750913acb660a7e64988c37f0e5cf25c3a2488..c5dfed2796a42a4d4b2f8c1c638c298ef40fb36f 100644 |
--- a/webrtc/modules/audio_processing/test/py_quality_assessment/quality_assessment/noise_generation_unittest.py |
+++ b/webrtc/modules/audio_processing/test/py_quality_assessment/quality_assessment/noise_generation_unittest.py |
@@ -20,6 +20,8 @@ from . import signal_processing |
class TestNoiseGen(unittest.TestCase): |
+ """Unit tests for the noise_generation module. |
+ """ |
def setUp(self): |
"""Create temporary folders.""" |
@@ -54,7 +56,7 @@ class TestNoiseGen(unittest.TestCase): |
self.assertTrue(os.path.exists(input_signal_filepath)) |
# Load input signal. |
- input_signal = signal_processing.SignalProcessingUtils.load_wav( |
+ input_signal = signal_processing.SignalProcessingUtils.LoadWav( |
input_signal_filepath) |
# Try each registered noise generator. |
@@ -72,7 +74,7 @@ class TestNoiseGen(unittest.TestCase): |
registered_classes[noise_generator_name]) |
# Generate the noisy input - reference pairs. |
- noise_generator.generate( |
+ noise_generator.Generate( |
input_signal_filepath=input_signal_filepath, |
input_noise_cache_path=self._input_noise_cache_path, |
base_output_path=self._base_output_path) |
@@ -92,7 +94,7 @@ class TestNoiseGen(unittest.TestCase): |
self.assertEqual(number_of_pairs, |
len(noise_generator.noisy_signal_filepaths)) |
self.assertEqual(number_of_pairs, |
- len(noise_generator.output_paths)) |
+ len(noise_generator.apm_output_paths)) |
self.assertEqual(number_of_pairs, |
len(noise_generator.reference_signal_filepaths)) |
@@ -108,30 +110,30 @@ class TestNoiseGen(unittest.TestCase): |
input_signal: AudioSegment instance. |
""" |
input_signal_length = ( |
- signal_processing.SignalProcessingUtils.count_samples(input_signal)) |
+ signal_processing.SignalProcessingUtils.CountSamples(input_signal)) |
# Iterate over the noisy signal - reference pairs. |
for noise_config_name in noise_generator.config_names: |
# Load the noisy input file. |
noisy_signal_filepath = noise_generator.noisy_signal_filepaths[ |
noise_config_name] |
- noisy_signal = signal_processing.SignalProcessingUtils.load_wav( |
+ noisy_signal = signal_processing.SignalProcessingUtils.LoadWav( |
noisy_signal_filepath) |
# Check noisy input signal length. |
noisy_signal_length = ( |
- signal_processing.SignalProcessingUtils.count_samples(noisy_signal)) |
+ signal_processing.SignalProcessingUtils.CountSamples(noisy_signal)) |
self.assertGreaterEqual(noisy_signal_length, input_signal_length) |
# Load the reference file. |
reference_signal_filepath = ( |
noise_generator.reference_signal_filepaths[noise_config_name]) |
- reference_signal = signal_processing.SignalProcessingUtils.load_wav( |
+ reference_signal = signal_processing.SignalProcessingUtils.LoadWav( |
reference_signal_filepath) |
# Check noisy input signal length. |
reference_signal_length = ( |
- signal_processing.SignalProcessingUtils.count_samples( |
+ signal_processing.SignalProcessingUtils.CountSamples( |
reference_signal)) |
self.assertGreaterEqual(reference_signal_length, input_signal_length) |
@@ -143,5 +145,5 @@ class TestNoiseGen(unittest.TestCase): |
""" |
# Iterate over the noisy signal - reference pairs. |
for noise_config_name in noise_generator.config_names: |
- output_path = noise_generator.output_paths[noise_config_name] |
+ output_path = noise_generator.apm_output_paths[noise_config_name] |
self.assertTrue(os.path.exists(output_path)) |