test(cli): cover local version in splash and --version flag

Author: harryswift01

tests/unit/CodeEntropy/config/argparse/test_argparse_build_parser.py

@@ -1,3 +1,6 @@
+import pytest
+
+from CodeEntropy import __version__
 from CodeEntropy.config.argparse import ConfigResolver
 
 
@@ -37,3 +40,13 @@ def test_build_parser_nargs_plus_parses_top_traj_file_list():
     args = parser.parse_args(["--top_traj_file", "a.tpr", "b.trr"])
 
     assert args.top_traj_file == ["a.tpr", "b.trr"]
+
+
+def test_version_flag_prints_installed_version_and_exits(capsys):
+    parser = ConfigResolver().build_parser()
+
+    with pytest.raises(SystemExit) as exc:
+        parser.parse_args(["--version"])
+
+    assert exc.value.code == 0
+    assert f"CodeEntropy {__version__}" in capsys.readouterr().out

tests/unit/CodeEntropy/config/runtime/test_show_splash.py

@@ -3,6 +3,8 @@
 
 from rich.console import Console
 
+from CodeEntropy import __version__
+
 
 def test_show_splash_with_citation(runner):
     citation = {
@@ -26,8 +28,8 @@ def test_show_splash_with_citation(runner):
     out = buf.getvalue()
 
     assert "Welcome to CodeEntropy" in out
-    assert "Version 1.0" in out
-    assert "2025-01-01" in out
+    assert f"Version {__version__}" in out
+    assert "Version 1.0" not in out
     assert "https://example.com" in out
     assert "This is a test abstract." in out
     assert "Alice Smith" in out
@@ -43,4 +45,7 @@ def test_show_splash_without_citation(runner):
     ):
         runner.show_splash()
 
-    assert "Welcome to CodeEntropy" in buf.getvalue()
+    out = buf.getvalue()
+
+    assert "Welcome to CodeEntropy" in out
+    assert f"Version {__version__}" in out