def test_set_then_get_sample_rate_hertz():
from model.configuration import Configuration
config = Configuration()
expected = 16000
config.set_sample_rate_hertz(expected)
result = config.get_sample_rate_hertz()
assert result == expected
def test_str():
from model.configuration import Configuration
config = Configuration()
expected_model = 'phone_call'
expected_language_code = 'hi-IN'
expected_use_enhanced = True
expected_sample_rate_hertz = 48000
expected_audio_channel_count = 5
expected_enable_separate_recognition_per_channel = False
expected_boost = 6
expected_phrases = ['testing', '$ADDRESSNUM']
config.set_model(expected_model)
config.set_language_code(expected_language_code)
config.set_enable_separate_recognition_per_channel(
expected_enable_separate_recognition_per_channel)
config.set_audio_channel_count(expected_audio_channel_count)
config.set_use_enhanced(expected_use_enhanced)
config.set_sample_rate_hertz(expected_sample_rate_hertz)
config.set_speech_context(expected_phrases, expected_boost)
result = config.__str__()
assert isinstance(result, str)
assert expected_model in result
assert expected_language_code in result
assert str(expected_use_enhanced) in result
assert str(expected_sample_rate_hertz) in result
assert str(expected_audio_channel_count) in result
assert str(expected_enable_separate_recognition_per_channel) in result
assert str(expected_boost) in result
def test_get_hypothesis():
from model.configuration import Configuration
from utilities.speech_to_text import SpeechToText
uri = 'gs://brb/test_audio_n_truth/1.wav'
configuration_object = Configuration()
configuration_object.set_language_code('en-US')
configuration_object.set_encoding('LINEAR16')
configuration_object.set_sample_rate_hertz(44100)
configuration_object.set_model('default')
speech = SpeechToText()
result = speech.get_hypothesis(uri, configuration_object)
expected = ' testing 1 2 3 hello hello testing one two three'
assert result == expected
if boost > 0:
string = f'Running with phrase hints, boost {boost}'
else:
string = 'No speech context applied'
print(string)
logger.debug(string)
if phrases:
configuration.set_speech_context(phrases, boost)
else:
configuration.set_speech_context([], 0)
if alt_run:
configuration.set_alternative_language_codes(alternative_language_codes)
configuration.set_model(model)
configuration.set_sample_rate_hertz(sample_rate_hertz)
configuration.set_language_code(language)
configuration.set_encoding(encoding)
if audio_channel_count > 1:
configuration.set_audio_channel_count(audio_channel_count)
configuration.set_enable_separate_recognition_per_channel(True)
logger.debug(f'CONFIGURATION: {configuration}')
print(f'STARTING')
msg = f'audio: {audio}, {configuration}'
logger.debug(msg)
print(msg)
# Generate hyp
def test_set_sample_rate_hertz():
from model.configuration import Configuration
config = Configuration()
config.set_sample_rate_hertz(16000)