Skip to content

Commit 47ec14f

Browse files
committed
Add tests for coordinate attributes
1 parent 5e91797 commit 47ec14f

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

tests/test_regrid.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -128,20 +128,25 @@ def test_conservative_nans(conservative_input_data, conservative_sample_grid):
128128
@pytest.mark.parametrize("method", ["linear", "nearest", "cubic"])
129129
def test_attrs_dataarray(sample_input_data, sample_grid_ds, method):
130130
regridder = getattr(sample_input_data["d2m"].regrid, method)
131-
assert regridder(sample_grid_ds).attrs == sample_input_data["d2m"].attrs
131+
da_regrid = regridder(sample_grid_ds)
132+
assert da_regrid.attrs == sample_input_data["d2m"].attrs
133+
assert da_regrid["longitude"].attrs == sample_input_data["longitude"].attrs
132134

133135

134136
def test_attrs_dataarray_conservative(sample_input_data, sample_grid_ds):
135137
da_regrid = sample_input_data["d2m"].regrid.conservative(
136138
sample_grid_ds, latitude_coord="latitude"
137139
)
138140
assert da_regrid.attrs == sample_input_data["d2m"].attrs
141+
assert da_regrid["longitude"].attrs == sample_input_data["longitude"].attrs
139142

140143

141144
@pytest.mark.parametrize("method", ["linear", "nearest", "cubic"])
142145
def test_attrs_dataset(sample_input_data, sample_grid_ds, method):
143146
regridder = getattr(sample_input_data.regrid, method)
144-
assert regridder(sample_grid_ds).attrs == sample_input_data.attrs
147+
ds_regrid = regridder(sample_grid_ds)
148+
assert ds_regrid.attrs == sample_input_data.attrs
149+
assert ds_regrid["longitude"].attrs == sample_input_data["longitude"].attrs
145150

146151

147152
def test_attrs_dataset_conservative(sample_input_data, sample_grid_ds):
@@ -150,3 +155,4 @@ def test_attrs_dataset_conservative(sample_input_data, sample_grid_ds):
150155
)
151156
assert ds_regrid.attrs == sample_input_data.attrs
152157
assert ds_regrid["d2m"].attrs == sample_input_data["d2m"].attrs
158+
assert ds_regrid["longitude"].attrs == sample_input_data["longitude"].attrs

0 commit comments

Comments
 (0)