Your IP : 216.73.216.196
���]�R � @ sx d d l m Z d d l Z d d l m Z m Z d d l m Z d d l m
Z
m Z m Z m
Z
d d l m Z d d d � Z Gd
d � d e j � Z Gd d
� d
e j � Z Gd d � d e j � Z Gd d � d e � Z Gd d � d e � Z Gd d � d e � Z Gd d � d e j � Z Gd d � d e j � Z Gd d � d e j � Z Gd d � d e j � Z d S) � )�support� N)�Node�Leaf)�
fixer_util)�Attr�Name�Call�Comma)�tokenc C s@ t j | � } x! t | � D] } | j d } q Wd | _ | S)Nr )r Z parse_string�range�children�parent)�code� strip_levelsZtree�i� r �./opt/alt/python35/lib64/python3.5/test_util.py�parse s
r c @ s e Z d Z d d � Z d S)�
MacroTestCasec C sD t | t t f � r* t t j j | � } | j t | � | � d S)N) �
isinstance�tuple�listr r ZsymsZ simple_stmt� assertEqual�str)�self�node�stringr r r � assertStr s zMacroTestCase.assertStrN)�__name__�
__module__� __qualname__r r r r r r s r c @ s4 e Z d Z d d � Z d d � Z d d � Z d S)�
Test_is_tuplec C s t j t | d d �� S)Nr � )r �is_tupler )r r r r r r$ s zTest_is_tuple.is_tuplec C sr | j | j d � � | j | j d � � | j | j d � � | j | j d � � | j | j d � � d S)Nz(a, b)z (a, (b, c))z((a, (b, c)),)z(a,)z())�
assertTruer$ )r r r r �
test_valid# s
zTest_is_tuple.test_validc C s0 | j | j d � � | j | j d � � d S)Nz(a)z('foo') % (b, c))� assertFalser$ )r r r r � test_invalid* s zTest_is_tuple.test_invalidN)r r r! r$ r& r( r r r r r" s r" c @ s4 e Z d Z d d � Z d d � Z d d � Z d S)� Test_is_listc C s t j t | d d �� S)Nr r# )r �is_listr )r r r r r r* 0 s zTest_is_list.is_listc C sr | j | j d � � | j | j d � � | j | j d � � | j | j d � � | j | j d � � d S)Nz[]z[a]z[a, b]z [a, [b, c]]z[[a, [b, c]],])r% r* )r r r r r& 3 s
zTest_is_list.test_validc C s | j | j d � � d S)Nz[]+[])r' r* )r r r r r( : s zTest_is_list.test_invalidN)r r r! r* r& r( r r r r r) / s r) c @ s( e Z d Z d d � Z d d � Z d S)� Test_Attrc C sZ t d d d �} | j t t d � t d � � d � | j t | t d � � d � d S)Nzfoo()r r# �a�bza.bzfoo().b)r r r r )r �callr r r �test? s %zTest_Attr.testc C s5 t t d � t d � � } | j t | � t � d S)Nr, r- )r r r �typer )r �attrr r r � test_returnsE s zTest_Attr.test_returnsN)r r r! r/ r2 r r r r r+ > s r+ c @ s e Z d Z d d � Z d S)� Test_Namec C sL | j t d � d � | j t d � d � | j t d d d �d � d S)Nr, z
foo.foo().bar�prefixr- Zba)r r )r r r r r/ K s zTest_Name.testN)r r r! r/ r r r r r3 J s r3 c @ s. e Z d Z d d d d � Z d d � Z d S)� Test_CallNc C sc g } t | t � rM x+ | D]# } | j | � | j t � � q W| j � t t | � | | � S)N)r r �appendr
�popr r )r �name�argsr4 r
�argr r r �_CallR s
zTest_Call._Callc C s" d t t j d � t t j d � t t j d � g t t j d � t t j d � t t j d � t t j d � g t t j d � t t j d d d �g g } | j | j d � d
� | j | j d | d � d � | j | j d | d � d
� | j | j d | d d d �d � d S)Nr r# � � r- �jr4 � �AzA()zb(1,2,3)za.b().cza.b().c(1,3,2,4)�dz d(b, j))r r �NUMBER�STRINGr r; )r Zkidsr r r r/ \ s !- zTest_Call.test)r r r! r; r/ r r r r r5 Q s
r5 c @ s@ e Z d Z d d � Z d d � Z d d � Z d d � Z d S)
�Test_does_tree_importc C sP t j | | � } | r | Sx- | j D]" } | j | | � } | r& | Sq& Wd S)N)r � find_bindingr
�_find_bind_rec)r r8 r �cZchildr r r rF k s z$Test_does_tree_import._find_bind_recc C s1 t | � } | j d | � } t j | | | � S)N�start)r rF r �does_tree_import)r �packager8 r r r r r rI t s z&Test_does_tree_import.does_tree_importc C s� d } xk | D]c \ } } } | j | | | d | � } | j | � | j | | | d | � } | j | � q
Wd } xk | D]c \ } } } | j | | | d | � } | j | � | j | | | d | � } | j | � q� Wd S)!Nr, �from a import b�a.d�from a.d import b�d.a�from d.a import b�import b�import b, c, d�
�from a import a�x�from x import a�from x import b, c, a, d�x.b�from x.b import a�from x.b import b, c, a, d�import a�import b, c, a, d�r, r, rK �rL r, rM �rN r, rO �Nr, rP �Nr, rQ )r\ r] r^ r_ r` �r, r, rS �rT r, rU �rT r, rV �rW r, rX �rW r, rY �Nr, rZ �Nr, r[ )ra rb rc rd re rf rg )rI r' r% )r r Z
failing_testsrJ r8 �import_�nZ
passing_testsr r r �try_withz s,
z Test_does_tree_import.try_withc C s | j d � d S)Nz def foo():
bar.baz()
start=3)rj )r r r r �test_in_function� s z&Test_does_tree_import.test_in_functionN)r r r! rF rI rj rk r r r r rD j s rD c @ s? e Z d Z d d d � Z d d � Z d d � Z d d � Z d
d � Z d d
� Z d d � Z d d � Z
d d � Z d d � Z d d � Z
d d � Z d d � Z d d � Z d d � Z d d! � Z d" d# � Z d$ d% � Z d& d' � Z d( d) � Z d* d+ � Z d, d- � Z d. d/ � Z d0 d1 � Z d2 d3 � Z d S)4�Test_find_bindingNc C s t j | t | � | � S)N)r rE r )r r8 r rJ r r r rE � s z Test_find_binding.find_bindingc C s� | j | j d d � � | j | j d d � � | j | j d d � � | j | j d d � � | j | j d d � � | j | j d d � � d S)Nr, za = bz
a = [b, c, d]z a = foo()za = foo().foo.foo[6][foo]zfoo = azfoo = (a, b, c))r% rE r' )r r r r �test_simple_assignment� s z(Test_find_binding.test_simple_assignmentc C s� | j | j d d � � | j | j d d � � | j | j d d � � | j | j d d � � | j | j d d � � | j | j d d � � d S)Nr, z(a,) = bz(a, b, c) = [b, c, d]z(c, (d, a), b) = foo()z(a, b) = foo().foo[6][foo]z(foo, b) = (b, a)z(foo, (b, c)) = (a, b, c))r% rE r' )r r r r �test_tuple_assignment� s z'Test_find_binding.test_tuple_assignmentc C s� | j | j d d � � | j | j d d � � | j | j d d � � | j | j d d � � | j | j d d � � | j | j d d � � d S)Nr, z[a] = bz[a, b, c] = [b, c, d]z[c, [d, a], b] = foo()z[a, b] = foo().foo[a][foo]z[foo, b] = (b, a)z[foo, [b, c]] = (a, b, c))r% rE r' )r r r r �test_list_assignment� s z&Test_find_binding.test_list_assignmentc C sh | j | j d d � � | j | j d d � � | j | j d d � � | j | j d d � � d S)Nr, z foo.a = 5z
foo[a] = 5z
foo(a) = 5z
foo(a, b) = 5)r' rE )r r r r �test_invalid_assignments� s z*Test_find_binding.test_invalid_assignmentsc C sh | j | j d d � � | j | j d d � � | j | j d d � � | j | j d d � � d S)Nr, zimport azimport b, c, a, dzimport bzimport b, c, d)r% rE r' )r r r r �test_simple_import� s z$Test_find_binding.test_simple_importc C s� | j | j d d � � | j | j d d � � | j | j d d � � | j | j d d � � | j | j d d � � | j | j d d � � | j | j d d � � | j | j d d � � d S)
Nr, zfrom x import azfrom a import azfrom x import b, c, a, dzfrom x.b import azfrom x.b import b, c, a, dzfrom a import bzfrom a.d import bzfrom d.a import b)r% rE r' )r r r r �test_from_import� s z"Test_find_binding.test_from_importc C sh | j | j d d � � | j | j d d � � | j | j d d � � | j | j d d � � d S)Nr, z
import b as azimport b as a, c, a as f, dz
import a as fzimport b, c as f, d as e)r% rE r' )r r r r �test_import_as� s z Test_find_binding.test_import_asc C s� | j | j d d � � | j | j d d � � | j | j d d � � | j | j d d � � | j | j d d � � | j | j d d � � | j | j d d � � d S) Nr, zfrom x import b as az from x import g as a, d as bzfrom x.b import t as azfrom x.b import g as a, dzfrom a import b as tzfrom a.d import b as tzfrom d.a import b as t)r% rE r' )r r r r �test_from_import_as� s z%Test_find_binding.test_from_import_asc C sn | j | j d d � � | j | j d d � � | j | j d d d � � | j | j d d d � � d S)Nr- zimport bzimport b, c, drG )r% rE r' )r r r r � test_simple_import_with_package� s z1Test_find_binding.test_simple_import_with_packagec C s8 | j | j d d d � � | j | j d d d � � | j | j d d d � � | j | j d d d � � | j | j d d d � � | j | j d d d � � | j | j d d
d � � | j | j d d d � � | j | j d d d
� � | j | j d d d
� � | j | j d d d � � d S)Nr, zfrom x import arT zfrom a import azfrom x import *zfrom x import b, c, a, dzfrom x.b import azx.bzfrom x.b import *zfrom x.b import b, c, a, dzfrom a import bzfrom a.d import bza.dzfrom d.a import bzfrom x.y import *za.b)r% rE r' )r r r r � test_from_import_with_package� s z/Test_find_binding.test_from_import_with_packagec C sX | j | j d d d � � | j | j d d d � � | j | j d d d � � d S)Nr, zimport b.c as azb.cz
import a as f�f)r' rE )r r r r �test_import_as_with_package� s z-Test_find_binding.test_import_as_with_packagec C s� | j | j d d d � � | j | j d d d � � | j | j d d d � � | j | j d d d � � | j | j d d d � � | j | j d d d � � | j | j d d d
� � d S) Nr, zfrom x import b as arT z from x import g as a, d as bzfrom x.b import t as azx.bzfrom x.b import g as a, dzfrom a import b as tr- �t)r' rE )r r r r � test_from_import_as_with_package� s z2Test_find_binding.test_from_import_as_with_packagec C s� | j | j d d � � | j | j d d � � | j | j d d � � | j | j d d � � | j | j d d � � | j | j d d � � | j | j d d � � d } | j | j d | � � d S)
Nr, z
def a(): passzdef a(b, c, d): passzdef a(): b = 7zdef d(b, (c, a), e): passzdef d(a=7): passzdef d(a): passzdef d(): a = 7zG
def d():
def a():
pass)r% rE r' )r �sr r r �test_function_def� s z#Test_find_binding.test_function_defc C s | j | j d d � � | j | j d d � � | j | j d d � � | j | j d d � � | j | j d d � � | j | j d d � � | j | j d d � � | j | j d d � � | j | j d d
� � | j | j d d � � d } | j | j d | � � d S)
Nr, z
class a: passzclass a(): passzclass a(b): passzclass a(b, c=8): passz
class d: passzclass d(a): passzclass d(b, a=7): passzclass d(b, *a): passzclass d(b, **a): passzclass d: a = 7zK
class d():
class a():
pass)r% rE r' )r r{ r r r �test_class_def
s z Test_find_binding.test_class_defc C s� | j | j d d � � | j | j d d � � | j | j d d � � | j | j d d � � | j | j d d � � | j | j d d � � | j | j d d � � d S) Nr, zfor a in r: passzfor a, b in r: passzfor (a, b) in r: passzfor c, (a,) in r: passzfor c, (a, b) in r: passzfor c in r: a = czfor c in a: pass)r% rE r' )r r r r �test_for s zTest_find_binding.test_forc C s� d } | j | j d | � � d } | j | j d | � � d } | j | j d | � � d } | j | j d | � � d } | j | j d | � � d } | j | j d | � � d } | j | j d | � � d } | j | j d | � � d S)
NzM
for b in r:
for a in b:
passr, zP
for b in r:
for a, c in b:
passzR
for b in r:
for (a, c) in b:
passzP
for b in r:
for (a,) in b:
passzU
for b in r:
for c, (a, d) in b:
passzN
for b in r:
for c in b:
a = 7zN
for b in r:
for c in b:
d = azN
for b in r:
for c in a:
d = 7)r% rE r' )r r{ r r r �test_for_nested% s z!Test_find_binding.test_for_nestedc C s6 | j | j d d � � | j | j d d � � d S)Nr, zif b in r: a = czif a in r: d = e)r% rE r' )r r r r �test_ifV s zTest_find_binding.test_ifc C sB d } | j | j d | � � d } | j | j d | � � d S)NzL
if b in r:
if c in d:
a = cr, zL
if b in r:
if c in d:
c = a)r% rE r' )r r{ r r r �test_if_nestedZ s z Test_find_binding.test_if_nestedc C s6 | j | j d d � � | j | j d d � � d S)Nr, zwhile b in r: a = czwhile a in r: d = e)r% rE r' )r r r r �
test_whileg s z Test_find_binding.test_whilec C sB d } | j | j d | � � d } | j | j d | � � d S)NzR
while b in r:
while c in d:
a = cr, zR
while b in r:
while c in d:
c = a)r% rE r' )r r{ r r r �test_while_nestedk s z#Test_find_binding.test_while_nestedc C s� d } | j | j d | � � d } | j | j d | � � d } | j | j d | � � d } | j | j d | � � d S)NzQ
try:
a = 6
except:
b = 8r, zQ
try:
b = 8
except:
a = 6z�
try:
b = 8
except KeyError:
pass
except:
a = 6zQ
try:
b = 8
except:
b = 6)r% rE r' )r r{ r r r �test_try_exceptx s z!Test_find_binding.test_try_exceptc C s� d } | j | j d | � � d } | j | j d | � � d } | j | j d | � � d } | j | j d | � � d } | j | j d | � � d } | j | j d | � � d } | j | j d | � � d S) Nz�
try:
try:
a = 6
except:
pass
except:
b = 8r, z�
try:
b = 8
except:
try:
a = 6
except:
passz�
try:
b = 8
except:
try:
pass
except:
a = 6z�
try:
try:
b = 8
except KeyError:
pass
except:
a = 6
except:
passz�
try:
pass
except:
try:
b = 8
except KeyError:
pass
except:
a = 6zQ
try:
b = 8
except:
b = 6a
try:
try:
b = 8
except:
c = d
except:
try:
b = 6
except:
t = 8
except:
o = y)r% rE r' )r r{ r r r �test_try_except_nested� s z(Test_find_binding.test_try_except_nestedc C s� d } | j | j d | � � d } | j | j d | � � d } | j | j d | � � d } | j | j d | � � d S)Nz|
try:
c = 6
except:
b = 8
finally:
a = 9r, zR
try:
b = 8
finally:
a = 6zR
try:
b = 8
finally:
b = 6z|
try:
b = 8
except:
b = 9
finally:
b = 6)r% rE r' )r r{ r r r �test_try_except_finally� s z)Test_find_binding.test_try_except_finallyc C sa d } | j | j d | � � d } | j | j d | � � d } | j | j d | � � d S)Nz�
try:
c = 6
except:
b = 8
finally:
try:
a = 9
except:
b = 9
finally:
c = 9r, z�
try:
b = 8
finally:
try:
pass
finally:
a = 6z�
try:
b = 8
finally:
try:
b = 6
finally:
b = 7)r% rE r' )r r{ r r r � test_try_except_finally_nested s z0Test_find_binding.test_try_except_finally_nested) r r r! rE rm rn ro rp rq rr rs rt ru rv rx rz r| r} r~ r r� r� r� r� r� r� r� r� r r r r rl � s2
1
M !rl c @ sL e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d S) �Test_touch_importc C s9 t d � } t j d d | � | j t | � d � d S)Nz"""foo"""
bar()�fooz """foo"""
import foo
bar()
)r r � touch_importr r )r r r r r �test_after_docstring* s z&Test_touch_import.test_after_docstringc C s9 t d � } t j d d | � | j t | � d � d S)Nz"""foo"""
import bar
bar()r� z'"""foo"""
import bar
import foo
bar()
)r r r� r r )r r r r r �test_after_imports/ s z$Test_touch_import.test_after_importsc C s9 t d � } t j d d | � | j t | � d � d S)Nzbar()r� zimport foo
bar()
)r r r� r r )r r r r r �test_beginning4 s z Test_touch_import.test_beginningc C s9 t d � } t j d d | � | j t | � d � d S)Nzbar()Zhtml�escapez from html import escape
bar()
)r r r� r r )r r r r r rr 9 s z"Test_touch_import.test_from_importc C s9 t d � } t j d d | � | j t | � d � d S)Nzbar()Zcgizimport cgi
bar()
)r r r� r r )r r r r r �test_name_import> s z"Test_touch_import.test_name_importN)r r r! r� r� r� rr r� r r r r r� ( s
r� c @ s( e Z d Z d d � Z d d � Z d S)�Test_find_indentationc C sQ t j } t d � } | j | | � d � t d � } | j | | � d � d S)Nznode()� )r �find_indentationr r )r �fir r r r � test_nothingE s
z"Test_find_indentation.test_nothingc C s� t j } t d � } | j | | � d � | j | | j d j d j d � d � t d � } | j | | j d j d j d � d � d S)Nzdef f():
x()r� r r= r# z zdef f():
x()
y())r r� r r r
)r r� r r r r � test_simpleL s + z!Test_find_indentation.test_simpleN)r r r! r� r� r r r r r� C s r� ) r� r Zos.path�osZlib2to3.pytreer r Zlib2to3r Zlib2to3.fixer_utilr r r r
Z
lib2to3.pgen2r r ZTestCaser r" r) r+ r3 r5 rD rl r� r� r r r r �<module> s" " ,� �