add pyaudio mock

2024-02-08 11:12:57 +05:30
parent 567ceb1246
commit 98399707a3
1 changed files with 5 additions and 1 deletions
@@ -77,7 +77,11 @@ class TestServerInferenceAccuracy(unittest.TestCase):
        cls.server_process.terminate()
        cls.server_process.wait()
    
-    def setUp(self):
+    @mock.patch('pyaudio.PyAudio')
+    def setUp(self, mock_pyaudio):
+        self.mock_pyaudio = mock_pyaudio.return_value
+        self.mock_stream = mock.MagicMock()
+        self.mock_pyaudio.open.return_value = self.mock_stream
        self.metric = evaluate.load("wer")
        self.normalizer = EnglishTextNormalizer()
        self.client  = TranscriptionClient(