Hopiu/llm
1
0
Fork 0
mirror of https://github.com/Hopiu/llm.git synced 2026-05-12 15:53:11 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Ensure test does not fail due to random sort order of plugins

Browse source
This commit is contained in:
Simon Willison 2025-05-19 19:03:26 -04:00
parent a31ae86c20
commit 4be2bbdc0e
1 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
tests/test_plugins.py
View file

@ -296,13 +296,16 @@ def test_plugins_command():
runner = CliRunner() runner = CliRunner()
result = runner.invoke(cli.cli, ["plugins"]) result = runner.invoke(cli.cli, ["plugins"])
assert result.exit_code == 0 assert result.exit_code == 0
assert json.loads(result.output) == [ expected = [
{"name": "EchoModelPlugin", "hooks": ["register_models"]}, {"name": "EchoModelPlugin", "hooks": ["register_models"]},
{ {
"name": "MockModelsPlugin", "name": "MockModelsPlugin",
"hooks": ["register_embedding_models", "register_models"], "hooks": ["register_embedding_models", "register_models"],
}, },
] ]
actual = json.loads(result.output)
actual.sort(key=lambda p: p["name"])
assert actual == expected
# Test the --hook option # Test the --hook option
result2 = runner.invoke(cli.cli, ["plugins", "--hook", "register_embedding_models"]) result2 = runner.invoke(cli.cli, ["plugins", "--hook", "register_embedding_models"])
assert result2.exit_code == 0 assert result2.exit_code == 0