Skip to content

Commit 04e54c2

Browse files
Fix test kmedoids (#86)
* change test kmedoids
1 parent a12f7e7 commit 04e54c2

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

sklearn_extra/cluster/tests/test_k_medoids.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@
2525
centers=np.array([[-1, -1], [1, 1]]),
2626
random_state=rng,
2727
shuffle=False,
28+
cluster_std=0.2,
2829
)
2930

3031

@@ -34,7 +35,7 @@
3435
)
3536
def test_kmedoid_results(method, init):
3637
expected = np.hstack([np.zeros(50), np.ones(50)])
37-
km = KMedoids(n_clusters=2, init=init, method=method)
38+
km = KMedoids(n_clusters=2, init=init, method=method, random_state=rng)
3839
km.fit(X_cc)
3940
# This test use data that are not perfectly separable so the
4041
# accuracy is not 1. Accuracy around 0.85

0 commit comments

Comments
 (0)