Datasetpublic
codekingpro/portable-devtools

sourceHugging Faceupdated 5mo agoView on Hugging Face
1likes14kdownloads
rootpythonuser_packagesPython313site-packagesnumpytypingtestsdatareveallinalg.pyi
linalg.pyi155 linesDownload Raw Back to reveal
1from typing import Any, assert_type
2
3import numpy as np
4import numpy.typing as npt
5from numpy.linalg._linalg import (
6    EighResult,
7    EigResult,
8    QRResult,
9    SlogdetResult,
10    SVDResult,
11)
12
13float_list_2d: list[list[float]]
14AR_i8: npt.NDArray[np.int64]
15AR_f4: npt.NDArray[np.float32]
16AR_f8: npt.NDArray[np.float64]
17AR_c8: npt.NDArray[np.complex64]
18AR_c16: npt.NDArray[np.complex128]
19AR_O: npt.NDArray[np.object_]
20AR_m: npt.NDArray[np.timedelta64]
21AR_S: npt.NDArray[np.str_]
22AR_b: npt.NDArray[np.bool]
23
24assert_type(np.linalg.tensorsolve(AR_i8, AR_i8), npt.NDArray[np.float64])
25assert_type(np.linalg.tensorsolve(AR_i8, AR_f8), npt.NDArray[np.floating])
26assert_type(np.linalg.tensorsolve(AR_c16, AR_f8), npt.NDArray[np.complexfloating])
27
28assert_type(np.linalg.solve(AR_i8, AR_i8), npt.NDArray[np.float64])
29assert_type(np.linalg.solve(AR_i8, AR_f8), npt.NDArray[np.floating])
30assert_type(np.linalg.solve(AR_c16, AR_f8), npt.NDArray[np.complexfloating])
31
32assert_type(np.linalg.tensorinv(AR_i8), npt.NDArray[np.float64])
33assert_type(np.linalg.tensorinv(AR_f8), npt.NDArray[np.floating])
34assert_type(np.linalg.tensorinv(AR_c16), npt.NDArray[np.complexfloating])
35
36assert_type(np.linalg.inv(AR_i8), npt.NDArray[np.float64])
37assert_type(np.linalg.inv(AR_f8), npt.NDArray[np.floating])
38assert_type(np.linalg.inv(AR_c16), npt.NDArray[np.complexfloating])
39
40assert_type(np.linalg.matrix_power(AR_i8, -1), npt.NDArray[Any])
41assert_type(np.linalg.matrix_power(AR_f8, 0), npt.NDArray[Any])
42assert_type(np.linalg.matrix_power(AR_c16, 1), npt.NDArray[Any])
43assert_type(np.linalg.matrix_power(AR_O, 2), npt.NDArray[Any])
44
45assert_type(np.linalg.cholesky(AR_i8), npt.NDArray[np.float64])
46assert_type(np.linalg.cholesky(AR_f8), npt.NDArray[np.floating])
47assert_type(np.linalg.cholesky(AR_c16), npt.NDArray[np.complexfloating])
48
49assert_type(np.linalg.outer(AR_i8, AR_i8), npt.NDArray[np.int64])
50assert_type(np.linalg.outer(AR_f8, AR_f8), npt.NDArray[np.float64])
51assert_type(np.linalg.outer(AR_c16, AR_c16), npt.NDArray[np.complex128])
52assert_type(np.linalg.outer(AR_b, AR_b), npt.NDArray[np.bool])
53assert_type(np.linalg.outer(AR_O, AR_O), npt.NDArray[np.object_])
54assert_type(np.linalg.outer(AR_i8, AR_m), npt.NDArray[np.timedelta64])
55
56assert_type(np.linalg.qr(AR_i8), QRResult)
57assert_type(np.linalg.qr(AR_f8), QRResult)
58assert_type(np.linalg.qr(AR_c16), QRResult)
59
60assert_type(np.linalg.eigvals(AR_i8), npt.NDArray[np.float64] | npt.NDArray[np.complex128])
61assert_type(np.linalg.eigvals(AR_f8), npt.NDArray[np.floating] | npt.NDArray[np.complexfloating])
62assert_type(np.linalg.eigvals(AR_c16), npt.NDArray[np.complexfloating])
63
64assert_type(np.linalg.eigvalsh(AR_i8), npt.NDArray[np.float64])
65assert_type(np.linalg.eigvalsh(AR_f8), npt.NDArray[np.floating])
66assert_type(np.linalg.eigvalsh(AR_c16), npt.NDArray[np.floating])
67
68assert_type(np.linalg.eig(AR_i8), EigResult)
69assert_type(np.linalg.eig(AR_f8), EigResult)
70assert_type(np.linalg.eig(AR_c16), EigResult)
71
72assert_type(np.linalg.eigh(AR_i8), EighResult)
73assert_type(np.linalg.eigh(AR_f8), EighResult)
74assert_type(np.linalg.eigh(AR_c16), EighResult)
75
76assert_type(np.linalg.svd(AR_i8), SVDResult)
77assert_type(np.linalg.svd(AR_f8), SVDResult)
78assert_type(np.linalg.svd(AR_c16), SVDResult)
79assert_type(np.linalg.svd(AR_i8, compute_uv=False), npt.NDArray[np.float64])
80assert_type(np.linalg.svd(AR_i8, True, False), npt.NDArray[np.float64])
81assert_type(np.linalg.svd(AR_f8, compute_uv=False), npt.NDArray[np.floating])
82assert_type(np.linalg.svd(AR_c16, compute_uv=False), npt.NDArray[np.floating])
83assert_type(np.linalg.svd(AR_c16, True, False), npt.NDArray[np.floating])
84
85assert_type(np.linalg.svdvals(AR_b), npt.NDArray[np.float64])
86assert_type(np.linalg.svdvals(AR_i8), npt.NDArray[np.float64])
87assert_type(np.linalg.svdvals(AR_f4), npt.NDArray[np.float32])
88assert_type(np.linalg.svdvals(AR_c8), npt.NDArray[np.float32])
89assert_type(np.linalg.svdvals(AR_f8), npt.NDArray[np.float64])
90assert_type(np.linalg.svdvals(AR_c16), npt.NDArray[np.float64])
91assert_type(np.linalg.svdvals([[1, 2], [3, 4]]), npt.NDArray[np.float64])
92assert_type(np.linalg.svdvals([[1.0, 2.0], [3.0, 4.0]]), npt.NDArray[np.float64])
93assert_type(np.linalg.svdvals([[1j, 2j], [3j, 4j]]), npt.NDArray[np.float64])
94
95assert_type(np.linalg.cond(AR_i8), Any)
96assert_type(np.linalg.cond(AR_f8), Any)
97assert_type(np.linalg.cond(AR_c16), Any)
98
99assert_type(np.linalg.matrix_rank(AR_i8), Any)
100assert_type(np.linalg.matrix_rank(AR_f8), Any)
101assert_type(np.linalg.matrix_rank(AR_c16), Any)
102
103assert_type(np.linalg.pinv(AR_i8), npt.NDArray[np.float64])
104assert_type(np.linalg.pinv(AR_f8), npt.NDArray[np.floating])
105assert_type(np.linalg.pinv(AR_c16), npt.NDArray[np.complexfloating])
106
107assert_type(np.linalg.slogdet(AR_i8), SlogdetResult)
108assert_type(np.linalg.slogdet(AR_f8), SlogdetResult)
109assert_type(np.linalg.slogdet(AR_c16), SlogdetResult)
110
111assert_type(np.linalg.det(AR_i8), Any)
112assert_type(np.linalg.det(AR_f8), Any)
113assert_type(np.linalg.det(AR_c16), Any)
114
115assert_type(np.linalg.lstsq(AR_i8, AR_i8), tuple[npt.NDArray[np.float64], npt.NDArray[np.float64], np.int32, npt.NDArray[np.float64]])
116assert_type(np.linalg.lstsq(AR_i8, AR_f8), tuple[npt.NDArray[np.floating], npt.NDArray[np.floating], np.int32, npt.NDArray[np.floating]])
117assert_type(np.linalg.lstsq(AR_f8, AR_c16), tuple[npt.NDArray[np.complexfloating], npt.NDArray[np.floating], np.int32, npt.NDArray[np.floating]])
118
119assert_type(np.linalg.norm(AR_i8), np.floating)
120assert_type(np.linalg.norm(AR_f8), np.floating)
121assert_type(np.linalg.norm(AR_c16), np.floating)
122assert_type(np.linalg.norm(AR_S), np.floating)
123assert_type(np.linalg.norm(AR_f8, axis=0), Any)
124
125assert_type(np.linalg.matrix_norm(AR_i8), np.floating)
126assert_type(np.linalg.matrix_norm(AR_f8), np.floating)
127assert_type(np.linalg.matrix_norm(AR_c16), np.floating)
128assert_type(np.linalg.matrix_norm(AR_S), np.floating)
129
130assert_type(np.linalg.vector_norm(AR_i8), np.floating)
131assert_type(np.linalg.vector_norm(AR_f8), np.floating)
132assert_type(np.linalg.vector_norm(AR_c16), np.floating)
133assert_type(np.linalg.vector_norm(AR_S), np.floating)
134
135assert_type(np.linalg.tensordot(AR_b, AR_b), npt.NDArray[np.bool])
136assert_type(np.linalg.tensordot(AR_i8, AR_i8), npt.NDArray[np.int64])
137assert_type(np.linalg.tensordot(AR_f8, AR_f8), npt.NDArray[np.float64])
138assert_type(np.linalg.tensordot(AR_c16, AR_c16), npt.NDArray[np.complex128])
139assert_type(np.linalg.tensordot(AR_m, AR_m), npt.NDArray[np.timedelta64])
140assert_type(np.linalg.tensordot(AR_O, AR_O), npt.NDArray[np.object_])
141
142assert_type(np.linalg.multi_dot([AR_i8, AR_i8]), Any)
143assert_type(np.linalg.multi_dot([AR_i8, AR_f8]), Any)
144assert_type(np.linalg.multi_dot([AR_f8, AR_c16]), Any)
145assert_type(np.linalg.multi_dot([AR_O, AR_O]), Any)
146assert_type(np.linalg.multi_dot([AR_m, AR_m]), Any)
147
148assert_type(np.linalg.cross(AR_i8, AR_i8), npt.NDArray[np.signedinteger])
149assert_type(np.linalg.cross(AR_f8, AR_f8), npt.NDArray[np.floating])
150assert_type(np.linalg.cross(AR_c16, AR_c16), npt.NDArray[np.complexfloating])
151
152assert_type(np.linalg.matmul(AR_i8, AR_i8), npt.NDArray[np.int64])
153assert_type(np.linalg.matmul(AR_f8, AR_f8), npt.NDArray[np.float64])
154assert_type(np.linalg.matmul(AR_c16, AR_c16), npt.NDArray[np.complex128])
155