wmt-2020-pl-en/tests/test_utils.py

from googletrans import utils
from pytest import raises


def test_format_json():
    text = '[,,"en",,,,0.96954316,,[["en"],,[0.96954316]]]'

    result = utils.format_json(text)

    assert result == [None, None, 'en', None, None, None, 0.96954316, None,
                      [['en'], None, [0.96954316]]]


def test_format_malformed_json():
    text = '[,,"en",,,,0.96954316,,[["en"],,0.96954316]]]'

    with raises(ValueError):
        utils.format_json(text)


def test_rshift():
    value, n = 1000, 3

    result = utils.rshift(value, n)

    assert result == 125


def test_build_params_with_override():
    params = utils.build_params(
        query='',
        src='',
        dest='',
        token='',
        override={
            'otf': '3',
        },
    )

    assert params['otf'] == '3'