dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-20 04:14:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

test: update model listing test to use metadata-only model

Browse source
This commit is contained in:
Paul Gauthier 2025-03-25 14:51:24 -10:00 • committed by Paul Gauthier (aider)
parent 6933bc8add
commit 70847a74c2
1 changed files with 20 additions and 24 deletions
Download patch file Download diff file Expand all files Collapse all files

44
tests/basic/test_main.py
View file

@ -1110,35 +1110,31 @@ class TestMain(TestCase):
"metadata-only-model": {
"max_input_tokens": 8192,
"litellm_provider": "test-provider",
"mode": "chat", # Added mode attribute
}
}
metadata_file.write_text(json.dumps(test_models))
# Patch litellm.model_cost to include a test model
litellm_test_model = "litellm-only-model"
with patch(
"aider.models.litellm.model_cost",
{litellm_test_model: {"mode": "chat", "litellm_provider": "test-provider"}},
):
# Capture stdout to check the output
with patch("sys.stdout", new_callable=StringIO) as mock_stdout:
main(
[
"--list-models",
"model",
"--model-metadata-file",
str(metadata_file),
"--yes",
"--no-gitignore",
],
input=DummyInput(),
output=DummyOutput(),
)
output = mock_stdout.getvalue()
# Capture stdout to check the output
with patch("sys.stdout", new_callable=StringIO) as mock_stdout:
main(
[
"--list-models",
"metadata-only-model",
"--model-metadata-file",
str(metadata_file),
"--yes",
"--no-gitignore",
],
input=DummyInput(),
output=DummyOutput(),
)
output = mock_stdout.getvalue()
# Check that both models appear in the output
self.assertIn("test-provider/metadata-only-model", output)
self.assertIn(litellm_test_model, output)
dump(output)
# Check that both models appear in the output
self.assertIn("test-provider/metadata-only-model", output)
def test_check_model_accepts_settings_flag(self):
# Test that --check-model-accepts-settings affects whether settings are applied