Explorar o código

chore : format

hellofinch hai 1 ano
pai
achega
a0a832276d
Modificáronse 1 ficheiros con 21 adicións e 5 borrados
  1. 21 5
      test/test_translator.py

+ 21 - 5
test/test_translator.py

@@ -73,6 +73,7 @@ class TestTranslator(unittest.TestCase):
         with self.assertRaises(NotImplementedError):
             translator.translate("Hello World")
 
+
 class TestOpenAIlikedTranslator(unittest.TestCase):
     def setUp(self) -> None:
         self.default_envs = {
@@ -80,10 +81,13 @@ class TestOpenAIlikedTranslator(unittest.TestCase):
             "OPENAILIKED_API_KEY": "test_api_key",
             "OPENAILIKED_MODEL": "test_model",
         }
+
     def test_missing_base_url_raises_error(self):
         """测试缺失 OPENAILIKED_BASE_URL 时抛出异常"""
         with self.assertRaises(ValueError) as context:
-            OpenAIlikedTranslator(lang_in="en", lang_out="zh", model="test_model", envs={})
+            OpenAIlikedTranslator(
+                lang_in="en", lang_out="zh", model="test_model", envs={}
+            )
         self.assertIn("The OPENAILIKED_BASE_URL is missing.", str(context.exception))
 
     def test_missing_model_raises_error(self):
@@ -93,7 +97,9 @@ class TestOpenAIlikedTranslator(unittest.TestCase):
             "OPENAILIKED_API_KEY": "test_api_key",
         }
         with self.assertRaises(ValueError) as context:
-            OpenAIlikedTranslator(lang_in="en", lang_out="zh", model=None, envs=envs_without_model)
+            OpenAIlikedTranslator(
+                lang_in="en", lang_out="zh", model=None, envs=envs_without_model
+            )
         self.assertIn("The OPENAILIKED_MODEL is missing.", str(context.exception))
 
     def test_initialization_with_valid_envs(self):
@@ -104,8 +110,14 @@ class TestOpenAIlikedTranslator(unittest.TestCase):
             model=None,
             envs=self.default_envs,
         )
-        self.assertEqual(translator.envs["OPENAILIKED_BASE_URL"], self.default_envs["OPENAILIKED_BASE_URL"])
-        self.assertEqual(translator.envs["OPENAILIKED_API_KEY"], self.default_envs["OPENAILIKED_API_KEY"])
+        self.assertEqual(
+            translator.envs["OPENAILIKED_BASE_URL"],
+            self.default_envs["OPENAILIKED_BASE_URL"],
+        )
+        self.assertEqual(
+            translator.envs["OPENAILIKED_API_KEY"],
+            self.default_envs["OPENAILIKED_API_KEY"],
+        )
         self.assertEqual(translator.model, self.default_envs["OPENAILIKED_MODEL"])
 
     def test_default_api_key_fallback(self):
@@ -120,8 +132,12 @@ class TestOpenAIlikedTranslator(unittest.TestCase):
             model=None,
             envs=envs_without_key,
         )
-        self.assertEqual(translator.envs["OPENAILIKED_BASE_URL"], self.default_envs["OPENAILIKED_BASE_URL"])
+        self.assertEqual(
+            translator.envs["OPENAILIKED_BASE_URL"],
+            self.default_envs["OPENAILIKED_BASE_URL"],
+        )
         self.assertEqual(translator.envs["OPENAILIKED_API_KEY"], None)
 
+
 if __name__ == "__main__":
     unittest.main()