diff options
| author | Charles Harris <charlesr.harris@gmail.com> | 2013-08-02 14:36:39 -0600 |
|---|---|---|
| committer | Charles Harris <charlesr.harris@gmail.com> | 2013-08-02 14:36:39 -0600 |
| commit | 645b9f572f0a22e9049fd736b8b91427be2c8402 (patch) | |
| tree | a2a043f988f655b5fa3e8093c6420897c1c554d1 /numpy/lib/tests/test_function_base.py | |
| parent | 196398fadb7fe402a67ccea0bc1546b36d066440 (diff) | |
| parent | ea768739dab69c0b67488179ffa67d57d63d59f8 (diff) | |
| download | numpy-645b9f572f0a22e9049fd736b8b91427be2c8402.tar.gz | |
Merge pull request 'gh-3497'
* gh-3497:
BUG: Make np.insert check for out of bounds axis arguments.
TST: add test for negative axis values in np.insert.
BUG: Fix bug in np.insert when axis=-1
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) |