| Index: webrtc/modules/audio_processing/test/py_quality_assessment/quality_assessment/test_data_generation.py
|
| diff --git a/webrtc/modules/audio_processing/test/py_quality_assessment/quality_assessment/test_data_generation.py b/webrtc/modules/audio_processing/test/py_quality_assessment/quality_assessment/test_data_generation.py
|
| index 9f4df941ad6e26f8cdb117debb44ee457f3192d8..76f82e00a97f8dd64aea7bf7137eaa6a8394494a 100644
|
| --- a/webrtc/modules/audio_processing/test/py_quality_assessment/quality_assessment/test_data_generation.py
|
| +++ b/webrtc/modules/audio_processing/test/py_quality_assessment/quality_assessment/test_data_generation.py
|
| @@ -297,10 +297,11 @@ class EnvironmentalNoiseTestDataGenerator(TestDataGenerator):
|
| _NOISY_SIGNAL_FILENAME_TEMPLATE = '{0}_{1:d}_SNR.wav'
|
|
|
| # TODO(alessiob): allow the user to store the noise tracks in a custom path.
|
| - _NOISE_TRACKS_PATH = os.path.join(os.getcwd(), 'noise_tracks')
|
| + _NOISE_TRACKS_PATH = os.path.join(
|
| + os.path.dirname(__file__), os.pardir, 'noise_tracks')
|
|
|
| - # TODO(alessiob): allow the user to have custom noise tracks.
|
| - # TODO(alessiob): exploit TestDataGeneratorFactory.GetInstance().
|
| + # TODO(alessiob): Allow the user to have custom noise tracks.
|
| + # TODO(alessiob): Exploit TestDataGeneratorFactory.GetInstance().
|
| _NOISE_TRACKS = [
|
| 'city.wav'
|
| ]
|
| @@ -436,7 +437,7 @@ class ReverberationTestDataGenerator(TestDataGenerator):
|
| # Load noise track.
|
| noise_signal = signal_processing.SignalProcessingUtils.LoadWav(
|
| noise_track_filepath)
|
| - except IOError: # File not found.
|
| + except exceptions.FileNotFoundError:
|
| # Generate noise track by applying the impulse response.
|
| impulse_response_filepath = os.path.join(
|
| self._aechen_ir_database_path,
|
|
|