From 98399707a3aac729c693035a29cf5a335c619886 Mon Sep 17 00:00:00 2001 From: makaveli10 Date: Thu, 8 Feb 2024 11:12:57 +0530 Subject: [PATCH] add pyaudio mock --- tests/test_server.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/tests/test_server.py b/tests/test_server.py index e72f646..34c1288 100644 --- a/tests/test_server.py +++ b/tests/test_server.py @@ -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(