xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-09-11 01:17:06 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update tests to match changes.

Browse Source
This commit is contained in:
filtered 2025-01-26 14:25:32 +11:00
parent 8a29e9329e
commit 87dcae08d9
1 changed files with 26 additions and 37 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

63
tests-unit/comfy_test/folder_path_test.py
View File

@ -108,45 +108,34 @@ def test_base_path_changes(clear_folder_paths):
assert os.path.join(test_dir, "custom_nodes") in folder_paths.get_folder_paths("custom_nodes") assert os.path.join(test_dir, "custom_nodes") in folder_paths.get_folder_paths("custom_nodes")
for name in ["checkpoints", "loras", "vae", "configs", "embeddings", "controlnet", "classifiers", "configs"]: for name in ["checkpoints", "loras", "vae", "configs", "embeddings", "controlnet", "classifiers"]:
assert folder_paths.get_folder_paths(name)[0] == os.path.join(test_dir, name) assert folder_paths.get_folder_paths(name)[0] == os.path.join(test_dir, "models", name)
def test_add_default_paths_preseves_dirs(clear_folder_paths): def test_base_path_change_clears_old(clear_folder_paths):
test_dir = os.path.abspath(os.path.join(os.path.curdir, "..", "..")) test_dir = "/test/path"
base_path = folder_paths.base_path folder_paths.reset_all_paths(test_dir)
models_dir = folder_paths.models_dir
input_directory = folder_paths.input_directory
folder_paths.add_default_model_paths(test_dir) assert len(folder_paths.get_folder_paths("custom_nodes")) == 1
assert folder_paths.base_path == base_path
assert folder_paths.models_dir == models_dir
assert folder_paths.input_directory == input_directory
single_model_paths = [
"checkpoints",
"loras",
"vae",
"configs",
"clip_vision",
"style_models",
"diffusers",
"vae_approx",
"gligen",
"upscale_models",
"embeddings",
"hypernetworks",
"photomaker",
"classifiers",
]
for name in single_model_paths:
assert len(folder_paths.get_folder_paths(name)) == 1
def test_add_default_paths_preseves_paths(clear_folder_paths): for name in ["controlnet", "diffusion_models", "text_encoders"]:
test_dir = os.path.abspath(os.path.join(os.path.curdir, "invalid")) assert len(folder_paths.get_folder_paths(name)) == 2
checkpoints = folder_paths.get_folder_paths("checkpoints")[0]
text_encoders = folder_paths.get_folder_paths("text_encoders")[0]
clip = folder_paths.get_folder_paths("clip")[1]
folder_paths.add_default_model_paths(test_dir)
assert not os.path.join(test_dir, "custom_nodes") in folder_paths.get_folder_paths("custom_nodes")
assert folder_paths.get_folder_paths("checkpoints")[0] == checkpoints
assert folder_paths.get_folder_paths("text_encoders")[0] == text_encoders
assert folder_paths.get_folder_paths("clip")[1] == clip
def test_add_default_model_paths(clear_folder_paths):
test_dir = os.path.abspath(os.path.join(os.path.curdir, "bad_path"))
folder_paths.add_default_model_paths(test_dir)
for name in ["checkpoints", "loras", "vae", "configs", "embeddings", "controlnet", "classifiers", "configs"]:
paths = folder_paths.get_folder_paths(name)
# Handle multiple default paths
assert len(paths) % 2 == 0
index = int(len(paths) / 2)
assert folder_paths.get_folder_paths(name)[index] == os.path.join(test_dir, name)
assert folder_paths.get_folder_paths("clip")[2] == os.path.join(test_dir, "text_encoders")
assert folder_paths.get_folder_paths("clip")[3] == os.path.join(test_dir, "clip")