dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-20 12:24:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

style: Format test_main.py with consistent line breaks

Browse source
This commit is contained in:
Paul Gauthier (aider) 2025-03-18 12:18:57 -07:00
parent 719324981d
commit 50d8a19397
1 changed files with 13 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

19
tests/basic/test_main.py
View file

@ -690,14 +690,21 @@ class TestMain(TestCase):
# Test model that accepts the thinking_tokens setting # Test model that accepts the thinking_tokens setting
with patch("aider.io.InputOutput.tool_warning") as mock_warning: with patch("aider.io.InputOutput.tool_warning") as mock_warning:
main( main(
["--model", "anthropic/claude-3-7-sonnet-20250219", "--thinking-tokens", "1000", "--yes", "--exit"], [
"--model",
"anthropic/claude-3-7-sonnet-20250219",
"--thinking-tokens",
"1000",
"--yes",
"--exit",
],
input=DummyInput(), input=DummyInput(),
output=DummyOutput(), output=DummyOutput(),
) )
# No warning should be shown as this model accepts thinking_tokens # No warning should be shown as this model accepts thinking_tokens
for call in mock_warning.call_args_list: for call in mock_warning.call_args_list:
self.assertNotIn("thinking_tokens", call[0][0]) self.assertNotIn("thinking_tokens", call[0][0])
# Test model that doesn't have accepts_settings for thinking_tokens # Test model that doesn't have accepts_settings for thinking_tokens
with patch("aider.io.InputOutput.tool_warning") as mock_warning: with patch("aider.io.InputOutput.tool_warning") as mock_warning:
main( main(
@ -711,7 +718,7 @@ class TestMain(TestCase):
if "thinking_tokens" in call[0][0]: if "thinking_tokens" in call[0][0]:
warning_shown = True warning_shown = True
self.assertTrue(warning_shown) self.assertTrue(warning_shown)
# Test model that accepts the reasoning_effort setting # Test model that accepts the reasoning_effort setting
with patch("aider.io.InputOutput.tool_warning") as mock_warning: with patch("aider.io.InputOutput.tool_warning") as mock_warning:
main( main(
@ -722,7 +729,7 @@ class TestMain(TestCase):
# No warning should be shown as this model accepts reasoning_effort # No warning should be shown as this model accepts reasoning_effort
for call in mock_warning.call_args_list: for call in mock_warning.call_args_list:
self.assertNotIn("reasoning_effort", call[0][0]) self.assertNotIn("reasoning_effort", call[0][0])
# Test model that doesn't have accepts_settings for reasoning_effort # Test model that doesn't have accepts_settings for reasoning_effort
with patch("aider.io.InputOutput.tool_warning") as mock_warning: with patch("aider.io.InputOutput.tool_warning") as mock_warning:
main( main(
@ -736,7 +743,7 @@ class TestMain(TestCase):
if "reasoning_effort" in call[0][0]: if "reasoning_effort" in call[0][0]:
warning_shown = True warning_shown = True
self.assertTrue(warning_shown) self.assertTrue(warning_shown)
@patch("aider.models.ModelInfoManager.set_verify_ssl") @patch("aider.models.ModelInfoManager.set_verify_ssl")
def test_no_verify_ssl_sets_model_info_manager(self, mock_set_verify_ssl): def test_no_verify_ssl_sets_model_info_manager(self, mock_set_verify_ssl):
with GitTemporaryDirectory(): with GitTemporaryDirectory():
@ -996,7 +1003,7 @@ class TestMain(TestCase):
self.assertEqual( self.assertEqual(
coder.main_model.extra_params.get("extra_body", {}).get("reasoning_effort"), "3" coder.main_model.extra_params.get("extra_body", {}).get("reasoning_effort"), "3"
) )
def test_thinking_tokens_option(self): def test_thinking_tokens_option(self):
coder = main( coder = main(
["--thinking-tokens", "1000", "--yes", "--exit"], ["--thinking-tokens", "1000", "--yes", "--exit"],