Skip to content

Commit fb845e0

Browse files
authored
🏷️ add test for np.put() with various modes (#271)
1 parent 9bf3100 commit fb845e0

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

test/static/accept/fromnumeric.pyi

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,14 @@ assert_type(np.repeat(f, 1), npt.NDArray[Any])
5050
assert_type(np.repeat(AR_b, 1), npt.NDArray[np.bool])
5151
assert_type(np.repeat(AR_f4, 1), npt.NDArray[np.float32])
5252

53-
# TODO: array_bdd tests for np.put()
53+
assert_type(np.put(AR_b, [0], [True]), None)
54+
assert_type(np.put(AR_f4, [0], [1.0]), None)
55+
assert_type(np.put(AR_b, [0], [True], mode="raise"), None)
56+
assert_type(np.put(AR_f4, [0], [1.0], mode="raise"), None)
57+
assert_type(np.put(AR_b, [0], [True], mode="wrap"), None)
58+
assert_type(np.put(AR_f4, [0], [1.0], mode="wrap"), None)
59+
assert_type(np.put(AR_b, [0], [True], mode="clip"), None)
60+
assert_type(np.put(AR_f4, [0], [1.0], mode="clip"), None)
5461

5562
assert_type(np.swapaxes([[0, 1]], 0, 0), npt.NDArray[Any])
5663
assert_type(np.swapaxes(AR_b, 0, 0), npt.NDArray[np.bool])

0 commit comments

Comments
 (0)