x>> g 4Q 7DDD g 4Q 7AAA g 4Q 7 III g 4Q 7BBBBBr c | d }| |j ddg | |j d t
j | |j d t
j dS )z
Test the common pitfall of missing `generic_visit` in visitors by
ensuring that nested nodes are reported
zdef f():
x = 1
r r N)r r r r r r r r r s r test_visitor_recursesz*TypeableVisitorTests.test_visitor_recurses( s
##
/!Q888 g 4Q 7 III g 4Q 7DDDDDr c l | d }| |j g d | |j d t
j | |j d t
j | |j d t
j dS )z@
Test that the PEP 492 node types are collected
zoasync def f(): # async def
async for x in y: pass # async for
async with a as b: pass # async with
)r r r r r r N) r r r r r r AsyncFunctionDefAsyncFor AsyncWithr s r test_py35_node_typesz)TypeableVisitorTests.test_py35_node_types7 s ##
/;;; g 4Q 79MNNN g 4Q 7 FFF g 4Q 7 GGGGGr c | d }| |j ddg | |j d j d j d dS )n
Test that when two typeable nodes are present on a line, the last
typeable one wins.
z x = 1; y = 1r r yN)r r r r targetsidr s r test_last_node_winsz(TypeableVisitorTests.test_last_node_winsG sc
##N33 /!Q888 /2:1=@#FFFFFr N)
__name__
__module__ __qualname____doc__ staticmethodr r! r# r( r. r r r r s \C C C$
E
E
E H H H G G G G Gr r c j e Zd ZdZed Zd Zd Zd Zd Z d Z
d Z d Z d
Z
d Zd Zd
ZdS )CollectTypeCommentsTestsz,
Tests of L{_collect_type_comments}
c t j | }t j | }t j || }d | D S )z
Run L{_collect_type_comments} on the parsed source and return the
mapping from nodes to comments. The return value is converted to
a set: {(node_type, tuple of comments), ...}
c d h | ]-\ }}t | t d |D f.S )c 3 K | ] \ }}|V
d S )Nr4 ).0_r s r z>CollectTypeCommentsTests._collect...b s& 0 0da 0 0 0 0 0 0r )typetuple)r: kvs r z4CollectTypeCommentsTests._collect..b s> JJJdaa% 0 0a 0 0 0001JJJr )r r r make_tokens_collect_type_commentsitems)r r
tokensrets r _collectz!CollectTypeCommentsTests._collectX sO y||$Q'',T6::JJciikkJJJJr c v | d }| |t j dfh dS )z@
Test that the function works for binary source
s x = 1 # type: int # type: intNrG assertSetEqualr r r rF s r
test_bytesz#CollectTypeCommentsTests.test_bytesd s= mm 122 C3:/?"@!ABBBBBr c v | d }| |t j dfh dS )z>
Test that the function works for text source
zx = 1 # type: intrI N)rG r r r rM s r test_textz"CollectTypeCommentsTests.test_textk s= mm 011
,< = >?????r c t | d }| |t dS )z9
Test that a non-type comment is ignored
z
x = 1 # noqaNrG rL setrM s r test_non_type_comment_ignoredz6CollectTypeCommentsTests.test_non_type_comment_ignoredr s5 mmO,, C'''''r c t | d }| |t dS )zP
Test that a type comment before something typeable is ignored.
z# type: int
x = 1NrR rM s r !test_type_comment_before_typeablez:CollectTypeCommentsTests.test_type_comment_before_typeabley s6 mm 011 C'''''r c t | d }| |t dS )zH
Test that `# type: ignore` comments are not collected.
zx = 1 # type: ignoreNrR rM s r test_type_ignore_comment_ignoredz9CollectTypeCommentsTests.test_type_ignore_comment_ignored s6 mm 344 C'''''r c | d }| |t | d }| |t dS )zg
Test that `# type: ignore` comments with more content are also not
collected.
z x = 1 # type: ignore # noqazx = 1 #type:ignore#noqaNrR rM s r *test_type_ignore_with_other_things_ignoredzCCollectTypeCommentsTests.test_type_ignore_with_other_things_ignored sb
mm :;; C'''mm 677 C'''''r c v | d }| |t j dfh d S )Nzx = 1 # type: int # noqa)z# type: int # noqarK rM s r ,test_type_comment_with_extra_still_collectedzECollectTypeCommentsTests.test_type_comment_with_extra_still_collected s; mm 899 C3:/G"H!IJJJJJr c v | d }| |t j dfh d S )Nzx = 1 #type:int)z #type:intrK rM s r $test_type_comment_without_whitespacez=CollectTypeCommentsTests.test_type_comment_without_whitespace s: mm -.. C3:~">!?@@@@@r c t | d }| |t d S )Nzx = 1 # type: ignore[T]rR rM s r )test_type_comment_starts_with_word_ignorezBCollectTypeCommentsTests.test_type_comment_starts_with_word_ignore s4 mm 566 C'''''r c v | d }| |t j dfh dS )r* zdef f(): x = 1 # type: intrI NrK rM s r r. z,CollectTypeCommentsTests.test_last_node_wins s=
mm 9:: C3:/?"@!ABBBBBr c z | d }t j dfh}| || dS )zw
Test that type comments for function arguments are all attributed to
the function definition.
z]def f(
a, # type: int
b, # type: str
):
# type: (...) -> None
pass
)rJ z # type: strz# type: (...) -> NoneN)rG r r rL )r rF expecteds r #test_function_def_assigned_commentszrg s
* * * * * *IG IG IG IG IG8 IG IG IGXe+ e+ e+ e+ e+x e+ e+ e+ e+ e+r