codekingpro/portable-devtools

1from typing import Any
2
3import numpy as np
4import numpy.typing as npt
5
6b_ = np.bool()
7dt = np.datetime64(0, "D")
8td = np.timedelta64(0, "D")
9
10AR_b: npt.NDArray[np.bool]
11AR_u: npt.NDArray[np.uint32]
12AR_i: npt.NDArray[np.int64]
13AR_f: npt.NDArray[np.longdouble]
14AR_c: npt.NDArray[np.complex128]
15AR_m: npt.NDArray[np.timedelta64]
16AR_M: npt.NDArray[np.datetime64]
17
18ANY: Any
19
20AR_LIKE_b: list[bool]
21AR_LIKE_u: list[np.uint32]
22AR_LIKE_i: list[int]
23AR_LIKE_f: list[float]
24AR_LIKE_c: list[complex]
25AR_LIKE_m: list[np.timedelta64]
26AR_LIKE_M: list[np.datetime64]
27
28# Array subtraction
29
30# NOTE: mypys `NoReturn` errors are, unfortunately, not that great
31_1 = AR_b - AR_LIKE_b  # type: ignore[var-annotated]
32_2 = AR_LIKE_b - AR_b  # type: ignore[var-annotated]
33AR_i - b""  # type: ignore[operator]
34
35AR_f - AR_LIKE_m  # type: ignore[operator]
36AR_f - AR_LIKE_M  # type: ignore[operator]
37AR_c - AR_LIKE_m  # type: ignore[operator]
38AR_c - AR_LIKE_M  # type: ignore[operator]
39
40AR_m - AR_LIKE_f  # type: ignore[operator]
41AR_M - AR_LIKE_f  # type: ignore[operator]
42AR_m - AR_LIKE_c  # type: ignore[operator]
43AR_M - AR_LIKE_c  # type: ignore[operator]
44
45AR_m - AR_LIKE_M  # type: ignore[operator]
46AR_LIKE_m - AR_M  # type: ignore[operator]
47
48# array floor division
49
50AR_M // AR_LIKE_b  # type: ignore[operator]
51AR_M // AR_LIKE_u  # type: ignore[operator]
52AR_M // AR_LIKE_i  # type: ignore[operator]
53AR_M // AR_LIKE_f  # type: ignore[operator]
54AR_M // AR_LIKE_c  # type: ignore[operator]
55AR_M // AR_LIKE_m  # type: ignore[operator]
56AR_M // AR_LIKE_M  # type: ignore[operator]
57
58AR_b // AR_LIKE_M  # type: ignore[operator]
59AR_u // AR_LIKE_M  # type: ignore[operator]
60AR_i // AR_LIKE_M  # type: ignore[operator]
61AR_f // AR_LIKE_M  # type: ignore[operator]
62AR_c // AR_LIKE_M  # type: ignore[operator]
63AR_m // AR_LIKE_M  # type: ignore[operator]
64AR_M // AR_LIKE_M  # type: ignore[operator]
65
66_3 = AR_m // AR_LIKE_b  # type: ignore[var-annotated]
67AR_m // AR_LIKE_c  # type: ignore[operator]
68
69AR_b // AR_LIKE_m  # type: ignore[operator]
70AR_u // AR_LIKE_m  # type: ignore[operator]
71AR_i // AR_LIKE_m  # type: ignore[operator]
72AR_f // AR_LIKE_m  # type: ignore[operator]
73AR_c // AR_LIKE_m  # type: ignore[operator]
74
75# regression tests for https://github.com/numpy/numpy/issues/28957
76AR_c // 2  # type: ignore[operator]
77AR_c // AR_i  # type: ignore[operator]
78AR_c // AR_c  # type: ignore[operator]
79
80# Array multiplication
81
82AR_b *= AR_LIKE_u  # type: ignore[arg-type]
83AR_b *= AR_LIKE_i  # type: ignore[arg-type]
84AR_b *= AR_LIKE_f  # type: ignore[arg-type]
85AR_b *= AR_LIKE_c  # type: ignore[arg-type]
86AR_b *= AR_LIKE_m  # type: ignore[arg-type]
87
88AR_u *= AR_LIKE_f  # type: ignore[arg-type]
89AR_u *= AR_LIKE_c  # type: ignore[arg-type]
90AR_u *= AR_LIKE_m  # type: ignore[arg-type]
91
92AR_i *= AR_LIKE_f  # type: ignore[arg-type]
93AR_i *= AR_LIKE_c  # type: ignore[arg-type]
94AR_i *= AR_LIKE_m  # type: ignore[arg-type]
95
96AR_f *= AR_LIKE_c  # type: ignore[arg-type]
97AR_f *= AR_LIKE_m  # type: ignore[arg-type]
98
99# Array power
100
101AR_b **= AR_LIKE_b  # type: ignore[misc]
102AR_b **= AR_LIKE_u  # type: ignore[misc]
103AR_b **= AR_LIKE_i  # type: ignore[misc]
104AR_b **= AR_LIKE_f  # type: ignore[misc]
105AR_b **= AR_LIKE_c  # type: ignore[misc]
106
107AR_u **= AR_LIKE_f  # type: ignore[arg-type]
108AR_u **= AR_LIKE_c  # type: ignore[arg-type]
109
110AR_i **= AR_LIKE_f  # type: ignore[arg-type]
111AR_i **= AR_LIKE_c  # type: ignore[arg-type]
112
113AR_f **= AR_LIKE_c  # type: ignore[arg-type]
114
115# Scalars
116
117b_ - b_  # type: ignore[operator]
118
119dt + dt  # type: ignore[operator]
120td - dt  # type: ignore[operator]
121td % 1  # type: ignore[operator]
122td / dt  # type: ignore[operator]
123td % dt  # type: ignore[operator]
124
125-b_  # type: ignore[operator]
126+b_  # type: ignore[operator]
127