mirror of
https://github.com/Aider-AI/aider.git
synced 2025-05-29 08:44:59 +00:00
test: Add tests for default and override model settings
This commit is contained in:
parent
644da2f9e4
commit
7c8f10e832
1 changed files with 28 additions and 0 deletions
|
@ -88,6 +88,34 @@ class TestModels(unittest.TestCase):
|
||||||
any("bogus-model" in msg for msg in warning_messages)
|
any("bogus-model" in msg for msg in warning_messages)
|
||||||
) # Check that one of the warnings mentions the bogus model
|
) # 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="diff",
|
||||||
|
use_repo_map=True,
|
||||||
|
),
|
||||||
|
ModelSettings(
|
||||||
|
name="aider/override",
|
||||||
|
use_temperature=False,
|
||||||
|
),
|
||||||
|
])
|
||||||
|
|
||||||
|
# Test that defaults are applied when no exact match
|
||||||
|
model = Model("unknown-model")
|
||||||
|
self.assertEqual(model.edit_format, "diff")
|
||||||
|
self.assertTrue(model.use_repo_map)
|
||||||
|
self.assertFalse(model.use_temperature) # Override should win
|
||||||
|
|
||||||
|
# Test that exact match overrides defaults but not overrides
|
||||||
|
model = Model("gpt-4")
|
||||||
|
self.assertNotEqual(model.edit_format, "diff") # Model setting should win over default
|
||||||
|
self.assertFalse(model.use_temperature) # Override should still win
|
||||||
|
|
||||||
|
# Clean up by removing test settings
|
||||||
|
MODEL_SETTINGS[:] = [ms for ms in MODEL_SETTINGS if ms.name not in ("aider/default", "aider/override")]
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
unittest.main()
|
unittest.main()
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue