diff --git a/tests/test_coder.py b/tests/test_coder.py index 52ffabdb9..f703969e3 100644 --- a/tests/test_coder.py +++ b/tests/test_coder.py @@ -1,11 +1,19 @@ import os import unittest -from unittest.mock import MagicMock +from unittest.mock import MagicMock, patch from aider.coder import Coder class TestCoder(unittest.TestCase): + def setUp(self): + self.patcher = patch("aider.coder.Coder.check_model_availability") + self.mock_check = self.patcher.start() + self.mock_check.return_value = True + + def tearDown(self): + self.patcher.stop() + def test_check_for_file_mentions(self): # Mock the IO object mock_io = MagicMock() diff --git a/tests/test_commands.py b/tests/test_commands.py index 5f3d596d1..223c33acd 100644 --- a/tests/test_commands.py +++ b/tests/test_commands.py @@ -2,6 +2,7 @@ import os import shutil import tempfile from unittest import TestCase +from unittest.mock import patch from aider.commands import Commands from aider.io import InputOutput @@ -13,10 +14,16 @@ class TestCommands(TestCase): self.tempdir = tempfile.mkdtemp() os.chdir(self.tempdir) + self.patcher = patch("aider.coder.Coder.check_model_availability") + self.mock_check = self.patcher.start() + self.mock_check.return_value = True + def tearDown(self): os.chdir(self.original_cwd) shutil.rmtree(self.tempdir) + self.patcher.stop() + def test_cmd_add(self): # Initialize the Commands and InputOutput objects io = InputOutput(pretty=False, yes=True)