diff options
Diffstat (limited to 'numpy/ma/tests/test_extras.py')
| -rw-r--r-- | numpy/ma/tests/test_extras.py | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/numpy/ma/tests/test_extras.py b/numpy/ma/tests/test_extras.py index d7bc765a9..8902d67dc 100644 --- a/numpy/ma/tests/test_extras.py +++ b/numpy/ma/tests/test_extras.py @@ -508,6 +508,10 @@ class TestMedian(TestCase): r = np.ma.median([[np.inf, np.inf], [np.inf, np.inf]], axis=-1) assert_equal(r, np.inf) + def test_non_masked(self): + assert_equal(np.ma.median(np.arange(9)), 4.) + assert_equal(np.ma.median(range(9)), 4) + def test_2d(self): # Tests median w/ 2D (n, p) = (101, 30) |