Update test_translator.py

sioaeko · Jun 30, 2024 · a3bab78 · a3bab78
1 parent a23590f
commit a3bab78
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/tests/test_translator.py b/tests/test_translator.py
@@ -3,9 +3,17 @@
 
 class TestTranslator(unittest.TestCase):
 
-    def test_translation(self):
-        result = translate("Hello, world!", "en", "fr")
-        self.assertEqual(result, "Bonjour, le monde!")
+    def test_english_to_french(self):
+        result = translate("Hello, world!", "eng_Latn", "fra_Latn")
+        self.assertIn("Bonjour", result)
+
+    def test_french_to_english(self):
+        result = translate("Bonjour le monde", "fra_Latn", "eng_Latn")
+        self.assertIn("Hello", result)
+
+    def test_error_handling(self):
+        result = translate("Test", "invalid_lang", "eng_Latn")
+        self.assertIn("error", result.lower())
 
 if __name__ == '__main__':
     unittest.main()