Index: webrtc/modules/audio_processing/test/py_quality_assessment/apm_quality_assessment_unittest.py |
diff --git a/webrtc/modules/audio_processing/test/py_quality_assessment/apm_quality_assessment_unittest.py b/webrtc/modules/audio_processing/test/py_quality_assessment/apm_quality_assessment_unittest.py |
index dd0eb073830f7f5207955586702c2e0b928fafee..f5240f86965234cbe11c463e928fbbe7a912157f 100644 |
--- a/webrtc/modules/audio_processing/test/py_quality_assessment/apm_quality_assessment_unittest.py |
+++ b/webrtc/modules/audio_processing/test/py_quality_assessment/apm_quality_assessment_unittest.py |
@@ -9,8 +9,16 @@ |
"""Unit tests for the apm_quality_assessment module. |
""" |
+import os |
+import sys |
import unittest |
+SRC = os.path.abspath(os.path.join( |
+ os.path.dirname((__file__)), os.pardir, os.pardir, os.pardir)) |
+sys.path.append(os.path.join(SRC, 'third_party', 'pymock')) |
+ |
+import mock |
+ |
import apm_quality_assessment |
class TestSimulationScript(unittest.TestCase): |
@@ -19,6 +27,7 @@ class TestSimulationScript(unittest.TestCase): |
def testMain(self): |
# Exit with error code if no arguments are passed. |
- with self.assertRaises(SystemExit) as cm: |
+ with self.assertRaises(SystemExit) as cm, mock.patch.object( |
+ sys, 'argv', ['apm_quality_assessment.py']): |
apm_quality_assessment.main() |
self.assertGreater(cm.exception.code, 0) |