def test_end(_mock): client = CloudClient(socket_url="", key_id="", key_secret="") client.connect() client._socket.recv.side_effect = [ json.dumps( { "error": None, "final": False, "hypotheses": [{"confidence": 0.5, "transcript": "this is a test"}], "status": "ok", } ), json.dumps( { "error": None, "final": True, "hypotheses": [{"confidence": 0.5, "transcript": "this is a test"}], "status": "ok", } ), ] client.initialize() client.end() client.receive() assert client.is_final
def test_send_audio(_mock): client = CloudClient(socket_url="", key_id="", key_secret="") frame = np.random.rand(160).astype(np.int16) with pytest.raises(ConnectionError): client.send(frame) with pytest.raises(ConnectionError): client.end() client.connect() client.end() for i in range(100): client.send(frame) client.end()