From 96a0127a2ec8b4a7a5528077f1aafa05d76a4d5f Mon Sep 17 00:00:00 2001 From: Robin Huang Date: Thu, 12 Sep 2024 09:46:36 +0900 Subject: [PATCH] Fix comments. --- tests-unit/utils/extra_config_test.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tests-unit/utils/extra_config_test.py b/tests-unit/utils/extra_config_test.py index a2c32c41a..06349560d 100644 --- a/tests-unit/utils/extra_config_test.py +++ b/tests-unit/utils/extra_config_test.py @@ -24,7 +24,7 @@ def yaml_config_with_appdata(): return """ test_config: base_path: '%APPDATA%/ComfyUI' - checkpoints: 'models' + checkpoints: 'models/checkpoints' """ @pytest.fixture @@ -95,7 +95,7 @@ def test_load_extra_model_paths_expands_appdata( yaml_config_with_appdata, mock_yaml_content_appdata ): - # Set the mock_file to return our YAML content + # Set the mock_file to return yaml with appdata as a variable mock_file.return_value.read.return_value = yaml_config_with_appdata # Attach mocks @@ -111,12 +111,12 @@ def test_load_extra_model_paths_expands_appdata( expected_base_path = 'C:/Users/TestUser/AppData/Roaming/ComfyUI' expected_calls = [ - ('checkpoints', os.path.join(expected_base_path, 'models')), + ('checkpoints', os.path.join(expected_base_path, 'models/checkpoints')), ] assert mock_add_model_folder_path.call_count == len(expected_calls) - # Check if add_model_folder_path was called with the correct arguments + # Check the base path variable was expanded for actual_call, expected_call in zip(mock_add_model_folder_path.call_args_list, expected_calls): assert actual_call.args == expected_call