mirror of
https://github.com/Aider-AI/aider.git
synced 2025-05-23 13:54:59 +00:00
style: Format Python code with linter
This commit is contained in:
parent
10a5250527
commit
333ddfb37a
1 changed files with 20 additions and 14 deletions
|
@ -1302,15 +1302,21 @@ class TestCommands(TestCase):
|
||||||
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():
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue