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)