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

test: Ensure partial_response_content contains only main content

Browse source
This commit is contained in:
Paul Gauthier (aider) 2025-03-07 17:20:03 -08:00
parent 779a266713
commit 37c7b81c95
1 changed files with 13 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

13
tests/basic/test_reasoning.py
View file

@ -57,6 +57,9 @@ class TestReasoning(unittest.TestCase):
# Output should include both reasoning and main content # Output should include both reasoning and main content
self.assertIn(reasoning_content, output) self.assertIn(reasoning_content, output)
self.assertIn(main_content, output) self.assertIn(main_content, output)
# Verify that partial_response_content only contains the main content
self.assertEqual(coder.partial_response_content.strip(), main_content.strip())
# Ensure proper order: reasoning first, then main content # Ensure proper order: reasoning first, then main content
reasoning_pos = output.find(reasoning_content) reasoning_pos = output.find(reasoning_content)
@ -163,6 +166,13 @@ class TestReasoning(unittest.TestCase):
self.assertLess( self.assertLess(
reasoning_pos, main_pos, "Reasoning content should appear before main content" reasoning_pos, main_pos, "Reasoning content should appear before main content"
) )
# Verify that partial_response_content only contains the main content
expected_content = "Final answer after reasoning"
self.assertEqual(coder.partial_response_content.strip(), expected_content)
# Verify that partial_response_content only contains the main content
self.assertEqual(coder.partial_response_content.strip(), "Final answer after reasoning")
def test_send_with_think_tags(self): def test_send_with_think_tags(self):
"""Test that <think> tags are properly processed and formatted.""" """Test that <think> tags are properly processed and formatted."""
@ -228,6 +238,9 @@ class TestReasoning(unittest.TestCase):
self.assertLess( self.assertLess(
reasoning_pos, main_pos, "Reasoning content should appear before main content" reasoning_pos, main_pos, "Reasoning content should appear before main content"
) )
# Verify that partial_response_content only contains the main content
self.assertEqual(coder.partial_response_content.strip(), main_content.strip())
def test_send_with_think_tags_stream(self): def test_send_with_think_tags_stream(self):
"""Test that streaming with <think> tags is properly processed and formatted.""" """Test that streaming with <think> tags is properly processed and formatted."""