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

fix: update test cases to use Model wrapper class

Browse source
This commit is contained in:
Paul Gauthier 2024-12-07 15:09:04 -08:00 • committed by Paul Gauthier (aider)
parent a74cdbfc28
commit 42ae279b91
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
tests/basic/test_sendchat.py
View file

@ -75,7 +75,7 @@ class TestSendChat(unittest.TestCase):
mock_completion.return_value.choices = None mock_completion.return_value.choices = None
# Should return None on AttributeError # Should return None on AttributeError
result = simple_send_with_retries(self.mock_model, self.mock_messages) result = simple_send_with_retries(Model(self.mock_model), self.mock_messages)
assert result is None assert result is None
@patch("litellm.completion") @patch("litellm.completion")
@ -89,7 +89,7 @@ class TestSendChat(unittest.TestCase):
message="Invalid request", llm_provider="test_provider", model="test_model" message="Invalid request", llm_provider="test_provider", model="test_model"
) )
result = simple_send_with_retries(self.mock_model, self.mock_messages) result = simple_send_with_retries(Model(self.mock_model), self.mock_messages)
assert result is None assert result is None
# Should only print the error message # Should only print the error message
assert mock_print.call_count == 1 assert mock_print.call_count == 1