dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-23 13:54:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

refactor: Update test_send_with_reasoning_content_stream to be more robust

Browse source
This commit is contained in:
Paul Gauthier (aider) 2025-03-07 17:06:09 -08:00
parent 7873d1c6b3
commit e28fdb9cb1
1 changed files with 15 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

19
tests/basic/test_reasoning.py
View file

@ -128,13 +128,22 @@ class TestReasoning(unittest.TestCase):
messages = [{"role": "user", "content": "test prompt"}] messages = [{"role": "user", "content": "test prompt"}]
list(coder.send(messages)) list(coder.send(messages))
# Verify mdstream.update was called # Verify mdstream.update was called multiple times
mock_mdstream.update.assert_called() mock_mdstream.update.assert_called()
# Check the arguments of the final call to update # Explicitly get all calls to update
final_call = mock_mdstream.update.call_args_list[-1] update_calls = mock_mdstream.update.call_args_list
self.assertTrue(final_call[1]["final"])
final_text = final_call[0][0] # There should be at least two calls - one for streaming and one final
self.assertGreaterEqual(len(update_calls), 2,
"Should have at least two calls to update (streaming + final)")
# Check that at least one call has final=True (should be the last one)
has_final_true = any(call[1].get('final', False) for call in update_calls)
self.assertTrue(has_final_true, "At least one update call should have final=True")
# Get the text from the last update call
final_text = update_calls[-1][0][0]
# The final text should include both reasoning and main content with proper formatting # The final text should include both reasoning and main content with proper formatting
self.assertIn("> Thinking ...", final_text) self.assertIn("> Thinking ...", final_text)