Add e2e test for create_iceflow_parquet

Author: trey-stafford

tests/integration/test_e2e.py

@@ -12,9 +12,10 @@
 
 import datetime as dt
 
+import dask.dataframe as dd
 import pandas as pd
 
-from nsidc.iceflow.api import fetch_iceflow_df
+from nsidc.iceflow.api import create_iceflow_parquet, fetch_iceflow_df
 from nsidc.iceflow.data.models import (
     BLATM1BDataset,
     BoundingBox,
@@ -144,3 +145,35 @@ def test_glah06(tmp_path):
     )
 
     assert (results.ITRF == "ITRF2008").all()
+
+
+def test_create_iceflow_parquet(tmp_path):
+    target_itrf = "ITRF2014"
+    common_bounding_box = BoundingBox(
+        lower_left_lon=-49.149,
+        lower_left_lat=69.186,
+        upper_right_lon=-48.949,
+        upper_right_lat=69.238,
+    )
+
+    # This should finds 4 results for ILATM1B v1 and 3 results for v2.
+    parquet_path = create_iceflow_parquet(
+        dataset_search_params=DatasetSearchParameters(
+            datasets=[ILATM1BDataset(version="1"), ILATM1BDataset(version="2")],
+            bounding_box=common_bounding_box,
+            temporal=((dt.date(2007, 1, 1), dt.date(2014, 10, 28))),
+        ),
+        output_dir=tmp_path,
+        target_itrf=target_itrf,
+    )
+
+    df = dd.read_parquet(parquet_path)  # type: ignore[attr-defined]
+
+    # Assert that the parquet data has the expected columns
+    expected_columns = sorted(["latitude", "longitude", "elevation", "dataset"])
+    assert expected_columns == sorted(df.columns)
+
+    # Assert that the two datasets we expect are present.
+    assert sorted(["ILATM1Bv1", "ILATM1Bv2"]) == sorted(
+        df.dataset.unique().compute().values
+    )