From 3d66b5379195043460ab90cce6c4a35a46554044 Mon Sep 17 00:00:00 2001 From: "Paul Gauthier (aider)" Date: Mon, 28 Oct 2024 14:27:20 -0700 Subject: [PATCH] test: add basic test for retry_exceptions function --- aider/sendchat.py | 1 - tests/basic/test_sendchat.py | 5 ++++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/aider/sendchat.py b/aider/sendchat.py index 02494af9f..d0baf55e4 100644 --- a/aider/sendchat.py +++ b/aider/sendchat.py @@ -15,7 +15,6 @@ CACHE = None RETRY_TIMEOUT = 60 -#ai def retry_exceptions(): import httpx diff --git a/tests/basic/test_sendchat.py b/tests/basic/test_sendchat.py index a1cb8bf5d..8c73c51ca 100644 --- a/tests/basic/test_sendchat.py +++ b/tests/basic/test_sendchat.py @@ -10,8 +10,11 @@ from aider.sendchat import simple_send_with_retries class PrintCalled(Exception): pass -#ai add a test that simply invokes retry_exceptions() and ensures no error raised! class TestSendChat(unittest.TestCase): + def test_retry_exceptions(self): + """Test that retry_exceptions() can be called without raising errors""" + from aider.sendchat import retry_exceptions + retry_exceptions() # Should not raise any exceptions @patch("litellm.completion") @patch("builtins.print") def test_simple_send_with_retries_rate_limit_error(self, mock_print, mock_completion):