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

style: Apply linter formatting to test_models.py

Browse source
This commit is contained in:
Paul Gauthier (aider) 2024-11-19 10:04:23 -08:00
parent 7c8f10e832
commit 210500ff3e
1 changed files with 16 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

28
tests/basic/test_models.py
View file

@ -90,17 +90,19 @@ class TestModels(unittest.TestCase):
def test_default_and_override_settings(self): def test_default_and_override_settings(self):
# Add default and override settings to MODEL_SETTINGS # Add default and override settings to MODEL_SETTINGS
MODEL_SETTINGS.extend([ MODEL_SETTINGS.extend(
ModelSettings( [
name="aider/default", ModelSettings(
edit_format="diff", name="aider/default",
use_repo_map=True, edit_format="diff",
), use_repo_map=True,
ModelSettings( ),
name="aider/override", ModelSettings(
use_temperature=False, name="aider/override",
), use_temperature=False,
]) ),
]
)
# Test that defaults are applied when no exact match # Test that defaults are applied when no exact match
model = Model("unknown-model") model = Model("unknown-model")
@ -114,7 +116,9 @@ class TestModels(unittest.TestCase):
self.assertFalse(model.use_temperature) # Override should still win self.assertFalse(model.use_temperature) # Override should still win
# Clean up by removing test settings # Clean up by removing test settings
MODEL_SETTINGS[:] = [ms for ms in MODEL_SETTINGS if ms.name not in ("aider/default", "aider/override")] MODEL_SETTINGS[:] = [
ms for ms in MODEL_SETTINGS if ms.name not in ("aider/default", "aider/override")
]
if __name__ == "__main__": if __name__ == "__main__":