| 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 909d7bad39c266c30473c192bf517d6f3e28cbf8..0bf7e1ab98595405d6bc7349b25939dfa7856a63 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
|
| @@ -29,7 +29,7 @@ class TestTestDataGenerators(unittest.TestCase):
|
| def setUp(self):
|
| """Create temporary folders."""
|
| self._base_output_path = tempfile.mkdtemp()
|
| - self._input_noise_cache_path = tempfile.mkdtemp()
|
| + self._test_data_cache_path = tempfile.mkdtemp()
|
| self._fake_air_db_path = tempfile.mkdtemp()
|
|
|
| # Fake AIR DB impulse responses.
|
| @@ -49,13 +49,13 @@ class TestTestDataGenerators(unittest.TestCase):
|
| def tearDown(self):
|
| """Recursively delete temporary folders."""
|
| shutil.rmtree(self._base_output_path)
|
| - shutil.rmtree(self._input_noise_cache_path)
|
| + shutil.rmtree(self._test_data_cache_path)
|
| shutil.rmtree(self._fake_air_db_path)
|
|
|
| def testTestDataGenerators(self):
|
| # Preliminary check.
|
| self.assertTrue(os.path.exists(self._base_output_path))
|
| - self.assertTrue(os.path.exists(self._input_noise_cache_path))
|
| + self.assertTrue(os.path.exists(self._test_data_cache_path))
|
|
|
| # Check that there is at least one registered test data generator.
|
| registered_classes = (
|
| @@ -66,6 +66,7 @@ class TestTestDataGenerators(unittest.TestCase):
|
| # Instance generators factory.
|
| generators_factory = (
|
| test_data_generation_factory.TestDataGeneratorFactory(
|
| + output_directory_prefix='datagen-',
|
| aechen_ir_database_path=self._fake_air_db_path))
|
|
|
| # Use a sample input file as clean input signal.
|
| @@ -86,7 +87,7 @@ class TestTestDataGenerators(unittest.TestCase):
|
| # Generate the noisy input - reference pairs.
|
| generator.Generate(
|
| input_signal_filepath=input_signal_filepath,
|
| - input_noise_cache_path=self._input_noise_cache_path,
|
| + test_data_cache_path=self._test_data_cache_path,
|
| base_output_path=self._base_output_path)
|
|
|
| # Perform checks.
|
|
|