andras/great-ai
1
0
Fork 0
Code Issues Pull requests 1 Projects Releases 4 Packages Wiki Activity Actions

Improve batch processing API

Browse source
This commit is contained in:
Andras Schmelczer 2022-07-08 09:36:36 +02:00
parent 1d30fd730a
commit 00714bef4e
4 changed files with 104 additions and 94 deletions
Download patch file Download diff file Expand all files Collapse all files

27
tests/test_great_ai.py
View file

@ -12,6 +12,19 @@ def test_process_batch() -> None:
assert [v.output for v in f.process_batch([3, 9, 34])] == [5, 11, 36]
def test_process_batch_unpacked() -> None:
@GreatAI.create
def f(a: int, b: str) -> str:
return b * a
assert [
v.output
for v in f.process_batch(
[(2, "aa"), (1, "fa"), (3, "b")], unpack_arguments=True
)
] == ["aaaa", "fa", "bbb"]
@pytest.mark.asyncio
async def test_process_batch_async() -> None:
@GreatAI.create
@ -20,3 +33,17 @@ async def test_process_batch_async() -> None:
return x + 2
assert [v.output for v in f.process_batch([3, 9, 34])] == [5, 11, 36]
@pytest.mark.asyncio
async def test_process_batch_async_unpacked() -> None:
@GreatAI.create
async def f(a: int, b: str) -> str:
return b * a
assert [
v.output
for v in f.process_batch(
[(2, "aa"), (1, "fa"), (3, "b")], unpack_arguments=True
)
] == ["aaaa", "fa", "bbb"]