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

fix(tests): fix batch tests [WIP]

Browse Source
This commit is contained in:
psychedelicious 2023-09-05 18:07:47 +10:00
parent 331743ca0c
commit 531c3bb1e2
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tests/nodes/test_invoker.py
View File

@ -72,8 +72,12 @@ def simple_batch():
@pytest.fixture @pytest.fixture
def graph_with_subgraph(): def graph_with_subgraph():
sub_g = Graph()
sub_g.add_node(PromptTestInvocation(id="1", prompt="Banana sushi"))
sub_g.add_node(TextToImageTestInvocation(id="2"))
sub_g.add_edge(create_edge("1", "prompt", "2", "prompt"))
g = Graph() g = Graph()
g.add_node(GraphInvocation(id="1", graph=simple_graph())) g.add_node(GraphInvocation(id="1", graph=sub_g))
return g return g
@ -221,10 +225,10 @@ def test_can_create_batch_with_subgraph(mock_invoker: Invoker, graph_with_subgra
graph=graph_with_subgraph, graph=graph_with_subgraph,
) )
assert batch_process_res.batch_id assert batch_process_res.batch_id
assert len(batch_process_res.session_ids) == 25
# TODO: without the mock events service emitting the `graph_execution_state` events, # TODO: without the mock events service emitting the `graph_execution_state` events,
# the batch sessions do not know when they have finished, so this logic will fail # the batch sessions do not know when they have finished, so this logic will fail
# assert len(batch_process_res.session_ids) == 25
# mock_invoker.services.batch_manager.run_batch_process(batch_process_res.batch_id) # mock_invoker.services.batch_manager.run_batch_process(batch_process_res.batch_id)
# def has_executed_all_batches(batch_id: str): # def has_executed_all_batches(batch_id: str):
@ -241,10 +245,10 @@ def test_can_create_batch(mock_invoker: Invoker, simple_graph, simple_batch):
graph=simple_graph, graph=simple_graph,
) )
assert batch_process_res.batch_id assert batch_process_res.batch_id
assert len(batch_process_res.session_ids) == 25
# TODO: without the mock events service emitting the `graph_execution_state` events, # TODO: without the mock events service emitting the `graph_execution_state` events,
# the batch sessions do not know when they have finished, so this logic will fail # the batch sessions do not know when they have finished, so this logic will fail
# assert len(batch_process_res.session_ids) == 25
# mock_invoker.services.batch_manager.run_batch_process(batch_process_res.batch_id) # mock_invoker.services.batch_manager.run_batch_process(batch_process_res.batch_id)
# def has_executed_all_batches(batch_id: str): # def has_executed_all_batches(batch_id: str):