ó 4¶†\c@snddlZddlZddlmZddlmZmZdefd„ƒYZdefd„ƒYZdS(iÿÿÿÿN(tchecker(tTestCasetskipIftTypeableVisitorTestscBsYeZdZed„ƒZd„Zd„Zeej d kdƒd„ƒZ d„Z RS( s& Tests of L{_TypeableVisitor} cCs,tj|ƒ}tjƒ}|j|ƒ|S(sV Run L{_TypeableVisitor} on the parsed source and return the visitor. (tasttparseRt_TypeableVisitortvisit(tsttreetvisitor((s9lib/python2.7/site-packages/pyflakes/test/test_checker.pyt _run_visitor s  cCsš|jdƒ}|j|jddddgƒ|j|jdtjƒ|j|jdtjƒ|j|jdtjƒ|j|jdtj ƒdS(sA Test that the typeable node types are collected sx = 1 # assignment for x in range(1): pass # for loop def f(): pass # function definition with a as b: pass # with statement iiiiN( R t assertEqualttypeable_linestassertIsInstancettypeable_nodesRtAssigntFort FunctionDeftWith(tselfR ((s9lib/python2.7/site-packages/pyflakes/test/test_checker.pyttest_node_typess cCs`|jdƒ}|j|jddgƒ|j|jdtjƒ|j|jdtjƒdS(sƒ Test the common pitfall of missing `generic_visit` in visitors by ensuring that nested nodes are reported sdef f(): x = 1 iiN(R R R RRRRR(RR ((s9lib/python2.7/site-packages/pyflakes/test/test_checker.pyttest_visitor_recurses)s  iisasync syntax introduced in py35cCs}|jdƒ}|j|jdddgƒ|j|jdtjƒ|j|jdtjƒ|j|jdtjƒdS(s@ Test that the PEP 492 node types are collected soasync def f(): # async def async for x in y: pass # async for async with a as b: pass # async with iiiN( R R R RRRtAsyncFunctionDeftAsyncFort AsyncWith(RR ((s9lib/python2.7/site-packages/pyflakes/test/test_checker.pyttest_py35_node_types8s  cCsM|jdƒ}|j|jddgƒ|j|jdjdjdƒdS(sn Test that when two typeable nodes are present on a line, the last typeable one wins. s x = 1; y = 1iityN(R R R Rttargetstid(RR ((s9lib/python2.7/site-packages/pyflakes/test/test_checker.pyttest_last_node_winsIs(ii( t__name__t __module__t__doc__t staticmethodR RRRtsyst version_infoRR(((s9lib/python2.7/site-packages/pyflakes/test/test_checker.pyRs   !tCollectTypeCommentsTestscBs€eZdZed„ƒZd„Zd„Zd„Zd„Zd„Z d„Z d„Z d „Z d „Z d „Zd „ZRS( s, Tests of L{_collect_type_comments} cCsDtj|ƒ}tj|ƒ}tj||ƒ}d„|jƒDƒS(sÒ 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), ...} cSs8h|].\}}t|ƒtd„|Dƒƒf’qS(css|]\}}|VqdS(N((t.0t_R((s9lib/python2.7/site-packages/pyflakes/test/test_checker.pys ds(ttypettuple(R&tktv((s9lib/python2.7/site-packages/pyflakes/test/test_checker.pys ds (RRRt make_tokenst_collect_type_commentstitems(RR ttokenstret((s9lib/python2.7/site-packages/pyflakes/test/test_checker.pyt_collectZscCs/|jdƒ}|j|tjdfhƒdS(s@ Test that the function works for binary source sx = 1 # type: ints # type: intN(s # type: int(R1tassertSetEqualRR(RR0((s9lib/python2.7/site-packages/pyflakes/test/test_checker.pyt test_bytesfscCs/|jdƒ}|j|tjdfhƒdS(s> Test that the function works for text source ux = 1 # type: ints # type: intN(s # type: int(R1R RR(RR0((s9lib/python2.7/site-packages/pyflakes/test/test_checker.pyt test_textmscCs&|jdƒ}|j|tƒƒdS(s9 Test that a non-type comment is ignored s x = 1 # noqaN(R1R2tset(RR0((s9lib/python2.7/site-packages/pyflakes/test/test_checker.pyttest_non_type_comment_ignoredtscCs&|jdƒ}|j|tƒƒdS(sP Test that a type comment before something typeable is ignored. s# type: int x = 1N(R1R2R5(RR0((s9lib/python2.7/site-packages/pyflakes/test/test_checker.pyt!test_type_comment_before_typeable{scCs&|jdƒ}|j|tƒƒdS(sH Test that `# type: ignore` comments are not collected. sx = 1 # type: ignoreN(R1R2R5(RR0((s9lib/python2.7/site-packages/pyflakes/test/test_checker.pyt test_type_ignore_comment_ignored‚scCsH|jdƒ}|j|tƒƒ|jdƒ}|j|tƒƒdS(sg Test that `# type: ignore` comments with more content are also not collected. sx = 1 # type: ignore # noqasx = 1 #type:ignore#noqaN(R1R2R5(RR0((s9lib/python2.7/site-packages/pyflakes/test/test_checker.pyt*test_type_ignore_with_other_things_ignored‰scCs/|jdƒ}|j|tjdfhƒdS(Nsx = 1 # type: int # noqas# type: int # noqa(s# type: int # noqa(R1R2RR(RR0((s9lib/python2.7/site-packages/pyflakes/test/test_checker.pyt,test_type_comment_with_extra_still_collected“scCs/|jdƒ}|j|tjdfhƒdS(Nsx = 1 #type:ints #type:int(s #type:int(R1R2RR(RR0((s9lib/python2.7/site-packages/pyflakes/test/test_checker.pyt$test_type_comment_without_whitespace—scCs/|jdƒ}|j|tjdfhƒdS(Nsx = 1 # type: ignore[T]s# type: ignore[T](s# type: ignore[T](R1R2RR(RR0((s9lib/python2.7/site-packages/pyflakes/test/test_checker.pyt)test_type_comment_starts_with_word_ignore›scCs/|jdƒ}|j|tjdfhƒdS(sn Test that when two typeable nodes are present on a line, the last typeable one wins. sdef f(): x = 1 # type: ints # type: intN(s # type: int(R1R2RR(RR0((s9lib/python2.7/site-packages/pyflakes/test/test_checker.pyRŸscCs5|jdƒ}tjdfh}|j||ƒdS(sw Test that type comments for function arguments are all attributed to the function definition. s]def f( a, # type: int b, # type: str ): # type: (...) -> None pass s # type: ints # type: strs# type: (...) -> NoneN(s # type: ints # type: strs# type: (...) -> None(R1RRR2(RR0texpected((s9lib/python2.7/site-packages/pyflakes/test/test_checker.pyt#test_function_def_assigned_comments§s   (RR R!R"R1R3R4R6R7R8R9R:R;R<RR>(((s9lib/python2.7/site-packages/pyflakes/test/test_checker.pyR%Us         ( RR#tpyflakesRtpyflakes.test.harnessRRRR%(((s9lib/python2.7/site-packages/pyflakes/test/test_checker.pyts  M