llm/tests/test_aliases.py
2023-08-11 23:18:46 -07:00

17 lines
502 B
Python

from click.testing import CliRunner
from llm.cli import cli
def test_aliases_list():
runner = CliRunner()
result = runner.invoke(cli, ["aliases", "list"])
assert result.exit_code == 0
assert result.output == (
"3.5 : gpt-3.5-turbo\n"
"chatgpt : gpt-3.5-turbo\n"
"chatgpt-16k : gpt-3.5-turbo-16k\n"
"3.5-16k : gpt-3.5-turbo-16k\n"
"4 : gpt-4\n"
"gpt4 : gpt-4\n"
"4-32k : gpt-4-32k\n"
)