dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-21 12:55:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

refac

Browse source
This commit is contained in:
Paul Gauthier 2023-06-20 17:46:47 -07:00
parent e22373b55a
commit 070f0a2919
7 changed files with 22 additions and 22 deletions
Download patch file Download diff file Expand all files Collapse all files

6
tests/test_coder.py
View file

@ -11,7 +11,7 @@ from aider.coders import Coder
class TestCoder(unittest.TestCase):
def setUp(self):
self.patcher = patch("aider.coders.base.check_model_availability")
self.patcher = patch("aider.coders.base_coder.check_model_availability")
self.mock_check = self.patcher.start()
self.mock_check.return_value = True
@ -121,7 +121,7 @@ class TestCoder(unittest.TestCase):
# Assert that the returned message is the expected one
self.assertEqual(result, 'a good "commit message"')
@patch("aider.coders.base.openai.ChatCompletion.create")
@patch("aider.coders.base_coder.openai.ChatCompletion.create")
@patch("builtins.print")
def test_send_with_retries_rate_limit_error(self, mock_print, mock_chat_completion_create):
# Mock the IO object
@ -143,7 +143,7 @@ class TestCoder(unittest.TestCase):
# Assert that print was called once
mock_print.assert_called_once()
@patch("aider.coders.base.openai.ChatCompletion.create")
@patch("aider.coders.base_coder.openai.ChatCompletion.create")
@patch("builtins.print")
def test_send_with_retries_connection_error(self, mock_print, mock_chat_completion_create):
# Mock the IO object