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

style: Format strings with double quotes in test_models.py

Browse source
This commit is contained in:
Paul Gauthier (aider) 2025-02-06 08:19:25 -08:00
parent 9094af565f
commit 016aa87e34
1 changed files with 12 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

10
tests/basic/test_models.py
View file

@ -208,8 +208,8 @@ class TestModels(unittest.TestCase):
except OSError: except OSError:
pass pass
@patch('aider.models.litellm.completion') @patch("aider.models.litellm.completion")
@patch.object(Model, 'token_count') @patch.object(Model, "token_count")
def test_ollama_num_ctx_set_when_missing(self, mock_token_count, mock_completion): def test_ollama_num_ctx_set_when_missing(self, mock_token_count, mock_completion):
mock_token_count.return_value = 1000 mock_token_count.return_value = 1000
@ -230,7 +230,7 @@ class TestModels(unittest.TestCase):
tool_choice=None, tool_choice=None,
) )
@patch('aider.models.litellm.completion') @patch("aider.models.litellm.completion")
def test_ollama_uses_existing_num_ctx(self, mock_completion): def test_ollama_uses_existing_num_ctx(self, mock_completion):
model = Model("ollama/llama3") model = Model("ollama/llama3")
model.extra_params = {"num_ctx": 4096} model.extra_params = {"num_ctx": 4096}
@ -249,7 +249,7 @@ class TestModels(unittest.TestCase):
tool_choice=None, tool_choice=None,
) )
@patch('aider.models.litellm.completion') @patch("aider.models.litellm.completion")
def test_non_ollama_no_num_ctx(self, mock_completion): def test_non_ollama_no_num_ctx(self, mock_completion):
model = Model("gpt-4") model = Model("gpt-4")
messages = [{"role": "user", "content": "Hello"}] messages = [{"role": "user", "content": "Hello"}]
@ -265,7 +265,7 @@ class TestModels(unittest.TestCase):
tools=None, tools=None,
tool_choice=None, tool_choice=None,
) )
self.assertNotIn('num_ctx', mock_completion.call_args.kwargs) self.assertNotIn("num_ctx", mock_completion.call_args.kwargs)
if __name__ == "__main__": if __name__ == "__main__":