Your IP : 216.73.216.213
3
� f�? � @ sj d Z ddlmZ ddlmZ ye W n ek
r@ dd� ZY nX G dd� dej�ZG d d
� d
ej�Z d S ) a 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.
� )�support� )�pytreec C s t | �}|j� |S )N)�list�sort)Zlst�l� r �>/opt/alt/python36/lib64/python3.6/lib2to3/tests/test_pytree.py�sorted s r
c @ s e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Zd0d1� Zd2d3� Z d4d5� Z d6d7� Z d8d9� Z d:d;� Z d<d=� Z!d>d?� Z"d@dA� Z#dBS )C� TestNodesz(Unit tests for nodes (Base, Leaf, Node).c C s d S )Nr )�selfr r r �test_instantiate_base s z TestNodes.test_instantiate_basec C s, t jdd�}| j|jd� | j|jd� d S )N�d �foo)r �Leaf� assertEqual�type�value)r �l1r r r � test_leaf" s zTestNodes.test_leafc C s t jdd�}| jt|�d� d S )Nr r zLeaf(100, 'foo'))r r r �repr)r r r r r �test_leaf_repr' s zTestNodes.test_leaf_reprc C s@ t jdd�}| jt|�d� t jddd d�}| jt|�d� d S )
Nr r � �
r )�contextz foo�r r )r r )r r r �str)r r �l2r r r �
test_leaf_str+ s zTestNodes.test_leaf_strc C s& t jdd�}d|_| jt|�d� d S )N� � Zfoo_Zfoo_5)r r �prefixr r )r r r r r �test_leaf_str_numeric_value1 s z%TestNodes.test_leaf_str_numeric_valuec C s\ t jdd�}t jddd
d�}| j||� t jdd�}t jdd�}| j||� | j||� d S ) Nr r r r r )r �e �bar�r r )r r% )r r r �assertNotEqual)r r r �l3�l4r r r �test_leaf_equality8 s z TestNodes.test_leaf_equalityc C sJ t jdd�}| j|jd� | j|j� d|_| j|jd� | j|j� d S )Nr r � z ##
)r r r r! � assertFalse� was_changed�
assertTrue)r r r r r �test_leaf_prefixA s zTestNodes.test_leaf_prefixc C sL t jdd�}t jdd�}t jd||g�}| j|jd� | j|j||g� d S )Nr r �� r$ i� )r r �Noder r �children)r r r �n1r r r � test_nodeI s
zTestNodes.test_nodec C sP t jdd�}t jddd d�}t jd||g�}| jt|�d t|�t|�f � d S ) Nr r r$ r r r )r i� zNode(1000, [%s, %s])�r r )r r4 )r r r0 r r )r r r r2 r r r �test_node_reprP s
zTestNodes.test_node_reprc C s@ t jdd�}t jddd d�}t jd||g�}| jt|�d � d S ) Nr r r$ r r r )r i� zfoo bar�r r )r r6 )r r r0 r r )r r r r2 r r r �
test_node_strW s zTestNodes.test_node_strc C s\ t jdd�}| j|jd� t jd|g�}| j|jd� d|_| j|jd� | j|jd� d S )Nr r r* i� r )r r r r! r0 )r r r2 r r r �test_node_prefix] s zTestNodes.test_node_prefixc C s� t jdddd�}t jdddd�}t jd||g�}| j|j� |j� | j|j� d� | j|j� d� t jddd d�}t jd||g�}| j|j� |j� | j|j� d� | j|j� d� d S )
Nr r �a)r! r$ �bi� r* �c)r r r0 r Z
get_suffixr! )r r r r2 r' �n2r r r �test_get_suffixf s zTestNodes.test_get_suffixc C sD t jdf �}t jdg dd�}| j||� t jdf �}| j||� d S ) Ni� r r r )r i� �r r )r r> )r r0 r r& )r r2 r< �n3r r r �test_node_equalityv s
z TestNodes.test_node_equalityc C sj t jdd�}t jdd�}t jd|g�}t jd|g�}| j||� t jdd�}t jd|g�}| j||� d S )Nr r i� r$ )r r r0 r r& )r r r r2 r< r' r? r r r � test_node_recursive_equality} s z&TestNodes.test_node_recursive_equalityc C s� t jdd�}t jdd�}t jdd�}t jd|||g�}| j|j|||g� | j|jt� | j|j� t jdd�}|j |� | j|j|||g� | j|jt� | j
|j� d S )Nr r �+r$ i� �-) r r r0 r r1 �assertIsInstancer r+ r, �replacer- )r r r r' r2 Zl2newr r r � test_replace� s
zTestNodes.test_replacec C sv t jdd�}t jdd�}t jdd�}t jd|||g�}|jt jdd�t jdd�g� | jt|�d� | j|jt� d S )Nr r rB r$ i� �*zfoo**bar) r r r0 rE r r rD r1 r )r r r r' r2 r r r �test_replace_with_list� s z TestNodes.test_replace_with_listc C sp t jdd�}t jdd�}t jdd�}t jd||g�}t jd|g�}t jd||g�}| jt|j� �|||g� d S )Nr r r$ �fooeyi� )r r r0 r r �leaves)r r r r' r< r? r2 r r r � test_leaves� s zTestNodes.test_leavesc C sx t jdd�}t jdd�}t jd||g�}t jdg �}t jd||g�}| j|j� d� | j|j� d� | j|j� d� d S )Nr r r$ i� r r r )r r r0 r Zdepth)r r r r< r? r2 r r r �
test_depth� s zTestNodes.test_depthc C sf t jdd�}t jdd�}t jdd�}t jd||g�}t jd||g�}| jt|j� �|||||g� d S )Nr r r$ rI i� )r r r0 r r Z
post_order)r r r r' �c1r2 r r r �test_post_order� s zTestNodes.test_post_orderc C sf t jdd�}t jdd�}t jdd�}t jd||g�}t jd||g�}| jt|j� �|||||g� d S )Nr r r$ rI i� )r r r0 r r Z pre_order)r r r r' rM r2 r r r �test_pre_order� s zTestNodes.test_pre_orderc C s� t jdd�}| j|j� |j� | j|j� t jdd�}t jd|g�}| j|j� |j� | j|j� t jdd�}t jdd�}t jdd�}t jd|||g�}t jd|g�}| j|j� | j|j� | j|j� |j� | j|j� | j|j� | j|j� d S )Nr �fi� r rB r$ )r r r+ r, Zchangedr- r0 )r r r2 r r' r< r r r � test_changed� s* zTestNodes.test_changedc C sD x>dD ]6}t jdd|d�}| jt|�|d � | j|j|� qW d S )N�xyz_r* r r )r! )rR r* )r r r- r r r! )r r! r r r r � test_leaf_constructor_prefix� s
z&TestNodes.test_leaf_constructor_prefixc C s� xzd
D ]r}t jdd�}t jdddd�}t jd||g|d�}| jt|�|d � | j|j|� | j|j|� | j|jd� qW d S ) NrR r* r r r �_)r! i� Zself_foo)rR r* )r r r0 r- r r r! )r r! r r r2 r r r � test_node_constructor_prefix� s
z&TestNodes.test_node_constructor_prefixc C s
t jdd�}t jdd�}t jd||g�}t jd|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� d S )Nr r i� r r )
r r r0 r �remover1 �parentr+ r, r- )r r r r2 r< r r r � test_remove� s& zTestNodes.test_removec C sH t jdg �}|j� | j|jd � t jdd�}|j� | j|jd � d S )Ni� r r )r r0 rV r rW r )r r2 r r r r �test_remove_parentless s z TestNodes.test_remove_parentlessc C s� t jdd�}t jd|g�}t jdd�}|jd|� | j|jd � | j|j|� | j|j|g� t jd|g�}|jd|� | j|jd � | j|j|� | j|jd � | j|j|g� | jt|jd|� | jt |jdt
� d S )Nr r i� r$ r � ) r r r0 Z set_childr rW r1 � assertRaises�
IndexError� Exceptionr )r r r2 r r< r r r �test_node_set_child s z TestNodes.test_node_set_childc C s� t jdd�}t jd|g�}t jdd�}|jd|� | j|j|� | j|j||g� t jdd�}|jd|� | j|j|||g� | jt|jdt � d S )Nr r i� r$ r �abcr )
r r r0 Z insert_childr rW r1 r[ r] r )r r r2 r r' r r r �test_node_insert_child# s z TestNodes.test_node_insert_childc C s� t jdg �}t jdd�}|j|� | j|j|� | j|j|g� t jdd�}|j|� | j|j|� | j|j||g� | jt|jt � d S )Ni� r r r$ )
r r0 r Z append_childr rW r1 r[ r] r )r r2 r r r r r �test_node_append_child3 s
z TestNodes.test_node_append_childc C sV t jdg �}t jdg �}t jd||g�}| j|j|� | j|jd � | j|jd � d S )Ni� )r r0 �assertIs� next_siblingr )r r2 r< �p1r r r �test_node_next_siblingC s z TestNodes.test_node_next_siblingc C sV t jdd�}t jdd�}t jd||g�}| j|j|� | j|jd � | j|jd � d S )Nr r9 r: i� )r r r0 rb rc r )r r r rd r r r �test_leaf_next_siblingL s z TestNodes.test_leaf_next_siblingc C sV t jdg �}t jdg �}t jd||g�}| j|j|� | j|jd � | j|jd � d S )Ni� )r r0 rb � prev_siblingr )r r2 r< rd r r r �test_node_prev_siblingU s z TestNodes.test_node_prev_siblingc C sV t jdd�}t jdd�}t jd||g�}| j|j|� | j|jd � | j|jd � d S )Nr r9 r: i� )r r r0 rb rg r )r r r rd r r r �test_leaf_prev_sibling^ s z TestNodes.test_leaf_prev_siblingN)$�__name__�
__module__� __qualname__�__doc__r
r r r r" r) r. r3 r5 r7 r8 r= r@ rA rF rH rK rL rN rO rQ rS rU rX rY r^ r` ra re rf rh ri r r r r r sB
r c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) � TestPatternsz&Unit tests for tree matching patterns.c
C s� t jdd�}t jdd�}t jdd�}t jd||g�}t jd|g�}t jd||g�}t jdddd�}i }| j|j||d�� | j|i � | j|j||d�� | j|i � | j|j||d�� | j|i � | j|j||d�� | j|d|i� i }| j|j||d�� | j|i � t jd|gdd�} | j| j||d�� | j|i � | j| j||d�� | j|i � | j| j||d�� | j|||d �� i }| j| j||d�� | j|i � | j| j||d�� | j|i � d S )
Nr r r$ i� �pl)�name)Zresults�pn)rq ro ) r r r0 � LeafPatternr+ �matchr r- � NodePattern)
r r r r' r2 r< �rootro �rrq r r r �test_basic_patternsl s> z TestPatterns.test_basic_patternsc C s� t jdd�}t jdd�}t jdd�}t jd||g�}t jd|g�}t jd||g�}t jdddd�}t jd|gdd�}t j|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|
|||gd �� i }
| j | j||g|
�� | j|
|||gd
�� | j |
d |� i }
d S ) Nr r r$ i� ro )rp rq �pw)ro rq rx )ro rx )
r r r0 rr rt �WildcardPatternr+ Z match_seqr r- r
�keysrb ) r r r r' r2 r< ru ro rq rx rv r r r �
test_wildcard� s2 zTestPatterns.test_wildcardc C s� t jdd�}t jdd�}t jdd�}t jdd�}t jdd�}t jdd�}||||||g}t jd|�}t jddd �} t jddd
�}
t jddd �} t jddd �} t jddd
�}
t jddd�}t j| |
| g| |
g| |
g| | g|
|ggdddd�}| jdd� |j|�D �dddddg� t jd|gdd�}tt j|g|g��}| jt |�d� |d \}}| j|d� | jt
|d �d� | j|d ||||||g� x*dD ]"}| j|d | t jd|�� �q�W d S ) Nr r9 r: r; �d�erP i� �pa�pb�pc�pd�pe�pfrZ rx )�min�maxrp c S s g | ] }|d �qS )r r )�.0�xr r r �
<listcomp>� s z6TestPatterns.test_generate_matches.<locals>.<listcomp>� r r � �pr)r Zcontentrp r Zabcdef�p) r r r0 rr ry r Zgenerate_matchesrt r �lenr )r ZlaZlbZlcZld�leZlfrJ ru r~ r r� r� r� r� rx r� Zmatchesr; rv r r r �test_generate_matches� s8
z"TestPatterns.test_generate_matchesc C s� t jdt jd�t jdd�t jd�f�}t jdd�}t jdd�}t jdd �}t jd|||g�}i }| j|j||�� | j|d |g� d S )
NiK � �args)rp � �(r� r� �)) r rt rr ry r r0 r- rs r )r �patternr r r' Znoderv r r r �test_has_key_example� s
z!TestPatterns.test_has_key_exampleN)rj rk rl rm rw r{ r� r� r r r r rn h s
$ rn N)
rm r* r Zlib2to3r r
� NameErrorZTestCaser rn r r r r �<module>
s Q