From 6acbd80ceea44c779d8fe2be515232126029d1e4 Mon Sep 17 00:00:00 2001 From: "Paul Gauthier (aider)" Date: Tue, 25 Mar 2025 13:34:34 -1000 Subject: [PATCH] style: Format test data with consistent indentation --- tests/basic/test_main.py | 21 +++++++++++++++++---- 1 file changed, 17 insertions(+), 4 deletions(-) diff --git a/tests/basic/test_main.py b/tests/basic/test_main.py index 109211f5d..866a4529b 100644 --- a/tests/basic/test_main.py +++ b/tests/basic/test_main.py @@ -1068,8 +1068,14 @@ class TestMain(TestCase): # Create a temporary model-metadata.json with test models metadata_file = Path(".aider.model.metadata.json") test_models = { - "unique-model-name": {"max_input_tokens": 8192, "litellm_provider": "test-provider"}, - "another-unique-model": {"max_input_tokens": 4096, "litellm_provider": "another-provider"}, + "unique-model-name": { + "max_input_tokens": 8192, + "litellm_provider": "test-provider", + }, + "another-unique-model": { + "max_input_tokens": 4096, + "litellm_provider": "another-provider", + }, } metadata_file.write_text(json.dumps(test_models)) @@ -1097,7 +1103,12 @@ class TestMain(TestCase): with GitTemporaryDirectory(): # Create a temporary model-metadata.json with test models metadata_file = Path(".aider.model.metadata.json") - test_models = {"metadata-only-model": {"max_input_tokens": 8192, "litellm_provider": "test-provider"}} + test_models = { + "metadata-only-model": { + "max_input_tokens": 8192, + "litellm_provider": "test-provider", + } + } metadata_file.write_text(json.dumps(test_models)) # Patch litellm.model_cost to include a test model @@ -1150,7 +1161,9 @@ class TestMain(TestCase): # Test that models from resources/model-metadata.json are included in list-models output with GitTemporaryDirectory(): # Mock the importlib.resources.open_text to return a custom model-metadata.json - test_resource_models = {"special-model": {"max_input_tokens": 8192, "litellm_provider": "resource-provider"}} + test_resource_models = { + "special-model": {"max_input_tokens": 8192, "litellm_provider": "resource-provider"} + } mock_file = MagicMock() mock_file.read.return_value = json.dumps(test_resource_models)