mirror of
https://github.com/deepset-ai/haystack.git
synced 2025-09-03 13:23:38 +00:00
165a0a5faa
* mock all translator tests and move one to e2e * typo * extract pipeline tests using translator * remove duplicate test * move generator test in e2e * Update e2e/pipelines/test_extractive_qa.py * pytest.mark.unit * black * remove model name as well * remove unused fixture * rename original and improve pipeline tests * fixes * pylint
14 lines
538 B
Python
14 lines
538 B
Python
from haystack import Document
|
|
from haystack.nodes import TransformersTranslator
|
|
|
|
|
|
def test_translator():
|
|
en_to_de_translator = TransformersTranslator(model_name_or_path="Helsinki-NLP/opus-mt-en-de")
|
|
|
|
original = "I live in Berlin"
|
|
translation = "Ich lebe in Berlin"
|
|
|
|
assert en_to_de_translator.translate(query=original) == translation
|
|
assert en_to_de_translator.translate(documents=[original])[0] == translation
|
|
assert en_to_de_translator.translate(documents=[Document(content=original)])[0].content == translation
|