🏷️ ufunc annotations for logical-{not,and,or,xor}

Author: guan404ming

src/numpy-stubs/__init__.pyi

@@ -7102,7 +7102,7 @@ log: Final[_ufunc_1_1] = ...
 log2: Final[_ufunc_1_1] = ...
 log10: Final[_ufunc_1_1] = ...
 log1p: Final[_ufunc_1_1] = ...
-logical_not: Final[_ufunc_1_1] = ...
+logical_not: Final[_ufunc_1_1[_Call11Bool]] = ...
 negative: Final[_ufunc_1_1] = ...
 positive: Final[_ufunc_1_1] = ...
 rad2deg: Final[_ufunc_1_1] = ...
@@ -7159,9 +7159,9 @@ ldexp: Final[_ufunc_2_1] = ...
 left_shift: Final[_ufunc_2_1] = ...
 logaddexp: Final[_ufunc_2_1] = ...
 logaddexp2: Final[_ufunc_2_1] = ...
-logical_and: Final[_ufunc_2_1] = ...
-logical_or: Final[_ufunc_2_1] = ...
-logical_xor: Final[_ufunc_2_1] = ...
+logical_and: Final[_ufunc_2_1[_Call21Bool]] = ...
+logical_or: Final[_ufunc_2_1[_Call21Bool]] = ...
+logical_xor: Final[_ufunc_2_1[_Call21Bool]] = ...
 maximum: Final[_ufunc_2_1] = ...
 minimum: Final[_ufunc_2_1] = ...
 mod: Final[_ufunc_2_1] = ...

test/static/accept/ufuncs.pyi

@@ -62,3 +62,25 @@ assert_type(np.isinf(AR_f8, out=AR_bool), npt.NDArray[np.bool_])
 assert_type(np.isfinite(f8), np.bool_)
 assert_type(np.isfinite(AR_f8), npt.NDArray[np.bool_])
 assert_type(np.isfinite(AR_f8, out=AR_bool), npt.NDArray[np.bool_])
+
+assert_type(np.logical_not(True), np.bool_)
+assert_type(np.logical_not(AR_bool), npt.NDArray[np.bool_])
+assert_type(np.logical_not(AR_bool, out=AR_bool), npt.NDArray[np.bool_])
+
+assert_type(np.logical_and(True, True), np.bool_)
+assert_type(np.logical_and(AR_bool, AR_bool), npt.NDArray[np.bool_])
+assert_type(np.logical_and(AR_bool, AR_bool, out=AR_bool), npt.NDArray[np.bool_])
+assert_type(np.logical_and(AR_i8, AR_bool, out=AR_bool), npt.NDArray[np.bool_])
+assert_type(np.logical_and(AR_bool, AR_i8), npt.NDArray[np.bool_])
+
+assert_type(np.logical_or(True, True), np.bool_)
+assert_type(np.logical_or(AR_bool, AR_bool), npt.NDArray[np.bool_])
+assert_type(np.logical_or(AR_bool, AR_bool, out=AR_bool), npt.NDArray[np.bool_])
+assert_type(np.logical_or(AR_i8, AR_bool, out=AR_bool), npt.NDArray[np.bool_])
+assert_type(np.logical_or(AR_bool, AR_i8), npt.NDArray[np.bool_])
+
+assert_type(np.logical_xor(True, True), np.bool_)
+assert_type(np.logical_xor(AR_bool, AR_bool), npt.NDArray[np.bool_])
+assert_type(np.logical_xor(AR_bool, AR_bool, out=AR_bool), npt.NDArray[np.bool_])
+assert_type(np.logical_xor(AR_i8, AR_bool, out=AR_bool), npt.NDArray[np.bool_])
+assert_type(np.logical_xor(AR_bool, AR_i8), npt.NDArray[np.bool_])