From aef2b95d413eaeaa3a1f4f9dae2128bb45fe743a Mon Sep 17 00:00:00 2001 From: "Paul Gauthier (aider)" Date: Thu, 6 Feb 2025 08:21:42 -0800 Subject: [PATCH] fix: Reset MODEL_SETTINGS between tests to prevent parameter leakage --- tests/basic/test_models.py | 11 +++++++++++ 1 file changed, 11 insertions(+) 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")