TST Fixes scikit-learn#4386

artsobolev · amueller · commit 5b2aba1598dd · 2015-03-25T17:31:47.000-04:00
diff --git a/sklearn/linear_model/tests/test_sgd.py b/sklearn/linear_model/tests/test_sgd.py
@@ -630,7 +630,7 @@ def test_weights_multiplied(self):
         clf1.fit(X4, Y4, sample_weight=sample_weights)
         clf2.fit(X4, Y4, sample_weight=multiplied_together)
 
-        assert_array_equal(clf1.coef_, clf2.coef_)
+        assert_almost_equal(clf1.coef_, clf2.coef_)
 
     def test_auto_weight(self):
         """Test class weights for imbalanced data"""
diff --git a/sklearn/neighbors/tests/test_approximate.py b/sklearn/neighbors/tests/test_approximate.py
@@ -471,15 +471,16 @@ def test_sparse_input():
 
     d_sparse, i_sparse = forest_sparse.kneighbors(X2, return_distance=True)
     d_dense, i_dense = forest_dense.kneighbors(X2.A, return_distance=True)
-    assert_array_equal(d_sparse, d_dense)
-    assert_array_equal(i_sparse, i_dense)
+
+    assert_almost_equal(d_sparse, d_dense)
+    assert_almost_equal(i_sparse, i_dense)
 
     d_sparse, i_sparse = forest_sparse.radius_neighbors(X2,
                                                         return_distance=True)
     d_dense, i_dense = forest_dense.radius_neighbors(X2.A,
                                                      return_distance=True)
     assert_equal(d_sparse.shape, d_dense.shape)
     for a, b in zip(d_sparse, d_dense):
-        assert_array_equal(a, b)
+        assert_almost_equal(a, b)
     for a, b in zip(i_sparse, i_dense):
-        assert_array_equal(a, b)
+        assert_almost_equal(a, b)
