From 779a266713a209cc7663a06d64e0d9657b9fd78c Mon Sep 17 00:00:00 2001 From: "Paul Gauthier (aider)" Date: Fri, 7 Mar 2025 17:18:38 -0800 Subject: [PATCH] style: Apply linter formatting to test_reasoning.py --- tests/basic/test_reasoning.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/tests/basic/test_reasoning.py b/tests/basic/test_reasoning.py index 1d0110e81..c04fb4668 100644 --- a/tests/basic/test_reasoning.py +++ b/tests/basic/test_reasoning.py @@ -120,8 +120,10 @@ class TestReasoning(unittest.TestCase): mock_hash.hexdigest.return_value = "mock_hash_digest" # Mock the model's send_completion to return the hash and completion - with patch.object(model, "send_completion", return_value=(mock_hash, chunks)), \ - patch.object(model, "token_count", return_value=10): # Mock token count to avoid serialization issues + with ( + patch.object(model, "send_completion", return_value=(mock_hash, chunks)), + patch.object(model, "token_count", return_value=10), + ): # Mock token count to avoid serialization issues # Set mdstream directly on the coder object coder.mdstream = mock_mdstream @@ -255,7 +257,7 @@ class TestReasoning(unittest.TestCase): else: # Need to handle attribute access that would raise AttributeError delattr(self.choices[0].delta, "content") - + # Set reasoning_content if provided if reasoning_content is not None: self.choices[0].delta.reasoning_content = reasoning_content