Revert last-minute refactor.

This commit is contained in:
filtered 2025-01-29 00:39:42 +11:00
parent a856e6067c
commit aa613db5af

View File

@ -28,7 +28,6 @@ def temp_dir():
@pytest.fixture @pytest.fixture
def set_base_dir(): def set_base_dir():
def _set_base_dir(base_dir): def _set_base_dir(base_dir):
base_dir = os.path.abspath(base_dir)
# Mock CLI args # Mock CLI args
with patch.object(sys, 'argv', ["main.py", "--base-directory", base_dir]): with patch.object(sys, 'argv', ["main.py", "--base-directory", base_dir]):
reload(comfy.cli_args) reload(comfy.cli_args)
@ -114,7 +113,7 @@ def test_get_save_image_path(temp_dir):
def test_base_path_changes(set_base_dir): def test_base_path_changes(set_base_dir):
test_dir = "/test/dir" test_dir = os.path.abspath("/test/dir")
set_base_dir(test_dir) set_base_dir(test_dir)
assert folder_paths.base_path == test_dir assert folder_paths.base_path == test_dir
@ -131,7 +130,7 @@ def test_base_path_changes(set_base_dir):
def test_base_path_change_clears_old(set_base_dir): def test_base_path_change_clears_old(set_base_dir):
test_dir = "/test/dir" test_dir = os.path.abspath("/test/dir")
set_base_dir(test_dir) set_base_dir(test_dir)
assert len(folder_paths.get_folder_paths("custom_nodes")) == 1 assert len(folder_paths.get_folder_paths("custom_nodes")) == 1