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
| import numpy as np
|
| # Note: we use dtype objects instead of dtype classes. The spec does not
| # require any behavior on dtypes other than equality.
| int8 = np.dtype("int8")
| int16 = np.dtype("int16")
| int32 = np.dtype("int32")
| int64 = np.dtype("int64")
| uint8 = np.dtype("uint8")
| uint16 = np.dtype("uint16")
| uint32 = np.dtype("uint32")
| uint64 = np.dtype("uint64")
| float32 = np.dtype("float32")
| float64 = np.dtype("float64")
| # Note: This name is changed
| bool = np.dtype("bool")
|
| _all_dtypes = (
| int8,
| int16,
| int32,
| int64,
| uint8,
| uint16,
| uint32,
| uint64,
| float32,
| float64,
| bool,
| )
| _boolean_dtypes = (bool,)
| _floating_dtypes = (float32, float64)
| _integer_dtypes = (int8, int16, int32, int64, uint8, uint16, uint32, uint64)
| _integer_or_boolean_dtypes = (
| bool,
| int8,
| int16,
| int32,
| int64,
| uint8,
| uint16,
| uint32,
| uint64,
| )
| _numeric_dtypes = (
| float32,
| float64,
| int8,
| int16,
| int32,
| int64,
| uint8,
| uint16,
| uint32,
| uint64,
| )
|
| _dtype_categories = {
| "all": _all_dtypes,
| "numeric": _numeric_dtypes,
| "integer": _integer_dtypes,
| "integer or boolean": _integer_or_boolean_dtypes,
| "boolean": _boolean_dtypes,
| "floating-point": _floating_dtypes,
| }
|
|
| # Note: the spec defines a restricted type promotion table compared to NumPy.
| # In particular, cross-kind promotions like integer + float or boolean +
| # integer are not allowed, even for functions that accept both kinds.
| # Additionally, NumPy promotes signed integer + uint64 to float64, but this
| # promotion is not allowed here. To be clear, Python scalar int objects are
| # allowed to promote to floating-point dtypes, but only in array operators
| # (see Array._promote_scalar) method in _array_object.py.
| _promotion_table = {
| (int8, int8): int8,
| (int8, int16): int16,
| (int8, int32): int32,
| (int8, int64): int64,
| (int16, int8): int16,
| (int16, int16): int16,
| (int16, int32): int32,
| (int16, int64): int64,
| (int32, int8): int32,
| (int32, int16): int32,
| (int32, int32): int32,
| (int32, int64): int64,
| (int64, int8): int64,
| (int64, int16): int64,
| (int64, int32): int64,
| (int64, int64): int64,
| (uint8, uint8): uint8,
| (uint8, uint16): uint16,
| (uint8, uint32): uint32,
| (uint8, uint64): uint64,
| (uint16, uint8): uint16,
| (uint16, uint16): uint16,
| (uint16, uint32): uint32,
| (uint16, uint64): uint64,
| (uint32, uint8): uint32,
| (uint32, uint16): uint32,
| (uint32, uint32): uint32,
| (uint32, uint64): uint64,
| (uint64, uint8): uint64,
| (uint64, uint16): uint64,
| (uint64, uint32): uint64,
| (uint64, uint64): uint64,
| (int8, uint8): int16,
| (int8, uint16): int32,
| (int8, uint32): int64,
| (int16, uint8): int16,
| (int16, uint16): int32,
| (int16, uint32): int64,
| (int32, uint8): int32,
| (int32, uint16): int32,
| (int32, uint32): int64,
| (int64, uint8): int64,
| (int64, uint16): int64,
| (int64, uint32): int64,
| (uint8, int8): int16,
| (uint16, int8): int32,
| (uint32, int8): int64,
| (uint8, int16): int16,
| (uint16, int16): int32,
| (uint32, int16): int64,
| (uint8, int32): int32,
| (uint16, int32): int32,
| (uint32, int32): int64,
| (uint8, int64): int64,
| (uint16, int64): int64,
| (uint32, int64): int64,
| (float32, float32): float32,
| (float32, float64): float64,
| (float64, float32): float64,
| (float64, float64): float64,
| (bool, bool): bool,
| }
|
|
| def _result_type(type1, type2):
| if (type1, type2) in _promotion_table:
| return _promotion_table[type1, type2]
| raise TypeError(f"{type1} and {type2} cannot be type promoted together")
|
|
