From 7d902d2f3ebe322d3e6f9a13e0b0f02b44fe1a44 Mon Sep 17 00:00:00 2001 From: Paul Gauthier Date: Tue, 11 Mar 2025 12:24:42 -0700 Subject: [PATCH] test: Update token budget parsing test cases --- tests/basic/test_commands.py | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/tests/basic/test_commands.py b/tests/basic/test_commands.py index 675a8f021..3bebf3f5a 100644 --- a/tests/basic/test_commands.py +++ b/tests/basic/test_commands.py @@ -1291,10 +1291,8 @@ class TestCommands(TestCase): # Test with various formats test_values = { "8k": 8192, # 8 * 1024 - "8K": 8192, # 8 * 1024 "10.5k": 10752, # 10.5 * 1024 - "0.5M": 524288, # 0.5 * 1024 * 1024 - "1000": 1000, + "512k": 524288, # 0.5 * 1024 * 1024 } 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 # Use the actual input_value (not normalized) in the assertion mock_tool_output.assert_any_call( - f"Set thinking token budget to {expected_tokens:,} tokens" - f" ({input_value})." + f"Set thinking token budget to {expected_tokens:,} tokens ({input_value})." ) # Test with no value provided - should display current value