patch check_model_availability in tests

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()

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)