style: Fix linting issues in test_reasoning.py

This commit is contained in:
Paul Gauthier (aider) 2025-03-07 17:20:11 -08:00
parent 37c7b81c95
commit 5f2cf75be8

View file

@ -57,7 +57,7 @@ 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 # Verify that partial_response_content only contains the main content
self.assertEqual(coder.partial_response_content.strip(), main_content.strip()) self.assertEqual(coder.partial_response_content.strip(), main_content.strip())
@ -166,11 +166,11 @@ 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 # Verify that partial_response_content only contains the main content
expected_content = "Final answer after reasoning" expected_content = "Final answer after reasoning"
self.assertEqual(coder.partial_response_content.strip(), expected_content) self.assertEqual(coder.partial_response_content.strip(), expected_content)
# Verify that partial_response_content only contains the main content # Verify that partial_response_content only contains the main content
self.assertEqual(coder.partial_response_content.strip(), "Final answer after reasoning") self.assertEqual(coder.partial_response_content.strip(), "Final answer after reasoning")
@ -238,7 +238,7 @@ 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 # Verify that partial_response_content only contains the main content
self.assertEqual(coder.partial_response_content.strip(), main_content.strip()) self.assertEqual(coder.partial_response_content.strip(), main_content.strip())