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

fix: Update test mocking to match send_completion method signature

Browse source
This commit is contained in:
Paul Gauthier (aider) 2025-03-07 16:52:51 -08:00
parent 82df218bcb
commit 54ef8a1e19
1 changed files with 2 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

14
tests/basic/test_reasoning.py
View file

@ -29,22 +29,12 @@ class TestReasoning(unittest.TestCase):
mock_completion = MockCompletion(main_content, reasoning_content)
# Mock the model's send_completion method
with patch.object(model, "send_completion", return_value=mock_completion):
# Mock the model's send_completion method to return the expected tuple format
with patch.object(model, "send_completion", return_value=(None, mock_completion)):
# Call send with a simple message
messages = [{"role": "user", "content": "test prompt"}]
list(coder.send(messages))
# Format the response as it would happen in the normal flow
#coder.partial_response_content = mock_completion.content
#coder.partial_response_reasoning_content = mock_completion.reasoning_content
#output = coder.get_multi_response_content_in_progress(final=True)
# Manually call ai_output to simulate the normal flow
#coder.io.ai_output(output)
print(coder.partial_response_content)
# Now verify ai_output was called with the right content
io.ai_output.assert_called_once()
output = io.ai_output.call_args[0][0]