�
P��ac@�s�dZd
dl
mZ
d
dlZd
dlZddlmZ
d
dlmZ
ye	
Wne
k
rr


d�Z	n
Xdejf
d	��YZd
ejf
d��YZ
dS(s
Unit tests for pytree.py.

NOTE: Please *don't* add doc strings to individual test methods!
In verbose mode, printing of the module, class and method name is much
more helpful than printing of (the first line of) the docstring,
especially when debugging a test.
i����(
twith_statementNi
(
tsupport(
tpytreec
C�st|�
}
|
j
�
|
S(
N(tlisttsort(tlstt
l((s5/usr/local/lib/python2.7/lib2to3/tests/test_pytree.pytsorteds



t	TestNodescB�sI
eZ
dZejd$kr'd�Znd�Zd�Zd�Zd�Z	d�Z
d	�Zd
�Zd�Z
d�Zd
�Zd�Zd�Zd�Zd�Zd�Zd�Zd�Zd�Zd�Zd�Zd�Zd�Zd�Zd�Zd�Zd�Zd�Z d�Z!d �Z"d!�Z#d"�Z$d#�Z%RS(%s(Unit tests for nodes (Base, Leaf, Node).iic
C�s�tj
d
d�}
tjdt�
�:}tjdt�
|j|
j�d�
|
j	d�

WdQX|j|
j
d�
|jt|�
d�
x$|D]}|j|j
tk�

q�W|jt|dj�
d	�
|jt|d
j�
d�
dS(Nidtfootrecordtalwaystthiiis3get_prefix() is deprecated; use the prefix propertyi
s3set_prefix() is deprecated; use the prefix property(RtLeaftwarningstcatch_warningstTruetsimplefiltertDeprecationWarningtassertEqualt
get_prefixt
set_prefixtprefixtlent
assertTruetcategorytstrtmessage(tselfRt
wtwarning((s5/usr/local/lib/python2.7/lib2to3/tests/test_pytree.pyttest_deprecated_prefix_methods$s










c


C�sdS(
N((
R((s5/usr/local/lib/python2.7/lib2to3/tests/test_pytree.pyttest_instantiate_base3s
c
C�s<tj
d
d�}
|j|
jd
�
|j|
jd�
dS(NidR	(RRRttypetvalue(Rtl1((s5/usr/local/lib/python2.7/lib2to3/tests/test_pytree.pyt	test_leaf8s


c
C�s,tj
d
d�}
|jt|
�
d�
dS(NidR	sLeaf(100, 'foo')(RRRtrepr(RR$((s5/usr/local/lib/python2.7/lib2to3/tests/test_pytree.pyttest_leaf_repr=s

c
C�s`tj
d
d�}
|jt|
�
d�
tj
d
ddddf�
}|jt|�
d�
dS(	NidR	tcontextt
 i
i
s foo(i
i
(RRRR(RR$tl2((s5/usr/local/lib/python2.7/lib2to3/tests/test_pytree.pyt
test_leaf_strAs



c
C�s5tj
d
d�}
d|
_|jt|
�
d�
dS(Niitfoo_tfoo_5(RRRRR(RR$((s5/usr/local/lib/python2.7/lib2to3/tests/test_pytree.pyttest_leaf_str_numeric_valueGs
	
c
C�s�tj
d
d�}
tj
d
dddd	f�
}|j|
|�
tj
dd�}tj
d
d�}|j|
|�
|j|
|�
dS(
NidR	R(R)i
iietbar(i
i(RRRtassertNotEqual(RR$R*tl3tl4((s5/usr/local/lib/python2.7/lib2to3/tests/test_pytree.pyttest_leaf_equalityNs






c
C�setj
d
d�}
|j|
jd�
|j|
j�

d|
_|j|
jd�
|j|
j�

dS(NidR	Rs  ##

(RRRRtassertFalsetwas_changedR(RR$((s5/usr/local/lib/python2.7/lib2to3/tests/test_pytree.pyttest_leaf_prefixWs



	

c
C�sltj
d
d�}
tj
dd�}tjd|
|g�}|j|jd�
|j|j|
|g�
dS(NidR	i�R/i�(RRtNodeRR"tchildren(RR$R*tn1((s5/usr/local/lib/python2.7/lib2to3/tests/test_pytree.pyt	test_node_s





c
C�sxtj
d
d�}
tj
d
dddd
f�
}tjd|
|g�}|jt|�
d	t|
�
t|�
f�
dS(NidR	R/R(R)i
ii�sNode(1000, [%s, %s])(i
i(RRR7RR&(RR$R*R9((s5/usr/local/lib/python2.7/lib2to3/tests/test_pytree.pyttest_node_reprfs





c
C�sbtj
d
d�}
tj
d
dddd
f�
}tjd|
|g�}|jt|�
d	�
dS(NidR	R/R(R)i
ii�sfoo bar(i
i(RRR7RR(RR$R*R9((s5/usr/local/lib/python2.7/lib2to3/tests/test_pytree.pyt
test_node_strms



c
C�s�tj
d
d�}
|j|
jd�
tjd|
g
�}|j|jd�
d|_|j|jd�
|j|
jd�
dS(NidR	Ri�R)(RRRRR7(RR$R9((s5/usr/local/lib/python2.7/lib2to3/tests/test_pytree.pyttest_node_prefixss




	

c
C�s
tj
d
ddd�
}
tj
d
ddd�
}tjd|
|g�}|j|
j�|j�
|j|j�d�
|j|j�d�
tj
d
ddd	�
}tjd||g�}|j|j�|j�
|j|j�d�
|j|j�d�
dS(
NidR	Rt
aR/t
bi�Rt
c(RRR7Rt
get_suffixR(RR$R*R9R1tn2((s5/usr/local/lib/python2.7/lib2to3/tests/test_pytree.pyttest_get_suffix|s







c
C�sftj
d
d�}
tj
d
gdddf�
}|j|
|�
tj
dd	�}|j|
|�
dS(
Ni�R(R)i
ii�((i
i((RR7RR0(RR9RBtn3((s5/usr/local/lib/python2.7/lib2to3/tests/test_pytree.pyttest_node_equality�s





c
C�s�tj
d
d�}
tj
d
d�}tjd|
g
�}tjd|g
�}|j||�
tj
d
d�}tjd|g
�}|j||�
dS(NidR	i�R/(RRR7RR0(RR$R*R9RBR1RD((s5/usr/local/lib/python2.7/lib2to3/tests/test_pytree.pyttest_node_recursive_equality�s







c
C�s�tj
d
d�}
tj
d
d�}tj
d
d�}tjd|
||g�}|j|j|
||g�
|j|jt�
|j|j�

tj
d
d�}|j	|�

|j|j|
||g�
|j|jt�
|j
|j�

dS(NidR	t
+R/i�t
-(RRR7RR8tassertIsInstanceRR4R5treplaceR(RR$R*R1R9tl2new((s5/usr/local/lib/python2.7/lib2to3/tests/test_pytree.pyttest_replace�s












c
C�s�tj
d
d�}
tj
d
d�}tj
d
d�}tjd|
||g�}|jtj
d
d�tj
d
d�g�

|jt|�
d�
|j|jt�
dS(NidR	RGR/i�t
*sfoo**bar(	RRR7RJRRRIR8R(RR$R*R1R9((s5/usr/local/lib/python2.7/lib2to3/tests/test_pytree.pyttest_replace_with_list�s



+

c
C�s�tj
d
d�}
tj
d
d�}tj
d
d�}tjd|
|g�}tjd|g
�}tjd||g�}|jt|j��
|
||g�
dS(NidR	R/tfooeyi�(RRR7RRtleaves(RR$R*R1RBRDR9((s5/usr/local/lib/python2.7/lib2to3/tests/test_pytree.pyttest_leaves�s





c
C�s�tj
d
d�}
tj
d
d�}tjd|
|g�}tjdg�}tjd||g�}|j|
j�d�
|j|j�d�
|j|j�d�
dS(NidR	R/i�ii
i(RRR7Rtdepth(RR$R*RBRDR9((s5/usr/local/lib/python2.7/lib2to3/tests/test_pytree.pyt
test_depth�s






c
C�s�tj
d
d�}
tj
d
d�}tj
d
d�}tjd|
|g�}tjd||g�}|jt|j��
|
||||g�
dS(NidR	R/ROi�(RRR7RRt
post_order(RR$R*R1tc1R9((s5/usr/local/lib/python2.7/lib2to3/tests/test_pytree.pyttest_post_order�s





c
C�s�tj
d
d�}
tj
d
d�}tj
d
d�}tjd|
|g�}tjd||g�}|jt|j��
|||
||g�
dS(NidR	R/ROi�(RRR7RRt	pre_order(RR$R*R1RUR9((s5/usr/local/lib/python2.7/lib2to3/tests/test_pytree.pyttest_pre_order�s





c
C�sa
tj
d
d�}
|j|
j�

|
j�
|j|
j�

tj
d
d�}
tjd|
g
�}|j|j�

|j�
|j|j�

tj
d
d�}
tj
d
d�}tj
d
d�}tjd|
||g�}tjd|g
�}|j|
j�

|j|j�

|j|j�

|j�
|j|j�

|j|j�

|j|
j�

dS(Nidt
fi�R	RGR/(RRR4R5tchangedRR7(RR$R9R*R1RB((s5/usr/local/lib/python2.7/lib2to3/tests/test_pytree.pyttest_changed�s*




















c
C�sZxSdD]K}
tj
ddd|
�
}|jt|�
|
d�
|j|j|
�
qWdS(Ntxyz_RidRR(R\R(RRRRRR(RRR$((s5/usr/local/lib/python2.7/lib2to3/tests/test_pytree.pyttest_leaf_constructor_prefix�s




c
C�s�x�d
D]�}
tj
dd�}tj
dddd�
}tjd||gd|
�
}|jt|�
|
d	�
|j|j|
�
|j|j|
�
|j|jd�
qWdS(NR\RidRR	Rt
_i�tself_foo(R\R(RRR7RRRR(RRR$R*R9((s5/usr/local/lib/python2.7/lib2to3/tests/test_pytree.pyttest_node_constructor_prefix�s








c
C�so
tj
d
d�}
tj
d
d�}tjd|
|g�}tjd|g
�}|j|j�d�
|j|jg�
|j|
j|�
|j|jd�
|j|jd�
|j|j	�

|j
|j	�

|j|j�d�
|j|
j�d�
|j|jg�
|j|
jd�
|j|jd�
|j|jd�
|j
|j	�

|j
|j	�

dS(NidR	i�ii
(RRR7RtremoveR8tparenttNoneR4R5R(RR$R*R9RB((s5/usr/local/lib/python2.7/lib2to3/tests/test_pytree.pyttest_remove
s&
















c
C�sbtj
d
g�}
|
j�
|j|
jd�
tjdd�}|j�
|j|jd�
dS(Ni�idR	(RR7RaRRbRcR(RR9R$((s5/usr/local/lib/python2.7/lib2to3/tests/test_pytree.pyttest_remove_parentless
s






c
C�s/
tj
d
d�}
tjd|
g
�}tj
d
d�}|jd|�
|j|
jd�
|j|j|�
|j|j|g
�
tjd|
g
�}|jd|�
|j|
jd�
|j|j|�
|j|jd�
|j|j|g
�
|jt	|jd|�
|jt
|jdt�
dS(NidR	i�R/ii(RRR7t	set_childRRbRcR8tassertRaisest
IndexErrort	ExceptionR(RR$R9R*RB((s5/usr/local/lib/python2.7/lib2to3/tests/test_pytree.pyttest_node_set_child$
s










c
C�s�tj
d
d�}
tjd|
g
�}tj
d
d�}|jd|�
|j|j|�
|j|j||
g�
tj
d
d�}|jd|�
|j|j||
|g�
|jt|jdt	�
dS(NidR	i�R/itabci(
RRR7tinsert_childRRbR8RgRiR(RR$R9R*R1((s5/usr/local/lib/python2.7/lib2to3/tests/test_pytree.pyttest_node_insert_child9
s






c
C�s�tj
d
g�}
tjdd�}|
j|�

|j|j|
�
|j|
j|g
�
tjdd�}|
j|�

|j|j|
�
|j|
j||g�
|jt|
jt	�
dS(Ni�idR	R/(
RR7Rtappend_childRRbR8RgRiR(RR9R$R*((s5/usr/local/lib/python2.7/lib2to3/tests/test_pytree.pyttest_node_append_childI
s








c
C�sytj
d
g�}
tj
d
g�}tj
d
|
|g�}|j|
j|�
|j|jd�
|j|jd�
dS(Ni�(RR7tassertIstnext_siblingRRc(RR9RBtp1((s5/usr/local/lib/python2.7/lib2to3/tests/test_pytree.pyttest_node_next_siblingY
s




c
C�sytj
d
d�}
tj
d
d�}tjd|
|g�}|j|
j|�
|j|jd�
|j|jd�
dS(NidR>R?i�(RRR7RpRqRRc(RR$R*Rr((s5/usr/local/lib/python2.7/lib2to3/tests/test_pytree.pyttest_leaf_next_siblingb
s




c
C�sytj
d
g�}
tj
d
g�}tj
d
|
|g�}|j|j|
�
|j|
jd�
|j|jd�
dS(Ni�(RR7Rptprev_siblingRRc(RR9RBRr((s5/usr/local/lib/python2.7/lib2to3/tests/test_pytree.pyttest_node_prev_siblingk
s




c
C�sytj
d
d�}
tj
d
d�}tjd|
|g�}|j|j|
�
|j|
jd�
|j|jd�
dS(NidR>R?i�(RRR7RpRuRRc(RR$R*Rr((s5/usr/local/lib/python2.7/lib2to3/tests/test_pytree.pyttest_leaf_prev_siblingt
s




(ii(&t__name__t
__module__t__doc__tsystversion_infoR R!R%R'R+R.R3R6R:R;R<R=RCRERFRLRNRQRSRVRXR[R]R`RdReRjRmRoRsRtRvRw(((s5/usr/local/lib/python2.7/lib2to3/tests/test_pytree.pyRsF																
		
	
						
												tTestPatternsc
B�s2eZ
dZd
�Zd�Zd�Zd�ZRS(s&Unit tests for tree matching patterns.c

C�s�tj
d
d�}
tj
d
d�}tj
d
d�}tjd|
|g�}tjd|g
�}tjd||g�}tjd
ddd�
}i}|j|j|d|�

�

|j|i�
|j|j|d|�

�

|j|i�
|j|j|d|�

�

|j|i�
|j|j|
d|�

�

|j|i
|
d6�
i}|j|j|d|�

�

|j|i�
tjd|g
dd�
}	|j|	j|d|�

�

|j|i�
|j|	j|d|�

�

|j|i�
|j|	j|d|�

�

|j|i|d6|d6�
i}|j|	j|
d|�

�

|j|i�
|j|	j|d|�

�

|j|i�
dS(	NidR	R/i�tnametpltresultstpn(	RRR7tLeafPatternR4tmatchRRtNodePattern(
RR$R*R1R9RBtrootRt
rR�((s5/usr/local/lib/python2.7/lib2to3/tests/test_pytree.pyttest_basic_patterns�
s>



























c
C�s?tj
d
d�}
tj
d
d�}tj
d
d�}tjd|
|g�}tjd|g
�}tjd||g�}tjd
ddd�
}tjd|g
dd�
}tj|g
||ggdd�

}	i}
|j|	j|g
|
��

|j|
i�
|j|	j|g
|
��

|j|
i�
|j	|	j|g
|
��

|jt
|
j��
dddg�
|j|
d|
�
|j|
d|�
|j|
d|g
�
|j|
i|
d6|d6|g
d6�
i}
|j	|	j|
|g|
��

|j|
i|d6|
|gd6�
|j|
d|�
i}
dS(	NidR	R/i�R~RR�tpw(
RRR7R�R�tWildcardPatternR4t	match_seqRRRtkeysRp(RR$R*R1R9RBR�RR�R�R�((s5/usr/local/lib/python2.7/lib2to3/tests/test_pytree.pyt
test_wildcard�
s2






$





%


(


$

c
C�s�tj
d
d�}
tj
d
d�}tj
d
d�}tj
d
d�}tj
d
d�}tj
d
d�}|
|||||g}tjd|�}tjd
dd	�}	tjd
dd
�}
tjd
dd�}tjd
dd�}tjd
dd
�}
tjd
dd�}tj|	|
|g||
g|	|
g||g|
|ggdd
dddd�
}|jg|j|�
D]}|d^q{
dddddg�
tjddd|g
dd�}ttj|g
|g
��
}|jt	|�
d
�
|d\}}|j|d
�
|jt
|d�
d�
|j|d|
|||||g�
x2dD]*}|j|d|tj
d
|��
q`WdS(Ni
R>R?R@t
dt
eRYi�tpatpbtpctpdtpetpftmintmaxiR~R�iiiiiR"tcontenttprtabcdeft
p(RRR7R�R�Rtgenerate_matchesR�RRR(RtlatlbtlctldtletlfRPR�R�R�R�R�R�R�R�t
xR�tmatchesR@R�((s5/usr/local/lib/python2.7/lib2to3/tests/test_pytree.pyttest_generate_matches�
s8















!

)

!





&


c
C�s�tj
d
tjd�
tjdd�
tjd�
f�}
tjdd�}tjdd�}tjdd	�}tjd
|||g�}i}|j|
j||��

|j|d|g
�
dS(
NiK
iR~targsit
(iR�t
)(	RR�R�R�RR7RR�R(RtpatternR$R*R1tnodeR�((s5/usr/local/lib/python2.7/lib2to3/tests/test_pytree.pyttest_has_key_example�
s
	









(RxRyRzR�R�R�R�(((s5/usr/local/lib/python2.7/lib2to3/tests/test_pytree.pyR}~
s
	$		(Rzt
__future__RR{RRR
tlib2to3RRt	NameErrortTestCaseRR}(((s5/usr/local/lib/python2.7/lib2to3/tests/test_pytree.pyt<module>
s





�a