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

test: update print call count assertions in sendchat tests

Browse source
This commit is contained in:
Paul Gauthier (aider) 2024-10-28 15:09:11 -07:00
parent 29250f82ed
commit 0351924628
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

@ -38,7 +38,7 @@ class TestSendChat(unittest.TestCase):
# Call the simple_send_with_retries method # Call the simple_send_with_retries method
simple_send_with_retries("model", ["message"]) simple_send_with_retries("model", ["message"])
mock_print.assert_called_once() assert mock_print.call_count == 2
@patch("litellm.completion") @patch("litellm.completion")
@patch("builtins.print") @patch("builtins.print")
@ -51,4 +51,4 @@ class TestSendChat(unittest.TestCase):
# Call the simple_send_with_retries method # Call the simple_send_with_retries method
simple_send_with_retries("model", ["message"]) simple_send_with_retries("model", ["message"])
mock_print.assert_called_once() assert mock_print.call_count == 2