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

Fix command line args for pytest

Browse Source
This commit is contained in:
enzymezoo-code 2023-09-05 15:42:53 -05:00
parent 3289db151f
commit 10dc6353f0
3 changed files with 16 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
pytest.ini
View File

@ -1,4 +1,5 @@
[pytest] [pytest]
markers = markers =
inference: mark as inference test (deselect with '-m "not inference"') inference: mark as inference test (deselect with '-m "not inference"')
testpaths = tests
addopts = -s addopts = -s

0
tests/inference/conftest.py → tests/conftest.py
View File

20
tests/inference/test_inference.py
View File

@ -17,7 +17,10 @@ import json
import urllib.request import urllib.request
import urllib.parse import urllib.parse
from comfy.samplers import KSampler # Currently causes an error when running pytest with built-in pytest args
# TODO: modify cli_args.py to not parse args on import
# We will hard-code sampler and scheduler lists for now
# from comfy.samplers import KSampler
""" """
These tests generate and save images through a range of parameters These tests generate and save images through a range of parameters
@ -130,9 +133,16 @@ comfy_graph_list = [DEFAULT_COMFY_GRAPH]
prompt_list = [ prompt_list = [
'a painting of a cat', 'a painting of a cat',
] ]
sampler_list = KSampler.SAMPLERS #TODO use sampler and scheduler list from comfy.samplers.KSampler
scheduler_list = KSampler.SCHEDULERS # sampler_list = KSampler.SAMPLERS
# scheduler_list = KSampler.SCHEDULERS
# Hard coded sampler and scheduler lists for now
SCHEDULERS = ["normal", "karras", "exponential", "sgm_uniform", "simple", "ddim_uniform"]
SAMPLERS = ["euler", "euler_ancestral", "heun", "dpm_2", "dpm_2_ancestral",
"lms", "dpm_fast", "dpm_adaptive", "dpmpp_2s_ancestral", "dpmpp_sde", "dpmpp_sde_gpu",
"dpmpp_2m", "dpmpp_2m_sde", "dpmpp_2m_sde_gpu", "dpmpp_3m_sde", "dpmpp_3m_sde_gpu", "ddim", "uni_pc", "uni_pc_bh2"]
sampler_list = [SAMPLERS[0]]
scheduler_list = [SCHEDULERS[0]]
@pytest.mark.inference @pytest.mark.inference
@pytest.mark.parametrize("sampler", sampler_list) @pytest.mark.parametrize("sampler", sampler_list)
@pytest.mark.parametrize("scheduler", scheduler_list) @pytest.mark.parametrize("scheduler", scheduler_list)
@ -217,7 +227,7 @@ class TestInference:
# Generate # Generate
images = client.get_images(comfy_graph.graph) images = client.get_images(comfy_graph.graph)
assert len(images) == 1, "No images generated" assert len(images) != 0, "No images generated"
# assert all images are not blank # assert all images are not blank
for images_output in images.values(): for images_output in images.values():
for image_data in images_output: for image_data in images_output: