| # Copyright (c) 2017 The WebRTC project authors. All Rights Reserved. |
| # |
| # Use of this source code is governed by a BSD-style license |
| # that can be found in the LICENSE file in the root of the source |
| # tree. An additional intellectual property rights grant can be found |
| # in the file PATENTS. All contributing project authors may |
| # be found in the AUTHORS file in the root of the source tree. |
| |
| """Unit tests for the echo path simulation module. |
| """ |
| |
| import shutil |
| import os |
| import tempfile |
| import unittest |
| |
| import pydub |
| |
| from . import echo_path_simulation |
| from . import echo_path_simulation_factory |
| from . import signal_processing |
| |
| |
| class TestEchoPathSimulators(unittest.TestCase): |
| """Unit tests for the eval_scores module. |
| """ |
| |
| def setUp(self): |
| """Creates temporary data.""" |
| self._tmp_path = tempfile.mkdtemp() |
| |
| # Create and save white noise. |
| silence = pydub.AudioSegment.silent(duration=1000, frame_rate=48000) |
| white_noise = signal_processing.SignalProcessingUtils.GenerateWhiteNoise( |
| silence) |
| self._audio_track_num_samples = ( |
| signal_processing.SignalProcessingUtils.CountSamples(white_noise)) |
| self._audio_track_filepath = os.path.join(self._tmp_path, 'white_noise.wav') |
| signal_processing.SignalProcessingUtils.SaveWav( |
| self._audio_track_filepath, white_noise) |
| |
| # Make a copy the white noise audio track file; it will be used by |
| # echo_path_simulation.RecordedEchoPathSimulator. |
| shutil.copy(self._audio_track_filepath, os.path.join( |
| self._tmp_path, 'white_noise_echo.wav')) |
| |
| def tearDown(self): |
| """Recursively deletes temporary folders.""" |
| shutil.rmtree(self._tmp_path) |
| |
| def testRegisteredClasses(self): |
| # Check that there is at least one registered echo path simulator. |
| registered_classes = ( |
| echo_path_simulation.EchoPathSimulator.REGISTERED_CLASSES) |
| self.assertIsInstance(registered_classes, dict) |
| self.assertGreater(len(registered_classes), 0) |
| |
| # Instance factory. |
| factory = echo_path_simulation_factory.EchoPathSimulatorFactory() |
| |
| # Try each registered echo path simulator. |
| for echo_path_simulator_name in registered_classes: |
| simulator = factory.GetInstance( |
| echo_path_simulator_class=registered_classes[ |
| echo_path_simulator_name], |
| render_input_filepath=self._audio_track_filepath) |
| |
| echo_filepath = simulator.Simulate(self._tmp_path) |
| if echo_filepath is None: |
| self.assertEqual(echo_path_simulation.NoEchoPathSimulator.NAME, |
| echo_path_simulator_name) |
| # No other tests in this case. |
| continue |
| |
| # Check that the echo audio track file exists and its length is greater or |
| # equal to that of the render audio track. |
| self.assertTrue(os.path.exists(echo_filepath)) |
| echo = signal_processing.SignalProcessingUtils.LoadWav(echo_filepath) |
| self.assertGreaterEqual( |
| signal_processing.SignalProcessingUtils.CountSamples(echo), |
| self._audio_track_num_samples) |
