summaryrefslogtreecommitdiff
path: root/numpy/lib/tests/test_function_base.py
diff options
context:
space:
mode:
authorChris Jordan-Squire <cjordan1@uw.edu>2011-08-05 10:43:51 -0500
committerCharles Harris <charlesr.harris@gmail.com>2011-08-29 20:19:04 -0600
commit180750b9a4a4dc46be32ed6d738329b2ab97abfa (patch)
tree331bb2eb603587c694ba50ae7d431eb2678bf274 /numpy/lib/tests/test_function_base.py
parentd91521e5fd858726998146e6055f677cc3aeb011 (diff)
downloadnumpy-180750b9a4a4dc46be32ed6d738329b2ab97abfa.tar.gz
ENH: Add function for adding docstrings to ufuncs.
Diffstat (limited to 'numpy/lib/tests/test_function_base.py')
-rw-r--r--numpy/lib/tests/test_function_base.py12
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()