Add tests to check if attrs are kept after regridding

Author: BSchilperoort

tests/test_most_common.py

@@ -33,6 +33,7 @@ def dummy_lc_data():
             "longitude": (["longitude"], lon_coords),
             "latitude": (["latitude"], lat_coords),
         },
+        attrs={"test": "not empty"},
     )
 
 
@@ -77,3 +78,20 @@ def test_most_common(dummy_lc_data, dummy_target_grid):
         dummy_lc_data.regrid.most_common(dummy_target_grid)["lc"],
         expected["lc"],
     )
+
+
+def test_attrs_dataarray(dummy_lc_data, dummy_target_grid):
+    dummy_lc_data["lc"].attrs = {"test": "testing"}
+    da_regrid = dummy_lc_data["lc"].regrid.most_common(
+        dummy_target_grid
+    )
+    assert da_regrid.attrs != {}
+    assert da_regrid.attrs == dummy_lc_data["lc"].attrs
+
+
+def test_attrs_dataset(dummy_lc_data, dummy_target_grid):
+    ds_regrid = dummy_lc_data.regrid.most_common(
+        dummy_target_grid,
+    )
+    assert ds_regrid.attrs != {}
+    assert ds_regrid.attrs == dummy_lc_data.attrs

tests/test_regrid.py

@@ -1,3 +1,4 @@
+from copy import deepcopy
 from pathlib import Path
 
 import pytest
@@ -123,3 +124,29 @@ def test_conservative_nans(conservative_input_data, conservative_sample_grid):
         rtol=0.002,
         atol=2e-6,
     )
+
+
+@pytest.mark.parametrize("method", ["linear", "nearest", "cubic"])
+def test_attrs_dataarray(sample_input_data, sample_grid_ds, method):
+    regridder = getattr(sample_input_data["d2m"].regrid, method)
+    assert regridder(sample_grid_ds).attrs == sample_input_data["d2m"].attrs
+
+
+def test_attrs_dataarray_conservative(sample_input_data, sample_grid_ds):
+    da_regrid = sample_input_data["d2m"].regrid.conservative(
+        sample_grid_ds, latitude_coord="latitude"
+    )
+    assert da_regrid.attrs == sample_input_data["d2m"].attrs
+
+
+@pytest.mark.parametrize("method", ["linear", "nearest", "cubic"])
+def test_attrs_dataset(sample_input_data, sample_grid_ds, method):
+    regridder = getattr(sample_input_data.regrid, method)
+    assert regridder(sample_grid_ds).attrs == sample_input_data.attrs
+
+
+def test_attrs_dataset_conservative(sample_input_data, sample_grid_ds):
+    ds_regrid = sample_input_data.regrid.conservative(
+        sample_grid_ds, latitude_coord="latitude"
+    )
+    assert ds_regrid.attrs == sample_input_data.attrs