diff options
author | H Lohaus <hlohaus@users.noreply.github.com> | 2024-01-26 07:54:13 +0100 |
---|---|---|
committer | GitHub <noreply@github.com> | 2024-01-26 07:54:13 +0100 |
commit | feb83c168b0a57ecd8c511aa654209c5f40da30e (patch) | |
tree | 84ef9a576064b7480e339426d9966b17a3102cad /etc | |
parent | Merge pull request #1510 from hlohaus/sort (diff) | |
download | gpt4free-feb83c168b0a57ecd8c511aa654209c5f40da30e.tar gpt4free-feb83c168b0a57ecd8c511aa654209c5f40da30e.tar.gz gpt4free-feb83c168b0a57ecd8c511aa654209c5f40da30e.tar.bz2 gpt4free-feb83c168b0a57ecd8c511aa654209c5f40da30e.tar.lz gpt4free-feb83c168b0a57ecd8c511aa654209c5f40da30e.tar.xz gpt4free-feb83c168b0a57ecd8c511aa654209c5f40da30e.tar.zst gpt4free-feb83c168b0a57ecd8c511aa654209c5f40da30e.zip |
Diffstat (limited to 'etc')
-rw-r--r-- | etc/unittest/asyncio.py | 8 | ||||
-rw-r--r-- | etc/unittest/backend.py | 12 |
2 files changed, 18 insertions, 2 deletions
diff --git a/etc/unittest/asyncio.py b/etc/unittest/asyncio.py index 74e29986..a31ce211 100644 --- a/etc/unittest/asyncio.py +++ b/etc/unittest/asyncio.py @@ -1,6 +1,10 @@ from .include import DEFAULT_MESSAGES import asyncio -import nest_asyncio +try: + import nest_asyncio + has_nest_asyncio = True +except: + has_nest_asyncio = False import unittest import g4f from g4f import ChatCompletion @@ -39,6 +43,8 @@ class TestChatCompletionAsync(unittest.IsolatedAsyncioTestCase): class TestChatCompletionNestAsync(unittest.IsolatedAsyncioTestCase): def setUp(self) -> None: + if not has_nest_asyncio: + self.skipTest('"nest_asyncio" not installed') nest_asyncio.apply() async def test_create(self): diff --git a/etc/unittest/backend.py b/etc/unittest/backend.py index f5961e2d..3be83f84 100644 --- a/etc/unittest/backend.py +++ b/etc/unittest/backend.py @@ -3,11 +3,17 @@ import unittest from unittest.mock import MagicMock from .mocks import ProviderMock import g4f -from g4f.gui.server.backend import Backend_Api, get_error_message +try: + from g4f.gui.server.backend import Backend_Api, get_error_message + has_requirements = True +except: + has_requirements = False class TestBackendApi(unittest.TestCase): def setUp(self): + if not has_requirements: + self.skipTest('"flask" not installed') self.app = MagicMock() self.api = Backend_Api(self.app) @@ -28,6 +34,10 @@ class TestBackendApi(unittest.TestCase): class TestUtilityFunctions(unittest.TestCase): + def setUp(self): + if not has_requirements: + self.skipTest('"flask" not installed') + def test_get_error_message(self): g4f.debug.last_provider = ProviderMock exception = Exception("Message") |