@@ -31,9 +31,13 @@ class D: #@
31
31
class E: #@
32
32
def f():
33
33
...
34
+
35
+ @decorator
36
+ class F: #@
37
+ ...
34
38
"""
35
39
).strip ()
36
- ast_nodes : List [nodes .NodeNG ] = builder .extract_node (code ) # type: ignore
40
+ ast_nodes : List [nodes .NodeNG ] = builder .extract_node (code ) # type: ignore[assignment]
37
41
38
42
a = ast_nodes [0 ]
39
43
assert isinstance (a , nodes .ClassDef )
@@ -55,6 +59,10 @@ def f():
55
59
assert isinstance (e , nodes .ClassDef )
56
60
assert e .position == (13 , 0 , 13 , 7 )
57
61
62
+ f = ast_nodes [5 ]
63
+ assert isinstance (f , nodes .ClassDef )
64
+ assert f .position == (18 , 0 , 18 , 7 )
65
+
58
66
@staticmethod
59
67
def test_position_function () -> None :
60
68
"""Position should only include keyword and name.
@@ -81,7 +89,7 @@ def e(): #@
81
89
...
82
90
"""
83
91
).strip ()
84
- ast_nodes : List [nodes .NodeNG ] = builder .extract_node (code ) # type: ignore
92
+ ast_nodes : List [nodes .NodeNG ] = builder .extract_node (code ) # type: ignore[assignment]
85
93
86
94
a = ast_nodes [0 ]
87
95
assert isinstance (a , nodes .FunctionDef )
@@ -129,7 +137,7 @@ async def e(): #@
129
137
...
130
138
"""
131
139
).strip ()
132
- ast_nodes : List [nodes .NodeNG ] = builder .extract_node (code ) # type: ignore
140
+ ast_nodes : List [nodes .NodeNG ] = builder .extract_node (code ) # type: ignore[assignment]
133
141
134
142
a = ast_nodes [0 ]
135
143
assert isinstance (a , nodes .FunctionDef )
0 commit comments