32 lines
1006 B
Python
32 lines
1006 B
Python
from multilang_translator.translator.llm_translator import translator_de_en, translator_de_fr, translator_de_it
|
|
from multilang_translator.translator.test_content import TESTSENTENCE_DE_BROKER, TESTSENTENCE_DE_RAINBOW
|
|
from multilang_translator.main_local import translate_from_german
|
|
|
|
|
|
import time
|
|
import logging as log
|
|
|
|
def test_translator():
|
|
|
|
|
|
start=time.time()
|
|
response = translator_de_en(TESTSENTENCE_DE_RAINBOW)
|
|
|
|
log.info("First query took %s", time.time() - start)
|
|
log.info(response)
|
|
assert 'The rainbow is an atmospheric optical phenomenon' in response
|
|
|
|
start=time.time()
|
|
response = translator_de_fr(TESTSENTENCE_DE_RAINBOW)
|
|
log.info("Second query took %s", time.time() - start)
|
|
log.info(response)
|
|
|
|
start=time.time()
|
|
response = translator_de_it(TESTSENTENCE_DE_RAINBOW)
|
|
log.info("Second query took %s", time.time() - start)
|
|
log.info(response)
|
|
|
|
def test_translate_from_german_and_encode():
|
|
|
|
translate_from_german(TESTSENTENCE_DE_RAINBOW)
|