1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
|
import sys
from typing import Any
import numpy as np
import numpy.typing as npt
if sys.version_info >= (3, 11):
from typing import assert_type
else:
from typing_extensions import assert_type
mat: np.matrix[Any, np.dtype[np.int64]]
ar_f8: npt.NDArray[np.float64]
assert_type(mat * 5, np.matrix[Any, Any])
assert_type(5 * mat, np.matrix[Any, Any])
mat *= 5
assert_type(mat**5, np.matrix[Any, Any])
mat **= 5
assert_type(mat.sum(), Any)
assert_type(mat.mean(), Any)
assert_type(mat.std(), Any)
assert_type(mat.var(), Any)
assert_type(mat.prod(), Any)
assert_type(mat.any(), np.bool_)
assert_type(mat.all(), np.bool_)
assert_type(mat.max(), np.int64)
assert_type(mat.min(), np.int64)
assert_type(mat.argmax(), np.intp)
assert_type(mat.argmin(), np.intp)
assert_type(mat.ptp(), np.int64)
assert_type(mat.sum(axis=0), np.matrix[Any, Any])
assert_type(mat.mean(axis=0), np.matrix[Any, Any])
assert_type(mat.std(axis=0), np.matrix[Any, Any])
assert_type(mat.var(axis=0), np.matrix[Any, Any])
assert_type(mat.prod(axis=0), np.matrix[Any, Any])
assert_type(mat.any(axis=0), np.matrix[Any, np.dtype[np.bool_]])
assert_type(mat.all(axis=0), np.matrix[Any, np.dtype[np.bool_]])
assert_type(mat.max(axis=0), np.matrix[Any, np.dtype[np.int64]])
assert_type(mat.min(axis=0), np.matrix[Any, np.dtype[np.int64]])
assert_type(mat.argmax(axis=0), np.matrix[Any, np.dtype[np.intp]])
assert_type(mat.argmin(axis=0), np.matrix[Any, np.dtype[np.intp]])
assert_type(mat.ptp(axis=0), np.matrix[Any, np.dtype[np.int64]])
assert_type(mat.sum(out=ar_f8), npt.NDArray[np.float64])
assert_type(mat.mean(out=ar_f8), npt.NDArray[np.float64])
assert_type(mat.std(out=ar_f8), npt.NDArray[np.float64])
assert_type(mat.var(out=ar_f8), npt.NDArray[np.float64])
assert_type(mat.prod(out=ar_f8), npt.NDArray[np.float64])
assert_type(mat.any(out=ar_f8), npt.NDArray[np.float64])
assert_type(mat.all(out=ar_f8), npt.NDArray[np.float64])
assert_type(mat.max(out=ar_f8), npt.NDArray[np.float64])
assert_type(mat.min(out=ar_f8), npt.NDArray[np.float64])
assert_type(mat.argmax(out=ar_f8), npt.NDArray[np.float64])
assert_type(mat.argmin(out=ar_f8), npt.NDArray[np.float64])
assert_type(mat.ptp(out=ar_f8), npt.NDArray[np.float64])
assert_type(mat.T, np.matrix[Any, np.dtype[np.int64]])
assert_type(mat.I, np.matrix[Any, Any])
assert_type(mat.A, npt.NDArray[np.int64])
assert_type(mat.A1, npt.NDArray[np.int64])
assert_type(mat.H, np.matrix[Any, np.dtype[np.int64]])
assert_type(mat.getT(), np.matrix[Any, np.dtype[np.int64]])
assert_type(mat.getI(), np.matrix[Any, Any])
assert_type(mat.getA(), npt.NDArray[np.int64])
assert_type(mat.getA1(), npt.NDArray[np.int64])
assert_type(mat.getH(), np.matrix[Any, np.dtype[np.int64]])
assert_type(np.bmat(ar_f8), np.matrix[Any, Any])
assert_type(np.bmat([[0, 1, 2]]), np.matrix[Any, Any])
assert_type(np.bmat("mat"), np.matrix[Any, Any])
assert_type(np.asmatrix(ar_f8, dtype=np.int64), np.matrix[Any, Any])
|
