codekingpro/portable-devtools

1from typing import Any, TypeAlias, assert_type
2
3import numpy as np
4import numpy.typing as npt
5
6_BytesCharArray: TypeAlias = np.char.chararray[tuple[Any, ...], np.dtype[np.bytes_]]
7_StrCharArray: TypeAlias = np.char.chararray[tuple[Any, ...], np.dtype[np.str_]]
8
9AR_U: _StrCharArray
10AR_S: _BytesCharArray
11
12assert_type(AR_U == AR_U, npt.NDArray[np.bool])
13assert_type(AR_S == AR_S, npt.NDArray[np.bool])
14
15assert_type(AR_U != AR_U, npt.NDArray[np.bool])
16assert_type(AR_S != AR_S, npt.NDArray[np.bool])
17
18assert_type(AR_U >= AR_U, npt.NDArray[np.bool])
19assert_type(AR_S >= AR_S, npt.NDArray[np.bool])
20
21assert_type(AR_U <= AR_U, npt.NDArray[np.bool])
22assert_type(AR_S <= AR_S, npt.NDArray[np.bool])
23
24assert_type(AR_U > AR_U, npt.NDArray[np.bool])
25assert_type(AR_S > AR_S, npt.NDArray[np.bool])
26
27assert_type(AR_U < AR_U, npt.NDArray[np.bool])
28assert_type(AR_S < AR_S, npt.NDArray[np.bool])
29
30assert_type(AR_U * 5, _StrCharArray)
31assert_type(AR_S * [5], _BytesCharArray)
32
33assert_type(AR_U % "test", _StrCharArray)
34assert_type(AR_S % b"test", _BytesCharArray)
35
36assert_type(AR_U.capitalize(), _StrCharArray)
37assert_type(AR_S.capitalize(), _BytesCharArray)
38
39assert_type(AR_U.center(5), _StrCharArray)
40assert_type(AR_S.center([2, 3, 4], b"a"), _BytesCharArray)
41
42assert_type(AR_U.encode(), _BytesCharArray)
43assert_type(AR_S.decode(), _StrCharArray)
44
45assert_type(AR_U.expandtabs(), _StrCharArray)
46assert_type(AR_S.expandtabs(tabsize=4), _BytesCharArray)
47
48assert_type(AR_U.join("_"), _StrCharArray)
49assert_type(AR_S.join([b"_", b""]), _BytesCharArray)
50
51assert_type(AR_U.ljust(5), _StrCharArray)
52assert_type(AR_S.ljust([4, 3, 1], fillchar=[b"a", b"b", b"c"]), _BytesCharArray)
53assert_type(AR_S.ljust([4, 3, 1], fillchar="a"), _BytesCharArray)
54assert_type(AR_U.rjust(5), _StrCharArray)
55assert_type(AR_S.rjust([4, 3, 1], fillchar=[b"a", b"b", b"c"]), _BytesCharArray)
56
57assert_type(AR_U.lstrip(), _StrCharArray)
58assert_type(AR_S.lstrip(chars=b"_"), _BytesCharArray)
59assert_type(AR_U.rstrip(), _StrCharArray)
60assert_type(AR_S.rstrip(chars=b"_"), _BytesCharArray)
61assert_type(AR_U.strip(), _StrCharArray)
62assert_type(AR_S.strip(chars=b"_"), _BytesCharArray)
63
64assert_type(AR_U.partition("\n"), _StrCharArray)
65assert_type(AR_S.partition([b"a", b"b", b"c"]), _BytesCharArray)
66assert_type(AR_U.rpartition("\n"), _StrCharArray)
67assert_type(AR_S.rpartition([b"a", b"b", b"c"]), _BytesCharArray)
68
69assert_type(AR_U.replace("_", "-"), _StrCharArray)
70assert_type(AR_S.replace([b"_", b""], [b"a", b"b"]), _BytesCharArray)
71
72assert_type(AR_U.split("_"), npt.NDArray[np.object_])
73assert_type(AR_S.split(maxsplit=[1, 2, 3]), npt.NDArray[np.object_])
74assert_type(AR_U.rsplit("_"), npt.NDArray[np.object_])
75assert_type(AR_S.rsplit(maxsplit=[1, 2, 3]), npt.NDArray[np.object_])
76
77assert_type(AR_U.splitlines(), npt.NDArray[np.object_])
78assert_type(AR_S.splitlines(keepends=[True, True, False]), npt.NDArray[np.object_])
79
80assert_type(AR_U.swapcase(), _StrCharArray)
81assert_type(AR_S.swapcase(), _BytesCharArray)
82
83assert_type(AR_U.title(), _StrCharArray)
84assert_type(AR_S.title(), _BytesCharArray)
85
86assert_type(AR_U.upper(), _StrCharArray)
87assert_type(AR_S.upper(), _BytesCharArray)
88
89assert_type(AR_U.zfill(5), _StrCharArray)
90assert_type(AR_S.zfill([2, 3, 4]), _BytesCharArray)
91
92assert_type(AR_U.count("a", start=[1, 2, 3]), npt.NDArray[np.int_])
93assert_type(AR_S.count([b"a", b"b", b"c"], end=9), npt.NDArray[np.int_])
94
95assert_type(AR_U.endswith("a", start=[1, 2, 3]), npt.NDArray[np.bool])
96assert_type(AR_S.endswith([b"a", b"b", b"c"], end=9), npt.NDArray[np.bool])
97assert_type(AR_U.startswith("a", start=[1, 2, 3]), npt.NDArray[np.bool])
98assert_type(AR_S.startswith([b"a", b"b", b"c"], end=9), npt.NDArray[np.bool])
99
100assert_type(AR_U.find("a", start=[1, 2, 3]), npt.NDArray[np.int_])
101assert_type(AR_S.find([b"a", b"b", b"c"], end=9), npt.NDArray[np.int_])
102assert_type(AR_U.rfind("a", start=[1, 2, 3]), npt.NDArray[np.int_])
103assert_type(AR_S.rfind([b"a", b"b", b"c"], end=9), npt.NDArray[np.int_])
104
105assert_type(AR_U.index("a", start=[1, 2, 3]), npt.NDArray[np.int_])
106assert_type(AR_S.index([b"a", b"b", b"c"], end=9), npt.NDArray[np.int_])
107assert_type(AR_U.rindex("a", start=[1, 2, 3]), npt.NDArray[np.int_])
108assert_type(AR_S.rindex([b"a", b"b", b"c"], end=9), npt.NDArray[np.int_])
109
110assert_type(AR_U.isalpha(), npt.NDArray[np.bool])
111assert_type(AR_S.isalpha(), npt.NDArray[np.bool])
112
113assert_type(AR_U.isalnum(), npt.NDArray[np.bool])
114assert_type(AR_S.isalnum(), npt.NDArray[np.bool])
115
116assert_type(AR_U.isdecimal(), npt.NDArray[np.bool])
117assert_type(AR_S.isdecimal(), npt.NDArray[np.bool])
118
119assert_type(AR_U.isdigit(), npt.NDArray[np.bool])
120assert_type(AR_S.isdigit(), npt.NDArray[np.bool])
121
122assert_type(AR_U.islower(), npt.NDArray[np.bool])
123assert_type(AR_S.islower(), npt.NDArray[np.bool])
124
125assert_type(AR_U.isnumeric(), npt.NDArray[np.bool])
126assert_type(AR_S.isnumeric(), npt.NDArray[np.bool])
127
128assert_type(AR_U.isspace(), npt.NDArray[np.bool])
129assert_type(AR_S.isspace(), npt.NDArray[np.bool])
130
131assert_type(AR_U.istitle(), npt.NDArray[np.bool])
132assert_type(AR_S.istitle(), npt.NDArray[np.bool])
133
134assert_type(AR_U.isupper(), npt.NDArray[np.bool])
135assert_type(AR_S.isupper(), npt.NDArray[np.bool])
136
137assert_type(AR_U.__array_finalize__(object()), None)
138assert_type(AR_S.__array_finalize__(object()), None)
139