mirror of
https://github.com/Aider-AI/aider.git
synced 2025-05-23 22:04:59 +00:00
style: Apply linter formatting to test_reasoning.py
This commit is contained in:
parent
ad8b5c9d29
commit
4e732d0379
1 changed files with 3 additions and 3 deletions
|
@ -34,15 +34,15 @@ class TestReasoning(unittest.TestCase):
|
||||||
# Call send with a simple message
|
# Call send with a simple message
|
||||||
messages = [{"role": "user", "content": "test prompt"}]
|
messages = [{"role": "user", "content": "test prompt"}]
|
||||||
result = coder.send(messages)
|
result = 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)
|
||||||
|
|
||||||
# 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()
|
||||||
output = io.ai_output.call_args[0][0]
|
output = io.ai_output.call_args[0][0]
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue