def test_cube_reduction_4d(self):
cube = setup_4d_cube()
dim_indices = {'dim 1 index': 2,
'dim 2 index': 0,
'sliced dim index': 1}
collapsed_indices = [1]
new_cube = cl.get_sub_cube(cube, dim_indices, collapsed_indices)
expected_cube = cube[:, :, :, 1]
self.assertEqual(new_cube, expected_cube)
def test_cube_reduction_7d(self):
cube = setup_7d_anonymous_cube()
dim_indices = {'dim 1 index': 4,
'dim 2 index': 0,
'sliced dim index': 6}
collapsed_indices = [0, 3, 4, 2]
new_cube = cl.get_sub_cube(cube, dim_indices, collapsed_indices)
expected_cube = cube[:, 0, 3, 4, :, 2, :]
self.assertEqual(new_cube, expected_cube)
