Skip to content

Commit 184a71b

Browse files
qinhanmin2014rth
authored andcommitted
TST Reduce time for test_mean_shift.py::test_parallel (scikit-learn#11162)
1 parent 1f85328 commit 184a71b

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

sklearn/cluster/tests/test_mean_shift.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -65,6 +65,10 @@ def test_estimate_bandwidth_with_sparse_matrix():
6565

6666

6767
def test_parallel():
68+
centers = np.array([[1, 1], [-1, -1], [1, -1]]) + 10
69+
X, _ = make_blobs(n_samples=50, n_features=2, centers=centers,
70+
cluster_std=0.4, shuffle=True, random_state=11)
71+
6872
ms1 = MeanShift(n_jobs=2)
6973
ms1.fit(X)
7074

0 commit comments

Comments
 (0)