From feb83c168b0a57ecd8c511aa654209c5f40da30e Mon Sep 17 00:00:00 2001 From: H Lohaus Date: Fri, 26 Jan 2024 07:54:13 +0100 Subject: New minimum requirements (#1515) * New minimum requirements * Add ConversationStyleOptionSets to Bing * Add image.ImageRequest * Improve python version support * Improve unittests --- etc/unittest/asyncio.py | 8 +++++++- etc/unittest/backend.py | 12 +++++++++++- 2 files changed, 18 insertions(+), 2 deletions(-) (limited to 'etc/unittest') 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") -- cgit v1.2.3