codekingpro/portable-devtools

1from typing import Any, Literal, NoReturn, assert_type
2
3import numpy as np
4import numpy.typing as npt
5
6i8: np.int64
7f8: np.float64
8AR_f8: npt.NDArray[np.float64]
9AR_i8: npt.NDArray[np.int64]
10
11assert_type(np.absolute.__doc__, str)
12assert_type(np.absolute.types, list[str])
13
14assert_type(np.absolute.__name__, Literal["absolute"])
15assert_type(np.absolute.__qualname__, Literal["absolute"])
16assert_type(np.absolute.ntypes, Literal[20])
17assert_type(np.absolute.identity, None)
18assert_type(np.absolute.nin, Literal[1])
19assert_type(np.absolute.nin, Literal[1])
20assert_type(np.absolute.nout, Literal[1])
21assert_type(np.absolute.nargs, Literal[2])
22assert_type(np.absolute.signature, None)
23assert_type(np.absolute(f8), Any)
24assert_type(np.absolute(AR_f8), npt.NDArray[Any])
25assert_type(np.absolute.at(AR_f8, AR_i8), None)
26
27assert_type(np.add.__name__, Literal["add"])
28assert_type(np.add.__qualname__, Literal["add"])
29assert_type(np.add.ntypes, Literal[22])
30assert_type(np.add.identity, Literal[0])
31assert_type(np.add.nin, Literal[2])
32assert_type(np.add.nout, Literal[1])
33assert_type(np.add.nargs, Literal[3])
34assert_type(np.add.signature, None)
35assert_type(np.add(f8, f8), Any)
36assert_type(np.add(AR_f8, f8), npt.NDArray[Any])
37assert_type(np.add.at(AR_f8, AR_i8, f8), None)
38assert_type(np.add.reduce(AR_f8, axis=0), Any)
39assert_type(np.add.accumulate(AR_f8), npt.NDArray[Any])
40assert_type(np.add.reduceat(AR_f8, AR_i8), npt.NDArray[Any])
41assert_type(np.add.outer(f8, f8), Any)
42assert_type(np.add.outer(AR_f8, f8), npt.NDArray[Any])
43
44assert_type(np.frexp.__name__, Literal["frexp"])
45assert_type(np.frexp.__qualname__, Literal["frexp"])
46assert_type(np.frexp.ntypes, Literal[4])
47assert_type(np.frexp.identity, None)
48assert_type(np.frexp.nin, Literal[1])
49assert_type(np.frexp.nout, Literal[2])
50assert_type(np.frexp.nargs, Literal[3])
51assert_type(np.frexp.signature, None)
52assert_type(np.frexp(f8), tuple[Any, Any])
53assert_type(np.frexp(AR_f8), tuple[npt.NDArray[Any], npt.NDArray[Any]])
54
55assert_type(np.divmod.__name__, Literal["divmod"])
56assert_type(np.divmod.__qualname__, Literal["divmod"])
57assert_type(np.divmod.ntypes, Literal[15])
58assert_type(np.divmod.identity, None)
59assert_type(np.divmod.nin, Literal[2])
60assert_type(np.divmod.nout, Literal[2])
61assert_type(np.divmod.nargs, Literal[4])
62assert_type(np.divmod.signature, None)
63assert_type(np.divmod(f8, f8), tuple[Any, Any])
64assert_type(np.divmod(AR_f8, f8), tuple[npt.NDArray[Any], npt.NDArray[Any]])
65
66assert_type(np.matmul.__name__, Literal["matmul"])
67assert_type(np.matmul.__qualname__, Literal["matmul"])
68assert_type(np.matmul.ntypes, Literal[19])
69assert_type(np.matmul.identity, None)
70assert_type(np.matmul.nin, Literal[2])
71assert_type(np.matmul.nout, Literal[1])
72assert_type(np.matmul.nargs, Literal[3])
73assert_type(np.matmul.signature, Literal["(n?,k),(k,m?)->(n?,m?)"])
74assert_type(np.matmul.identity, None)
75assert_type(np.matmul(AR_f8, AR_f8), Any)
76assert_type(np.matmul(AR_f8, AR_f8, axes=[(0, 1), (0, 1), (0, 1)]), Any)
77
78assert_type(np.vecdot.__name__, Literal["vecdot"])
79assert_type(np.vecdot.__qualname__, Literal["vecdot"])
80assert_type(np.vecdot.ntypes, Literal[19])
81assert_type(np.vecdot.identity, None)
82assert_type(np.vecdot.nin, Literal[2])
83assert_type(np.vecdot.nout, Literal[1])
84assert_type(np.vecdot.nargs, Literal[3])
85assert_type(np.vecdot.signature, Literal["(n),(n)->()"])
86assert_type(np.vecdot.identity, None)
87assert_type(np.vecdot(AR_f8, AR_f8), Any)
88
89assert_type(np.bitwise_count.__name__, Literal["bitwise_count"])
90assert_type(np.bitwise_count.__qualname__, Literal["bitwise_count"])
91assert_type(np.bitwise_count.ntypes, Literal[11])
92assert_type(np.bitwise_count.identity, None)
93assert_type(np.bitwise_count.nin, Literal[1])
94assert_type(np.bitwise_count.nout, Literal[1])
95assert_type(np.bitwise_count.nargs, Literal[2])
96assert_type(np.bitwise_count.signature, None)
97assert_type(np.bitwise_count.identity, None)
98assert_type(np.bitwise_count(i8), Any)
99assert_type(np.bitwise_count(AR_i8), npt.NDArray[Any])
100
101def test_absolute_outer_invalid() -> None:
102    assert_type(np.absolute.outer(AR_f8, AR_f8), NoReturn)  # type: ignore[arg-type]
103def test_frexp_outer_invalid() -> None:
104    assert_type(np.frexp.outer(AR_f8, AR_f8), NoReturn)  # type: ignore[arg-type]
105def test_divmod_outer_invalid() -> None:
106    assert_type(np.divmod.outer(AR_f8, AR_f8), NoReturn)  # type: ignore[arg-type]
107def test_matmul_outer_invalid() -> None:
108    assert_type(np.matmul.outer(AR_f8, AR_f8), NoReturn)  # type: ignore[arg-type]
109
110def test_absolute_reduceat_invalid() -> None:
111    assert_type(np.absolute.reduceat(AR_f8, AR_i8), NoReturn)  # type: ignore[arg-type]
112def test_frexp_reduceat_invalid() -> None:
113    assert_type(np.frexp.reduceat(AR_f8, AR_i8), NoReturn)  # type: ignore[arg-type]
114def test_divmod_reduceat_invalid() -> None:
115    assert_type(np.divmod.reduceat(AR_f8, AR_i8), NoReturn)  # type: ignore[arg-type]
116def test_matmul_reduceat_invalid() -> None:
117    assert_type(np.matmul.reduceat(AR_f8, AR_i8), NoReturn)  # type: ignore[arg-type]
118
119def test_absolute_reduce_invalid() -> None:
120    assert_type(np.absolute.reduce(AR_f8), NoReturn)  # type: ignore[arg-type]
121def test_frexp_reduce_invalid() -> None:
122    assert_type(np.frexp.reduce(AR_f8), NoReturn)  # type: ignore[arg-type]
123def test_divmod_reduce_invalid() -> None:
124    assert_type(np.divmod.reduce(AR_f8), NoReturn)  # type: ignore[arg-type]
125def test_matmul_reduce_invalid() -> None:
126    assert_type(np.matmul.reduce(AR_f8), NoReturn)  # type: ignore[arg-type]
127
128def test_absolute_accumulate_invalid() -> None:
129    assert_type(np.absolute.accumulate(AR_f8), NoReturn)  # type: ignore[arg-type]
130def test_frexp_accumulate_invalid() -> None:
131    assert_type(np.frexp.accumulate(AR_f8), NoReturn)  # type: ignore[arg-type]
132def test_divmod_accumulate_invalid() -> None:
133    assert_type(np.divmod.accumulate(AR_f8), NoReturn)  # type: ignore[arg-type]
134def test_matmul_accumulate_invalid() -> None:
135    assert_type(np.matmul.accumulate(AR_f8), NoReturn)  # type: ignore[arg-type]
136
137def test_frexp_at_invalid() -> None:
138    assert_type(np.frexp.at(AR_f8, i8), NoReturn)  # type: ignore[arg-type]
139def test_divmod_at_invalid() -> None:
140    assert_type(np.divmod.at(AR_f8, i8, AR_f8), NoReturn)  # type: ignore[arg-type]
141def test_matmul_at_invalid() -> None:
142    assert_type(np.matmul.at(AR_f8, i8, AR_f8), NoReturn)  # type: ignore[arg-type]
143