refactor: Use temporary YAML file for model settings in test_default_and_override_settings

This commit is contained in:
Paul Gauthier (aider) 2024-11-19 11:49:57 -08:00
parent 6133fa8384
commit 7b9a76c5ea

View file

@ -91,20 +91,28 @@ class TestModels(unittest.TestCase):
) # Check that one of the warnings mentions the bogus model
def test_default_and_override_settings(self):
# Add default and override settings to MODEL_SETTINGS
MODEL_SETTINGS.extend(
[
ModelSettings(
name="aider/default",
edit_format="fake",
use_repo_map=True,
),
ModelSettings(
name="aider/override",
use_temperature=False,
),
import tempfile
import yaml
# Create temporary YAML file with test settings
test_settings = [
{
"name": "aider/default",
"edit_format": "fake",
"use_repo_map": True,
},
{
"name": "aider/override",
"use_temperature": False,
}
]
)
with tempfile.NamedTemporaryFile(mode='w', suffix='.yml') as tmp:
yaml.dump(test_settings, tmp)
tmp.flush()
# Register the test settings
register_models([tmp.name])
# Test that defaults are applied when no exact match
model = Model("unknown-model")