Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add a batch validation test

Browse Source
This commit is contained in:
Brandon Rising 2023-08-16 15:42:45 -04:00
parent 29fceb960d
commit 796ee1246b
1 changed files with 77 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

77
tests/nodes/test_invoker.py
View File

@ -182,3 +182,80 @@ def test_can_create_batch(mock_invoker: Invoker, simple_graph, simple_batch):
assert len(batch_process_res.session_ids) == 25
for session in batch_process_res.session_ids:
assert mock_invoker.services.graph_execution_manager.get(session)
def test_can_create_bad_batches():
batch = None
try:
batch = Batch( # This batch has a duplicate node_id|fieldname combo
data=[
[
BatchData(
node_id="1",
field_name="prompt",
items=[
"Tomato sushi",
],
)
],
[
BatchData(
node_id="1",
field_name="prompt",
items=[
"Ume sushi",
],
)
],
]
)
except Exception as e:
assert e
try:
batch = Batch( # This batch has different item list lengths in the same group
data=[
[
BatchData(
node_id="1",
field_name="prompt",
items=[
"Tomato sushi",
],
),
BatchData(
node_id="1",
field_name="prompt",
items=[
"Tomato sushi",
"Courgette sushi",
],
),
],
[
BatchData(
node_id="1",
field_name="prompt",
items=[
"Ume sushi",
],
)
],
]
)
except Exception as e:
assert e
try:
batch = Batch( # This batch has a type mismatch in single items list
data=[
[
BatchData(
node_id="1",
field_name="prompt",
items=["Tomato sushi", 5],
),
],
]
)
except Exception as e:
assert e
assert not batch