def test_get_enabled_4d_filled_contour_qplt(self): cube = tcl.setup_4d_cube() dim_1_name = 'time' dim_2_name = 'grid_longitude' status = {'cube loaded': True, 'plot method': 'using quickplot', 'plot type': 'Filled Contour', 'cube': cube, 'dim 1 name': dim_1_name, 'dim 2 name': dim_2_name, 'projection': 'Automatic'} state = gl.get_enabled(status) expected_state = {'previous': True, 'next': True, 'source code': True, 'cartographic': False, 'labels': False, 'colorbar': True, 'plot type': True, 'colormap': True, 'contour slider': True, 'third dim': True, 'central longitude': False, 'update': True} difference = compare_dict(state, expected_state) self.assertIsNone(difference)
def test_get_enabled_4d_filled_contour_qplt(self): cube = tcl.setup_4d_cube() dim_1_name = 'time' dim_2_name = 'grid_longitude' status = { 'cube loaded': True, 'plot method': 'using quickplot', 'plot type': 'Filled Contour', 'cube': cube, 'dim 1 name': dim_1_name, 'dim 2 name': dim_2_name, 'projection': 'Automatic' } state = gl.get_enabled(status) expected_state = { 'previous': True, 'next': True, 'source code': True, 'cartographic': False, 'labels': False, 'colorbar': True, 'plot type': True, 'colormap': True, 'contour slider': True, 'third dim': True, 'central longitude': False, 'update': True } difference = compare_dict(state, expected_state) self.assertIsNone(difference)
def test_get_dim_index(self): cube = tcl.setup_4d_cube() names = gl.get_dim_names(cube) dim = "model_level_number" index = gl.get_dim_index(dim, names) expected_index = 1 self.assertEqual(index, expected_index)
def test_get_dim_values_date_time(self): cube = tcl.setup_4d_cube() names = gl.get_dim_names(cube) values = gl.get_coord_values(cube, "time", names) points = cube.coord("time").points coord = cube.coord("time") expected_values = [coord.units.num2date(point) for point in points] self.assertEqual(all(values), all(expected_values))
def test_can_draw_map_false(self): cube = tcl.setup_4d_cube() dim_1_name = "grid_longitude" dim_2_name = "time" can_draw_map = gl.get_can_draw_map(cube, dim_2_name, dim_1_name) self.assertFalse(can_draw_map)
def test_can_draw_map_reversed(self): cube = tcl.setup_4d_cube() dim_1_name = "grid_longitude" dim_2_name = "grid_latitude" can_draw_map = gl.get_can_draw_map(cube, dim_2_name, dim_1_name) self.assertTrue(can_draw_map)
def test_get_dim_values_4d(self): cube = tcl.setup_4d_cube() names = gl.get_dim_names(cube) values = gl.get_coord_values(cube, "grid_latitude", names) expected_values = cube.coord("grid_latitude").points self.assertEqual(all(values), all(expected_values))
def test_get_remaining_dims_4d(self): cube = tcl.setup_4d_cube() names = gl.get_dim_names(cube) used_dims = (names[2], names[0], names[3]) remaining_dims = gl.get_remaining_dims(names, used_dims) self.assertEqual(remaining_dims, [names[1]])