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

test: Update token budget parsing test cases

Browse source
This commit is contained in:
Paul Gauthier 2025-03-11 12:24:42 -07:00 • committed by Paul Gauthier (aider)
parent 3d05007024
commit 7d902d2f3e
1 changed files with 2 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

7
tests/basic/test_commands.py
View file

@ -1291,10 +1291,8 @@ class TestCommands(TestCase):
# Test with various formats # Test with various formats
test_values = { test_values = {
"8k": 8192, # 8 * 1024 "8k": 8192, # 8 * 1024
"8K": 8192, # 8 * 1024
"10.5k": 10752, # 10.5 * 1024 "10.5k": 10752, # 10.5 * 1024
"0.5M": 524288, # 0.5 * 1024 * 1024 "512k": 524288, # 0.5 * 1024 * 1024
"1000": 1000,
} }
for input_value, expected_tokens in test_values.items(): for input_value, expected_tokens in test_values.items():
@ -1309,8 +1307,7 @@ class TestCommands(TestCase):
# Check that the tool output shows the correct value with format # Check that the tool output shows the correct value with format
# Use the actual input_value (not normalized) in the assertion # Use the actual input_value (not normalized) in the assertion
mock_tool_output.assert_any_call( mock_tool_output.assert_any_call(
f"Set thinking token budget to {expected_tokens:,} tokens" f"Set thinking token budget to {expected_tokens:,} tokens ({input_value})."
f" ({input_value})."
) )
# Test with no value provided - should display current value # Test with no value provided - should display current value