示例#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
示例#2
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
    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'
示例#4
0
def test_set_use_enhanced():
    from model.configuration import Configuration
    config = Configuration()
    config.set_use_enhanced(True)