fixed tests

This commit is contained in:
Paul Gauthier 2023-06-21 21:36:36 -07:00
parent 0aafff4bc5
commit 2edbaa6e7a

View file

@ -138,7 +138,7 @@ class TestCoder(unittest.TestCase):
]
# Call the send_with_retries method
coder.send_with_retries("model", ["message"])
coder.send_with_retries("model", ["message"], None)
# Assert that print was called once
mock_print.assert_called_once()
@ -160,7 +160,7 @@ class TestCoder(unittest.TestCase):
]
# Call the send_with_retries method
coder.send_with_retries("model", ["message"])
coder.send_with_retries("model", ["message"], None)
# Assert that print was called once
mock_print.assert_called_once()