summaryrefslogtreecommitdiff
path: root/numpy/lib/tests/test_function_base.py
diff options
context:
space:
mode:
authorSebastian Berg <sebastian@sipsolutions.net>2020-03-23 09:55:11 -0500
committerGitHub <noreply@github.com>2020-03-23 09:55:11 -0500
commit9ae4a0d1ebc52a556ed13248172e8280ad9fc6bd (patch)
tree71c37df0211023fbe54604c6bb36f815598d4d10 /numpy/lib/tests/test_function_base.py
parentad0049026f4afa5d6e8312c1a46b3777be396104 (diff)
parent849756ff1582ad46d14b7ff621a6e524337304f9 (diff)
downloadnumpy-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.py12
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: