Please fix the random seed by using rng = np.random.RandomState(42) and also fix the random_state param of MiniBatchKMeans. It would be great to find a way to make this test in less than 1s.