Fix: mock pyaudio for ci to pass the server tests
This commit is contained in:
@@ -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()
|
||||||
|
|||||||
Reference in New Issue
Block a user