diff options
Diffstat (limited to '')
-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") |