From 93cb615ffde4af2c868acb84d85b57a8acf1ff6a Mon Sep 17 00:00:00 2001 From: "Paul Gauthier (aider)" Date: Wed, 2 Oct 2024 11:15:27 -0700 Subject: [PATCH] test: add test for sanity check with bogus editor model --- tests/basic/test_models.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/basic/test_models.py b/tests/basic/test_models.py index efb9bacd1..397b31485 100644 --- a/tests/basic/test_models.py +++ b/tests/basic/test_models.py @@ -1,7 +1,7 @@ import unittest from unittest.mock import MagicMock, patch -from aider.models import Model, get_model_info, sanity_check_model, sanity_check_models +from aider.models import Model, get_model_info, sanity_check_model class TestModels(unittest.TestCase): @@ -74,7 +74,7 @@ class TestModels(unittest.TestCase): # Set up mock to return False for main model and True (problem) for editor model mock_sanity_check_model.side_effect = [False, True] - result = models.sanity_check_models(mock_io, main_model) + result = sanity_check_models(mock_io, main_model) self.assertTrue( result