Example #1
0
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
Example #2
0
def test_set_get_speech_context():
    from model.configuration import Configuration
    config = Configuration()
    expected_boost = 3
    expected_phrases = ['foo', 'bar', 'baz']
    config.set_speech_context(expected_phrases, expected_boost)
    result = config.get_speech_context()
    assert result[0]['boost'] == expected_boost
    assert result[0]['phrases'] == expected_phrases
Example #3
0
def test_create_unique_root_2():
    from utilities.utilities import Utilities
    from model.configuration import Configuration
    from model.nlp import NLPModel
    u = Utilities()
    configuration = Configuration()
    nlp_model = NLPModel()
    root = '12345'
    configuration.set_model('video')
    configuration.set_use_enhanced(False)
    configuration.set_language_code('fr_FR')
    configuration.set_alternative_language_codes(['en-US', 'ru-RU'])
    configuration.set_speech_context('hi', 5)
    nlp_model.set_remove_stop_words(True)
    nlp_model.set_apply_stemming(False)
    nlp_model.set_expand_contractions(True)
    nlp_model.set_n2w(True)
    result = u.create_unique_root(root, configuration, nlp_model)
    expected = '12345_video_fr_FR_alts_applied_speech_adaptation_applied_boost_5_stop_words_removed_contractions_expanded_numbers_converted_2_words'
    assert result == expected
                    if not local_files_path:
                        ref = gcs.read_ref(cloud_store_uri, root + '.txt')
                    else:
                        ref = io_handler.read_file(local_files_path + '/' + root + '.txt')
                    logger.debug(f'INIT REF: {ref}')
                    #for speech_run in speech_context_runs:
                    for boost in boosts:
                        for language in language_codes:
                            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}')