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