summaryrefslogtreecommitdiffstats
path: root/etc/unittest/main.py
diff options
context:
space:
mode:
authorHeiner Lohaus <hlohaus@users.noreply.github.com>2024-01-20 18:16:45 +0100
committerHeiner Lohaus <hlohaus@users.noreply.github.com>2024-01-20 18:16:45 +0100
commit782edbdb6a0bea10b48deedd4a4d3f3afdf81a7a (patch)
tree96d449ae8ec7398431cce339aa1a394f0b9b953b /etc/unittest/main.py
parentAdd copilot github action (diff)
downloadgpt4free-782edbdb6a0bea10b48deedd4a4d3f3afdf81a7a.tar
gpt4free-782edbdb6a0bea10b48deedd4a4d3f3afdf81a7a.tar.gz
gpt4free-782edbdb6a0bea10b48deedd4a4d3f3afdf81a7a.tar.bz2
gpt4free-782edbdb6a0bea10b48deedd4a4d3f3afdf81a7a.tar.lz
gpt4free-782edbdb6a0bea10b48deedd4a4d3f3afdf81a7a.tar.xz
gpt4free-782edbdb6a0bea10b48deedd4a4d3f3afdf81a7a.tar.zst
gpt4free-782edbdb6a0bea10b48deedd4a4d3f3afdf81a7a.zip
Diffstat (limited to '')
-rw-r--r--etc/unittest/main.py9
1 files changed, 5 insertions, 4 deletions
diff --git a/etc/unittest/main.py b/etc/unittest/main.py
index 4b92a6a6..ad1fe02d 100644
--- a/etc/unittest/main.py
+++ b/etc/unittest/main.py
@@ -39,10 +39,11 @@ class TestBackendApi(unittest.TestCase):
class TestChatCompletion(unittest.TestCase):
- def test_create(self):
+ def test_create_default(self):
messages = [{'role': 'user', 'content': 'Hello'}]
result = ChatCompletion.create(g4f.models.default, messages)
- self.assertTrue("Hello" in result or "Good" in result)
+ if "Good" not in result and "Hi" not in result:
+ self.assertIn("Hello", result)
def test_get_last_provider(self):
messages = [{'role': 'user', 'content': 'Hello'}]
@@ -53,14 +54,14 @@ class TestChatCompletion(unittest.TestCase):
messages = [{'role': 'user', 'content': 'Hello'}]
provider = g4f.Provider.Bing
result = ChatCompletion.create(g4f.models.default, messages, provider)
- self.assertTrue("Bing" in result)
+ self.assertIn("Bing", result)
class TestChatCompletionAsync(unittest.IsolatedAsyncioTestCase):
async def test_async(self):
messages = [{'role': 'user', 'content': 'Hello'}]
result = await ChatCompletion.create_async(g4f.models.default, messages, MockProvider)
- self.assertTrue("Mock" in result)
+ self.assertEqual("Mock", result)
class TestUtilityFunctions(unittest.TestCase):