def test_esmpy_normalisation():
"""
Integration test for :meth:`~esmf_regrid.esmf_regridder.Regridder`.
Checks against ESMF to ensure results are consistent.
"""
src_data = np.array(
[
[1.0, 1.0],
[1.0, 0.0],
[1.0, 0.0],
],
)
src_mask = np.array(
[
[True, False],
[False, False],
[False, False],
]
)
src_array = ma.array(src_data, mask=src_mask)
lon, lat, lon_bounds, lat_bounds = make_grid_args(2, 3)
src_grid = GridInfo(lon, lat, lon_bounds, lat_bounds)
src_esmpy_grid = src_grid._make_esmf_grid()
src_esmpy_grid.add_item(ESMF.GridItem.MASK, staggerloc=ESMF.StaggerLoc.CENTER)
src_esmpy_grid.mask[0][...] = src_mask
src_field = ESMF.Field(src_esmpy_grid)
src_field.data[...] = src_data
lon, lat, lon_bounds, lat_bounds = make_grid_args(3, 2)
tgt_grid = GridInfo(lon, lat, lon_bounds, lat_bounds)
tgt_field = tgt_grid.make_esmf_field()
regridder = Regridder(src_grid, tgt_grid)
regridding_kwargs = {
"ignore_degenerate": True,
"regrid_method": ESMF.RegridMethod.CONSERVE,
"unmapped_action": ESMF.UnmappedAction.IGNORE,
"factors": True,
"src_mask_values": [1],
}
esmpy_fracarea_regridder = ESMF.Regrid(
src_field, tgt_field, norm_type=ESMF.NormType.FRACAREA, **regridding_kwargs
)
esmpy_dstarea_regridder = ESMF.Regrid(
src_field, tgt_field, norm_type=ESMF.NormType.DSTAREA, **regridding_kwargs
)
tgt_field_dstarea = esmpy_dstarea_regridder(src_field, tgt_field)
result_esmpy_dstarea = tgt_field_dstarea.data
result_dstarea = regridder.regrid(src_array, norm_type="dstarea")
assert ma.allclose(result_esmpy_dstarea, result_dstarea)
tgt_field_fracarea = esmpy_fracarea_regridder(src_field, tgt_field)
result_esmpy_fracarea = tgt_field_fracarea.data
result_fracarea = regridder.regrid(src_array, norm_type="fracarea")
assert ma.allclose(result_esmpy_fracarea, result_fracarea)
class TimeGridInfo:
def setup(self):
lon, lat, lon_bounds, lat_bounds = _make_small_grid_args()
self.grid = GridInfo(lon, lat, lon_bounds, lat_bounds)
def time_make_grid(self):
"""Basic test for :meth:`~esmf_regrid.esmf_regridder.GridInfo.make_esmf_field`."""
esmf_grid = self.grid.make_esmf_field()
esmf_grid.data[:] = 0
time_make_grid.version = 1
def test_make_grid():
"""Basic test for :meth:`~esmf_regrid.esmf_regridder.GridInfo.make_esmf_field`."""
lon, lat, lon_bounds, lat_bounds = _make_small_grid_args()
grid = GridInfo(lon, lat, lon_bounds, lat_bounds)
esmf_grid = grid.make_esmf_field()
esmf_grid.data[:] = 0
relative_path = ("esmf_regridder", "test_GridInfo", "small_grid.txt")
fname = tests.get_result_path(relative_path)
with open(fname) as fi:
expected_repr = fi.read()
assert esmf_grid.__repr__() == expected_repr
