Tests for --table arg to contents

Author: dwhswenson

paths_cli/commands/contents.py

@@ -1,13 +1,8 @@
 import click
 from paths_cli.parameters import INPUT_FILE
 
-UNNAMED_SECTIONS = {
-    'Steps': lambda storage: storage.steps,
-    'Move Changes': lambda storage: storage.movechanges,
-    'SampleSets': lambda storage: storage.samplesets,
-    'Trajectories': lambda storage: storage.trajectories,
-    'Snapshots': lambda storage: storage.snapshots
-}
+UNNAMED_SECTIONS = ['steps', 'movechanges', 'samplesets', 'trajectories',
+                    'snapshots']
 
 @click.command(
     'contents',
@@ -52,7 +47,15 @@ def report_all_tables(storage):
     print(get_section_string_nameable('Tags', storage.tags, _get_named_tags))
 
     print("\nData Objects:")
-    for section, store_func in UNNAMED_SECTIONS.items():
+    data_object_mapping = {
+        'Steps': lambda storage: storage.steps,
+        'Move Changes': lambda storage: storage.movechanges,
+        'SampleSets': lambda storage: storage.samplesets,
+        'Trajectories': lambda storage: storage.trajectories,
+        'Snapshots': lambda storage: storage.snapshots
+    }
+
+    for section, store_func in data_object_mapping.items():
         store = store_func(storage)
         print(get_unnamed_section_string(section, store))
 

paths_cli/tests/commands/test_contents.py

@@ -40,3 +40,30 @@ def test_contents(tps_fixture):
         assert results.output.split('\n') == expected
         for truth, beauty in zip(expected, results.output.split('\n')):
             assert truth == beauty
+
+@pytest.mark.parametrize('table', ['volumes', 'trajectories'])
+def test_contents_table(tps_fixture, table):
+    scheme, network, engine, init_conds = tps_fixture
+    runner = CliRunner()
+    with runner.isolated_filesystem():
+        storage = paths.Storage("setup.nc", 'w')
+        for obj in tps_fixture:
+            storage.save(obj)
+        storage.tags['initial_conditions'] = init_conds
+
+        results = runner.invoke(contents, ['setup.nc', '--table', table])
+        cwd = os.getcwd()
+        expected = {
+            'volumes': [
+                f"Storage @ '{cwd}/setup.nc'",
+                "volumes: 8 items", "* A", "* B", "* plus 6 unnamed items",
+                ""
+            ],
+            'trajectories': [
+                f"Storage @ '{cwd}/setup.nc'",
+                "trajectories: 1 unnamed item",
+                ""
+            ]
+        }[table]
+        assert results.output.split("\n") == expected
+        assert results.exit_code == 0