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

test: Update test_reasoning to use generator and print partial response

Browse source
This commit is contained in:
Paul Gauthier 2025-03-07 16:52:49 -08:00 • committed by Paul Gauthier (aider)
parent f613ad6c05
commit 82df218bcb
1 changed files with 7 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

12
tests/basic/test_reasoning.py
View file

@ -33,15 +33,17 @@ class TestReasoning(unittest.TestCase):
with patch.object(model, "send_completion", return_value=mock_completion): with patch.object(model, "send_completion", return_value=mock_completion):
# Call send with a simple message # Call send with a simple message
messages = [{"role": "user", "content": "test prompt"}] messages = [{"role": "user", "content": "test prompt"}]
coder.send(messages) list(coder.send(messages))
# Format the response as it would happen in the normal flow # Format the response as it would happen in the normal flow
coder.partial_response_content = mock_completion.content #coder.partial_response_content = mock_completion.content
coder.partial_response_reasoning_content = mock_completion.reasoning_content #coder.partial_response_reasoning_content = mock_completion.reasoning_content
output = coder.get_multi_response_content_in_progress(final=True) #output = coder.get_multi_response_content_in_progress(final=True)
# Manually call ai_output to simulate the normal flow # Manually call ai_output to simulate the normal flow
coder.io.ai_output(output) #coder.io.ai_output(output)
print(coder.partial_response_content)
# Now verify ai_output was called with the right content # Now verify ai_output was called with the right content
io.ai_output.assert_called_once() io.ai_output.assert_called_once()