# Sebastian Raschka 2014-2018
# mlxtend Machine Learning Library Extensions
# Author: Sebastian Raschka <sebastianraschka.com>
#
# License: BSD 3 clause
from mlxtend.data import three_blobs_data
from mlxtend.cluster import Kmeans
from mlxtend.utils import assert_raises
import numpy as np
X, y = three_blobs_data()
def test_nonfitted():
km = Kmeans(k=3,
max_iter=50,
random_seed=1,
print_progress=0)
km._is_fitted = False
assert_raises(AttributeError,
'Model is not fitted, yet.',
km.predict,
X)
def test_three_blobs_multi():
km = Kmeans(k=3,
max_iter=50,
def test_import_three_blobs_data():
X, y = three_blobs_data()
assert (X.shape[0] == 150), X.shape
assert (X.shape[1] == 2)
assert (y.shape[0] == 150)
def test_import_three_blobs_data():
X, y = three_blobs_data()
assert(X.shape[0] == 150), X.shape
assert(X.shape[1] == 2)
assert(y.shape[0] == 150)
