Index: webrtc/modules/audio_processing/test/py_quality_assessment/quality_assessment/test_data_generation_factory.py |
diff --git a/webrtc/modules/audio_processing/test/py_quality_assessment/quality_assessment/test_data_generation_factory.py b/webrtc/modules/audio_processing/test/py_quality_assessment/quality_assessment/test_data_generation_factory.py |
index b5193f2cc5503d99adc3e235962b165a084dfe0a..b42d3af273ab0e50ad3cc6f6f7d9830f760fc083 100644 |
--- a/webrtc/modules/audio_processing/test/py_quality_assessment/quality_assessment/test_data_generation_factory.py |
+++ b/webrtc/modules/audio_processing/test/py_quality_assessment/quality_assessment/test_data_generation_factory.py |
@@ -21,7 +21,8 @@ class TestDataGeneratorFactory(object): |
generators will be produced. |
""" |
- def __init__(self, aechen_ir_database_path): |
+ def __init__(self, output_directory_prefix, aechen_ir_database_path): |
+ self._output_directory_prefix = output_directory_prefix |
self._aechen_ir_database_path = aechen_ir_database_path |
def GetInstance(self, test_data_generators_class): |
@@ -30,12 +31,14 @@ class TestDataGeneratorFactory(object): |
Args: |
test_data_generators_class: TestDataGenerator class object (not an |
instance). |
+ |
+ Returns: |
+ TestDataGenerator instance. |
""" |
logging.debug('factory producing %s', test_data_generators_class) |
if test_data_generators_class == ( |
test_data_generation.ReverberationTestDataGenerator): |
return test_data_generation.ReverberationTestDataGenerator( |
- self._aechen_ir_database_path) |
+ self._output_directory_prefix, self._aechen_ir_database_path) |
else: |
- # By default, no arguments in the constructor. |
- return test_data_generators_class() |
+ return test_data_generators_class(self._output_directory_prefix) |