cjones/WhisperLive
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

add pyaudio mock

Browse Source
This commit is contained in:
makaveli10
2024-02-08 11:12:57 +05:30
parent 567ceb1246
commit 98399707a3
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_server.py
+5 -1
View File
@@ -77,7 +77,11 @@ class TestServerInferenceAccuracy(unittest.TestCase):
cls.server_process.terminate() cls.server_process.terminate()
cls.server_process.wait() 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.metric = evaluate.load("wer")
self.normalizer = EnglishTextNormalizer() self.normalizer = EnglishTextNormalizer()
self.client = TranscriptionClient( self.client = TranscriptionClient(