Fix h5py shape testing