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_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_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]])
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]])
