Index: webrtc/modules/audio_processing/test/py_quality_assessment/quality_assessment/test_noisegen.py |
diff --git a/webrtc/modules/audio_processing/test/py_quality_assessment/apm_quality_assessment-export.py b/webrtc/modules/audio_processing/test/py_quality_assessment/quality_assessment/test_noisegen.py |
similarity index 53% |
copy from webrtc/modules/audio_processing/test/py_quality_assessment/apm_quality_assessment-export.py |
copy to webrtc/modules/audio_processing/test/py_quality_assessment/quality_assessment/test_noisegen.py |
index 1b526c1b054918518b39370902d82a1fd53aff29..2d80ecbf7047cae5abb770b7a4510f247381533c 100644 |
--- a/webrtc/modules/audio_processing/test/py_quality_assessment/apm_quality_assessment-export.py |
+++ b/webrtc/modules/audio_processing/test/py_quality_assessment/quality_assessment/test_noisegen.py |
@@ -1,4 +1,3 @@ |
-#!/usr/bin/env python3 |
# Copyright (c) 2017 The WebRTC project authors. All Rights Reserved. |
# |
# Use of this source code is governed by a BSD-style license |
@@ -6,3 +5,15 @@ |
# 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. |
+ |
+import unittest |
+ |
+from . import noise_generation |
+ |
+class TestNoiseGen(unittest.TestCase): |
+ |
+ def test_registered_classes(self): |
+ # Check that there is at least one registered noise generator. |
+ classes = noise_generation.NoiseGenerator.REGISTERED_CLASSES |
+ self.assertIsInstance(classes, dict) |
+ self.assertGreater(len(classes), 0) |