From dfebefba4ff24627c0be0c97dc198a16573f2866 Mon Sep 17 00:00:00 2001 From: mdroe Date: Mon, 7 Dec 2009 21:12:20 +0000 Subject: Cleanup r7978 (add err_msg support to assert_mask_equal --- numpy/ma/testutils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'numpy/ma/testutils.py') diff --git a/numpy/ma/testutils.py b/numpy/ma/testutils.py index 5234e0db4..1918e97a0 100644 --- a/numpy/ma/testutils.py +++ b/numpy/ma/testutils.py @@ -228,10 +228,10 @@ def assert_array_less(x, y, err_msg='', verbose=True): header='Arrays are not less-ordered') -def assert_mask_equal(m1, m2): +def assert_mask_equal(m1, m2, err_msg=''): """Asserts the equality of two masks.""" if m1 is nomask: assert(m2 is nomask) if m2 is nomask: assert(m1 is nomask) - assert_array_equal(m1, m2) + assert_array_equal(m1, m2, err_msg=err_msg) -- cgit v1.2.1