diff options
| author | H Lohaus <hlohaus@users.noreply.github.com> | 2024-01-23 12:18:32 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-01-23 12:18:32 +0100 |
| commit | 984c3e1f891b1242954413d7abf81d0a45aa1110 (patch) | |
| tree | 1f1bee0d403b70866b2bf5f71cfc5a0fb2fe4efd /etc/unittest/main.py | |
| parent | Added CodeLlama and misc models from DeepInfra (#1506) (diff) | |
| parent | Fix: Model sometimes not converted #1507 (diff) | |
| download | gpt4free-984c3e1f891b1242954413d7abf81d0a45aa1110.tar gpt4free-984c3e1f891b1242954413d7abf81d0a45aa1110.tar.gz gpt4free-984c3e1f891b1242954413d7abf81d0a45aa1110.tar.bz2 gpt4free-984c3e1f891b1242954413d7abf81d0a45aa1110.tar.lz gpt4free-984c3e1f891b1242954413d7abf81d0a45aa1110.tar.xz gpt4free-984c3e1f891b1242954413d7abf81d0a45aa1110.tar.zst gpt4free-984c3e1f891b1242954413d7abf81d0a45aa1110.zip | |
Diffstat (limited to 'etc/unittest/main.py')
| -rw-r--r-- | etc/unittest/main.py | 8 |
1 files changed, 3 insertions, 5 deletions
diff --git a/etc/unittest/main.py b/etc/unittest/main.py index 9b9ca011..f5eb5138 100644 --- a/etc/unittest/main.py +++ b/etc/unittest/main.py @@ -1,4 +1,3 @@ -from .include import DEFAULT_MESSAGES import unittest import asyncio import g4f @@ -6,6 +5,8 @@ from g4f import ChatCompletion, get_last_provider from g4f.Provider import RetryProvider from .mocks import ProviderMock +DEFAULT_MESSAGES = [{'role': 'user', 'content': 'Hello'}] + class NoTestChatCompletion(unittest.TestCase): def no_test_create_default(self): @@ -31,7 +32,4 @@ class TestGetLastProvider(unittest.TestCase): def test_get_last_provider_async(self): coroutine = ChatCompletion.create_async(g4f.models.default, DEFAULT_MESSAGES, ProviderMock) asyncio.run(coroutine) - self.assertEqual(get_last_provider(), ProviderMock) - -if __name__ == '__main__': - unittest.main()
\ No newline at end of file + self.assertEqual(get_last_provider(), ProviderMock)
\ No newline at end of file |