def test_assert_cube_illegal_coord_var(self):
cube = new_cube(variables=dict(precipitation=0.5))
cube = cube.assign_coords(
lat=xr.DataArray(np.outer(cube.lat, np.ones(cube.lon.size)),
dims=("y", "x")),
lon=xr.DataArray(np.outer(np.ones(cube.lat.size), cube.lon),
dims=("y", "x")))
with self.assertRaises(ValueError) as cm:
assert_cube(cube)
self.assertEqual(
"Dataset is not a valid data cube, because:\n"
"- coordinate variable 'lat' must have a single dimension 'lat';\n"
"- coordinate variable 'lon' must have a single dimension 'lon'.",
f"{cm.exception}")
def test_assert_cube_illegal_data_var(self):
cube = new_cube(variables=dict(precipitation=0.5))
shape = cube.dims["lat"], cube.dims["lon"]
cube["chl"] = xr.DataArray(np.random.rand(*shape),
dims=("lat", "lon"),
coords=dict(lat=cube.lat, lon=cube.lon))
with self.assertRaises(ValueError) as cm:
assert_cube(cube)
self.assertEqual(
"Dataset is not a valid data cube, because:\n"
"- dimensions of data variable 'chl' must be"
" ('time', ..., 'lat', 'lon'), but were ('lat', 'lon') for 'chl';\n"
"- dimensions of all data variables must be same,"
" but found ('time', 'lat', 'lon') for 'precipitation'"
" and ('lat', 'lon') for 'chl'.", f"{cm.exception}")
def test_assert_cube_illegal_coord_bounds_var(self):
cube = new_cube(variables=dict(precipitation=0.5))
lat_bnds = np.zeros((cube.time.size, cube.lat.size, 2))
lon_bnds = np.zeros((cube.time.size, cube.lon.size, 2),
dtype=np.float16)
lat_bnds[:, :, :] = cube.lat_bnds
lon_bnds[:, :, :] = cube.lon_bnds
cube = cube.assign_coords(lat_bnds=xr.DataArray(lat_bnds,
dims=("time", "lat",
"bnds")),
lon_bnds=xr.DataArray(lon_bnds,
dims=("time", "lon",
"bnds")))
with self.assertRaises(ValueError) as cm:
assert_cube(cube)
self.assertEqual(
"Dataset is not a valid data cube, because:\n"
"- bounds coordinate variable 'lat_bnds' must have dimensions ('lat', <bounds_dim>);\n"
"- shape of bounds coordinate variable 'lat_bnds' must be (180, 2) but was (5, 180, 2);\n"
"- bounds coordinate variable 'lon_bnds' must have dimensions ('lon', <bounds_dim>);\n"
"- shape of bounds coordinate variable 'lon_bnds' must be (360, 2) but was (5, 360, 2);\n"
"- type of bounds coordinate variable 'lon_bnds' must be dtype('float64')"
" but was dtype('float16').", f"{cm.exception}")
def test_assert_cube_without_bounds(self):
cube = new_cube(variables=dict(precipitation=0.5), drop_bounds=True)
self.assertIs(cube, assert_cube(cube))
def test_assert_cube_ok(self):
cube = new_cube(variables=dict(precipitation=0.5))
self.assertIs(cube, assert_cube(cube))
