From 0351924628e48c5ac8654d3688d9536671fdd18f Mon Sep 17 00:00:00 2001 From: "Paul Gauthier (aider)" Date: Mon, 28 Oct 2024 15:09:11 -0700 Subject: [PATCH] test: update print call count assertions in sendchat tests --- tests/basic/test_sendchat.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/basic/test_sendchat.py b/tests/basic/test_sendchat.py index 56c5ea7b7..459937b1e 100644 --- a/tests/basic/test_sendchat.py +++ b/tests/basic/test_sendchat.py @@ -38,7 +38,7 @@ class TestSendChat(unittest.TestCase): # Call the simple_send_with_retries method simple_send_with_retries("model", ["message"]) - mock_print.assert_called_once() + assert mock_print.call_count == 2 @patch("litellm.completion") @patch("builtins.print") @@ -51,4 +51,4 @@ class TestSendChat(unittest.TestCase): # Call the simple_send_with_retries method simple_send_with_retries("model", ["message"]) - mock_print.assert_called_once() + assert mock_print.call_count == 2