def test_recognize(self): # Setup Expected Response expected_response = {} expected_response = cloud_speech_pb2.RecognizeResponse( **expected_response) # Mock the API response channel = ChannelStub(responses=[expected_response]) patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = speech_v1.SpeechClient() # Setup Request encoding = enums.RecognitionConfig.AudioEncoding.FLAC sample_rate_hertz = 44100 language_code = "en-US" config = { "encoding": encoding, "sample_rate_hertz": sample_rate_hertz, "language_code": language_code, } uri = "gs://bucket_name/file_name.flac" audio = {"uri": uri} response = client.recognize(config, audio) assert expected_response == response assert len(channel.requests) == 1 expected_request = cloud_speech_pb2.RecognizeRequest(config=config, audio=audio) actual_request = channel.requests[0][1] assert expected_request == actual_request
def test_streaming_recognize(self): # Setup Expected Response expected_response = {} expected_response = cloud_speech_pb2.StreamingRecognizeResponse( **expected_response) # Mock the API response channel = ChannelStub(responses=[iter([expected_response])]) patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = speech_v1.SpeechClient() # Setup Request request = {} request = cloud_speech_pb2.StreamingRecognizeRequest(**request) requests = [request] response = client.streaming_recognize(requests) resources = list(response) assert len(resources) == 1 assert expected_response == resources[0] assert len(channel.requests) == 1 actual_requests = channel.requests[0][1] assert len(actual_requests) == 1 actual_request = list(actual_requests)[0] assert request == actual_request
def test_streaming_recognize(self): # Setup Expected Response expected_response = {} expected_response = cloud_speech_pb2.StreamingRecognizeResponse( **expected_response) # Mock the API response channel = ChannelStub(responses=[iter([expected_response])]) client = speech_v1.SpeechClient(channel=channel) # Setup Request request = {} request = cloud_speech_pb2.StreamingRecognizeRequest(**request) requests = [request] response = client.streaming_recognize(requests) resources = list(response) assert len(resources) == 1 assert expected_response == resources[0] assert len(channel.requests) == 1 actual_requests = channel.requests[0][1] assert len(actual_requests) == 1 actual_request = list(actual_requests)[0] assert request == actual_request
def test_long_running_recognize_exception(self): # Setup Response error = status_pb2.Status() operation = operations_pb2.Operation( name="operations/test_long_running_recognize_exception", done=True) operation.error.CopyFrom(error) # Mock the API response channel = ChannelStub(responses=[operation]) patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = speech_v1.SpeechClient() # Setup Request encoding = enums.RecognitionConfig.AudioEncoding.FLAC sample_rate_hertz = 44100 language_code = "en-US" config = { "encoding": encoding, "sample_rate_hertz": sample_rate_hertz, "language_code": language_code, } uri = "gs://bucket_name/file_name.flac" audio = {"uri": uri} response = client.long_running_recognize(config, audio) exception = response.exception() assert exception.errors[0] == error
def test_recognize(self): # Setup Expected Response expected_response = {} expected_response = cloud_speech_pb2.RecognizeResponse( **expected_response) # Mock the API response channel = ChannelStub(responses=[expected_response]) client = speech_v1.SpeechClient(channel=channel) # Setup Request encoding = enums.RecognitionConfig.AudioEncoding.FLAC sample_rate_hertz = 44100 language_code = 'en-US' config = { 'encoding': encoding, 'sample_rate_hertz': sample_rate_hertz, 'language_code': language_code } uri = 'gs://bucket_name/file_name.flac' audio = {'uri': uri} response = client.recognize(config, audio) assert expected_response == response assert len(channel.requests) == 1 expected_request = cloud_speech_pb2.RecognizeRequest( config=config, audio=audio) actual_request = channel.requests[0][1] assert expected_request == actual_request
def test_long_running_recognize_exception(self): # Setup Response error = status_pb2.Status() operation = operations_pb2.Operation( name='operations/test_long_running_recognize_exception', done=True) operation.error.CopyFrom(error) # Mock the API response channel = ChannelStub(responses=[operation]) client = speech_v1.SpeechClient(channel=channel) # Setup Request encoding = enums.RecognitionConfig.AudioEncoding.FLAC sample_rate_hertz = 44100 language_code = 'en-US' config = { 'encoding': encoding, 'sample_rate_hertz': sample_rate_hertz, 'language_code': language_code } uri = 'gs://bucket_name/file_name.flac' audio = {'uri': uri} response = client.long_running_recognize(config, audio) exception = response.exception() assert exception.errors[0] == error
def test_long_running_recognize(self): # Setup Expected Response expected_response = {} expected_response = cloud_speech_pb2.LongRunningRecognizeResponse( **expected_response) operation = operations_pb2.Operation( name='operations/test_long_running_recognize', done=True) operation.response.Pack(expected_response) # Mock the API response channel = ChannelStub(responses=[operation]) client = speech_v1.SpeechClient(channel=channel) # Setup Request encoding = enums.RecognitionConfig.AudioEncoding.FLAC sample_rate_hertz = 44100 language_code = 'en-US' config = { 'encoding': encoding, 'sample_rate_hertz': sample_rate_hertz, 'language_code': language_code } uri = 'gs://bucket_name/file_name.flac' audio = {'uri': uri} response = client.long_running_recognize(config, audio) result = response.result() assert expected_response == result assert len(channel.requests) == 1 expected_request = cloud_speech_pb2.LongRunningRecognizeRequest( config=config, audio=audio) actual_request = channel.requests[0][1] assert expected_request == actual_request
def test_streaming_recognize_exception(self): # Mock the API response channel = ChannelStub(responses=[CustomException()]) client = speech_v1.SpeechClient(channel=channel) # Setup request request = {} request = cloud_speech_pb2.StreamingRecognizeRequest(**request) requests = [request] with pytest.raises(CustomException): client.streaming_recognize(requests)
def test_streaming_recognize_exception(self): # Mock the API response channel = ChannelStub(responses=[CustomException()]) patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = speech_v1.SpeechClient() # Setup request request = {} request = cloud_speech_pb2.StreamingRecognizeRequest(**request) requests = [request] with pytest.raises(CustomException): client.streaming_recognize(requests)
def test_recognize_exception(self): # Mock the API response channel = ChannelStub(responses=[CustomException()]) client = speech_v1.SpeechClient(channel=channel) # Setup request encoding = enums.RecognitionConfig.AudioEncoding.FLAC sample_rate_hertz = 44100 language_code = 'en-US' config = { 'encoding': encoding, 'sample_rate_hertz': sample_rate_hertz, 'language_code': language_code } uri = 'gs://bucket_name/file_name.flac' audio = {'uri': uri} with pytest.raises(CustomException): client.recognize(config, audio)
def test_recognize_exception(self): # Mock the API response channel = ChannelStub(responses=[CustomException()]) patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = speech_v1.SpeechClient() # Setup request encoding = enums.RecognitionConfig.AudioEncoding.FLAC sample_rate_hertz = 44100 language_code = "en-US" config = { "encoding": encoding, "sample_rate_hertz": sample_rate_hertz, "language_code": language_code, } uri = "gs://bucket_name/file_name.flac" audio = {"uri": uri} with pytest.raises(CustomException): client.recognize(config, audio)
def test_recognize_exception(self): # Mock the API response channel = ChannelStub(responses=[CustomException()]) patch = mock.patch('google.api_core.grpc_helpers.create_channel') with patch as create_channel: create_channel.return_value = channel client = speech_v1.SpeechClient() # Setup request encoding = enums.RecognitionConfig.AudioEncoding.FLAC sample_rate_hertz = 44100 language_code = 'en-US' config = { 'encoding': encoding, 'sample_rate_hertz': sample_rate_hertz, 'language_code': language_code } uri = 'gs://bucket_name/file_name.flac' audio = {'uri': uri} with pytest.raises(CustomException): client.recognize(config, audio)