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

refactor: Update test to use assistant_output instead of ai_output

Browse source
This commit is contained in:
Paul Gauthier 2025-03-07 17:00:17 -08:00 • committed by Paul Gauthier (aider)
parent 0cd8e3701d
commit f9b6501af1
1 changed files with 6 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

10
tests/basic/test_reasoning.py
View file

@ -4,14 +4,14 @@ from unittest.mock import MagicMock, patch
from aider.coders.base_coder import Coder
from aider.io import InputOutput
from aider.models import Model
from aider.dump import dump # noqa
class TestReasoning(unittest.TestCase):
def test_send_with_reasoning_content(self):
"""Test that reasoning content is properly formatted and output."""
# Setup IO with no pretty
io = InputOutput(pretty=False)
io.ai_output = MagicMock()
io.assistant_output = MagicMock()
# Setup model and coder
model = Model("gpt-3.5-turbo")
@ -44,8 +44,10 @@ class TestReasoning(unittest.TestCase):
list(coder.send(messages))
# Now verify ai_output was called with the right content
io.ai_output.assert_called_once()
output = io.ai_output.call_args[0][0]
io.assistant_output.assert_called_once()
output = io.assistant_output.call_args[0][0]
dump(output)
# Output should contain formatted reasoning tags
self.assertIn("Thinking ...", output)