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

remove whisper_live from patch to mock websocket

Browse Source
This commit is contained in:
makaveli10
2024-02-07 23:47:29 +05:30
parent ca2634bbb6
commit 84a97f5fdd
2 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_client.py
+3 -3
View File
@@ -8,7 +8,7 @@ from whisper_live.client import TranscriptionClient, resample
class TestClientWebSocketCommunication(unittest.TestCase):
@patch('whisper_live.client.websocket.WebSocketApp')
@patch('websocket.WebSocketApp')
def test_websocket_communication(self, mock_websocket):
mock_ws_instance = MagicMock()
mock_websocket.return_value = mock_ws_instance
@@ -23,7 +23,7 @@ class TestClientWebSocketCommunication(unittest.TestCase):
class TestClientCallbacks(unittest.TestCase):
@patch('whisper_live.client.websocket.WebSocketApp')
@patch('websocket.WebSocketApp')
def setUp(self, mock_websocket):
self.mock_ws_app = mock_websocket.return_value
self.mock_ws_app.send = MagicMock()
@@ -97,7 +97,7 @@ class TestAudioResampling(unittest.TestCase):
class TestSendingAudioPacket(unittest.TestCase):
@patch('whisper_live.client.websocket.WebSocketApp')
@patch('websocket.WebSocketApp')
def setUp(self, mock_websocket):
self.transcription_client = TranscriptionClient("localhost", "9090")
self.client = self.transcription_client.client