diff options
author | Sebastian Berg <sebastian@sipsolutions.net> | 2020-03-23 09:55:11 -0500 |
---|---|---|
committer | GitHub <noreply@github.com> | 2020-03-23 09:55:11 -0500 |
commit | 9ae4a0d1ebc52a556ed13248172e8280ad9fc6bd (patch) | |
tree | 71c37df0211023fbe54604c6bb36f815598d4d10 /numpy/lib/tests/test_function_base.py | |
parent | ad0049026f4afa5d6e8312c1a46b3777be396104 (diff) | |
parent | 849756ff1582ad46d14b7ff621a6e524337304f9 (diff) | |
download | numpy-9ae4a0d1ebc52a556ed13248172e8280ad9fc6bd.tar.gz |
Merge pull request #15805 from eric-wieser/expired-insert-delete-TypeError
DEP: Forbid passing non-integral index arrays to `insert` and `delete`
Diffstat (limited to 'numpy/lib/tests/test_function_base.py')
-rw-r--r-- | numpy/lib/tests/test_function_base.py | 12 |
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 860cf452b..a5988a719 100644 --- a/numpy/lib/tests/test_function_base.py +++ b/numpy/lib/tests/test_function_base.py @@ -544,6 +544,12 @@ class TestInsert: b = np.insert(a, [0, 2], val) assert_array_equal(b[[0, 3]], np.array(val, dtype=b.dtype)) + def test_index_floats(self): + with pytest.raises(IndexError): + np.insert([0, 1, 2], np.array([1.0, 2.0]), [10, 20]) + with pytest.raises(IndexError): + np.insert([0, 1, 2], np.array([], dtype=float), []) + class TestAmax: @@ -868,6 +874,12 @@ class TestDelete: assert_equal(m.flags.c_contiguous, k.flags.c_contiguous) assert_equal(m.flags.f_contiguous, k.flags.f_contiguous) + def test_index_floats(self): + with pytest.raises(IndexError): + np.delete([0, 1, 2], np.array([1.0, 2.0])) + with pytest.raises(IndexError): + np.delete([0, 1, 2], np.array([], dtype=float)) + class TestGradient: |