From 9cbe9c1ccb2381e37402a36297f11a0f96b1b557 Mon Sep 17 00:00:00 2001 From: Heiner Lohaus Date: Sun, 21 Jan 2024 02:20:23 +0100 Subject: Improve tests --- etc/unittest/asyncio.py | 57 +++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 57 insertions(+) create mode 100644 etc/unittest/asyncio.py (limited to 'etc/unittest/asyncio.py') diff --git a/etc/unittest/asyncio.py b/etc/unittest/asyncio.py new file mode 100644 index 00000000..74e29986 --- /dev/null +++ b/etc/unittest/asyncio.py @@ -0,0 +1,57 @@ +from .include import DEFAULT_MESSAGES +import asyncio +import nest_asyncio +import unittest +import g4f +from g4f import ChatCompletion +from .mocks import ProviderMock, AsyncProviderMock, AsyncGeneratorProviderMock + +class TestChatCompletion(unittest.TestCase): + + async def run_exception(self): + return ChatCompletion.create(g4f.models.default, DEFAULT_MESSAGES, AsyncProviderMock) + + def test_exception(self): + self.assertRaises(g4f.errors.NestAsyncioError, asyncio.run, self.run_exception()) + + def test_create(self): + result = ChatCompletion.create(g4f.models.default, DEFAULT_MESSAGES, AsyncProviderMock) + self.assertEqual("Mock",result) + + def test_create_generator(self): + result = ChatCompletion.create(g4f.models.default, DEFAULT_MESSAGES, AsyncGeneratorProviderMock) + self.assertEqual("Mock",result) + +class TestChatCompletionAsync(unittest.IsolatedAsyncioTestCase): + + async def test_base(self): + result = await ChatCompletion.create_async(g4f.models.default, DEFAULT_MESSAGES, ProviderMock) + self.assertEqual("Mock",result) + + async def test_async(self): + result = await ChatCompletion.create_async(g4f.models.default, DEFAULT_MESSAGES, AsyncProviderMock) + self.assertEqual("Mock",result) + + async def test_create_generator(self): + result = await ChatCompletion.create_async(g4f.models.default, DEFAULT_MESSAGES, AsyncGeneratorProviderMock) + self.assertEqual("Mock",result) + +class TestChatCompletionNestAsync(unittest.IsolatedAsyncioTestCase): + + def setUp(self) -> None: + nest_asyncio.apply() + + async def test_create(self): + result = await ChatCompletion.create_async(g4f.models.default, DEFAULT_MESSAGES, ProviderMock) + self.assertEqual("Mock",result) + + async def test_nested(self): + result = ChatCompletion.create(g4f.models.default, DEFAULT_MESSAGES, AsyncProviderMock) + self.assertEqual("Mock",result) + + async def test_nested_generator(self): + result = ChatCompletion.create(g4f.models.default, DEFAULT_MESSAGES, AsyncGeneratorProviderMock) + self.assertEqual("Mock",result) + +if __name__ == '__main__': + unittest.main() \ No newline at end of file -- cgit v1.2.3