From d58e5421b08a8193497c03c5d2e1a040e9f511e7 Mon Sep 17 00:00:00 2001 From: Paul Gauthier Date: Tue, 20 Jun 2023 17:36:24 -0700 Subject: [PATCH] fixed tests --- tests/test_coder.py | 9 +++++---- tests/test_utils.py | 2 +- 2 files changed, 6 insertions(+), 5 deletions(-) diff --git a/tests/test_coder.py b/tests/test_coder.py index 4125ee373..3ccca462a 100644 --- a/tests/test_coder.py +++ b/tests/test_coder.py @@ -5,12 +5,13 @@ from unittest.mock import MagicMock, patch import openai import requests -from aider.coder import Coder +from aider import models +from aider.coders import Coder class TestCoder(unittest.TestCase): def setUp(self): - self.patcher = patch("aider.coder.Coder.check_model_availability") + self.patcher = patch("aider.coders.base.check_model_availability") self.mock_check = self.patcher.start() self.mock_check.return_value = True @@ -120,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.coder.openai.ChatCompletion.create") + @patch("aider.coders.base.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 @@ -142,7 +143,7 @@ class TestCoder(unittest.TestCase): # Assert that print was called once mock_print.assert_called_once() - @patch("aider.coder.openai.ChatCompletion.create") + @patch("aider.coders.base.openai.ChatCompletion.create") @patch("builtins.print") def test_send_with_retries_connection_error(self, mock_print, mock_chat_completion_create): # Mock the IO object diff --git a/tests/test_utils.py b/tests/test_utils.py index a5ca4f40c..2444a4a1f 100644 --- a/tests/test_utils.py +++ b/tests/test_utils.py @@ -2,7 +2,7 @@ import unittest -from aider import utils +import aider.coders.editblock as utils class TestUtils(unittest.TestCase):