mirror of
https://github.com/Aider-AI/aider.git
synced 2025-05-21 21:04:59 +00:00
fix: Reset MODEL_SETTINGS between tests to prevent parameter leakage
This commit is contained in:
parent
11a233da84
commit
aef2b95d41
1 changed files with 11 additions and 0 deletions
|
@ -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")
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue