feat(tests): add parquet size checks and GeoJSON parity validation for HETA 1.2.0

Author: ari-nz

tests/aignostics/application/cli_test.py

@@ -1111,8 +1111,8 @@ def test_cli_run_execute(runner: CliRunner, tmp_path: Path, record_property) ->
     results_dir = tmp_path / SPOT_1_FILENAME.replace(".tiff", "")
     assert results_dir.is_dir(), f"Expected directory {results_dir} not found"
     files_in_dir = list(results_dir.glob("*"))
-    assert len(files_in_dir) == 9, (
-        f"Expected 9 files in {results_dir}, but found {len(files_in_dir)}: {[f.name for f in files_in_dir]}"
+    assert len(files_in_dir) == 12, (
+        f"Expected 12 files in {results_dir}, but found {len(files_in_dir)}: {[f.name for f in files_in_dir]}"
     )
     print(f"Found files in {results_dir}:")
     for filename, expected_size, tolerance_percent in SPOT_1_EXPECTED_RESULT_FILES:
@@ -1133,6 +1133,25 @@ def test_cli_run_execute(runner: CliRunner, tmp_path: Path, record_property) ->
             f"({min_size} to {max_size} bytes, ±{tolerance_percent}% of {expected_size})"
         )
 
+    # Validate parquet <-> GeoJSON row count parity for the 3 paired outputs
+    import pandas as pd
+
+    parquet_geojson_pairs = [
+        ("tissue_qc_parquet_polygons.parquet", "tissue_qc_geojson_polygons.json"),
+        ("tissue_segmentation_parquet_polygons.parquet", "tissue_segmentation_geojson_polygons.json"),
+        ("cell_classification_parquet_polygons.parquet", "cell_classification_geojson_polygons.json"),
+    ]
+    for parquet_filename, geojson_filename in parquet_geojson_pairs:
+        parquet_path = results_dir / parquet_filename
+        geojson_path = results_dir / geojson_filename
+        parquet_row_count = len(pd.read_parquet(parquet_path))
+        with geojson_path.open() as f:
+            geojson_feature_count = len(json.load(f)["features"])
+        assert parquet_row_count == geojson_feature_count, (
+            f"Row count mismatch between {parquet_filename} ({parquet_row_count} rows) "
+            f"and {geojson_filename} ({geojson_feature_count} features)"
+        )
+
     # Validate the execute command exited successfully
     assert result.exit_code == 0
 

tests/constants_test.py

@@ -105,6 +105,9 @@
             ("tissue_segmentation_segmentation_map_image.tiff", 1783952, 10),
             ("tissue_segmentation_csv_class_information.csv", 446, 10),
             ("tissue_qc_csv_class_information.csv", 290, 10),
+            ("tissue_qc_parquet_polygons.parquet", 29049, 10),
+            ("tissue_segmentation_parquet_polygons.parquet", 56682, 10),
+            ("cell_classification_parquet_polygons.parquet", 838533, 10),
         ]
 
     case "staging":
@@ -146,6 +149,9 @@
             ("tissue_segmentation_segmentation_map_image.tiff", 1783952, 10),
             ("tissue_segmentation_csv_class_information.csv", 446, 10),
             ("tissue_qc_csv_class_information.csv", 290, 10),
+            ("tissue_qc_parquet_polygons.parquet", 29049, 10),
+            ("tissue_segmentation_parquet_polygons.parquet", 56682, 10),
+            ("cell_classification_parquet_polygons.parquet", 838533, 10),
         ]
 
     case _: