Updated version output check

smart-on-fhir · Feb 11, 2025 · b5893d7 · b5893d7
1 parent 658ff62
commit b5893d7
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/cumulus_library/cli.py b/cumulus_library/cli.py
@@ -441,7 +441,7 @@ def main(cli_args=None):
         sys.exit(1)
 
     if args["action"] == "version":
-        table = rich.table.Table(title=f"cumulus library version: {__version__}")
+        table = rich.table.Table(title=f"cumulus-library version: {__version__}")
         table.add_column("Study Name", style="green")
         table.add_column("Version", style="blue")
         studies = get_study_dict(args.get("study_dir"))

diff --git a/tests/test_cli.py b/tests/test_cli.py
@@ -866,8 +866,13 @@ def test_version(capfd):
         )
     out, _ = capfd.readouterr()
     assert f"cumulus-library version: {__version__}" in out
-    assert "study_valid: 1.0.0" in out
-    assert "study_invalid_bad_query: no version defined" in out
+    out = out.split("\n")
+    valid = list(filter(lambda x: " study_valid " in x, out))
+    assert len(valid) == 1
+    assert " 1.0.0 " in valid[0]
+    invalid = list(filter(lambda x: " study_invalid_bad_query " in x, out))
+    assert len(invalid) == 1
+    assert " No version defined " in invalid[0]
 
 
 @mock.patch.dict(