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 | j ttj� d S )N)� assertRaises�AssertionErrorr ZBase)�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 r6 )r r r2 r r )r r r r4 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 r8 )r r r2 r r )r r r r4 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# r2 )r r r4 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 r2 r Z
get_suffixr# )r r r r4 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 r2 r r( )r r4 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 r2 r r( )r r r r4 r> r) rA 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 r2 r r3 �assertIsInstancer r- r. �replacer/ )r r r r) r4 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 rD r& i� �*zfoo**bar) r r r2 rG r r rF r3 r )r r r r) r4 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 r2 r r �leaves)r r r r) r> rA r4 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 r2 r Zdepth)r r r r> rA r4 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& rK i� )r r r2 r r Z
post_order)r r r r) �c1r4 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& rK i� )r r r2 r r Z pre_order)r r r r) rO r4 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 rD r& )r r r- r. Zchangedr/ r2 )r r r4 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# )rT 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 ) NrT r, r r r �_)r# i� Zself_foo)rT r, )r r r2 r/ r r r# )r r# r r r4 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 r2 r �remover3 �parentr- r. r/ )r r r r4 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 r2 rX r rY r )r r4 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 r2 Z set_childr rY r3 r �
IndexError� Exceptionr )r r r4 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 r2 Z insert_childr rY r3 r r^ r )r r r4 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 r2 r Z append_childr rY r3 r r^ r )r r4 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 r2 �assertIs� next_siblingr )r r4 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 r; r< i� )r r r2 rc rd r )r r r re 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 r2 rc � prev_siblingr )r r4 r> re 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 r; r< i� )r r r2 rc rh r )r r r re r r r �test_leaf_prev_sibling^ s z TestNodes.test_leaf_prev_siblingN)$�__name__�
__module__� __qualname__�__doc__r r r r r$ r+ r0 r5 r7 r9 r: r? rB rC rH rJ rM rN rP rQ rS rU rW rZ r[ r_ ra rb rf rg ri rj 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)rr rp ) r r r2 � LeafPatternr- �matchr r/ � NodePattern)
r r r r) r4 r> �rootrp �rrr 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� rp )rq rr �pw)rp rr ry )rp ry )
r r r2 rs ru �WildcardPatternr- Z match_seqr r/ r
�keysrc ) r r r r) r4 r> rv rp rr ry rw 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 r; r< r= �d�erR i� �pa�pb�pc�pd�pe�pfr\ ry )�min�maxrq c S s g | ] }|d �qS )r r )�.0�xr r r �
<listcomp>� s z6TestPatterns.test_generate_matches.<locals>.<listcomp>� r" r! � �pr)r Zcontentrq r Zabcdef�p) r r r2 rs rz r Zgenerate_matchesru r �lenr )r ZlaZlbZlcZld�leZlfrL rv r r� r� r� r� r� ry r� Zmatchesr= rw 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)rq � �(r� r� �)) r ru rs rz r r2 r/ rt r )r �patternr r r) Znoderw r r r �test_has_key_example� s
z!TestPatterns.test_has_key_exampleN)rk rl rm rn rx r| r� r� r r r r ro h s
$ ro N)
rn r, r Zlib2to3r r
� NameErrorZTestCaser ro r r r r �<module>
s Q