mirror of
https://github.com/Aider-AI/aider.git
synced 2025-05-21 21:04:59 +00:00
style: Apply linter formatting to test_models.py
This commit is contained in:
parent
7c8f10e832
commit
210500ff3e
1 changed files with 16 additions and 12 deletions
|
@ -90,17 +90,19 @@ class TestModels(unittest.TestCase):
|
|||
|
||||
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,
|
||||
),
|
||||
])
|
||||
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")
|
||||
|
@ -114,7 +116,9 @@ class TestModels(unittest.TestCase):
|
|||
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")]
|
||||
MODEL_SETTINGS[:] = [
|
||||
ms for ms in MODEL_SETTINGS if ms.name not in ("aider/default", "aider/override")
|
||||
]
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue