dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-22 21:34:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

style: Apply linter formatting to test_reasoning.py

Browse source
This commit is contained in:
Paul Gauthier (aider) 2025-03-07 17:18:38 -08:00
parent ebaedc6f05
commit 779a266713
1 changed files with 5 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

8
tests/basic/test_reasoning.py
View file

@ -120,8 +120,10 @@ class TestReasoning(unittest.TestCase):
mock_hash.hexdigest.return_value = "mock_hash_digest" mock_hash.hexdigest.return_value = "mock_hash_digest"
# Mock the model's send_completion to return the hash and completion # Mock the model's send_completion to return the hash and completion
with patch.object(model, "send_completion", return_value=(mock_hash, chunks)), \ with (
patch.object(model, "token_count", return_value=10): # Mock token count to avoid serialization issues 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 # Set mdstream directly on the coder object
coder.mdstream = mock_mdstream coder.mdstream = mock_mdstream
@ -255,7 +257,7 @@ class TestReasoning(unittest.TestCase):
else: else:
# Need to handle attribute access that would raise AttributeError # Need to handle attribute access that would raise AttributeError
delattr(self.choices[0].delta, "content") delattr(self.choices[0].delta, "content")
# Set reasoning_content if provided # Set reasoning_content if provided
if reasoning_content is not None: if reasoning_content is not None:
self.choices[0].delta.reasoning_content = reasoning_content self.choices[0].delta.reasoning_content = reasoning_content