diff --git a/tests/basic/test_reasoning.py b/tests/basic/test_reasoning.py index bb6d6ce43..62c2a4310 100644 --- a/tests/basic/test_reasoning.py +++ b/tests/basic/test_reasoning.py @@ -5,7 +5,7 @@ from aider.coders.base_coder import Coder from aider.dump import dump # noqa from aider.io import InputOutput from aider.models import Model -from aider.reasoning_tags import remove_reasoning_content +from aider.reasoning_tags import remove_reasoning_content, REASONING_START, REASONING_END class TestReasoning(unittest.TestCase): @@ -52,8 +52,8 @@ class TestReasoning(unittest.TestCase): dump(output) # Output should contain formatted reasoning tags - self.assertIn("Thinking ...", output) - self.assertIn("... done thinking", output) + self.assertIn(REASONING_START, output) + self.assertIn(REASONING_END, output) # Output should include both reasoning and main content self.assertIn(reasoning_content, output) @@ -157,9 +157,9 @@ class TestReasoning(unittest.TestCase): final_text = update_calls[-1][0][0] # The final text should include both reasoning and main content with proper formatting - self.assertIn("> Thinking ...", final_text) + self.assertIn(REASONING_START, final_text) self.assertIn("My step-by-step reasoning process", final_text) - self.assertIn("> ... done thinking", final_text) + self.assertIn(REASONING_END, final_text) self.assertIn("Final answer after reasoning", final_text) # Ensure proper order: reasoning first, then main content @@ -225,8 +225,8 @@ class TestReasoning(unittest.TestCase): dump(output) # Output should contain formatted reasoning tags - self.assertIn("Thinking ...", output) - self.assertIn("... done thinking", output) + self.assertIn(REASONING_START, output) + self.assertIn(REASONING_END, output) # Output should include both reasoning and main content self.assertIn(reasoning_content, output) @@ -330,9 +330,9 @@ class TestReasoning(unittest.TestCase): final_text = update_calls[-1][0][0] # The final text should include both reasoning and main content with proper formatting - self.assertIn("> Thinking ...", final_text) + self.assertIn(REASONING_START, final_text) self.assertIn("My step-by-step reasoning process", final_text) - self.assertIn("> ... done thinking", final_text) + self.assertIn(REASONING_END, final_text) self.assertIn("Final answer after reasoning", final_text) # Ensure proper order: reasoning first, then main content