llm/tests/test_llm.py

11 lines
291 B
Python
Raw Normal View History

2023-04-01 21:28:24 +00:00
from click.testing import CliRunner
2023-04-01 22:00:16 +00:00
from llm.cli import cli
2023-04-01 21:28:24 +00:00
2023-04-01 22:00:16 +00:00
def test_version():
2023-04-01 21:28:24 +00:00
runner = CliRunner()
with runner.isolated_filesystem():
2023-04-01 22:00:16 +00:00
result = runner.invoke(cli, ["--version"])
assert result.exit_code == 0
assert result.output.startswith("cli, version ")