mirror of
https://github.com/Aider-AI/aider.git
synced 2025-05-22 21:34: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)
|
||||
|
||||
# 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
|
||||
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
|
||||
with mock.patch.object(io, "tool_error") as mock_tool_error:
|
||||
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):
|
||||
with GitTemporaryDirectory():
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue