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

jnothman · jnothman · commit ae6f5808dc2d · 2019-02-19T15:59:10.000+11:00
diff --git a/sklearn/cluster/tests/test_mean_shift.py b/sklearn/cluster/tests/test_mean_shift.py
@@ -37,7 +37,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():
