def test_allvars_fix_metadata(cubes): fix = AllVars(None) out_cubes = fix.fix_metadata(cubes) assert cubes is out_cubes for cube in out_cubes: if cube.var_name == 'ps': assert cube.standard_name == 'air_pressure' assert cube.long_name == 'Air pressure' elif cube.var_name == 'tas' or cube.var_name == 'ta': assert cube.standard_name == 'air_temperature' assert cube.long_name == 'Air Temperature' else: assert False, "Invalid var_name" try: lat_coord = cube.coord('latitude') except iris.exceptions.CoordinateNotFoundError: assert cube.var_name == 'ps' else: assert lat_coord.var_name == 'lat' assert lat_coord.standard_name == 'latitude' assert lat_coord.long_name == 'latitude' try: lon_coord = cube.coord('longitude') except iris.exceptions.CoordinateNotFoundError: assert cube.var_name == 'ps' else: assert lon_coord.var_name == 'lon' assert lon_coord.standard_name == 'longitude' assert lon_coord.long_name == 'longitude' if 'parent_time_units' in cube.attributes: assert cube.attributes['parent_time_units'] == ( 'days since 0000-00-00')
def test_thetao_fix_metadata(thetao_cubes): """Test ``fix_metadata`` for ``thetao``.""" vardef = get_var_info('CMIP6', 'Omon', 'thetao') fix_omon = Omon(vardef) fix_allvars = AllVars(vardef) out_cubes = fix_omon.fix_metadata(thetao_cubes) out_cubes = fix_allvars.fix_metadata(out_cubes) assert out_cubes is thetao_cubes assert len(out_cubes) == 1 out_cube = out_cubes[0] # Check data of cube np.testing.assert_allclose(out_cube.data, [[[[1, 0], [3, 2]], [[5, 4], [7, 6]]], [[[9, 8], [11, 10]], [[13, 12], [15, 14]]]]) # Check data of longitude lon_coord = out_cube.coord('longitude') np.testing.assert_allclose(lon_coord.points, [357.1875, 359.0625]) np.testing.assert_allclose(lon_coord.bounds, [[356.25, 358.125], [358.125, 360.0]]) # Check metadata of depth coordinate depth_coord = out_cube.coord('depth') assert depth_coord.standard_name == 'depth' assert depth_coord.var_name == 'lev' assert depth_coord.long_name == 'ocean depth coordinate' assert depth_coord.units == 'm' assert depth_coord.attributes == {'positive': 'down'}
def test_allvars_fix_lon_bounds(cubes_bounds): fix = AllVars(None) out_cubes = fix.fix_metadata(cubes_bounds) assert cubes_bounds is out_cubes for cube in out_cubes: try: lon_coord = cube.coord('longitude') except iris.exceptions.CoordinateNotFoundError: pass else: assert lon_coord.bounds[-1][-1] == 358.125
def test_allvars_fix_metadata(cubes): fix = AllVars(None) out_cubes = fix.fix_metadata(cubes) assert cubes is out_cubes for cube in out_cubes: try: lat_coord = cube.coord('latitude') except iris.exceptions.CoordinateNotFoundError: pass else: assert lat_coord.var_name == 'lat' assert lat_coord.standard_name == 'latitude' try: lon_coord = cube.coord('longitude') except iris.exceptions.CoordinateNotFoundError: pass else: assert lon_coord.var_name == 'lon' assert lon_coord.standard_name == 'longitude' if 'parent_time_units' in cube.attributes: assert cube.attributes['parent_time_units'] == ( 'days since 0000-00-00')