codekingpro/portable-devtools
114k
1from typing import assert_type
2
3import numpy as np
4import numpy.typing as npt
5
6AR_LIKE_b: list[bool]
7AR_LIKE_u: list[np.uint32]
8AR_LIKE_i: list[int]
9AR_LIKE_f: list[float]
10AR_LIKE_O: list[np.object_]
11
12AR_U: npt.NDArray[np.str_]
13
14assert_type(np.fix(AR_LIKE_b), npt.NDArray[np.floating]) # type: ignore[deprecated]
15assert_type(np.fix(AR_LIKE_u), npt.NDArray[np.floating]) # type: ignore[deprecated]
16assert_type(np.fix(AR_LIKE_i), npt.NDArray[np.floating]) # type: ignore[deprecated]
17assert_type(np.fix(AR_LIKE_f), npt.NDArray[np.floating]) # type: ignore[deprecated]
18assert_type(np.fix(AR_LIKE_O), npt.NDArray[np.object_]) # type: ignore[deprecated]
19assert_type(np.fix(AR_LIKE_f, out=AR_U), npt.NDArray[np.str_]) # type: ignore[deprecated]
20
21assert_type(np.isposinf(AR_LIKE_b), npt.NDArray[np.bool])
22assert_type(np.isposinf(AR_LIKE_u), npt.NDArray[np.bool])
23assert_type(np.isposinf(AR_LIKE_i), npt.NDArray[np.bool])
24assert_type(np.isposinf(AR_LIKE_f), npt.NDArray[np.bool])
25assert_type(np.isposinf(AR_LIKE_f, out=AR_U), npt.NDArray[np.str_])
26
27assert_type(np.isneginf(AR_LIKE_b), npt.NDArray[np.bool])
28assert_type(np.isneginf(AR_LIKE_u), npt.NDArray[np.bool])
29assert_type(np.isneginf(AR_LIKE_i), npt.NDArray[np.bool])
30assert_type(np.isneginf(AR_LIKE_f), npt.NDArray[np.bool])
31assert_type(np.isneginf(AR_LIKE_f, out=AR_U), npt.NDArray[np.str_])
32 