dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-23 13:54:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

fix: Reset MODEL_SETTINGS between tests to prevent parameter leakage

Browse source
This commit is contained in:
Paul Gauthier (aider) 2025-02-06 08:21:42 -08:00
parent 11a233da84
commit aef2b95d41
1 changed files with 11 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

11
tests/basic/test_models.py
View file

@ -12,6 +12,17 @@ from aider.models import (
class TestModels(unittest.TestCase): 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): def test_get_model_info_nonexistent(self):
manager = ModelInfoManager() manager = ModelInfoManager()
info = manager.get_model_info("non-existent-model") info = manager.get_model_info("non-existent-model")