from multilang_translator.translator.llm_translator import translator_de_en
from multilang_translator.translator.llm_translator import translator_de_fr
import time
import logging as log

def test_translator():

    TESTSENTENCE_DE_BROKER = 'Ein Broker (oder Makler) ist eine Person oder ein Unternehmen, das sich zwischen dem Kauf- und Verkaufsberechtigten einer Wirtschaftsgüter (z.B. Aktien, Optionen, Derivate, Währungen, Rohstoffe usw.) stellt und als Vermittler fungiert. Sein Hauptziel ist es, Transaktionen zu erleichtern und Geld für sich selbst zu verdienen.'

    start=time.time()
    response = translator_de_en(TESTSENTENCE_DE_BROKER)

    log.info("First query took %s", time.time() - start) 
    log.info(response)
    assert 'A broker (or agent) is a person or company' in response

    start=time.time()
    response = translator_de_fr(TESTSENTENCE_DE_BROKER)
    log.info("Second query took %s", time.time() - start) 
    log.info(response)