summaryrefslogtreecommitdiff
path: root/numpy
diff options
context:
space:
mode:
authorFélix Hartmann <felix.hartmann@crans.org>2013-07-03 15:15:55 +0200
committerCharles Harris <charlesr.harris@gmail.com>2013-08-02 14:36:24 -0600
commit496813f1a23363bbd50a62a60c37f6bd4e10649b (patch)
treeed6eb99f9b1abd5a914a40468f1cc310ddc56c6f /numpy
parentcb6fe848bf1a6a046fd473b72b1350ea40c8644e (diff)
downloadnumpy-496813f1a23363bbd50a62a60c37f6bd4e10649b.tar.gz
TST: add test for negative axis values in np.insert.
Diffstat (limited to 'numpy')
-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 a23e406e3..13f907d5a 100644
--- a/numpy/lib/tests/test_function_base.py
+++ b/numpy/lib/tests/test_function_base.py
@@ -211,6 +211,13 @@ class TestInsert(TestCase):
assert_equal(insert(a[:,:1], 1, a[:,1], axis=1), a)
assert_equal(insert(a[:1,:], 1, a[1,:], axis=0), a)
+ # 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)