From 513b223ef6ce1a5d8a32e31fce91f033993a68cf Mon Sep 17 00:00:00 2001 From: psychedelicious <4822129+psychedelicious@users.noreply.github.com> Date: Wed, 26 Jul 2023 18:49:29 +1000 Subject: [PATCH] fix(test): fix test_graph_subgraph_t2i needed to be updated after adding the nsfw checker node to the graph --- tests/nodes/test_node_graph.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/nodes/test_node_graph.py b/tests/nodes/test_node_graph.py index cd995141ab..e2a4c8b343 100644 --- a/tests/nodes/test_node_graph.py +++ b/tests/nodes/test_node_graph.py @@ -462,16 +462,16 @@ def test_graph_subgraph_t2i(): n4 = ShowImageInvocation(id = "4") g.add_node(n4) - g.add_edge(create_edge("1.7","image","4","image")) + g.add_edge(create_edge("1.8","image","4","image")) # Validate dg = g.nx_graph_flat() - assert set(dg.nodes) == set(['1.width', '1.height', '1.seed', '1.3', '1.4', '1.5', '1.6', '1.7', '2', '3', '4']) + assert set(dg.nodes) == set(['1.width', '1.height', '1.seed', '1.3', '1.4', '1.5', '1.6', '1.7', '1.8', '2', '3', '4']) expected_edges = [(f'1.{e.source.node_id}',f'1.{e.destination.node_id}') for e in lg.graph.edges] expected_edges.extend([ ('2','1.width'), ('3','1.height'), - ('1.7','4') + ('1.8','4') ]) print(expected_edges) print(list(dg.edges))