Commit513b223

committed

fix(test): fix test_graph_subgraph_t2i

needed to be updated after adding the nsfw checker node to the graph

1 parentdb05445 commit513b223Copy full SHA for 513b223

File tree

-3

lines changed

-3

lines changed

Lines changed: 3 additions & 3 deletions

Original file line number	Diff line number	Diff line change
`@@ -462,16 +462,16 @@ def test_graph_subgraph_t2i():`
`462`	`462`
`463`	`463`	`n4=ShowImageInvocation(id="4")`
`464`	`464`	`g.add_node(n4)`
`465`		`-g.add_edge(create_edge("1.7","image","4","image"))`
	`465`	`+g.add_edge(create_edge("1.8","image","4","image"))`
`466`	`466`
`467`	`467`	`# Validate`
`468`	`468`	`dg=g.nx_graph_flat()`
`469`		`-assertset(dg.nodes)==set(['1.width','1.height','1.seed','1.3','1.4','1.5','1.6','1.7','2','3','4'])`
	`469`	`+assertset(dg.nodes)==set(['1.width','1.height','1.seed','1.3','1.4','1.5','1.6','1.7','1.8','2','3','4'])`
`470`	`470`	`expected_edges= [(f'1.{e.source.node_id}',f'1.{e.destination.node_id}')foreinlg.graph.edges]`
`471`	`471`	`expected_edges.extend([`
`472`	`472`	`('2','1.width'),`
`473`	`473`	`('3','1.height'),`
`474`		`- ('1.7','4')`
	`474`	`+ ('1.8','4')`
`475`	`475`	`])`
`476`	`476`	`print(expected_edges)`
`477`	`477`	`print(list(dg.edges))`

Comments

(0)