Skip to content

Commit fe066a3

Browse files
[FIX] bug test kmedoids (#159)
* fix bug random test * black
1 parent 58c2a54 commit fe066a3

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

sklearn_extra/cluster/tests/test_k_medoids.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -224,8 +224,11 @@ def test_precomputed():
224224
kmedoids.fit(D_1)
225225

226226
assert_allclose(kmedoids.inertia_, 0.2)
227-
assert_array_equal(kmedoids.medoid_indices_, [2, 0])
228-
assert_array_equal(kmedoids.labels_, [1, 1, 0, 0])
227+
assert_array_equal(
228+
np.sort(kmedoids.medoid_indices_), np.sort([2, 0])
229+
) # the order may change
230+
assert kmedoids.labels_[0] == kmedoids.labels_[1]
231+
assert kmedoids.labels_[2] == kmedoids.labels_[3]
229232
assert kmedoids.cluster_centers_ is None
230233

231234
med_1, med_2 = tuple(kmedoids.medoid_indices_)

0 commit comments

Comments
 (0)