From 34a69029861f0b13299b60a896225510877c8dbd Mon Sep 17 00:00:00 2001 From: "Paul Gauthier (aider)" Date: Fri, 31 Jan 2025 09:01:14 -0800 Subject: [PATCH] style: Format test code with consistent line breaks and indentation --- tests/basic/test_main.py | 24 ++++++++++++++++++------ 1 file changed, 18 insertions(+), 6 deletions(-) diff --git a/tests/basic/test_main.py b/tests/basic/test_main.py index 823a89391..7f494f55d 100644 --- a/tests/basic/test_main.py +++ b/tests/basic/test_main.py @@ -749,31 +749,41 @@ class TestMain(TestCase): with GitTemporaryDirectory(): # Test Anthropic API key os.environ["ANTHROPIC_API_KEY"] = "test-key" - coder = main(["--exit", "--yes"], input=DummyInput(), output=DummyOutput(), return_coder=True) + coder = main( + ["--exit", "--yes"], input=DummyInput(), output=DummyOutput(), return_coder=True + ) self.assertEqual(coder.main_model.name, "sonnet") del os.environ["ANTHROPIC_API_KEY"] # Test DeepSeek API key os.environ["DEEPSEEK_API_KEY"] = "test-key" - coder = main(["--exit", "--yes"], input=DummyInput(), output=DummyOutput(), return_coder=True) + coder = main( + ["--exit", "--yes"], input=DummyInput(), output=DummyOutput(), return_coder=True + ) self.assertEqual(coder.main_model.name, "deepseek") del os.environ["DEEPSEEK_API_KEY"] # Test OpenRouter API key os.environ["OPENROUTER_API_KEY"] = "test-key" - coder = main(["--exit", "--yes"], input=DummyInput(), output=DummyOutput(), return_coder=True) + coder = main( + ["--exit", "--yes"], input=DummyInput(), output=DummyOutput(), return_coder=True + ) self.assertEqual(coder.main_model.name, "openrouter/anthropic/claude-3.5-sonnet") del os.environ["OPENROUTER_API_KEY"] # Test OpenAI API key os.environ["OPENAI_API_KEY"] = "test-key" - coder = main(["--exit", "--yes"], input=DummyInput(), output=DummyOutput(), return_coder=True) + coder = main( + ["--exit", "--yes"], input=DummyInput(), output=DummyOutput(), return_coder=True + ) self.assertEqual(coder.main_model.name, "gpt-4o") del os.environ["OPENAI_API_KEY"] # Test Gemini API key os.environ["GEMINI_API_KEY"] = "test-key" - coder = main(["--exit", "--yes"], input=DummyInput(), output=DummyOutput(), return_coder=True) + coder = main( + ["--exit", "--yes"], input=DummyInput(), output=DummyOutput(), return_coder=True + ) self.assertEqual(coder.main_model.name, "flash") del os.environ["GEMINI_API_KEY"] @@ -786,7 +796,9 @@ class TestMain(TestCase): # Test that earlier API keys take precedence os.environ["ANTHROPIC_API_KEY"] = "test-key" os.environ["OPENAI_API_KEY"] = "test-key" - coder = main(["--exit", "--yes"], input=DummyInput(), output=DummyOutput(), return_coder=True) + coder = main( + ["--exit", "--yes"], input=DummyInput(), output=DummyOutput(), return_coder=True + ) self.assertEqual(coder.main_model.name, "sonnet") del os.environ["ANTHROPIC_API_KEY"] del os.environ["OPENAI_API_KEY"]