diff options
Diffstat (limited to 'numpy/lib/tests')
| -rw-r--r-- | numpy/lib/tests/test_function_base.py | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/numpy/lib/tests/test_function_base.py b/numpy/lib/tests/test_function_base.py index 0b6b1e19d..ba6b336ff 100644 --- a/numpy/lib/tests/test_function_base.py +++ b/numpy/lib/tests/test_function_base.py @@ -1185,5 +1185,17 @@ def test_median(): assert_allclose(np.median(a2, axis=1), [1, 4]) +class TestAdd_newdoc_ufunc(TestCase): + + def test_ufunc_arg(self): + assert_raises(TypeError, add_newdoc_ufunc, 2, "blah") + assert_raises(ValueError, add_newdoc_ufunc,np.add, "blah") + + def test_string_arg(self): + assert_raises(TypeError, add_newdoc_ufunc,np.add, 3) + + + + if __name__ == "__main__": run_module_suite() |