mirror of
https://github.com/Aider-AI/aider.git
synced 2025-05-29 08:44:59 +00:00
style: wrap long lines in test docstrings and comments
This commit is contained in:
parent
eadb8d5d0a
commit
a1f104cb4d
1 changed files with 6 additions and 3 deletions
|
@ -400,7 +400,8 @@ End"""
|
||||||
self.assertEqual(remove_reasoning_content(text, "think"), text)
|
self.assertEqual(remove_reasoning_content(text, "think"), text)
|
||||||
|
|
||||||
def test_send_with_reasoning(self):
|
def test_send_with_reasoning(self):
|
||||||
"""Test that reasoning content from the 'reasoning' attribute is properly formatted and output."""
|
"""Test that reasoning content from the 'reasoning' attribute is properly formatted
|
||||||
|
and output."""
|
||||||
# Setup IO with no pretty
|
# Setup IO with no pretty
|
||||||
io = InputOutput(pretty=False)
|
io = InputOutput(pretty=False)
|
||||||
io.assistant_output = MagicMock()
|
io.assistant_output = MagicMock()
|
||||||
|
@ -463,7 +464,8 @@ End"""
|
||||||
)
|
)
|
||||||
|
|
||||||
def test_send_with_reasoning_stream(self):
|
def test_send_with_reasoning_stream(self):
|
||||||
"""Test that streaming reasoning content from the 'reasoning' attribute is properly formatted and output."""
|
"""Test that streaming reasoning content from the 'reasoning' attribute is properly
|
||||||
|
formatted and output."""
|
||||||
# Setup IO with pretty output for streaming
|
# Setup IO with pretty output for streaming
|
||||||
io = InputOutput(pretty=True)
|
io = InputOutput(pretty=True)
|
||||||
mock_mdstream = MagicMock()
|
mock_mdstream = MagicMock()
|
||||||
|
@ -506,7 +508,8 @@ End"""
|
||||||
# Need to handle attribute access that would raise AttributeError
|
# Need to handle attribute access that would raise AttributeError
|
||||||
delattr(self.choices[0].delta, "reasoning")
|
delattr(self.choices[0].delta, "reasoning")
|
||||||
|
|
||||||
# Create chunks to simulate streaming - using reasoning attribute instead of reasoning_content
|
# Create chunks to simulate streaming - using reasoning attribute instead of
|
||||||
|
# reasoning_content
|
||||||
chunks = [
|
chunks = [
|
||||||
# First chunk with reasoning content starts the tag
|
# First chunk with reasoning content starts the tag
|
||||||
MockStreamingChunk(reasoning="My step-by-step "),
|
MockStreamingChunk(reasoning="My step-by-step "),
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue