summaryrefslogtreecommitdiff
path: root/numpy/typing/tests/data/reveal/mod.pyi
blob: b2790b7f3973f1412681c35ba32534b3a7027d46 (plain)
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
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
from typing import Any
import numpy as np

f8 = np.float64()
i8 = np.int64()
u8 = np.uint64()

f4 = np.float32()
i4 = np.int32()
u4 = np.uint32()

td = np.timedelta64(0, "D")
b_ = np.bool_()

b = bool()
f = float()
i = int()

AR_b: np.ndarray[Any, np.dtype[np.bool_]]
AR_m: np.ndarray[Any, np.dtype[np.timedelta64]]

# Time structures

reveal_type(td % td)  # E: timedelta64
reveal_type(AR_m % td)  # E: Any
reveal_type(td % AR_m)  # E: Any

reveal_type(divmod(td, td))  # E: Tuple[{int64}, timedelta64]
reveal_type(divmod(AR_m, td))  # E: Tuple[ndarray[Any, dtype[signedinteger[typing._64Bit]]], ndarray[Any, dtype[timedelta64]]]
reveal_type(divmod(td, AR_m))  # E: Tuple[ndarray[Any, dtype[signedinteger[typing._64Bit]]], ndarray[Any, dtype[timedelta64]]]

# Bool

reveal_type(b_ % b)  # E: {int8}
reveal_type(b_ % i)  # E: {int_}
reveal_type(b_ % f)  # E: {float64}
reveal_type(b_ % b_)  # E: {int8}
reveal_type(b_ % i8)  # E: {int64}
reveal_type(b_ % u8)  # E: {uint64}
reveal_type(b_ % f8)  # E: {float64}
reveal_type(b_ % AR_b)  # E: ndarray[Any, dtype[{int8}]]

reveal_type(divmod(b_, b))  # E: Tuple[{int8}, {int8}]
reveal_type(divmod(b_, i))  # E: Tuple[{int_}, {int_}]
reveal_type(divmod(b_, f))  # E: Tuple[{float64}, {float64}]
reveal_type(divmod(b_, b_))  # E: Tuple[{int8}, {int8}]
reveal_type(divmod(b_, i8))  # E: Tuple[{int64}, {int64}]
reveal_type(divmod(b_, u8))  # E: Tuple[{uint64}, {uint64}]
reveal_type(divmod(b_, f8))  # E: Tuple[{float64}, {float64}]
reveal_type(divmod(b_, AR_b))  # E: ndarray[Any, dtype[{int8}]], ndarray[Any, dtype[{int8}]]]

reveal_type(b % b_)  # E: {int8}
reveal_type(i % b_)  # E: {int_}
reveal_type(f % b_)  # E: {float64}
reveal_type(b_ % b_)  # E: {int8}
reveal_type(i8 % b_)  # E: {int64}
reveal_type(u8 % b_)  # E: {uint64}
reveal_type(f8 % b_)  # E: {float64}
reveal_type(AR_b % b_)  # E: ndarray[Any, dtype[{int8}]]

reveal_type(divmod(b, b_))  # E: Tuple[{int8}, {int8}]
reveal_type(divmod(i, b_))  # E: Tuple[{int_}, {int_}]
reveal_type(divmod(f, b_))  # E: Tuple[{float64}, {float64}]
reveal_type(divmod(b_, b_))  # E: Tuple[{int8}, {int8}]
reveal_type(divmod(i8, b_))  # E: Tuple[{int64}, {int64}]
reveal_type(divmod(u8, b_))  # E: Tuple[{uint64}, {uint64}]
reveal_type(divmod(f8, b_))  # E: Tuple[{float64}, {float64}]
reveal_type(divmod(AR_b, b_))  # E: ndarray[Any, dtype[{int8}]], ndarray[Any, dtype[{int8}]]]

# int

reveal_type(i8 % b)  # E: {int64}
reveal_type(i8 % i)  # E: {int64}
reveal_type(i8 % f)  # E: {float64}
reveal_type(i8 % i8)  # E: {int64}
reveal_type(i8 % f8)  # E: {float64}
reveal_type(i4 % i8)  # E: {int64}
reveal_type(i4 % f8)  # E: {float64}
reveal_type(i4 % i4)  # E: {int32}
reveal_type(i4 % f4)  # E: {float32}
reveal_type(i8 % AR_b)  # E: ndarray[Any, dtype[signedinteger[Any]]]

reveal_type(divmod(i8, b))  # E: Tuple[{int64}, {int64}]
reveal_type(divmod(i8, i))  # E: Tuple[{int64}, {int64}]
reveal_type(divmod(i8, f))  # E: Tuple[{float64}, {float64}]
reveal_type(divmod(i8, i8))  # E: Tuple[{int64}, {int64}]
reveal_type(divmod(i8, f8))  # E: Tuple[{float64}, {float64}]
reveal_type(divmod(i8, i4))  # E: Tuple[{int64}, {int64}]
reveal_type(divmod(i8, f4))  # E: Tuple[{float64}, {float64}]
reveal_type(divmod(i4, i4))  # E: Tuple[{int32}, {int32}]
reveal_type(divmod(i4, f4))  # E: Tuple[{float32}, {float32}]
reveal_type(divmod(i8, AR_b))  # E: Tuple[ndarray[Any, dtype[signedinteger[Any]]], ndarray[Any, dtype[signedinteger[Any]]]]

reveal_type(b % i8)  # E: {int64}
reveal_type(i % i8)  # E: {int64}
reveal_type(f % i8)  # E: {float64}
reveal_type(i8 % i8)  # E: {int64}
reveal_type(f8 % i8)  # E: {float64}
reveal_type(i8 % i4)  # E: {int64}
reveal_type(f8 % i4)  # E: {float64}
reveal_type(i4 % i4)  # E: {int32}
reveal_type(f4 % i4)  # E: {float32}
reveal_type(AR_b % i8)  # E: ndarray[Any, dtype[signedinteger[Any]]]

reveal_type(divmod(b, i8))  # E: Tuple[{int64}, {int64}]
reveal_type(divmod(i, i8))  # E: Tuple[{int64}, {int64}]
reveal_type(divmod(f, i8))  # E: Tuple[{float64}, {float64}]
reveal_type(divmod(i8, i8))  # E: Tuple[{int64}, {int64}]
reveal_type(divmod(f8, i8))  # E: Tuple[{float64}, {float64}]
reveal_type(divmod(i4, i8))  # E: Tuple[{int64}, {int64}]
reveal_type(divmod(f4, i8))  # E: Tuple[{float64}, {float64}]
reveal_type(divmod(i4, i4))  # E: Tuple[{int32}, {int32}]
reveal_type(divmod(f4, i4))  # E: Tuple[{float32}, {float32}]
reveal_type(divmod(AR_b, i8))  # E: Tuple[ndarray[Any, dtype[signedinteger[Any]]], ndarray[Any, dtype[signedinteger[Any]]]]

# float

reveal_type(f8 % b)  # E: {float64}
reveal_type(f8 % i)  # E: {float64}
reveal_type(f8 % f)  # E: {float64}
reveal_type(i8 % f4)  # E: {float64}
reveal_type(f4 % f4)  # E: {float32}
reveal_type(f8 % AR_b)  # E: ndarray[Any, dtype[floating[Any]]]

reveal_type(divmod(f8, b))  # E: Tuple[{float64}, {float64}]
reveal_type(divmod(f8, i))  # E: Tuple[{float64}, {float64}]
reveal_type(divmod(f8, f))  # E: Tuple[{float64}, {float64}]
reveal_type(divmod(f8, f8))  # E: Tuple[{float64}, {float64}]
reveal_type(divmod(f8, f4))  # E: Tuple[{float64}, {float64}]
reveal_type(divmod(f4, f4))  # E: Tuple[{float32}, {float32}]
reveal_type(divmod(f8, AR_b))  # E: Tuple[ndarray[Any, dtype[floating[Any]]], ndarray[Any, dtype[floating[Any]]]]

reveal_type(b % f8)  # E: {float64}
reveal_type(i % f8)  # E: {float64}
reveal_type(f % f8)  # E: {float64}
reveal_type(f8 % f8)  # E: {float64}
reveal_type(f8 % f8)  # E: {float64}
reveal_type(f4 % f4)  # E: {float32}
reveal_type(AR_b % f8)  # E: ndarray[Any, dtype[floating[Any]]]

reveal_type(divmod(b, f8))  # E: Tuple[{float64}, {float64}]
reveal_type(divmod(i, f8))  # E: Tuple[{float64}, {float64}]
reveal_type(divmod(f, f8))  # E: Tuple[{float64}, {float64}]
reveal_type(divmod(f8, f8))  # E: Tuple[{float64}, {float64}]
reveal_type(divmod(f4, f8))  # E: Tuple[{float64}, {float64}]
reveal_type(divmod(f4, f4))  # E: Tuple[{float32}, {float32}]
reveal_type(divmod(AR_b, f8))  # E: Tuple[ndarray[Any, dtype[floating[Any]]], ndarray[Any, dtype[floating[Any]]]]