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
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
|
from __future__ import annotations
import sys
from typing import Generic, List, Optional, Set, TypeVar
import pytest
from utils import temp_module, package_not_installed
from msgspec._utils import get_class_annotations
PY310 = sys.version_info[:2] >= (3, 10)
T = TypeVar("T")
S = TypeVar("S")
U = TypeVar("U")
class Base(Generic[T]):
x: T
class Base2(Generic[T, S]):
a: T
b: S
class TestGetClassAnnotations:
@pytest.mark.parametrize("future_annotations", [False, True])
def test_eval_scopes(self, future_annotations):
header = "from __future__ import annotations" if future_annotations else ""
source = f"""
{header}
STR = str
class Ex:
LOCAL = float
x: int
y: LOCAL
z: STR
"""
with temp_module(source) as mod:
assert get_class_annotations(mod.Ex) == {"x": int, "y": float, "z": str}
def test_none_to_nonetype(self):
class Ex:
x: None
assert get_class_annotations(Ex) == {"x": type(None)}
def test_subclass(self):
class Base:
x: int
y: str
class Sub(Base):
x: float
z: list
class Base2:
a: int
class Sub2(Sub, Base2):
b: float
y: list
assert get_class_annotations(Base) == {"x": int, "y": str}
assert get_class_annotations(Sub) == {"x": float, "y": str, "z": list}
assert get_class_annotations(Sub2) == {
"x": float,
"y": list,
"z": list,
"a": int,
"b": float,
}
def test_simple_generic(self):
class Test(Generic[T]):
x: T
y: List[T]
z: int
assert get_class_annotations(Test) == {"x": T, "y": List[T], "z": int}
assert get_class_annotations(Test[int]) == {"x": int, "y": List[int], "z": int}
assert get_class_annotations(Test[Set[T]]) == {
"x": Set[T],
"y": List[Set[T]],
"z": int,
}
def test_generic_sub1(self):
class Sub(Base):
y: int
assert get_class_annotations(Sub) == {"x": T, "y": int}
def test_generic_sub2(self):
class Sub(Base, Generic[T]):
y: List[T]
assert get_class_annotations(Sub) == {"x": T, "y": List[T]}
assert get_class_annotations(Sub[int]) == {"x": T, "y": List[int]}
def test_generic_sub3(self):
class Sub(Base[int], Generic[T]):
y: List[T]
assert get_class_annotations(Sub) == {"x": int, "y": List[T]}
assert get_class_annotations(Sub[float]) == {"x": int, "y": List[float]}
def test_generic_sub4(self):
class Sub(Base[T]):
y: List[T]
assert get_class_annotations(Sub) == {"x": T, "y": List[T]}
assert get_class_annotations(Sub[int]) == {"x": int, "y": List[int]}
def test_generic_sub5(self):
class Sub(Base[T], Generic[T]):
y: List[T]
assert get_class_annotations(Sub) == {"x": T, "y": List[T]}
assert get_class_annotations(Sub[int]) == {"x": int, "y": List[int]}
def test_generic_sub6(self):
class Sub(Base[S]):
y: List[S]
assert get_class_annotations(Sub) == {"x": S, "y": List[S]}
assert get_class_annotations(Sub[int]) == {"x": int, "y": List[int]}
def test_generic_sub7(self):
class Sub(Base[List[T]]):
y: Set[T]
assert get_class_annotations(Sub) == {"x": List[T], "y": Set[T]}
assert get_class_annotations(Sub[int]) == {"x": List[int], "y": Set[int]}
def test_generic_sub8(self):
class Sub(Base[int], Base2[float, str]):
pass
assert get_class_annotations(Sub) == {"x": int, "a": float, "b": str}
def test_generic_sub9(self):
class Sub(Base[U], Base2[List[U], U]):
y: str
assert get_class_annotations(Sub) == {"y": str, "x": U, "a": List[U], "b": U}
assert get_class_annotations(Sub[int]) == {
"y": str,
"x": int,
"a": List[int],
"b": int,
}
class Sub2(Sub[int]):
x: list
assert get_class_annotations(Sub2) == {
"x": list,
"y": str,
"a": List[int],
"b": int,
}
def test_generic_sub10(self):
class Sub(Base[U], Base2[List[U], U]):
y: str
class Sub3(Sub[List[T]]):
c: T
assert get_class_annotations(Sub3) == {
"c": T,
"y": str,
"x": List[T],
"a": List[List[T]],
"b": List[T],
}
assert get_class_annotations(Sub3[int]) == {
"c": int,
"y": str,
"x": List[int],
"a": List[List[int]],
"b": List[int],
}
def test_generic_sub11(self):
class Sub(Base[int]):
y: float
class Sub2(Sub, Base[int]):
z: str
assert get_class_annotations(Sub2) == {"x": int, "y": float, "z": str}
def test_generic_invalid_parameters(self):
class Invalid:
@property
def __parameters__(self):
pass
class Sub(Base[Invalid]):
pass
assert get_class_annotations(Sub) == {"x": Invalid}
@pytest.mark.skipif(PY310, reason="<3.10 only")
def test_union_backport_not_installed(self):
class Ex:
x: int | None = None
with package_not_installed("eval_type_backport"):
with pytest.raises(
TypeError, match=r"or install the `eval_type_backport` package."
):
get_class_annotations(Ex)
@pytest.mark.skipif(PY310, reason="<3.10 only")
def test_union_backport_installed(self):
class Ex:
x: int | None = None
pytest.importorskip("eval_type_backport")
assert get_class_annotations(Ex) == {"x": Optional[int]}
|
