def test_basic(self, single_chunk_blobs):
X, y = single_chunk_blobs
a = cluster.PartialMiniBatchKMeans(n_clusters=3, random_state=0)
b = cluster_.MiniBatchKMeans(n_clusters=3, random_state=0)
a.fit(X)
b.partial_fit(X)
assert_estimator_equal(a, b, exclude=['random_state_'])
from dask_ml import cluster
import numpy as np
if __name__ == '__main__':
X = np.random.rand(10000, 16)
kmeans = cluster.PartialMiniBatchKMeans(n_clusters=500, init='k-means++').partial_fit(X)
print(kmeans.labels_)