wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-02-14 23:42:35 +08:00
Code Issues Actions 14 Packages Projects Releases Wiki Activity

Assert images are generated

Browse Source
This commit is contained in:
enzymezoo-code 2023-09-04 22:11:01 -05:00
parent 2efe168ea2
commit 633243c983
1 changed files with 12 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
tests/inference/test_inference.py
View File

@ -1,4 +1,5 @@
from copy import deepcopy from copy import deepcopy
from io import BytesIO
from urllib import request from urllib import request
import numpy import numpy
import os import os
@ -128,8 +129,8 @@ comfy_graph_list = [DEFAULT_COMFY_GRAPH]
prompt_list = [ prompt_list = [
'a painting of a cat', 'a painting of a cat',
] ]
sampler_list = KSampler.SAMPLERS[0:2] sampler_list = KSampler.SAMPLERS
scheduler_list = [KSampler.SCHEDULERS[0]] scheduler_list = KSampler.SCHEDULERS
@pytest.mark.inference @pytest.mark.inference
@pytest.mark.parametrize("sampler", sampler_list) @pytest.mark.parametrize("sampler", sampler_list)
@ -208,5 +209,13 @@ class TestInference:
comfy_graph.set_prompt(prompt) comfy_graph.set_prompt(prompt)
# Generate # Generate
client.get_images(comfy_graph.graph) images = client.get_images(comfy_graph.graph)
assert len(images) == 1, "No images generated"
# assert all images are not blank
for images_output in images.values():
for image_data in images_output:
pil_image = Image.open(BytesIO(image_data))
assert numpy.array(pil_image).any() != 0, "Image is blank"