mirror of
https://github.com/Aider-AI/aider.git
synced 2025-05-29 08:44:59 +00:00
style: Organize and clean up import statements in test_reasoning.py
This commit is contained in:
parent
b230fea66f
commit
3b9c2b9729
1 changed files with 4 additions and 3 deletions
|
@ -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}
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue