Add a test to make sure the ```collapsed_cube.data.shape``` is as expected i.e. ```(1,)``` in this case.