codekingpro/portable-devtools

1from typing import Literal as L, TypeAlias, assert_type
2
3import numpy as np
4import numpy.typing as npt
5
6FalseType: TypeAlias = L[False]
7TrueType: TypeAlias = L[True]
8
9i4: np.int32
10i8: np.int64
11
12u4: np.uint32
13u8: np.uint64
14
15b_: np.bool[bool]
16b0_: np.bool[FalseType]
17b1_: np.bool[TrueType]
18
19b: bool
20b0: FalseType
21b1: TrueType
22
23i: int
24
25AR: npt.NDArray[np.int32]
26
27assert_type(i8 << i8, np.int64)
28assert_type(i8 >> i8, np.int64)
29assert_type(i8 | i8, np.int64)
30assert_type(i8 ^ i8, np.int64)
31assert_type(i8 & i8, np.int64)
32
33assert_type(i8 << AR, npt.NDArray[np.signedinteger])
34assert_type(i8 >> AR, npt.NDArray[np.signedinteger])
35assert_type(i8 | AR, npt.NDArray[np.signedinteger])
36assert_type(i8 ^ AR, npt.NDArray[np.signedinteger])
37assert_type(i8 & AR, npt.NDArray[np.signedinteger])
38
39assert_type(i4 << i4, np.int32)
40assert_type(i4 >> i4, np.int32)
41assert_type(i4 | i4, np.int32)
42assert_type(i4 ^ i4, np.int32)
43assert_type(i4 & i4, np.int32)
44
45assert_type(i8 << i4, np.signedinteger)
46assert_type(i8 >> i4, np.signedinteger)
47assert_type(i8 | i4, np.signedinteger)
48assert_type(i8 ^ i4, np.signedinteger)
49assert_type(i8 & i4, np.signedinteger)
50
51assert_type(i8 << b_, np.int64)
52assert_type(i8 >> b_, np.int64)
53assert_type(i8 | b_, np.int64)
54assert_type(i8 ^ b_, np.int64)
55assert_type(i8 & b_, np.int64)
56
57assert_type(i8 << b, np.int64)
58assert_type(i8 >> b, np.int64)
59assert_type(i8 | b, np.int64)
60assert_type(i8 ^ b, np.int64)
61assert_type(i8 & b, np.int64)
62
63assert_type(u8 << u8, np.uint64)
64assert_type(u8 >> u8, np.uint64)
65assert_type(u8 | u8, np.uint64)
66assert_type(u8 ^ u8, np.uint64)
67assert_type(u8 & u8, np.uint64)
68
69assert_type(u8 << AR, npt.NDArray[np.signedinteger])
70assert_type(u8 >> AR, npt.NDArray[np.signedinteger])
71assert_type(u8 | AR, npt.NDArray[np.signedinteger])
72assert_type(u8 ^ AR, npt.NDArray[np.signedinteger])
73assert_type(u8 & AR, npt.NDArray[np.signedinteger])
74
75assert_type(u4 << u4, np.uint32)
76assert_type(u4 >> u4, np.uint32)
77assert_type(u4 | u4, np.uint32)
78assert_type(u4 ^ u4, np.uint32)
79assert_type(u4 & u4, np.uint32)
80
81assert_type(u4 << i4, np.signedinteger)
82assert_type(u4 >> i4, np.signedinteger)
83assert_type(u4 | i4, np.signedinteger)
84assert_type(u4 ^ i4, np.signedinteger)
85assert_type(u4 & i4, np.signedinteger)
86
87assert_type(u4 << i, np.uint32)
88assert_type(u4 >> i, np.uint32)
89assert_type(u4 | i, np.uint32)
90assert_type(u4 ^ i, np.uint32)
91assert_type(u4 & i, np.uint32)
92
93assert_type(u8 << b_, np.uint64)
94assert_type(u8 >> b_, np.uint64)
95assert_type(u8 | b_, np.uint64)
96assert_type(u8 ^ b_, np.uint64)
97assert_type(u8 & b_, np.uint64)
98
99assert_type(u8 << b, np.uint64)
100assert_type(u8 >> b, np.uint64)
101assert_type(u8 | b, np.uint64)
102assert_type(u8 ^ b, np.uint64)
103assert_type(u8 & b, np.uint64)
104
105assert_type(b_ << b_, np.int8)
106assert_type(b_ >> b_, np.int8)
107assert_type(b_ | b_, np.bool)
108assert_type(b_ ^ b_, np.bool)
109assert_type(b_ & b_, np.bool)
110
111assert_type(b_ << AR, npt.NDArray[np.signedinteger])
112assert_type(b_ >> AR, npt.NDArray[np.signedinteger])
113assert_type(b_ | AR, npt.NDArray[np.signedinteger])
114assert_type(b_ ^ AR, npt.NDArray[np.signedinteger])
115assert_type(b_ & AR, npt.NDArray[np.signedinteger])
116
117assert_type(b_ << b, np.int8)
118assert_type(b_ >> b, np.int8)
119assert_type(b_ | b, np.bool)
120assert_type(b_ ^ b, np.bool)
121assert_type(b_ & b, np.bool)
122
123assert_type(b_ << i, np.int_)
124assert_type(b_ >> i, np.int_)
125assert_type(b_ | i, np.bool | np.int_)
126assert_type(b_ ^ i, np.bool | np.int_)
127assert_type(b_ & i, np.bool | np.int_)
128
129assert_type(~i8, np.int64)
130assert_type(~i4, np.int32)
131assert_type(~u8, np.uint64)
132assert_type(~u4, np.uint32)
133assert_type(~b_, np.bool)
134assert_type(~b0_, np.bool[TrueType])
135assert_type(~b1_, np.bool[FalseType])
136assert_type(~AR, npt.NDArray[np.int32])
137
138assert_type(b_ | b0_, np.bool)
139assert_type(b0_ | b_, np.bool)
140assert_type(b_ | b1_, np.bool[TrueType])
141assert_type(b1_ | b_, np.bool[TrueType])
142
143assert_type(b_ ^ b0_, np.bool)
144assert_type(b0_ ^ b_, np.bool)
145assert_type(b_ ^ b1_, np.bool)
146assert_type(b1_ ^ b_, np.bool)
147
148assert_type(b_ & b0_, np.bool[FalseType])
149assert_type(b0_ & b_, np.bool[FalseType])
150assert_type(b_ & b1_, np.bool)
151assert_type(b1_ & b_, np.bool)
152
153assert_type(b0_ | b0_, np.bool[FalseType])
154assert_type(b0_ | b1_, np.bool[TrueType])
155assert_type(b1_ | b0_, np.bool[TrueType])
156assert_type(b1_ | b1_, np.bool[TrueType])
157
158assert_type(b0_ ^ b0_, np.bool[FalseType])
159assert_type(b0_ ^ b1_, np.bool[TrueType])
160assert_type(b1_ ^ b0_, np.bool[TrueType])
161assert_type(b1_ ^ b1_, np.bool[FalseType])
162
163assert_type(b0_ & b0_, np.bool[FalseType])
164assert_type(b0_ & b1_, np.bool[FalseType])
165assert_type(b1_ & b0_, np.bool[FalseType])
166assert_type(b1_ & b1_, np.bool[TrueType])
167