From f5d3d96b3b48ba8b8d15a09334491f3f19804693 Mon Sep 17 00:00:00 2001 From: empeeu Date: Mon, 22 Jun 2015 19:51:28 -0400 Subject: Updating unit tests for removed warning. --- numpy/lib/tests/test_function_base.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'numpy/lib/tests/test_function_base.py') diff --git a/numpy/lib/tests/test_function_base.py b/numpy/lib/tests/test_function_base.py index 397505bd0..1684406ab 100644 --- a/numpy/lib/tests/test_function_base.py +++ b/numpy/lib/tests/test_function_base.py @@ -2335,7 +2335,7 @@ class TestMedian(TestCase): with warnings.catch_warnings(record=True) as w: warnings.filterwarnings('always', '', RuntimeWarning) assert_equal(np.median(a, 0), b) - assert_equal(len(w), 2) + assert_equal(len(w), 1) #axis1 b = np.median(np.arange(24, dtype=float).reshape(2, 3, 4), 1) @@ -2343,7 +2343,7 @@ class TestMedian(TestCase): with warnings.catch_warnings(record=True) as w: warnings.filterwarnings('always', '', RuntimeWarning) assert_equal(np.median(a, 1), b) - assert_equal(len(w), 2) + assert_equal(len(w), 1) #axis02 b = np.median(np.arange(24, dtype=float).reshape(2, 3, 4), (0, 2)) @@ -2351,7 +2351,7 @@ class TestMedian(TestCase): with warnings.catch_warnings(record=True) as w: warnings.filterwarnings('always', '', RuntimeWarning) assert_equal(np.median(a, (0, 2)), b) - assert_equal(len(w), 2) + assert_equal(len(w), 1) def test_object(self): o = np.arange(7.); -- cgit v1.2.1