diff options
Diffstat (limited to 'numpy/lib/tests/test_function_base.py')
| -rw-r--r-- | numpy/lib/tests/test_function_base.py | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/numpy/lib/tests/test_function_base.py b/numpy/lib/tests/test_function_base.py index a23e406e3..de561e55a 100644 --- a/numpy/lib/tests/test_function_base.py +++ b/numpy/lib/tests/test_function_base.py @@ -211,6 +211,17 @@ 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)) + + # invalid axis value + assert_raises(IndexError, insert, a, 1, a[:,2,:], axis=3) + assert_raises(IndexError, insert, a, 1, a[:,2,:], axis=-4) + def test_0d(self): # This is an error in the future a = np.array(1) |