summaryrefslogtreecommitdiff
path: root/numpy/lib/tests/test_function_base.py
diff options
context:
space:
mode:
authorFélix Hartmann <felix.hartmann@crans.org>2013-07-03 15:15:55 +0200
committerSebastian Berg <sebastian@sipsolutions.net>2013-12-02 13:54:59 +0100
commit9d6caf9710e98af2a99cc72650b95dd7ea0727ee (patch)
tree884bd54dc63cc268291c6d1cd6648cf242c5a7e4 /numpy/lib/tests/test_function_base.py
parent50fad03b2ab81d9518e95138f2425ecd82740be3 (diff)
downloadnumpy-9d6caf9710e98af2a99cc72650b95dd7ea0727ee.tar.gz
TST: add test for negative axis values in np.insert.
Diffstat (limited to 'numpy/lib/tests/test_function_base.py')
-rw-r--r--numpy/lib/tests/test_function_base.py7
1 files changed, 7 insertions, 0 deletions
diff --git a/numpy/lib/tests/test_function_base.py b/numpy/lib/tests/test_function_base.py
index 6829e06ae..d1069cf92 100644
--- a/numpy/lib/tests/test_function_base.py
+++ b/numpy/lib/tests/test_function_base.py
@@ -226,6 +226,13 @@ class TestInsert(TestCase):
assert_raises(IndexError, insert, a, 1, a[:, 2, :], axis=3)
assert_raises(IndexError, insert, a, 1, a[:, 2, :], axis=-4)
+ # negative axis value
+ a = np.arange(24).reshape((2,3,4))
+ assert_equal(insert(a, 1, a[:,:,3], axis=-1),
+ insert(a, 1, a[:,:,3], axis=2))
+ assert_equal(insert(a, 1, a[:,2,:], axis=-2),
+ insert(a, 1, a[:,2,:], axis=1))
+
def test_0d(self):
# This is an error in the future
a = np.array(1)