diff --git a/tests/basic/test_models.py b/tests/basic/test_models.py index 66bcf3095..ce560a30c 100644 --- a/tests/basic/test_models.py +++ b/tests/basic/test_models.py @@ -12,6 +12,17 @@ from aider.models import ( class TestModels(unittest.TestCase): + def setUp(self): + """Reset MODEL_SETTINGS before each test""" + from aider.models import MODEL_SETTINGS + self._original_settings = MODEL_SETTINGS.copy() + + def tearDown(self): + """Restore original MODEL_SETTINGS after each test""" + from aider.models import MODEL_SETTINGS + MODEL_SETTINGS.clear() + MODEL_SETTINGS.extend(self._original_settings) + def test_get_model_info_nonexistent(self): manager = ModelInfoManager() info = manager.get_model_info("non-existent-model")