返回提交历史
Modified
etc/unittest/__main__.py
+2
-2
Deleted
etc/unittest/async_client.py
+0
-56
Modified
etc/unittest/client.py
+20
-18
XFEstudio/gpt4free
refactor(etc/unittest): update async client test structure
e1e37b34
代码差异
3 个文件
+22
-76
@@ -4,8 +4,8 @@ from .backend import *
4
4
from .main import *
5
5
from .model import *
6
6
from .client import *
7
from .async_client import *
7
from .client import *
8
8
from .include import *
9
9
from .integration import *
10
10
11
unittest.main()
11
unittest.main()
@@ -1,56 +0,0 @@
1
import unittest
2
3
from g4f.client import AsyncClient, ChatCompletion, ChatCompletionChunk
4
from .mocks import AsyncGeneratorProviderMock, ModelProviderMock, YieldProviderMock
5
6
DEFAULT_MESSAGES = [{'role': 'user', 'content': 'Hello'}]
7
8
class AsyncTestPassModel(unittest.IsolatedAsyncioTestCase):
9
10
async def test_response(self):
11
client = AsyncClient(provider=AsyncGeneratorProviderMock)
12
response = await client.chat.completions.create(DEFAULT_MESSAGES, "")
13
self.assertIsInstance(response, ChatCompletion)
14
self.assertEqual("Mock", response.choices[0].message.content)
15
16
async def test_pass_model(self):
17
client = AsyncClient(provider=ModelProviderMock)
18
response = await client.chat.completions.create(DEFAULT_MESSAGES, "Hello")
19
self.assertIsInstance(response, ChatCompletion)
20
self.assertEqual("Hello", response.choices[0].message.content)
21
22
async def test_max_tokens(self):
23
client = AsyncClient(provider=YieldProviderMock)
24
messages = [{'role': 'user', 'content': chunk} for chunk in ["How ", "are ", "you", "?"]]
25
response = await client.chat.completions.create(messages, "Hello", max_tokens=1)
26
self.assertIsInstance(response, ChatCompletion)
27
self.assertEqual("How ", response.choices[0].message.content)
28
response = await client.chat.completions.create(messages, "Hello", max_tokens=2)
29
self.assertIsInstance(response, ChatCompletion)
30
self.assertEqual("How are ", response.choices[0].message.content)
31
32
async def test_max_stream(self):
33
client = AsyncClient(provider=YieldProviderMock)
34
messages = [{'role': 'user', 'content': chunk} for chunk in ["How ", "are ", "you", "?"]]
35
response = await client.chat.completions.create(messages, "Hello", stream=True)
36
async for chunk in response:
37
self.assertIsInstance(chunk, ChatCompletionChunk)
38
if chunk.choices[0].delta.content is not None:
39
self.assertIsInstance(chunk.choices[0].delta.content, str)
40
messages = [{'role': 'user', 'content': chunk} for chunk in ["You ", "You ", "Other", "?"]]
41
response = await client.chat.completions.create(messages, "Hello", stream=True, max_tokens=2)
42
response = [chunk async for chunk in response]
43
self.assertEqual(len(response), 3)
44
for chunk in response:
45
if chunk.choices[0].delta.content is not None:
46
self.assertEqual(chunk.choices[0].delta.content, "You ")
47
48
async def test_stop(self):
49
client = AsyncClient(provider=YieldProviderMock)
50
messages = [{'role': 'user', 'content': chunk} for chunk in ["How ", "are ", "you", "?"]]
51
response = await client.chat.completions.create(messages, "Hello", stop=["and"])
52
self.assertIsInstance(response, ChatCompletion)
53
self.assertEqual("How are you?", response.choices[0].message.content)
54
55
if __name__ == '__main__':
56
unittest.main()
@@ -5,52 +5,54 @@ from .mocks import AsyncGeneratorProviderMock, ModelProviderMock, YieldProviderM
5
5
6
6
DEFAULT_MESSAGES = [{'role': 'user', 'content': 'Hello'}]
7
7
8
class TestPassModel(unittest.TestCase):
8
class AsyncTestPassModel(unittest.IsolatedAsyncioTestCase):
9
9
10
def test_response(self):
10
async def test_response(self):
11
11
client = Client(provider=AsyncGeneratorProviderMock)
12
response = client.chat.completions.create(DEFAULT_MESSAGES, "")
12
response = await client.chat.completions.async_create(DEFAULT_MESSAGES, "")
13
13
self.assertIsInstance(response, ChatCompletion)
14
14
self.assertEqual("Mock", response.choices[0].message.content)
15
15
16
def test_pass_model(self):
16
async def test_pass_model(self):
17
17
client = Client(provider=ModelProviderMock)
18
response = client.chat.completions.create(DEFAULT_MESSAGES, "Hello")
18
response = await client.chat.completions.async_create(DEFAULT_MESSAGES, "Hello")
19
19
self.assertIsInstance(response, ChatCompletion)
20
20
self.assertEqual("Hello", response.choices[0].message.content)
21
21
22
def test_max_tokens(self):
22
async def test_max_tokens(self):
23
23
client = Client(provider=YieldProviderMock)
24
24
messages = [{'role': 'user', 'content': chunk} for chunk in ["How ", "are ", "you", "?"]]
25
response = client.chat.completions.create(messages, "Hello", max_tokens=1)
25
response = await client.chat.completions.async_create(messages, "Hello", max_tokens=1)
26
26
self.assertIsInstance(response, ChatCompletion)
27
27
self.assertEqual("How ", response.choices[0].message.content)
28
response = client.chat.completions.create(messages, "Hello", max_tokens=2)
28
response = await client.chat.completions.async_create(messages, "Hello", max_tokens=2)
29
29
self.assertIsInstance(response, ChatCompletion)
30
30
self.assertEqual("How are ", response.choices[0].message.content)
31
31
32
def test_max_stream(self):
32
async def test_max_stream(self):
33
33
client = Client(provider=YieldProviderMock)
34
34
messages = [{'role': 'user', 'content': chunk} for chunk in ["How ", "are ", "you", "?"]]
35
response = client.chat.completions.create(messages, "Hello", stream=True)
36
for chunk in response:
35
response = await client.chat.completions.async_create(messages, "Hello", stream=True)
36
async for chunk in response:
37
37
self.assertIsInstance(chunk, ChatCompletionChunk)
38
38
if chunk.choices[0].delta.content is not None:
39
39
self.assertIsInstance(chunk.choices[0].delta.content, str)
40
40
messages = [{'role': 'user', 'content': chunk} for chunk in ["You ", "You ", "Other", "?"]]
41
response = client.chat.completions.create(messages, "Hello", stream=True, max_tokens=2)
42
response = list(response)
43
self.assertEqual(len(response), 3)
44
for chunk in response:
41
response = await client.chat.completions.async_create(messages, "Hello", stream=True, max_tokens=2)
42
response_list = []
43
async for chunk in response:
44
response_list.append(chunk)
45
self.assertEqual(len(response_list), 3)
46
for chunk in response_list:
45
47
if chunk.choices[0].delta.content is not None:
46
48
self.assertEqual(chunk.choices[0].delta.content, "You ")
47
49
48
def test_stop(self):
50
async def test_stop(self):
49
51
client = Client(provider=YieldProviderMock)
50
52
messages = [{'role': 'user', 'content': chunk} for chunk in ["How ", "are ", "you", "?"]]
51
response = client.chat.completions.create(messages, "Hello", stop=["and"])
53
response = await client.chat.completions.async_create(messages, "Hello", stop=["and"])
52
54
self.assertIsInstance(response, ChatCompletion)
53
55
self.assertEqual("How are you?", response.choices[0].message.content)
54
56
55
57
if __name__ == '__main__':
56
unittest.main()
58
unittest.main()