def test_nearest_aux_factories(self):
src = self.src
xy_dim_len, = src.coord(axis='X').shape
z_dim_len, = src.coord('levels').shape
src.add_aux_coord(
iris.coords.AuxCoord(np.arange(z_dim_len) + 40,
long_name="level_height",
units="m"), 1)
src.add_aux_coord(
iris.coords.AuxCoord(np.arange(z_dim_len) + 50,
long_name="sigma",
units="1"), 1)
src.add_aux_coord(
iris.coords.AuxCoord(np.arange(xy_dim_len) + 100,
long_name="surface_altitude",
units="m"), 2)
src.add_aux_factory(
iris.aux_factory.HybridHeightFactory(
delta=src.coord("level_height"),
sigma=src.coord("sigma"),
orography=src.coord("surface_altitude")))
res = src.regrid(self.global_grid, ProjectedUnstructuredNearest())
self.assertArrayShapeStats(res, (1, 6, 73, 96),
315.8913582,
11.000639227334,
rtol=1e-8)
self.assertArrayShapeStats(res[:, 0], (1, 73, 96), 299.99993826,
3.9226378869e-5)
self.assertEqual(res.coord('altitude').shape, (6, 73, 96))
def test_nearest(self):
res = self.src.regrid(self.global_grid, ProjectedUnstructuredNearest())
self.assertArrayShapeStats(res, (1, 6, 73, 96),
315.8913582,
11.00063922733,
rtol=1e-8)
self.assertArrayShapeStats(res[:, 0], (1, 73, 96), 299.99993826,
3.9226378869e-5)
def test_nearest_sinusoidal(self):
crs = ccrs.Sinusoidal()
res = self.src.regrid(self.global_grid,
ProjectedUnstructuredNearest(crs))
self.assertArrayShapeStats(res, (1, 6, 73, 96),
315.891358296,
11.000639227,
rtol=1e-8)
self.assertArrayShapeStats(res[:, 0], (1, 73, 96), 299.99993826,
3.9223839688e-5)
def test_nearest_gnomonic_uk_domain(self):
crs = ccrs.Gnomonic(central_latitude=60.0)
uk_grid = self.global_grid.intersection(longitude=(-20, 20),
latitude=(40, 80))
res = self.src.regrid(uk_grid, ProjectedUnstructuredNearest(crs))
self.assertArrayShapeStats(res, (1, 6, 17, 11),
315.8873266,
11.0006664668,
rtol=1e-8)
self.assertArrayShapeStats(res[:, 0], (1, 17, 11), 299.99993826,
4.1356150388e-5)
expected_subset = np.array([[318.936829, 318.936829, 318.936829],
[318.936829, 318.936829, 318.936829],
[318.935163, 318.935163, 318.935163]])
self.assertArrayAlmostEqual(expected_subset, res.data[0, 3, 5:8,
4:7].data)