��Qc@�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((s//sys/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((s//sys/lib/python2.7/lib2to3/tests/test_pytree.pyttest_deprecated_prefix_methods$s










c

C�s|jt
tj�
dS(
N(tassertRaisestAssertionErrorRtBase(
R((s//sys/lib/python2.7/lib2to3/tests/test_pytree.pyttest_instantiate_base3sc
C�s<tj
d
d�}
|j|
jd
�
|j|
jd�
dS(NidR	(RRRttypetvalue(Rtl1((s//sys/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'((s//sys/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((s//sys/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'((s//sys/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((s//sys/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'((s//sys/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�R2i�(RRtNodeRR%tchildren(RR'R-tn1((s//sys/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	R2R+R,i
ii�sNode(1000, [%s, %s])(i
i(RRR:RR)(RR'R-R<((s//sys/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	R2R+R,i
ii�sfoo bar(i
i(RRR:RR(RR'R-R<((s//sys/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,(RRRRR:(RR'R<((s//sys/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
aR2t
bi�Rt
c(RRR:Rt
get_suffixR(RR'R-R<R4tn2((s//sys/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((RR:RR3(RR<REtn3((s//sys/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�R2(RRR:RR3(RR'R-R<RER4RG((s//sys/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�
|jt|jt��
|j|j	�
tj
d
d�}|j
|�
|j|j|
||g�
|jt|jt��
|j|j	�
dS(NidR	t
+R2i�t
-(RRR:RR;Rt
isinstanceRR7R8treplace(RR'R-R4R<tl2new((s//sys/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�
|jt|jt	��
dS(NidR	RJR2i�t
*sfoo**bar(
RRR:RMRRRRLR;R(RR'R-R4R<((s//sys/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	R2tfooeyi�(RRR:RRtleaves(RR'R-R4RERGR<((s//sys/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	R2i�ii
i(RRR:Rtdepth(RR'R-RERGR<((s//sys/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	R2RRi�(RRR:RRt
post_order(RR'R-R4tc1R<((s//sys/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	R2RRi�(RRR:RRt	pre_order(RR'R-R4RXR<((s//sys/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	RJR2(RRR7R8tchangedRR:(RR'R<R-R4RE((s//sys/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'((s//sys/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(sxyz_R(RRR:RRRR(RRR'R-R<((s//sys/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
(RRR:RtremoveR;tparenttNoneR7R8R(RR'R-R<RE((s//sys/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	(RR:RdRReRfR(RR<R'((s//sys/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�R2ii(RRR:t	set_childRReRfR;R!t
IndexErrort	ExceptionR(RR'R<R-RE((s//sys/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�R2itabci(
RRR:tinsert_childRReR;R!RkR(RR'R<R-R4((s//sys/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	R2(
RR:Rtappend_childRReR;R!RkR(RR<R'R-((s//sys/lib/python2.7/lib2to3/tests/test_pytree.pyttest_node_append_childI
s








c
C�s|tj
d
g�}
tj
d
g�}tj
d
|
|g�}|j|
j|k�
|j|jd�
|j|jd�
dS(Ni�(RR:Rtnext_siblingRRf(RR<REtp1((s//sys/lib/python2.7/lib2to3/tests/test_pytree.pyttest_node_next_siblingY
s




c
C�s|tj
d
d�}
tj
d
d�}tjd|
|g�}|j|
j|k�
|j|jd�
|j|jd�
dS(NidRARBi�(RRR:RRrRRf(RR'R-Rs((s//sys/lib/python2.7/lib2to3/tests/test_pytree.pyttest_leaf_next_siblingb
s




c
C�s|tj
d
g�}
tj
d
g�}tj
d
|
|g�}|j|j|
k�
|j|
jd�
|j|jd�
dS(Ni�(RR:Rtprev_siblingRRf(RR<RERs((s//sys/lib/python2.7/lib2to3/tests/test_pytree.pyttest_node_prev_siblingk
s




c
C�s|tj
d
d�}
tj
d
d�}tjd|
|g�}|j|j|
k�
|j|
jd�
|j|jd�
dS(NidRARBi�(RRR:RRvRRf(RR'R-Rs((s//sys/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.R1R6R9R=R>R?R@RFRHRIRORQRTRVRYR[R^R`RcRgRhRlRoRqRtRuRwRx(((s//sys/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	R2i�tnametpltresultstpn(	RRR:tLeafPatternR7tmatchRRtNodePattern(
RR'R-R4R<REtrootR�t
rR�((s//sys/lib/python2.7/lib2to3/tests/test_pytree.pyttest_basic_patterns�s>



























c
C�sBtj
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|k�
i}
dS(	NidR	R2i�RR�R�tpw(RRR:R�R�tWildcardPatternR7t	match_seqRRRtkeys(RR'R-R4R<RER�R�R�R�R�((s//sys/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
RARBRCt
dt
eR\i�tpatpbtpctpdtpetpftmintmaxiRR�iiiiiR%tcontenttprtabcdeft
p(RRR:R�R�Rtgenerate_matchesR�RRR(RtlatlbtlctldtletlfRSR�R�R�R�R�R�R�R�t
xR�tmatchesRCR�((s//sys/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
iRtargsit
(iR�t
)(	RR�R�R�RR:RR�R(RtpatternR'R-R4tnodeR�((s//sys/lib/python2.7/lib2to3/tests/test_pytree.pyttest_has_key_example�s
	









(RyRzR{R�R�R�R�(((s//sys/lib/python2.7/lib2to3/tests/test_pytree.pyR~~
s
	$		(R{t
__future__RR|RRR
tlib2to3RRt	NameErrortTestCaseRR~(((s//sys/lib/python2.7/lib2to3/tests/test_pytree.pyt<module>
s