style: Apply linter formatting to test_models.py

This commit is contained in:
Paul Gauthier (aider) 2024-11-19 10:04:23 -08:00
parent 7c8f10e832
commit 210500ff3e

View file

@ -90,7 +90,8 @@ 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( ModelSettings(
name="aider/default", name="aider/default",
edit_format="diff", edit_format="diff",
@ -100,7 +101,8 @@ class TestModels(unittest.TestCase):
name="aider/override", name="aider/override",
use_temperature=False, 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__":