fix

comfyanonymous · Dec 9, 2023 · 080ef75 · 080ef75
1 parent 9aaf368
commit 080ef75
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/tests-ui/tests/groupNode.test.js b/tests-ui/tests/groupNode.test.js
@@ -405,9 +405,14 @@ describe("group node", () => {
 		group1.outputs.LATENT.connectTo(group2.inputs.LATENT);
 
 		const decode = ez.VAEDecode(group2.outputs.LATENT, group2.outputs.VAE);
-		ez.PreviewImage(decode.outputs[0]);
+		const preview = ez.PreviewImage(decode.outputs[0]);
 
-		expect((await graph.toPrompt()).output).toEqual({});
+		expect((await graph.toPrompt()).output).toEqual({
+			[latent.id]: { inputs: { width: 512, height: 512, batch_size: 1 }, class_type: "EmptyLatentImage" },
+			[vae.id]: { inputs: { vae_name: "vae1.safetensors" }, class_type: "VAELoader" },
+			[decode.id]: { inputs: { samples: [latent.id + "", 0], vae: [vae.id + "", 0] }, class_type: "VAEDecode" },
+			[preview.id]: { inputs: { images: [decode.id + "", 0] }, class_type: "PreviewImage" },
+		});
 	});
 	test("displays generated image on group node", async () => {
 		const { ez, graph, app } = await start();