Fix: mock pyaudio for ci to pass the server tests

This commit is contained in:
makaveli
2024-03-15 12:29:32 +05:30
committed by GitHub
parent 37d7f2ed66
commit 7c7a446478
+6 -5
View File
@@ -69,6 +69,10 @@ class TestServerConnection(unittest.TestCase):
class TestServerInferenceAccuracy(unittest.TestCase): class TestServerInferenceAccuracy(unittest.TestCase):
@classmethod @classmethod
def setUpClass(cls): def setUpClass(cls):
cls.mock_pyaudio_patch = mock.patch('pyaudio.PyAudio')
cls.mock_pyaudio = cls.mock_pyaudio_patch.start()
cls.mock_pyaudio.return_value.open.return_value = mock.MagicMock()
cls.server_process = subprocess.Popen(["python", "run_server.py"]) cls.server_process = subprocess.Popen(["python", "run_server.py"])
time.sleep(2) time.sleep(2)
@@ -77,11 +81,7 @@ class TestServerInferenceAccuracy(unittest.TestCase):
cls.server_process.terminate() cls.server_process.terminate()
cls.server_process.wait() cls.server_process.wait()
@mock.patch('pyaudio.PyAudio') def setUp(self):
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.metric = evaluate.load("wer")
self.normalizer = EnglishTextNormalizer() self.normalizer = EnglishTextNormalizer()
@@ -117,6 +117,7 @@ class TestServerInferenceAccuracy(unittest.TestCase):
self.check_prediction("transcript1.srt") self.check_prediction("transcript1.srt")
self.check_prediction("transcript2.srt") self.check_prediction("transcript2.srt")
class TestExceptionHandling(unittest.TestCase): class TestExceptionHandling(unittest.TestCase):
def setUp(self): def setUp(self):
self.server = TranscriptionServer() self.server = TranscriptionServer()