Add a test for the MiniBatchKMeans with many reassignments