Revert "FIX remove FutureWarning in _object_dtype_isnan and add test (scikit-learn#12567)"

Xing · web-flow · commit 633f907436e2 · 2019-04-28T15:16:17.000-04:00
This reverts commit 178f87c.
diff --git a/sklearn/utils/fixes.py b/sklearn/utils/fixes.py
@@ -309,13 +309,18 @@ def nanmedian(a, axis=None):
 # Fix for behavior inconsistency on numpy.equal for object dtypes.
 # For numpy versions < 1.13, numpy.equal tests element-wise identity of objects
 # instead of equality. This fix returns the mask of NaNs in an array of
-# numerical or object values for all numpy versions.
-if np_version < (1, 13):
+# numerical or object values for all nupy versions.
+
+_nan_object_array = np.array([np.nan], dtype=object)
+_nan_object_mask = _nan_object_array != _nan_object_array
+
+if np.array_equal(_nan_object_mask, np.array([True])):
     def _object_dtype_isnan(X):
-        return np.frompyfunc(lambda x: x != x, 1, 1)(X).astype(bool)
+        return X != X
+
 else:
     def _object_dtype_isnan(X):
-        return X != X
+        return np.frompyfunc(lambda x: x != x, 1, 1)(X).astype(bool)
 
 
 # To be removed once this fix is included in six
diff --git a/sklearn/utils/tests/test_fixes.py b/sklearn/utils/tests/test_fixes.py
@@ -17,7 +17,6 @@
 from sklearn.utils.fixes import nanmedian
 from sklearn.utils.fixes import nanpercentile
 from sklearn.utils.fixes import _joblib_parallel_args
-from sklearn.utils.fixes import _object_dtype_isnan
 
 
 def test_divide():
@@ -89,18 +88,3 @@ def test_joblib_parallel_args(monkeypatch, joblib_version):
             _joblib_parallel_args(verbose=True)
     else:
         raise ValueError
-
-
-@pytest.mark.parametrize("dtype, val", ([object, 1],
-                                        [object, "a"],
-                                        [float, 1]))
-def test_object_dtype_isnan(dtype, val):
-    X = np.array([[val, np.nan],
-                  [np.nan, val]], dtype=dtype)
-
-    expected_mask = np.array([[False, True],
-                              [True, False]])
-
-    mask = _object_dtype_isnan(X)
-
-    assert_array_equal(mask, expected_mask)
