Ideally the only thing that should be inside the context manager is the call to `cube.collapsed()`. (i.e. Not this initialisation stuff, and not the assertions afterwards.)