andras/great-ai
1
0
Fork 0
Code Issues Pull requests 1 Projects Releases 4 Packages Wiki Activity Actions

Reformat asserts

Browse source
This commit is contained in:
Andras Schmelczer 2022-06-25 10:39:31 +02:00
parent fcdd25c974
commit 2ee246d133
10 changed files with 77 additions and 77 deletions
Download patch file Download diff file Expand all files Collapse all files

24
great_ai/tests/utilities/test_language.py
View file

@ -9,11 +9,11 @@ from src.great_ai.utilities.language import (
class TestLanguage(unittest.TestCase):
def test_predict_language(self) -> None:
self.assertEqual(predict_language("This is an English text."), "en")
self.assertEqual(predict_language("Ez egy magyar szöveg."), "hu")
self.assertEqual(predict_language("此處按原典，應為「黃武元年」，而電子稿此處為「黃初元年」。"), "zh-TW")
self.assertEqual(predict_language("32"), "und")
self.assertEqual(predict_language(""), "und")
assert predict_language("This is an English text.") == "en"
assert predict_language("Ez egy magyar szöveg.") == "hu"
assert predict_language("此處按原典，應為「黃武元年」，而電子稿此處為「黃初元年」。") == "zh-TW"
assert predict_language("32") == "und"
assert predict_language("") == "und"
def test_is_english(self) -> None:
self.assertTrue(is_english("en"))
@ -27,10 +27,10 @@ class TestLanguage(unittest.TestCase):
self.assertFalse(is_english(None))
def english_name_of_language(self) -> None:
self.assertEqual(english_name_of_language("en"), "English")
self.assertEqual(english_name_of_language("hu"), "Hungarian")
self.assertEqual(english_name_of_language("zh"), "Chinese")
self.assertEqual(english_name_of_language("zh-TW"), "Chinese")
self.assertEqual(english_name_of_language("und"), "Unknown language")
self.assertEqual(english_name_of_language(""), "Unknown language")
self.assertEqual(english_name_of_language(None), "Unknown language")
assert english_name_of_language("en") == "English"
assert english_name_of_language("hu") == "Hungarian"
assert english_name_of_language("zh") == "Chinese"
assert english_name_of_language("zh-TW") == "Chinese"
assert english_name_of_language("und") == "Unknown language"
assert english_name_of_language("") == "Unknown language"
assert english_name_of_language(None) == "Unknown language"