mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-07-20 21:38:30 +08:00
Update models-directory unit tests to remove orphaned getter/setter references and implement robust module state restoration.
This commit is contained in:
parent
56737f49b0
commit
13a2cf53df
@ -171,7 +171,6 @@ def test_models_directory_cli_and_getters(temp_dir):
|
|||||||
reload(comfy.cli_args)
|
reload(comfy.cli_args)
|
||||||
reload(folder_paths)
|
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.models_dir == os.path.abspath(temp_dir)
|
||||||
|
|
||||||
with pytest.raises(Exception):
|
with pytest.raises(Exception):
|
||||||
@ -181,12 +180,3 @@ def test_models_directory_cli_and_getters(temp_dir):
|
|||||||
reload(comfy.cli_args)
|
reload(comfy.cli_args)
|
||||||
reload(folder_paths)
|
reload(folder_paths)
|
||||||
|
|
||||||
|
|
||||||
def test_models_dir_getter_setter():
|
|
||||||
orig_models_dir = folder_paths.get_models_directory()
|
|
||||||
try:
|
|
||||||
folder_paths.set_models_directory("/custom/models/path")
|
|
||||||
assert folder_paths.get_models_directory() == "/custom/models/path"
|
|
||||||
finally:
|
|
||||||
folder_paths.set_models_directory(orig_models_dir)
|
|
||||||
|
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user