wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-07-13 01:47:14 +08:00
Code Issues Actions 10 Packages Projects Releases Wiki Activity

Updated the test case in tests-unit/comfy_test/folder_path_test.py to prevent any module-level state mutation as suggested by CodeRabbit review.

Browse Source
This commit is contained in:
silveroxides 2026-07-07 13:29:39 +02:00
parent 088006b0f4
commit 0c509600a4
1 changed files with 12 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -166,15 +166,20 @@ def test_base_path_change_clears_old(set_base_dir):
def test_models_directory_cli_and_getters(temp_dir):
with patch.object(sys, 'argv', ["main.py", "--models-directory", temp_dir]):
reload(comfy.cli_args)
reload(folder_paths)
try:
with patch.object(sys, 'argv', ["main.py", "--models-directory", temp_dir]):
reload(comfy.cli_args)
reload(folder_paths)
assert folder_paths.get_models_directory() == os.path.abspath(temp_dir)
assert folder_paths.models_dir == os.path.abspath(temp_dir)
assert folder_paths.get_models_directory() == os.path.abspath(temp_dir)
assert folder_paths.models_dir == os.path.abspath(temp_dir)
with pytest.raises(Exception):
comfy.cli_args.is_valid_directory(os.path.join(temp_dir, "non_existent_folder_path"))
with pytest.raises(Exception):
comfy.cli_args.is_valid_directory(os.path.join(temp_dir, "non_existent_folder_path"))
finally:
with patch.object(sys, 'argv', ["main.py"]):
reload(comfy.cli_args)
reload(folder_paths)
def test_models_dir_getter_setter():