|
|
@@ -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()
|