wmt-2020-pl-en/tests.py

# -*- coding: utf-8 -*-
import unittest
from googletrans import translator, conversion

class TranslateTests(unittest.TestCase):
    def setUp(self):
        pass

    def test_to_json(self):
        text = '[,,"en",,,,0.96954316,,[["en"],,[0.96954316]]]'
        approx = [None, None, 'en', None, None, None, 0.96954316, None, [['en'], None, [0.96954316]]]
        assert conversion.format_json(text) == approx

    def test_latin_to_english(self):
        result = translator.translate('veritas lux mea', src='la', dest='en')
        assert result.text == 'The truth is my light'

    def test_unicode(self):
        result = translator.translate('안녕하세요.', src='ko', dest='ja')
        assert result.text == u'こんにちは。'

    def test_list_translation(self):
        translations = translator.translate(['The quick brown fox', 'jumps over', 'the lazy dog'],
            src='en', dest='ko')

        assert translations[0].text == u'빠른 갈색 여우'
        assert translations[1].text == u'이상 점프'
        assert translations[2].text == u'게으른 개'

    def test_language_detection(self):
        ko = translator.detect('한국어')
        assert ko.lang == 'ko'

        en = translator.detect('English')
        assert en.lang == 'en'

if __name__ == '__main__':
    unittest.main()