def test_coord_does_not_exist(self):
cube_a = stock.simple_2d_w_multidim_and_scalars()
cube_b = cube_a.copy()
coord = cube_b.coord('dim1').copy()
coord.rename('new')
demote_dim_coord_to_aux_coord(cube_b, coord)
self.assertEqual(cube_a, cube_b)
def test_argument_is_coord_instance(self):
cube_a = stock.realistic_4d()
cube_b = cube_a.copy()
coord = cube_b.coord('model_level_number').copy()
demote_dim_coord_to_aux_coord(cube_b, coord)
self.assertEqual(cube_b.dim_coords,
(cube_a.coord('time'), cube_a.coord('grid_latitude'),
cube_a.coord('grid_longitude')))
def test_argument_is_basestring(self):
cube_a = stock.simple_3d()
cube_b = cube_a.copy()
demote_dim_coord_to_aux_coord(cube_b, cube_b.coord("wibble"))
self.assertEqual(
cube_b.dim_coords,
(cube_a.coord("latitude"), cube_a.coord("longitude")),
)
def test_argument_is_coord_instance(self):
cube_a = stock.realistic_4d()
cube_b = cube_a.copy()
coord = cube_b.coord('model_level_number').copy()
demote_dim_coord_to_aux_coord(cube_b, coord)
self.assertEqual(cube_b.dim_coords,
(cube_a.coord('time'),
cube_a.coord('grid_latitude'),
cube_a.coord('grid_longitude')))
def test_trying_to_demote_a_scalar_coord(self):
cube_a = stock.simple_2d_w_multidim_and_scalars()
cube_b = cube_a.copy()
demote_dim_coord_to_aux_coord(cube_b, 'an_other')
self.assertEqual(cube_a, cube_b)
def test_argument_is_wrong_type(self):
cube_a = stock.simple_1d()
with self.assertRaises(TypeError):
demote_dim_coord_to_aux_coord(cube_a, 0.0)
def test_coord_of_that_name_does_not_exist(self):
cube_a = stock.simple_2d_w_multidim_and_scalars()
with self.assertRaises(iris.exceptions.CoordinateNotFoundError):
demote_dim_coord_to_aux_coord(cube_a, 'wibble')
def test_old_dim_coord_is_now_aux_coord(self):
cube_a = stock.hybrid_height()
cube_b = cube_a.copy()
demote_dim_coord_to_aux_coord(cube_b, 'level_height')
self.assertTrue(cube_a.coord('level_height') in cube_b.aux_coords)
def test_argument_is_basestring(self):
cube_a = stock.simple_3d()
cube_b = cube_a.copy()
demote_dim_coord_to_aux_coord(cube_b, cube_b.coord('wibble'))
self.assertEqual(cube_b.dim_coords,
(cube_a.coord('latitude'), cube_a.coord('longitude')))
