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)