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

style: Format test_models.py with linter

Browse source
This commit is contained in:
Paul Gauthier (aider) 2025-02-06 08:36:38 -08:00
parent 856006a68d
commit 51938affc2
1 changed files with 3 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

8
tests/basic/test_models.py
View file

@ -301,16 +301,14 @@ End"""
@patch("aider.models.litellm.completion") @patch("aider.models.litellm.completion")
def test_simple_send_with_retries_removes_reasoning(self, mock_completion): def test_simple_send_with_retries_removes_reasoning(self, mock_completion):
model = Model("deepseek-r1") # This model has remove_reasoning="think" model = Model("deepseek-r1") # This model has remove_reasoning="think"
# Mock the completion response # Mock the completion response
mock_response = MagicMock() mock_response = MagicMock()
mock_response.choices = [ mock_response.choices = [MagicMock(message=MagicMock(content="""Here is some text
MagicMock(message=MagicMock(content="""Here is some text
<think> <think>
This reasoning should be removed This reasoning should be removed
</think> </think>
And this text should remain""")) And this text should remain"""))]
]
mock_completion.return_value = mock_response mock_completion.return_value = mock_response
messages = [{"role": "user", "content": "test"}] messages = [{"role": "user", "content": "test"}]