diff --git a/pyproject.toml b/pyproject.toml index 612739d..758082e 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -24,11 +24,11 @@ dependencies = [ "openai==1.3.8", "prompt-toolkit==3.0.41", "pytest==7.3.1", - "PyYAML==6.0", + "PyYAML==6.0.1", "rich==13.7.0", "tiktoken==0.5.2", "tokenizers==0.15.0", - "typing_extensions==4.5.0", + "typing_extensions==4.7.1", ] [project.optional-dependencies] diff --git a/tests/test_gpt.py b/tests/test_gpt.py new file mode 100644 index 0000000..b0f6314 --- /dev/null +++ b/tests/test_gpt.py @@ -0,0 +1,18 @@ + +import pytest +import gptcli +import sys +from gptcli.gpt import main + +def test_version_argument(monkeypatch, capsys): + monkeypatch.setattr(sys, "argv", ["gpt", "-v"]) + monkeypatch.setattr("gptcli.gpt.choose_config_file", lambda x: None) + + with pytest.raises(SystemExit) as e: + main() + + assert e.type == SystemExit + assert e.value.code == 0 + + captured = capsys.readouterr() + assert f"gpt-cli v{gptcli.__version__}\n" == captured.out