diff options
Diffstat (limited to 'numpy/ma/tests/test_extras.py')
| -rw-r--r-- | numpy/ma/tests/test_extras.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/numpy/ma/tests/test_extras.py b/numpy/ma/tests/test_extras.py index fb68bd261..547a3fb81 100644 --- a/numpy/ma/tests/test_extras.py +++ b/numpy/ma/tests/test_extras.py @@ -1039,14 +1039,14 @@ class TestMedian(TestCase): # axis 0 and 1 b = np.ma.masked_array(np.array([], dtype=float, ndmin=2)) - assert_equal(np.median(a, axis=0), b) - assert_equal(np.median(a, axis=1), b) + assert_equal(np.ma.median(a, axis=0), b) + assert_equal(np.ma.median(a, axis=1), b) # axis 2 b = np.ma.masked_array(np.array(np.nan, dtype=float, ndmin=2)) with warnings.catch_warnings(record=True) as w: warnings.filterwarnings('always', '', RuntimeWarning) - assert_equal(np.median(a, axis=2), b) + assert_equal(np.ma.median(a, axis=2), b) assert_(w[0].category is RuntimeWarning) def test_object(self): |