valid_month, and tests

ecmwf · Jul 17, 2024 · ba56258 · ba56258
1 parent fb5eb09
commit ba56258
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 0 deletions.
diff --git a/tests/sample-data/cams-egg4-monthly.grib b/tests/sample-data/cams-egg4-monthly.grib
diff --git a/tests/sample-data/era5-levels-members.nc b/tests/sample-data/era5-levels-members.nc
diff --git a/tests/test_50_sample_data.py b/tests/test_50_sample_data.py
@@ -120,3 +120,12 @@ def test_dataset_missing_field_values() -> None:
     t2 = res.variables["t2m"]
     assert np.isclose(np.nanmean(t2[0, :, :]), 268.375)
     assert np.isclose(np.nanmean(t2[1, :, :]), 270.716)
+
+
+def test_valid_month_time_dim() -> None:
+
+    test_file = os.path.join(SAMPLE_DATA_FOLDER, "cams-egg4-monthly.grib")
+    ds = xr.open_dataset(test_file, time_dims=["valid_month"])
+
+    assert "valid_month" in ds.dims
+