返回提交历史
Modified
etc/unittest/main.py
+5
-4
XFEstudio/gpt4free
Improve asserts in unittests
782edbdb
代码差异
1 个文件
+5
-4
@@ -39,10 +39,11 @@ class TestBackendApi(unittest.TestCase):
39
39
40
40
class TestChatCompletion(unittest.TestCase):
41
41
42
def test_create(self):
42
def test_create_default(self):
43
43
messages = [{'role': 'user', 'content': 'Hello'}]
44
44
result = ChatCompletion.create(g4f.models.default, messages)
45
self.assertTrue("Hello" in result or "Good" in result)
45
if "Good" not in result and "Hi" not in result:
46
self.assertIn("Hello", result)
46
47
47
48
def test_get_last_provider(self):
48
49
messages = [{'role': 'user', 'content': 'Hello'}]
@@ -53,14 +54,14 @@ class TestChatCompletion(unittest.TestCase):
53
54
messages = [{'role': 'user', 'content': 'Hello'}]
54
55
provider = g4f.Provider.Bing
55
56
result = ChatCompletion.create(g4f.models.default, messages, provider)
56
self.assertTrue("Bing" in result)
57
self.assertIn("Bing", result)
57
58
58
59
class TestChatCompletionAsync(unittest.IsolatedAsyncioTestCase):
59
60
60
61
async def test_async(self):
61
62
messages = [{'role': 'user', 'content': 'Hello'}]
62
63
result = await ChatCompletion.create_async(g4f.models.default, messages, MockProvider)
63
self.assertTrue("Mock" in result)
64
self.assertEqual("Mock", result)
64
65
65
66
class TestUtilityFunctions(unittest.TestCase):
66
67