TST More tolerance in float equality for 0 bandwidth (#13130)

jnothman · qinhanmin2014 · commit 24b69346632e · 2019-02-19T12:53:39.000+08:00
diff --git a/sklearn/cluster/tests/test_mean_shift.py b/sklearn/cluster/tests/test_mean_shift.py
@@ -36,7 +36,7 @@ def test_estimate_bandwidth_1sample():
     # Test estimate_bandwidth when n_samples=1 and quantile<1, so that
     # n_neighbors is set to 1.
     bandwidth = estimate_bandwidth(X, n_samples=1, quantile=0.3)
-    assert_equal(bandwidth, 0.)
+    assert_array_almost_equal(bandwidth, 0., decimal=5)
 
 
 def test_mean_shift():
