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_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
queue_string = io_handler.read_queue_file() queue = queue_string.split(',') queue.remove('') logger.debug(f'QUEUE: {queue}') for model in models: logger.debug(f'CURRENT MODEL: {model}') if enhance and model == 'phone_call' or enhance and model == 'video': enhanced_runs = [True, False] else: enhanced_runs = [False] for use_enhanced in enhanced_runs: configuration.set_use_enhanced(use_enhanced) logger.debug(f'ENAHANCED: {use_enhanced}') if alternative_language_codes: alternative_runs = [True, False] else: alternative_runs = [False] for alt_run in alternative_runs: logger.debug(f'ALT RUN: {alt_run}') for audio in audio_list: root = utilities.get_root_filename(audio) #read reference if not only_transcribe: msg = f'READING: Reference file {cloud_store_uri}/{root}.txt'
def test_set_use_enhanced(): from model.configuration import Configuration config = Configuration() config.set_use_enhanced(True)