def METHOD_NAME(max_iter):
    clusters1, _ = mean_shift(X, max_iter=max_iter)
    ms = MeanShift(max_iter=max_iter).fit(X)
    clusters2 = ms.cluster_centers_
    assert ms.n_iter_ <= ms.max_iter
    assert len(clusters1) == len(clusters2)
    for c1, c2 in zip(clusters1, clusters2):
        assert np.allclose(c1, c2)