import pytest

from api import translate


@pytest.mark.parametrize(
    "q,trans,source,target",
    [
        ("hi there\n", "hola.\n", "en", "es"),
        ("привет", "hi", "auto", "en")
    ]
)
def test_translate(q, trans, source, target):
    assert translate(q, source, target) == trans