tests/test_linguee.py

#!/usr/bin/env python

"""Tests for `deep_translator` package."""

import pytest

from deep_translator import LingueeTranslator, exceptions


@pytest.fixture
def linguee():
    return LingueeTranslator(source="english", target="german")


def test_content(linguee):
    """Sample pytest test function with the pytest fixture as an argument."""
    # from bs4 import BeautifulSoup
    # assert 'GitHub' in BeautifulSoup(response.content).title.string
    assert linguee.translate(word="good") is not None


def test_inputs():
    with pytest.raises(exceptions.InvalidSourceOrTargetLanguage):
        LingueeTranslator(source="", target="")

    with pytest.raises(exceptions.InvalidSourceOrTargetLanguage):
        LingueeTranslator(source="auto", target="")

    with pytest.raises(exceptions.InvalidSourceOrTargetLanguage):
        LingueeTranslator(source="", target="en")

    l1 = LingueeTranslator("en", "fr")
    l2 = LingueeTranslator("english", "french")
    assert l1._source == l2._source
    assert l1._target == l2._target


def test_payload(linguee):

    with pytest.raises(exceptions.NotValidPayload):
        linguee.translate(123)

    with pytest.raises(exceptions.NotValidPayload):
        linguee.translate({})

    with pytest.raises(exceptions.NotValidPayload):
        linguee.translate([])

    with pytest.raises(exceptions.NotValidLength):
        linguee.translate("a" * 51)