Add tests to run CLI

Author: clane9

tests/test_indexing.py

@@ -4,6 +4,7 @@
 import pytest
 
 import bids2table._indexing as indexing
+from bids2table._pathlib import cloudpathlib_is_available
 
 BIDS_EXAMPLES = Path(__file__).parents[1] / "bids-examples"
 
@@ -54,6 +55,16 @@ def test_index_dataset(root: str, expected_count: int):
     assert len(table) == expected_count
 
 
+@pytest.mark.skipif(
+    not cloudpathlib_is_available(), reason="cloudpathlib not installed"
+)
+def test_index_dataset_s3():
+    root = "s3://openneuro.org/ds000102"
+    expected_count = 130
+    table = indexing.index_dataset(root)
+    assert len(table) == expected_count
+
+
 @pytest.mark.parametrize("max_workers", [0, 2])
 def test_index_dataset_parallel(max_workers: int):
     root, expected_count = "ds102", 130

tests/test_main.py

@@ -0,0 +1,51 @@
+import shlex
+import sys
+from contextlib import contextmanager
+from pathlib import Path
+from typing import List
+
+import pytest
+
+from bids2table import __main__ as cli
+
+BIDS_EXAMPLES = Path(__file__).parents[1] / "bids-examples"
+
+COMMANDS = [
+    "find {examples}",
+    "index -o {out_dir}/ds102.parquet {examples}/ds102",
+    "index -o {out_dir}/ds101_ds102.parquet {examples}/ds101 {examples}/ds102",
+    "index -o {out_dir}/ds10N.parquet '{examples}/ds10?'",
+]
+
+
+@contextmanager
+def patch_argv(argv: List[str]):
+    old_argv = sys.argv
+    try:
+        sys.argv = argv.copy()
+        yield
+    finally:
+        sys.argv = old_argv
+
+
+@pytest.mark.parametrize(
+    "cmd,output",
+    [
+        ("find {examples}", None),
+        ("index -o {out_dir}/ds102.parquet {examples}/ds102", "ds102.parquet"),
+        (
+            "index -o {out_dir}/ds101_ds102.parquet {examples}/ds101 {examples}/ds102",
+            "ds101_ds102.parquet",
+        ),
+        ("index -o {out_dir}/ds10N.parquet '{examples}/ds10?'", "ds10N.parquet"),
+    ],
+)
+def test_main(cmd: str, output: str | None, tmp_path: Path):
+    cmd_fmt = cmd.format(out_dir=tmp_path, examples=BIDS_EXAMPLES)
+    prog = str(Path(cli.__file__).absolute())
+    argv = [prog] + shlex.split(cmd_fmt)
+    with patch_argv(argv):
+        cli.main()
+
+    if output:
+        assert (tmp_path / output).exists()