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 linter

Browse source
This commit is contained in:
Paul Gauthier (aider) 2025-03-18 17:33:58 -07:00
parent dd2efac3ae
commit 6cce7c34c2
1 changed files with 45 additions and 23 deletions
Download patch file Download diff file Expand all files Collapse all files

68
tests/basic/test_main.py
View file

@ -688,8 +688,10 @@ class TestMain(TestCase):
# Test that appropriate warnings are shown based on accepts_settings configuration # Test that appropriate warnings are shown based on accepts_settings configuration
with GitTemporaryDirectory(): with GitTemporaryDirectory():
# 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.models.Model.set_thinking_tokens") as mock_set_thinking: patch("aider.io.InputOutput.tool_warning") as mock_warning,
patch("aider.models.Model.set_thinking_tokens") as mock_set_thinking,
):
main( main(
[ [
"--model", "--model",
@ -709,8 +711,10 @@ class TestMain(TestCase):
mock_set_thinking.assert_called_once_with("1000") mock_set_thinking.assert_called_once_with("1000")
# 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.models.Model.set_thinking_tokens") as mock_set_thinking: patch("aider.io.InputOutput.tool_warning") as mock_warning,
patch("aider.models.Model.set_thinking_tokens") as mock_set_thinking,
):
main( main(
["--model", "gpt-4o", "--thinking-tokens", "1000", "--yes", "--exit"], ["--model", "gpt-4o", "--thinking-tokens", "1000", "--yes", "--exit"],
input=DummyInput(), input=DummyInput(),
@ -726,8 +730,10 @@ class TestMain(TestCase):
mock_set_thinking.assert_called_once_with("1000") mock_set_thinking.assert_called_once_with("1000")
# 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.models.Model.set_reasoning_effort") as mock_set_reasoning: patch("aider.io.InputOutput.tool_warning") as mock_warning,
patch("aider.models.Model.set_reasoning_effort") as mock_set_reasoning,
):
main( main(
["--model", "o1", "--reasoning-effort", "3", "--yes", "--exit"], ["--model", "o1", "--reasoning-effort", "3", "--yes", "--exit"],
input=DummyInput(), input=DummyInput(),
@ -740,8 +746,10 @@ class TestMain(TestCase):
mock_set_reasoning.assert_called_once_with("3") mock_set_reasoning.assert_called_once_with("3")
# 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.models.Model.set_reasoning_effort") as mock_set_reasoning: patch("aider.io.InputOutput.tool_warning") as mock_warning,
patch("aider.models.Model.set_reasoning_effort") as mock_set_reasoning,
):
main( main(
["--model", "gpt-3.5-turbo", "--reasoning-effort", "3", "--yes", "--exit"], ["--model", "gpt-3.5-turbo", "--reasoning-effort", "3", "--yes", "--exit"],
input=DummyInput(), input=DummyInput(),
@ -1026,7 +1034,7 @@ class TestMain(TestCase):
self.assertEqual( self.assertEqual(
coder.main_model.extra_params.get("thinking", {}).get("budget_tokens"), 1000 coder.main_model.extra_params.get("thinking", {}).get("budget_tokens"), 1000
) )
def test_check_model_accepts_settings_flag(self): def test_check_model_accepts_settings_flag(self):
# Test that --check-model-accepts-settings affects whether settings are applied # Test that --check-model-accepts-settings affects whether settings are applied
with GitTemporaryDirectory(): with GitTemporaryDirectory():
@ -1034,31 +1042,38 @@ class TestMain(TestCase):
with patch("aider.models.Model.set_thinking_tokens") as mock_set_thinking: with patch("aider.models.Model.set_thinking_tokens") as mock_set_thinking:
main( main(
[ [
"--model", "gpt-4o", "--model",
"--thinking-tokens", "1000", "gpt-4o",
"--thinking-tokens",
"1000",
"--check-model-accepts-settings", "--check-model-accepts-settings",
"--yes", "--exit" "--yes",
"--exit",
], ],
input=DummyInput(), input=DummyInput(),
output=DummyOutput(), output=DummyOutput(),
) )
# Method should not be called because model doesn't support it and flag is on # Method should not be called because model doesn't support it and flag is on
mock_set_thinking.assert_not_called() mock_set_thinking.assert_not_called()
# When flag is off, setting should be applied regardless of support # When flag is off, setting should be applied regardless of support
with patch("aider.models.Model.set_reasoning_effort") as mock_set_reasoning: with patch("aider.models.Model.set_reasoning_effort") as mock_set_reasoning:
main( main(
[ [
"--model", "gpt-3.5-turbo", "--model",
"--reasoning-effort", "3", "gpt-3.5-turbo",
"--reasoning-effort",
"3",
"--no-check-model-accepts-settings", "--no-check-model-accepts-settings",
"--yes", "--exit" "--yes",
"--exit",
], ],
input=DummyInput(), input=DummyInput(),
output=DummyOutput(), output=DummyOutput(),
) )
# Method should be called because flag is off # Method should be called because flag is off
mock_set_reasoning.assert_called_once_with("3") mock_set_reasoning.assert_called_once_with("3")
def test_model_accepts_settings_attribute(self): def test_model_accepts_settings_attribute(self):
with GitTemporaryDirectory(): with GitTemporaryDirectory():
# Test with a model where we override the accepts_settings attribute # Test with a model where we override the accepts_settings attribute
@ -1067,24 +1082,31 @@ class TestMain(TestCase):
mock_instance = MockModel.return_value mock_instance = MockModel.return_value
mock_instance.name = "test-model" mock_instance.name = "test-model"
mock_instance.accepts_settings = ["reasoning_effort"] mock_instance.accepts_settings = ["reasoning_effort"]
mock_instance.validate_environment.return_value = {"missing_keys": [], "keys_in_environment": []} mock_instance.validate_environment.return_value = {
"missing_keys": [],
"keys_in_environment": [],
}
mock_instance.info = {} mock_instance.info = {}
mock_instance.weak_model_name = None mock_instance.weak_model_name = None
mock_instance.get_weak_model.return_value = None mock_instance.get_weak_model.return_value = None
# Run with both settings, but model only accepts reasoning_effort # Run with both settings, but model only accepts reasoning_effort
main( main(
[ [
"--model", "test-model", "--model",
"--reasoning-effort", "3", "test-model",
"--thinking-tokens", "1000", "--reasoning-effort",
"3",
"--thinking-tokens",
"1000",
"--check-model-accepts-settings", "--check-model-accepts-settings",
"--yes", "--exit" "--yes",
"--exit",
], ],
input=DummyInput(), input=DummyInput(),
output=DummyOutput(), output=DummyOutput(),
) )
# Only set_reasoning_effort should be called, not set_thinking_tokens # Only set_reasoning_effort should be called, not set_thinking_tokens
mock_instance.set_reasoning_effort.assert_called_once_with("3") mock_instance.set_reasoning_effort.assert_called_once_with("3")
mock_instance.set_thinking_tokens.assert_not_called() mock_instance.set_thinking_tokens.assert_not_called()