We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
np.put()
1 parent 9bf3100 commit fb845e0Copy full SHA for fb845e0
test/static/accept/fromnumeric.pyi
@@ -50,7 +50,14 @@ assert_type(np.repeat(f, 1), npt.NDArray[Any])
50
assert_type(np.repeat(AR_b, 1), npt.NDArray[np.bool])
51
assert_type(np.repeat(AR_f4, 1), npt.NDArray[np.float32])
52
53
-# TODO: array_bdd tests for np.put()
+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)
61
62
assert_type(np.swapaxes([[0, 1]], 0, 0), npt.NDArray[Any])
63
assert_type(np.swapaxes(AR_b, 0, 0), npt.NDArray[np.bool])
0 commit comments