From a77709c67051a1d19ed4caa83b0b7686ba22bfc4 Mon Sep 17 00:00:00 2001 From: Eric Wieser Date: Thu, 27 Apr 2017 23:02:43 +0100 Subject: BUG: Fix incorrect behavior of nanfunctions on object arrays Fixes gh-8974 and gh-9008 --- numpy/lib/tests/test_nanfunctions.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) (limited to 'numpy/lib/tests/test_nanfunctions.py') diff --git a/numpy/lib/tests/test_nanfunctions.py b/numpy/lib/tests/test_nanfunctions.py index 1678e1091..466ceefb5 100644 --- a/numpy/lib/tests/test_nanfunctions.py +++ b/numpy/lib/tests/test_nanfunctions.py @@ -152,6 +152,18 @@ class TestNanFunctions_MinMax(TestCase): assert_(res != np.nan) assert_(len(w) == 0) + def test_object_array(self): + arr = np.array([[1.0, 2.0], [np.nan, 4.0], [np.nan, np.nan]], dtype=object) + assert_equal(np.nanmin(arr), 1.0) + assert_equal(np.nanmin(arr, axis=0), [1.0, 2.0]) + + with warnings.catch_warnings(record=True) as w: + warnings.simplefilter('always') + # assert_equal does not work on object arrays of nan + assert_equal(list(np.nanmin(arr, axis=1)), [1.0, 4.0, np.nan]) + assert_(len(w) == 1, 'no warning raised') + assert_(issubclass(w[0].category, RuntimeWarning)) + class TestNanFunctions_ArgminArgmax(TestCase): -- cgit v1.2.1