From c4721f36d8e2bc59c6663c75ef60df6d3f42ac41 Mon Sep 17 00:00:00 2001 From: Gael Varoquaux Date: Sun, 30 Aug 2015 16:00:57 +0100 Subject: [PATCH] TST: No n_jobs=-1 in the test Use n_jobs=2, to avoid blowing up on hugely parallel machines. --- sklearn/cluster/tests/test_mean_shift.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/sklearn/cluster/tests/test_mean_shift.py b/sklearn/cluster/tests/test_mean_shift.py index 67b8d1dce9b..89bfa50cb3a 100644 --- a/sklearn/cluster/tests/test_mean_shift.py +++ b/sklearn/cluster/tests/test_mean_shift.py @@ -46,13 +46,14 @@ def test_mean_shift(): n_clusters_ = len(labels_unique) assert_equal(n_clusters_, n_clusters) + def test_parallel(): - ms1 = MeanShift(n_jobs=-1) + ms1 = MeanShift(n_jobs=2) ms1.fit(X) - + ms2 = MeanShift() ms2.fit(X) - + assert_array_equal(ms1.cluster_centers_,ms2.cluster_centers_) assert_array_equal(ms1.labels_,ms2.labels_)