dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-28 08:14:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

aider: Replaced patching of aider.coder.backoff.expo with patching of the print function and updated the assertions accordingly.

Browse source 
# Context:
USER: instead of patching and checking that aider.coder.backoff.expo, patch and check "print"
ASSISTANT: To patch and check the `print` function instead of `aider.coder.backoff.expo`, you can modify the `test_send_with_retries_rate_limit_error` and `test_send_with_retries_connection_error` test cases in the `tests/test_coder.py` file. Here are the changes:

```python
tests/test_coder.py
<<<<<<< ORIGINAL
    @patch("aider.coder.openai.ChatCompletion.create")
    @patch("aider.coder.backoff.expo")
    def test_send_with_retries_rate_limit_error(self, mock_backoff_expo, mock_chat_completion_create):
=======
    @patch("aider.coder.openai.ChatCompletion.create")
    @patch("builtins.print")
    def test_send_with_retries_rate_limit_error(self, mock_print, mock_chat_completion_create):
>>>>>>> UPDATED
```

```python
tests/test_coder.py
<<<<<<< ORIGINAL
        # Assert that backoff.expo was called once
        mock_backoff_expo.assert_called_once()
=======
        # Assert that print was called once
        mock_print.assert_called_once()
>>>>>>> UPDATED
```

```python
tests/test_coder.py
<<<<<<< ORIGINAL
    @patch("aider.coder.openai.ChatCompletion.create")
    @patch("aider.coder.backoff.expo")
    def test_send_with_retries_connection_error(self, mock_backoff_expo, mock_chat_completion_create):
=======
    @patch("aider.coder.openai.ChatCompletion.create")
    @patch("builtins.print")
    def test_send_with_retries_connection_error(self, mock_print, mock_chat_completion_create):
>>>>>>> UPDATED
```

```python
tests/test_coder.py
<<<<<<< ORIGINAL
        # Assert that backoff.expo was called once
        mock_backoff_expo.assert_called_once()
=======
        # Assert that print was called once
        mock_print.assert_called_once()
>>>>>>> UPDATED
```

These changes replace the patching of `aider.coder.backoff.expo` with patching of the `print` function and update the assertions accordingly.
This commit is contained in:
Paul Gauthier 2023-06-04 06:05:48 -07:00
parent 0abbf43466
commit 84421d8651
1 changed files with 8 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

16
tests/test_coder.py
View file

@ -121,8 +121,8 @@ class TestCoder(unittest.TestCase):
self.assertEqual(result, 'a good "commit message"')
@patch("aider.coder.openai.ChatCompletion.create")
@patch("aider.coder.backoff.expo")
def test_send_with_retries_rate_limit_error(self, mock_backoff_expo, mock_chat_completion_create):
@patch("builtins.print")
def test_send_with_retries_rate_limit_error(self, mock_print, mock_chat_completion_create):
# Mock the IO object
mock_io = MagicMock()
@ -139,12 +139,12 @@ class TestCoder(unittest.TestCase):
# Call the send_with_retries method
coder.send_with_retries("model", ["message"])
# Assert that backoff.expo was called once
mock_backoff_expo.assert_called_once()
# Assert that print was called once
mock_print.assert_called_once()
@patch("aider.coder.openai.ChatCompletion.create")
@patch("aider.coder.backoff.expo")
def test_send_with_retries_connection_error(self, mock_backoff_expo, mock_chat_completion_create):
@patch("builtins.print")
def test_send_with_retries_connection_error(self, mock_print, mock_chat_completion_create):
# Mock the IO object
mock_io = MagicMock()
@ -161,8 +161,8 @@ class TestCoder(unittest.TestCase):
# Call the send_with_retries method
coder.send_with_retries("model", ["message"])
# Assert that backoff.expo was called once
mock_backoff_expo.assert_called_once()
# Assert that print was called once
mock_print.assert_called_once()
if __name__ == "__main__":