diff options
| author | Charles Harris <charlesr.harris@gmail.com> | 2017-04-28 22:02:48 -0600 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2017-04-28 22:02:48 -0600 |
| commit | ef2cfe12c49ab377cfbf2f54993699b92229b907 (patch) | |
| tree | 5e1285dd53774d1b158ef9136d2f8956fa285dd8 /numpy/lib/tests | |
| parent | e87523089129d50428b9a539600899d5c408bbcf (diff) | |
| parent | a77709c67051a1d19ed4caa83b0b7686ba22bfc4 (diff) | |
| download | numpy-ef2cfe12c49ab377cfbf2f54993699b92229b907.tar.gz | |
Merge pull request #9013 from eric-wieser/fix-nanfuncs
BUG: Fix np.lib.nanfunctions on object arrays
Diffstat (limited to 'numpy/lib/tests')
| -rw-r--r-- | numpy/lib/tests/test_nanfunctions.py | 12 |
1 files changed, 12 insertions, 0 deletions
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): |