From 782edbdb6a0bea10b48deedd4a4d3f3afdf81a7a Mon Sep 17 00:00:00 2001 From: Heiner Lohaus Date: Sat, 20 Jan 2024 18:16:45 +0100 Subject: Improve asserts in unittests --- etc/unittest/main.py | 9 +++++---- 1 file 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): -- cgit v1.2.3