summaryrefslogtreecommitdiff
path: root/numpy/lib
diff options
context:
space:
mode:
authorCharles Harris <charlesr.harris@gmail.com>2014-10-16 11:29:07 -0600
committerCharles Harris <charlesr.harris@gmail.com>2014-10-16 11:29:07 -0600
commit443a49a271e6d7c4c1dda72fe721eaed318ff764 (patch)
treed632e65899dab05e22458723371eeb00bbf2968e /numpy/lib
parent4c0747d5c2cb3c961f7246cd9222b89bdcf2ce42 (diff)
parent2f1e27610453f901e81cd892b2aa3b3a2ac30819 (diff)
downloadnumpy-443a49a271e6d7c4c1dda72fe721eaed318ff764.tar.gz
Merge pull request #5182 from juliantaylor/nanmedian-inf
BUG: fix nanmedian on arrays containing inf
Diffstat (limited to 'numpy/lib')
-rw-r--r--numpy/lib/tests/test_nanfunctions.py16
1 files changed, 16 insertions, 0 deletions
diff --git a/numpy/lib/tests/test_nanfunctions.py b/numpy/lib/tests/test_nanfunctions.py
index 3da6b5149..35ae86c20 100644
--- a/numpy/lib/tests/test_nanfunctions.py
+++ b/numpy/lib/tests/test_nanfunctions.py
@@ -645,6 +645,22 @@ class TestNanFunctions_Median(TestCase):
assert_raises(IndexError, np.nanmedian, d, axis=(0, 4))
assert_raises(ValueError, np.nanmedian, d, axis=(1, 1))
+ def test_float_special(self):
+ with warnings.catch_warnings(record=True):
+ warnings.simplefilter('ignore', RuntimeWarning)
+ a = np.array([[np.inf, np.nan], [np.nan, np.nan]])
+ assert_equal(np.nanmedian(a, axis=0), [np.inf, np.nan])
+ assert_equal(np.nanmedian(a, axis=1), [np.inf, np.nan])
+ assert_equal(np.nanmedian(a), np.inf)
+
+ # minimum fill value check
+ a = np.array([[np.nan, np.nan, np.inf], [np.nan, np.nan, np.inf]])
+ assert_equal(np.nanmedian(a, axis=1), np.inf)
+
+ # no mask path
+ a = np.array([[np.inf, np.inf], [np.inf, np.inf]])
+ assert_equal(np.nanmedian(a, axis=1), np.inf)
+
class TestNanFunctions_Percentile(TestCase):