add test case for cli to improve test coverage

ourownstory · Nov 22, 2022 · d568df2 · d568df2
1 parent 8e85474
commit d568df2
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/tests/test_cli.py b/tests/test_cli.py
@@ -0,0 +1,12 @@
+import pytest
+from neuralprophet._version import __version__
+from neuralprophet.__main__ import parser
+
+
+def test_main_file(capsys):
+    with pytest.raises(SystemExit) as exit_info:
+        parser.parse_args(["--version"])
+
+    out, _ = capsys.readouterr()
+    assert "<ExceptionInfo SystemExit(0) tblen=8>" == str(exit_info)
+    assert __version__ in out