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

style: Format Python code with linter

Browse source
This commit is contained in:
Paul Gauthier (aider) 2025-03-11 11:48:21 -07:00
parent 10a5250527
commit 333ddfb37a
1 changed files with 20 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

34
tests/basic/test_commands.py
View file

@ -1282,12 +1282,12 @@ class TestCommands(TestCase):
# Verify the file was not added # Verify the file was not added
self.assertEqual(len(coder.abs_fnames), 0) self.assertEqual(len(coder.abs_fnames), 0)
def test_cmd_think_tokens(self): def test_cmd_think_tokens(self):
io = InputOutput(pretty=False, fancy_input=False, yes=True) io = InputOutput(pretty=False, fancy_input=False, yes=True)
coder = Coder.create(self.GPT35, None, io) coder = Coder.create(self.GPT35, None, io)
commands = Commands(io, coder) commands = Commands(io, coder)
# Test with various formats # Test with various formats
test_values = { test_values = {
"8k": 8000, "8k": 8000,
@ -1296,21 +1296,27 @@ class TestCommands(TestCase):
"0.5M": 500000, "0.5M": 500000,
"1000": 1000, "1000": 1000,
} }
for input_value, expected_tokens in test_values.items(): for input_value, expected_tokens in test_values.items():
with mock.patch.object(io, "tool_output") as mock_tool_output: with mock.patch.object(io, "tool_output") as mock_tool_output:
commands.cmd_think_tokens(input_value) commands.cmd_think_tokens(input_value)
# Check that the model's thinking tokens were updated # Check that the model's thinking tokens were updated
self.assertEqual(coder.main_model.extra_params["thinking"]["budget_tokens"], expected_tokens) self.assertEqual(
coder.main_model.extra_params["thinking"]["budget_tokens"], expected_tokens
)
# Check that the tool output shows the correct value # Check that the tool output shows the correct value
mock_tool_output.assert_any_call(f"Set thinking token budget to {expected_tokens:,} tokens.") mock_tool_output.assert_any_call(
f"Set thinking token budget to {expected_tokens:,} tokens."
)
# Test with no value provided # Test with no value provided
with mock.patch.object(io, "tool_error") as mock_tool_error: with mock.patch.object(io, "tool_error") as mock_tool_error:
commands.cmd_think_tokens("") commands.cmd_think_tokens("")
mock_tool_error.assert_called_once_with("Please specify a token budget (e.g., 8k, 10k, 0.5M).") mock_tool_error.assert_called_once_with(
"Please specify a token budget (e.g., 8k, 10k, 0.5M)."
)
def test_cmd_add_aiderignored_file(self): def test_cmd_add_aiderignored_file(self):
with GitTemporaryDirectory(): with GitTemporaryDirectory():
@ -1750,28 +1756,28 @@ class TestCommands(TestCase):
del coder del coder
del commands del commands
def test_cmd_reasoning_effort(self): def test_cmd_reasoning_effort(self):
io = InputOutput(pretty=False, fancy_input=False, yes=True) io = InputOutput(pretty=False, fancy_input=False, yes=True)
coder = Coder.create(self.GPT35, None, io) coder = Coder.create(self.GPT35, None, io)
commands = Commands(io, coder) commands = Commands(io, coder)
# Test with numeric values # Test with numeric values
with mock.patch.object(io, "tool_output") as mock_tool_output: with mock.patch.object(io, "tool_output") as mock_tool_output:
commands.cmd_reasoning_effort("0.8") commands.cmd_reasoning_effort("0.8")
mock_tool_output.assert_any_call("Set reasoning effort to 0.8") mock_tool_output.assert_any_call("Set reasoning effort to 0.8")
# Test with text values (low/medium/high) # Test with text values (low/medium/high)
for effort_level in ["low", "medium", "high"]: for effort_level in ["low", "medium", "high"]:
with mock.patch.object(io, "tool_output") as mock_tool_output: with mock.patch.object(io, "tool_output") as mock_tool_output:
commands.cmd_reasoning_effort(effort_level) commands.cmd_reasoning_effort(effort_level)
mock_tool_output.assert_any_call(f"Set reasoning effort to {effort_level}") mock_tool_output.assert_any_call(f"Set reasoning effort to {effort_level}")
# Check model's reasoning effort was updated # Check model's reasoning effort was updated
with mock.patch.object(coder.main_model, "set_reasoning_effort") as mock_set_effort: with mock.patch.object(coder.main_model, "set_reasoning_effort") as mock_set_effort:
commands.cmd_reasoning_effort("0.5") commands.cmd_reasoning_effort("0.5")
mock_set_effort.assert_called_once_with("0.5") mock_set_effort.assert_called_once_with("0.5")
# Test with no value provided # Test with no value provided
with mock.patch.object(io, "tool_error") as mock_tool_error: with mock.patch.object(io, "tool_error") as mock_tool_error:
commands.cmd_reasoning_effort("") commands.cmd_reasoning_effort("")