Skip to content

Commit 900bb44

Browse files
committed
✅ add test for is{finite,inf,nan.nat}
1 parent 1b362b6 commit 900bb44

File tree

2 files changed

+18
-1
lines changed

2 files changed

+18
-1
lines changed

src/numpy-stubs/_typing/_ufunc.pyi

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -209,7 +209,7 @@ class _Call11Bool(Protocol):
209209
*,
210210
dtype: DTypeLike | None = None,
211211
**kwds: Unpack[_Kwargs2],
212-
) -> NDArray[np.bool] | np.bool: ...
212+
) -> NDArray[np.bool]: ...
213213
@overload # (array-like) -> Array[bool] | bool
214214
def __call__(
215215
self,

test/static/accept/ufuncs.pyi

Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,7 @@ i8: np.int64
88
f8: np.float64
99
AR_f8: npt.NDArray[np.float64]
1010
AR_i8: npt.NDArray[np.int64]
11+
AR_bool: npt.NDArray[np.bool_]
1112

1213
assert_type(np.absolute.types, list[str])
1314

@@ -36,3 +37,19 @@ assert_type(np.vecdot(AR_f8, AR_f8), Any)
3637

3738
assert_type(np.bitwise_count(i8), Any)
3839
assert_type(np.bitwise_count(AR_i8), npt.NDArray[Any])
40+
41+
assert_type(np.isnan(f8), np.bool_)
42+
assert_type(np.isnan(AR_f8), npt.NDArray[np.bool_])
43+
assert_type(np.isnan(AR_f8, out=AR_bool), npt.NDArray[np.bool_])
44+
45+
assert_type(np.isnat(f8), np.bool_)
46+
assert_type(np.isnat(AR_f8), npt.NDArray[np.bool_])
47+
assert_type(np.isnat(AR_f8, out=AR_bool), npt.NDArray[np.bool_])
48+
49+
assert_type(np.isinf(f8), np.bool_)
50+
assert_type(np.isinf(AR_f8), npt.NDArray[np.bool_])
51+
assert_type(np.isinf(AR_f8, out=AR_bool), npt.NDArray[np.bool_])
52+
53+
assert_type(np.isfinite(f8), np.bool_)
54+
assert_type(np.isfinite(AR_f8), npt.NDArray[np.bool_])
55+
assert_type(np.isfinite(AR_f8, out=AR_bool), npt.NDArray[np.bool_])

0 commit comments

Comments
 (0)