style: Organize and clean up import statements in test_reasoning.py

This commit is contained in:
Paul Gauthier (aider) 2025-03-07 17:00:25 -08:00
parent b230fea66f
commit 3b9c2b9729

View file

@ -2,9 +2,10 @@ import unittest
from unittest.mock import MagicMock, patch from unittest.mock import MagicMock, patch
from aider.coders.base_coder import Coder from aider.coders.base_coder import Coder
from aider.dump import dump # noqa
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):
@ -63,7 +64,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"
) )
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."""
# Setup IO with no pretty # Setup IO with no pretty
@ -78,7 +79,7 @@ class TestReasoning(unittest.TestCase):
# Test data # Test data
reasoning_content = "My step-by-step reasoning process" reasoning_content = "My step-by-step reasoning process"
main_content = "Final answer after reasoning" main_content = "Final answer after reasoning"
# Create content with think tags # Create content with think tags
combined_content = f"""<think> combined_content = f"""<think>
{reasoning_content} {reasoning_content}