Add a test for the validate CLI command

sdss · May 7, 2024 · 87f2897 · 87f2897
1 parent 80cdbaf
commit 87f2897
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/tests/test_cli.py b/tests/test_cli.py
@@ -113,3 +113,12 @@ def test_cli_dump(tmp_path: pathlib.Path):
 
     assert result.exit_code == 0
     assert tmp_file.exists()
+
+
+def test_cli_validate(files_path: pathlib.Path, too_mock: polars.DataFrame):
+    too_mock[150000:200000].clone().write_parquet(files_path / "too.parquet")
+
+    runner = CliRunner()
+    result = runner.invoke(too_cli, ["validate", str(files_path / "too.parquet")])
+
+    assert result.exit_code == 0