style: wrap long lines in test docstrings and comments

This commit is contained in:
Paul Gauthier (aider) 2025-03-09 08:43:45 -07:00
parent eadb8d5d0a
commit a1f104cb4d

View file

@ -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 "),