From 51938affc2c53c47499198757e4846ebebecd362 Mon Sep 17 00:00:00 2001 From: "Paul Gauthier (aider)" Date: Thu, 6 Feb 2025 08:36:38 -0800 Subject: [PATCH] style: Format test_models.py with linter --- tests/basic/test_models.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/tests/basic/test_models.py b/tests/basic/test_models.py index 3c62339dd..d8750ddb7 100644 --- a/tests/basic/test_models.py +++ b/tests/basic/test_models.py @@ -301,16 +301,14 @@ End""" @patch("aider.models.litellm.completion") def test_simple_send_with_retries_removes_reasoning(self, mock_completion): model = Model("deepseek-r1") # This model has remove_reasoning="think" - + # Mock the completion response mock_response = MagicMock() - mock_response.choices = [ - MagicMock(message=MagicMock(content="""Here is some text + mock_response.choices = [MagicMock(message=MagicMock(content="""Here is some text This reasoning should be removed -And this text should remain""")) - ] +And this text should remain"""))] mock_completion.return_value = mock_response messages = [{"role": "user", "content": "test"}]