Index: webrtc/modules/audio_processing/test/py_quality_assessment/quality_assessment/test_data_generation_unittest.py |
diff --git a/webrtc/modules/audio_processing/test/py_quality_assessment/quality_assessment/test_data_generation_unittest.py b/webrtc/modules/audio_processing/test/py_quality_assessment/quality_assessment/test_data_generation_unittest.py |
index 0bf7e1ab98595405d6bc7349b25939dfa7856a63..6239d516b0fbc67ca62f6228292ea1c0a6efa7be 100644 |
--- a/webrtc/modules/audio_processing/test/py_quality_assessment/quality_assessment/test_data_generation_unittest.py |
+++ b/webrtc/modules/audio_processing/test/py_quality_assessment/quality_assessment/test_data_generation_unittest.py |
@@ -52,6 +52,25 @@ class TestTestDataGenerators(unittest.TestCase): |
shutil.rmtree(self._test_data_cache_path) |
shutil.rmtree(self._fake_air_db_path) |
+ def testInputSignalCreation(self): |
+ # Init. |
+ generator = test_data_generation.IdentityTestDataGenerator('tmp') |
+ input_signal_filepath = os.path.join( |
+ self._test_data_cache_path, 'pure_tone-440_1000.wav') |
+ |
+ # Check that the input signal is generated. |
+ self.assertFalse(os.path.exists(input_signal_filepath)) |
+ generator.Generate( |
+ input_signal_filepath=input_signal_filepath, |
+ test_data_cache_path=self._test_data_cache_path, |
+ base_output_path=self._base_output_path) |
+ self.assertTrue(os.path.exists(input_signal_filepath)) |
+ |
+ # Check input signal properties. |
+ input_signal = signal_processing.SignalProcessingUtils.LoadWav( |
+ input_signal_filepath) |
+ self.assertEqual(1000, len(input_signal)) |
+ |
def testTestDataGenerators(self): |
# Preliminary check. |
self.assertTrue(os.path.exists(self._base_output_path)) |