File: //usr/local/lib/python3.7/test/__pycache__/test_weakset.cpython-37.pyc
B
��g�; � @ sz d dl Z d dlmZ d dlZd dlmZ d dlZd dlZG dd� d�Z G dd� d�Z
G dd � d e j �Z e
d
krve �� dS ) � N)�WeakSet)�
UserStringc @ s e Zd ZdS )�FooN)�__name__�
__module__� __qualname__� r r �-/usr/local/lib/python3.7/test/test_weakset.pyr s r c @ s e Zd Zdd� ZdS )�RefCyclec C s
| | _ d S )N)�cycle)�selfr r r �__init__
s zRefCycle.__init__N)r r r r
r r r r r
s r
c @ sl e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� 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/d0� Zd1d2� Zd3d4� Z d5d6� Z d7d8� Z d9d:� Z d;d<� Z d=d>� Z!d?d@� Z"dAdB� Z#dCdD� Z$dEdF� Z%dGdH� Z&dIdJ� Z'dKdL� Z(dMdN� Z)dOdP� Z*dQdR� Z+dSdT� Z,dUdV� Z-dWdX� Z.dYS )Z� TestWeakSetc C s� dd� dD �| _ dd� dD �| _dd� dD �| _dd� d D �| _d
d� d D �| _t| j�| _t| j�| _t| j�| _d d� t j
D �| _ t| j �| _ t
�| j �| _td
�| _t| jg�| _d S )Nc S s g | ] }t |��qS r )�ustr)�.0�cr r r �
<listcomp> s z%TestWeakSet.setUp.<locals>.<listcomp>)�a�br c S s g | ] }t |��qS r )r )r r r r r r s )�x�y�zc S s g | ] }t |��qS r )r )r r r r r r s Zabc S s g | ] }t |��qS r )r )r r r r r r s Zabcdec S s g | ] }t |��qS r )r )r r r r r r s �defc S s g | ] }t |��qS r )r )r r r r r r s �F)�items�items2Zab_itemsZ abcde_itemsZ def_itemsr �
ab_weakset�
abcde_weakset� def_weakset�string�
ascii_letters�letters�s�dict�fromkeys�dr �obj�fs)r r r r �setUp s
zTestWeakSet.setUpc C sD t t�}x6t t�D ]*}|dks|�d�r*q| �||d| � qW d S )NZ
test_c_api�_zWeakSet missing method )�dirr �set�
startswith�assertIn)r Zweaksetmethods�methodr r r � test_methods# s zTestWeakSet.test_methodsc C s | � ttg d� d S )N� )� assertRaises� TypeErrorr )r r r r �test_new_or_init+ s z TestWeakSet.test_new_or_initc C sD | � t| j�t| j�� | � t| j�d� | `| � t| j�d� d S )N� r )� assertEqual�lenr" r% r' r& )r r r r �test_len. s zTestWeakSet.test_lenc C s` x&| j D ] }| �|| jk|| jk� qW | �d| j� | �| j| j� | `| �td�| j� d S )Nr4 r ) r! r5 r" r% � assertNotInr- r& r' r )r r r r r �
test_contains4 s zTestWeakSet.test_containsc C s | j �| j�}x.| jD ]$}| �||k|| jkp6|| jk� qW | �| j t| j�� | �t|�t� | � t
| j jg g� xHt t t
jttfD ]4}t| j| j �}|| j�}| �| j �|�|� ~q�W | �t|�t| j�t| j� � | j�� t�� | �t|�t| j�t| j� � d S )N)r" �unionr r! r5 r% r r �typer1 r2 r+ � frozensetr# r$ �list�tupler6 �pop�gc�collect)r �ur �Cr r r r �
test_union= s $
zTestWeakSet.test_unionc C sB | j �| j�}| �| j t| j�B |� | �| j t| j�B |� d S )N)r" r: r r5 r+ r< )r �ir r r �test_orN s zTestWeakSet.test_orc C s� t | j�}|�| j�}x.| jD ]$}| �||k|| jko>|| jk� q W | �|t | j�� | �t|�t � x8tttj t
t fD ]$}t g �}| �|�|| j ��|� qzW | �t
|�t
| j�� | j�� t�� | �t
|�t
| j�� d S )N)r r! � intersectionr r5 r; r+ r< r# r$ r= r> r r6 r? r@ rA )r r" rE r rC r r r r �test_intersectionS s
$
z TestWeakSet.test_intersectionc C s6 | � | j�t| j��� | � | j�t| j�� � d S )N)�
assertTruer" �
isdisjointr r r! )r r r r �test_isdisjointb s zTestWeakSet.test_isdisjointc C sB | j �| j�}| �| j t| j�@ |� | �| j t| j�@ |� d S )N)r" rG r r5 r+ r< )r rE r r r �test_andf s zTestWeakSet.test_andc C sz | j �| j�}x.| jD ]$}| �||k|| jko6|| jk� qW | �| j t| j�� | �t|�t� | � t
| j jg g� d S )N) r" �
differencer r! r5 r% r r r; r1 r2 )r rE r r r r �test_differencek s $zTestWeakSet.test_differencec C sB | j �| j�}| �| j t| j� |� | �| j t| j� |� d S )N)r" rM r r5 r+ r< )r rE r r r �test_subs s zTestWeakSet.test_subc C s� | j �| j�}x.| jD ]$}| �||k|| jk|| jkA � qW | �| j t| j�� | �t|�t� | � t
| j jg g� | �t |�t | j�t | j� � | j� � t
�� | �t |�t | j�t | j� � d S )N)r" �symmetric_differencer r! r5 r% r r r; r1 r2 r6 r? r@ rA )r rE r r r r �test_symmetric_differencex s $
z%TestWeakSet.test_symmetric_differencec C sB | j �| j�}| �| j t| j�A |� | �| j t| j�A |� d S )N)r" rP r r5 r+ r< )r rE r r r �test_xor� s zTestWeakSet.test_xorc C s� | � | j| jk� | � | j| jk� | � | j| jk� | �| j| jk� | �| j| jk� | � td��d�� | � td��d�� | �td��d�� | �td��d�� d S )Nr �abcZcbs)rI r r � assertFalser r+ �issubset�
issuperset)r r r r �test_sub_and_super� s z TestWeakSet.test_sub_and_superc C sL | � | j| jk � | �| j| jk � | �| j| jk � | �t� t� k � d S )N)rI r r rT r r )r r r r �test_lt� s zTestWeakSet.test_ltc C sL | � | j| jk� | �| j| jk� | �| j| jk� | �t� t� k� d S )N)rI r r rT r r )r r r r �test_gt� s zTestWeakSet.test_gtc C s@ t dd� td�D ��}x$|D ] }||_||_t |g�|_q W d S )Nc s s | ] }t � V qd S )N)r )r rE r r r � <genexpr>� s z&TestWeakSet.test_gc.<locals>.<genexpr>i� )r �ranger �subr+ )r r" �elemr r r �test_gc� s
zTestWeakSet.test_gcc C sT G dd� dt �}|� }t� }|�|� | �||� |�|� |�|� |�|� d S )Nc @ s e Zd Zdd� ZdS )z5TestWeakSet.test_subclass_with_custom_hash.<locals>.Hc S s t t| �d@ �S )Ni���)�int�id)r r r r �__hash__� s z>TestWeakSet.test_subclass_with_custom_hash.<locals>.H.__hash__N)r r r ra r r r r �H� s rb )r r+ �addr- �remove�discard)r rb r" �fr r r � test_subclass_with_custom_hash� s
z*TestWeakSet.test_subclass_with_custom_hashc C sd t � }|�| j� | �|| j� |�| j� | �|t | j�� | �t|j|d� | �t|jd� d S )Nr0 r4 )r r
r r5 r" r r1 r2 )r r" r r r � test_init� s zTestWeakSet.test_initc C s* t | j�}t |�}| �t|�t|�� d S )N)r r �assertNotEqualr` )r r" �tr r r �test_constructor_identity� s
z%TestWeakSet.test_constructor_identityc C s | � tt| j� d S )N)r1 r2 �hashr" )r r r r � test_hash� s zTestWeakSet.test_hashc C s2 | j �� | �| j tg �� | �t| j �d� d S )Nr )r" �clearr5 r r6 )r r r r �
test_clear� s
zTestWeakSet.test_clearc C s2 | j �� }| �| j |� | �t| j �t|�� d S )N)r" �copyr5 ri r` )r �dupr r r � test_copy� s
zTestWeakSet.test_copyc C s� t d�}| j�|� | �|| j� | j�� }| j�|� | �| j|� | �t| jjg � | j�t � � | �
t | j�dk� | j�| j � | �
t | j�dk� d S )N�Qr4 )
r r" rc r- rp r5 r1 r2 r' r rI r6 r& )r r rq r r r �test_add� s
zTestWeakSet.test_addc C sJ t d�}| j�|� | �|| j� | �t| jj|� | �t| jjg � d S )Nr )r r" rd r8 r1 �KeyErrorr2 )r r r r r � test_remove� s
zTestWeakSet.test_removec C sN t d�t d� }}| j�|� | �|| j� | j�|� | �t| jjg � d S )Nr rs )r r" re r8 r1 r2 )r r �qr r r � test_discard� s
zTestWeakSet.test_discardc C sD x.t t| j��D ] }| j�� }| �|| j� qW | �t| jj� d S )N)r[ r6 r" r? r8 r1 ru )r rE r] r r r �test_pop� s
zTestWeakSet.test_popc C sV | j �| j�}| �|d � x"| j| j D ]}| �|| j � q(W | �t| j jg g� d S )N)r" �updater r5 r r- r1 r2 )r �retvalr r r r � test_update� s
zTestWeakSet.test_updatec C s: | j �t| j�� x"| j| j D ]}| �|| j � q W d S )N)r" rz r+ r r r- )r r r r r �test_update_set� s zTestWeakSet.test_update_setc C s< | j t| j�O _ x"| j| j D ]}| �|| j � q"W d S )N)r" r+ r r r- )r r r r r �test_ior� s zTestWeakSet.test_iorc C sz | j �| j�}| �|d � xF| j| j D ]6}|| jkrP|| jkrP| �|| j � q(| �|| j � q(W | �t| j jg g� d S )N) r" �intersection_updater r5 r r- r8 r1 r2 )r r{ r r r r �test_intersection_update s z$TestWeakSet.test_intersection_updatec C s` | j t| j�M _ xF| j| j D ]6}|| jkrJ|| jkrJ| �|| j � q"| �|| j � q"W d S )N)r" r+ r r r- r8 )r r r r r � test_iand s
zTestWeakSet.test_iandc C s� | j �| j�}| �|d � xF| j| j D ]6}|| jkrP|| jkrP| �|| j � q(| �|| j � q(W | �t| j jg g� | �t| j j g g� d S )N)
r" �difference_updater r5 r r- r8 r1 r2 �symmetric_difference_update)r r{ r r r r �test_difference_update s z"TestWeakSet.test_difference_updatec C s` | j t| j�8 _ xF| j| j D ]6}|| jkrJ|| jkrJ| �|| j � q"| �|| j � q"W d S )N)r" r+ r r r- r8 )r r r r r � test_isub s
zTestWeakSet.test_isubc C sz | j �| j�}| �|d � xF| j| j D ]6}|| jk|| jkA rP| �|| j � q(| �|| j � q(W | �t| j jg g� d S )N) r" r� r r5 r r- r8 r1 r2 )r r{ r r r r � test_symmetric_difference_update' s z,TestWeakSet.test_symmetric_difference_updatec C s` | j t| j�N _ xF| j| j D ]6}|| jk|| jkA rJ| �|| j � q"| �|| j � q"W d S )N)r" r+ r r r- r8 )r r r r r � test_ixor1 s
zTestWeakSet.test_ixorc C sp | j �� }||O }| �|| j � ||M }| �|| j � ||8 }| �|t� � | j �� }||N }| �|t� � d S )N)r" rp r5 r )r rj r r r �test_inplace_on_self9 s
z TestWeakSet.test_inplace_on_selfc C s� | � | j| jk� | � | jt| j�k� | �| jt| j�k� | �| jt| j�k� | �| jt| j�k� | �| jttg�k� | �| jdk� d S )Nr4 ) rI r" r r rT r+ r= r> r )r r r r �test_eqE s zTestWeakSet.test_eqc C s4 | � | jt| j�k� t� }t� }| �||k� d S )N)rI r" r+ r r rT )r �s1�s2r r r �test_neO s zTestWeakSet.test_nec C sr dd� dD �}t |�}t|�}t|� |d= t�� | �tt|��t|�t|�d g� ~| �t|�t|�� d S )Nc S s g | ] }t |��qS r )r )r r r r r r X s zATestWeakSet.test_weak_destroy_while_iterating.<locals>.<listcomp>)r r r ���r4 ) r �iter�nextr@ rA r- r6 r= r5 )r r r" �itr r r �!test_weak_destroy_while_iteratingU s $z-TestWeakSet.test_weak_destroy_while_iteratingc s dd� t jD �� t� ��tj� �fdd��}|� �}| �|�� W d Q R X |� �}| �t�j|� W d Q R X | �|�� |� �}�� |� W d Q R X | �
|�� �� � }|� �}�� |� W d Q R X | �
t��t|�� |� �}��� W d Q R X | �
t��d� d S )Nc S s g | ] }t |��qS r )r )r r r r r r g s zLTestWeakSet.test_weak_destroy_and_mutate_while_iterating.<locals>.<listcomp>c 3 sV zJt ��} ttt| ���}tt� �� ��}||kr:t| � t�� |V W d d } X d S )N)r� r �strr� r? r@ rA )r� ZyieldedrB )r r" r r � testcontexti s
zMTestWeakSet.test_weak_destroy_and_mutate_while_iterating.<locals>.testcontextr )r r r �
contextlib�contextmanagerr8 r1 ru rd rc r- rp rz r5 r6 rn )r r� rB rj r )r r" r �,test_weak_destroy_and_mutate_while_iteratinge s$ z8TestWeakSet.test_weak_destroy_and_mutate_while_iteratingc C s� d}dd� t |�D �}t|�}~t|�}y t|� W n tk
rH Y nX t�� t|�}~t�� t|�}| �|d� | � |d� d S )N� c S s g | ]
}t � �qS r )r
)r rE r r r r � s z/TestWeakSet.test_len_cycles.<locals>.<listcomp>)r r4 r )
r[ r r� r� �
StopIterationr@ rA r6 r- r5 )r �Nr r" r� �n1�n2r r r �test_len_cycles� s zTestWeakSet.test_len_cyclesc C s� | j tjft�� �� x�tdd�D ]�}d}t�d� t�|||� dd� t|�D �}t|�}~t|�}y t|� W n t k
r� Y nX t
|�}~t
|�}| � |d� | � ||� | � |d� | � ||� q"W d S )Nr4 �d r� r c S s g | ]
}t � �qS r )r
)r rE r r r r � s z-TestWeakSet.test_len_race.<locals>.<listcomp>)
Z
addCleanupr@ Z
set_thresholdZ
get_thresholdr[ rA r r� r� r� r6 ZassertGreaterEqualZassertLessEqual)r �thr� r r" r� r� r� r r r �
test_len_race� s(
zTestWeakSet.test_len_raceN)/r r r r( r/ r3 r7 r9 rD rF rH rK rL rN rO rQ rR rW rX rY r^ rg rh rk rm ro rr rt rv rx ry r| r} r~ r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r sX
%r �__main__)Zunittest�weakrefr r � collectionsr r r@ r� r r
ZTestCaser r �mainr r r r �<module> s (