Please don't use raw `np.random` in tests, it screws with reproducibility. (Not that it matters much in this case, but still). Use `np.random.RandomState(0).shuffle` instead.