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

test: update print call count assertions in sendchat tests

Browse source
This commit is contained in:
Paul Gauthier (aider) 2024-12-07 13:44:08 -08:00
parent c9c2d5ab6f
commit a56fa567dd
1 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
tests/basic/test_sendchat.py
View file

@ -37,8 +37,9 @@ class TestSendChat(unittest.TestCase):
]
# Call the simple_send_with_retries method
simple_send_with_retries("model", ["message"])
assert mock_print.call_count == 3
simple_send_with_retries(self.mock_model, self.mock_messages)
# Should print: error message, description (if any), and retry message
assert mock_print.call_count == 2 # Error message + retry message
@patch("litellm.completion")
def test_send_completion_basic(self, mock_completion):
@ -90,4 +91,5 @@ class TestSendChat(unittest.TestCase):
result = simple_send_with_retries(self.mock_model, self.mock_messages)
assert result is None
# Should only print the error message
assert mock_print.call_count == 1