File: //lib64/python3.6/test/__pycache__/test_heapq.cpython-36.pyc
3
ai�> � @ s� d Z ddlZddlZddlmZ ddlmZmZ ddlmZ ej ddgd�Z
ej ddgd �Z d
d d d
ddddgZ G dd� de�Z
G dd� d�ZG dd� dee�Zee d�G dd� dee��ZG dd � d �ZG d d � d �ZG d d � d �Zd!d"� ZG d#d$� d$�ZG d%d&� d&�ZG d'd(� d(�ZG d)d*� d*�ZG d+d,� d,�ZG d-d.� d.�ZG d/d0� d0�Zdd1l m Z d2d3� Z G d4d5� d5�Z G d6d7� d7�Z G d8d9� d9e e�Z!ee d�G d:d;� d;e e��Z"e#d<k�r�ej$� dS )=zUnittests for heapq.� N)�support)�TestCase�
skipUnless)�
itemgetter�heapq�_heapq)Zblocked)Zfresh�heapify�heappop�heappush� heappushpop� heapreplace� _heappop_max�_heapreplace_max� _heapify_maxc @ s& e Zd Zdd� Zeed�dd� �ZdS )� TestModulesc C s& x t D ]}| jtt|�jd� qW d S )Nr )�
func_names� assertEqual�getattr�py_heapq�
__module__)�self�fname� r �"/usr/lib64/python3.6/test_heapq.py�test_py_functions s
z TestModules.test_py_functionszrequires _heapqc C s& x t D ]}| jtt|�jd� qW d S )Nr )r r r �c_heapqr )r r r r r �test_c_functions s
z TestModules.test_c_functionsN)�__name__r � __qualname__r r r r r r r r r s r c @ s� 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 S ) �TestHeapc C s� g }g }| j |� x:td�D ].}tj� }|j|� | jj||� | j |� q W g }x(|rz| jj|�}| j |� |j|� qTW |d d � }|j� | j||� | j |� | j t
| jjg � y*| j t
| jjd d � | j t
| jjd � W n t k
r� Y nX d S )N� ) �check_invariant�range�random�append�moduler
r �sortr � assertRaises� TypeError�AttributeError)r �heap�data�i�item�resultsZ data_sortedr r r �
test_push_pop s.
zTestHeap.test_push_popc C s< x6t |�D ]*\}}|r
|d d? }| j|| |k� q
W d S )N� )� enumerateZ
assertTrue)r r* �posr- � parentposr r r r! ; s zTestHeap.check_invariantc C sZ xBt td��dg D ],}dd� t|�D �}| jj|� | j|� qW | jt| jjd � d S )N� i N c S s g | ] }t j � �qS r )r# )�.0Zdummyr r r �
<listcomp>D s z)TestHeap.test_heapify.<locals>.<listcomp>)�listr" r% r r! r' r( )r �sizer* r r r � test_heapifyB s
zTestHeap.test_heapifyc C sn dd� t d�D �}g }x2|D ]*}| jj||� t|�dkr | jj|� q W |j� | j|t|�dd � � d S )Nc S s g | ]}t jd ��qS )i� )r# � randrange)r5 r, r r r r6 K s z-TestHeap.test_naive_nbest.<locals>.<listcomp>i� �
i����)r" r% r
�lenr r& r �sorted)r r+ r* r- r r r �test_naive_nbestJ s
zTestHeap.test_naive_nbestc c s2 yx| j j|�V qW W n tk
r, Y nX d S )N)r% r �
IndexError)r r* r r r �heapiterT s
zTestHeap.heapiterc C s� dd� t d�D �}|d d� }| jj|� x.|dd � D ] }||d kr8| jj||� q8W | jt| j|��t|�dd � � | jt | jjd � | jt | jjd d � | jt
| jjg d � d S )Nc S s g | ]}t jd ��qS )i� )r# r: )r5 r, r r r r6 b s z'TestHeap.test_nbest.<locals>.<listcomp>i� r; r i����) r" r% r r r r7 r@ r= r' r( r? )r r+ r* r- r r r �
test_nbest\ s "zTestHeap.test_nbestc C s� dd� t d�D �}|d d� }| jj|� x"|dd � D ]}| jj||� q8W | jt| j|��t|�dd � � | j| jjg d�d� d S )Nc S s g | ]}t jd ��qS )i� )r# r: )r5 r, r r r r6 o s z4TestHeap.test_nbest_with_pushpop.<locals>.<listcomp>i� r; �xi����)r" r% r r r r7 r@ r= )r r+ r* r- r r r �test_nbest_with_pushpopn s "z TestHeap.test_nbest_with_pushpopc C s� g }| j j|d�}| j||fg df� dg}| j j|d�}| j||fdgdf� | jt|d �t� | jt|�t� dg}| j j|d�}| j||fdgdf� dg}| j j|d�}| j||fdgdf� d S )Nr; g $@r � � )r% r r �type�int�float)r �hrB r r r �test_heappushpopw s zTestHeap.test_heappushpopc s� x�t d�D ]�}tjd�}dd� t |�D �}|d@ rL|d d � � �jj� � n g � x|D ]}�jj� |� qVW � �fdd�t |�D �}�j|t|�� q
W d S )N�d �2 c S s g | ]}t jd ��qS )� )r# r: )r5 r, r r r r6 � s z*TestHeap.test_heapsort.<locals>.<listcomp>r0 c s g | ]}�j j� ��qS r )r% r )r5 r, )r* r r r r6 � s )r" r# r: r% r r
r r= )r Ztrialr8 r+ r- Z heap_sortedr )r* r r �
test_heapsort� s
zTestHeap.test_heapsortc
C s g }x\t tjd��D ]J}g }x6t tjd��D ]$}tjd�tjd
d�f}|j|� q,W |j|� qW x�d td�td�tdd�gD ]|}xvd D ]n}g }x |D ]} |jt| ||d �� q�W | jtt|� ||d �t | j
j |||d ���� | jt | j
j � �g � q�W q~W d S ) NrM rK �ABCi� r r0 FT)�key�reversei ���)FT) r" r# r: Zchoicer$ r r= r �chainr7 r% �merge)
r �inputsr, �row�j�tuprP rQ Zseqs�seqr r r �
test_merge� s
zTestHeap.test_mergec C s8 dd� }| j t�� t| jj|� |� �� W d Q R X d S )Nc s s, t td��} xtd�D ]}| | V qW d S )Nr; � )r7 r" )�sr, r r r �iterable� s zCTestHeap.test_merge_does_not_suppress_index_error.<locals>.iterable)r' r? r7 r% rS )r r\ r r r �(test_merge_does_not_suppress_index_error� s z1TestHeap.test_merge_does_not_suppress_index_errorc C s� G dd� dt �}g g g g g}xDtd�D ]8}tjd�}tjd�}||�}||f|_|| j|� q&W x|D ] }|j� qhW dd� | jj|� D �}| j |t
|�� d S )Nc @ s e Zd ZdS )z*TestHeap.test_merge_stability.<locals>.IntN)r r r r r r r �Int� s r^ i N � i� c S s g | ]
}|j �qS r )�pair)r5 r, r r r r6 � s z1TestHeap.test_merge_stability.<locals>.<listcomp>) rG r" r# r: r` r$ r& r% rS r r= )r r^ rT r, �streamrB �obj�resultr r r �test_merge_stability� s
z TestHeap.test_merge_stabilityc
C s� dd� t d�D �}xvd dd� fD ]f}x`dD ]X}| jt| jj||��t|�d |� � | jt| jj|||d��t||d�d |� � q*W q W d S )Nc S s g | ]}t jd �|f�qS )i� )r# r: )r5 r, r r r r6 � s z+TestHeap.test_nsmallest.<locals>.<listcomp>�� c S s | d d d S )Nr i# i� r )rB r r r �<lambda>� s z)TestHeap.test_nsmallest.<locals>.<lambda>r r0 � r; rK � �� �L )rP ) r r0 rg r; rK rh ri re rj )r" r r7 r% � nsmallestr= )r r+ �f�nr r r �test_nsmallest� s
zTestHeap.test_nsmallestc
C s� dd� t d�D �}x|d dd� fD ]l}xfdD ]^}| jt| jj||��t|dd�d |� � | jt| jj|||d��t||dd�d |� � q*W q W d S )Nc S s g | ]}t jd �|f�qS )i� )r# r: )r5 r, r r r r6 � s z*TestHeap.test_nlargest.<locals>.<listcomp>�� c S s | d d d S )Nr i# i� r )rB r r r rf � s z(TestHeap.test_nlargest.<locals>.<lambda>r r0 rg r; rK � �� �L T)rQ )rP )rP rQ ) r r0 rg r; rK rp rq ro rr )r" r r7 r% �nlargestr= )r r+ rl rm r r r �
test_nlargest� s
zTestHeap.test_nlargestc sj � fdd�}G dd� d�}G dd� d�}dd� t d �D �}t|d
d �}� j|||�|� � jt||� d S ) Nc s: � fdd��D ���j j�� ��fdd�tt���D �S )Nc s g | ] }� |��qS r r )r5 rB )�compr r r6 � s zDTestHeap.test_comparison_operator.<locals>.hsort.<locals>.<listcomp>c s g | ]}�j j� �j�qS r )r% r rB )r5 r, )r+ r r r r6 � s )r% r r" r< )r+ ru )r )ru r+ r �hsort� s z0TestHeap.test_comparison_operator.<locals>.hsortc @ s e Zd Zdd� Zdd� ZdS )z-TestHeap.test_comparison_operator.<locals>.LTc S s
|| _ d S )N)rB )r rB r r r �__init__� s z6TestHeap.test_comparison_operator.<locals>.LT.__init__c S s | j |j kS )N)rB )r �otherr r r �__lt__� s z4TestHeap.test_comparison_operator.<locals>.LT.__lt__N)r r r rw ry r r r r �LT� s rz c @ s e Zd Zdd� Zdd� ZdS )z-TestHeap.test_comparison_operator.<locals>.LEc S s
|| _ d S )N)rB )r rB r r r rw � s z6TestHeap.test_comparison_operator.<locals>.LE.__init__c S s | j |j kS )N)rB )r rx r r r �__le__� s z4TestHeap.test_comparison_operator.<locals>.LE.__le__N)r r r rw r{ r r r r �LE� s r| c S s g | ] }t j � �qS r )r# )r5 r, r r r r6 � s z5TestHeap.test_comparison_operator.<locals>.<listcomp>rK T)rQ )r" r= r r' r( )r rv rz r| r+ �targetr )r r �test_comparison_operator� s z!TestHeap.test_comparison_operatorN)r r r r/ r! r9 r> r@ rA rC rJ rN rY r] rd rn rt r~ r r r r r s
r c @ s e Zd ZeZdS )�TestHeapPythonN)r r r r r% r r r r r � s r zrequires _heapqc @ s e Zd ZeZdS )� TestHeapCN)r r r r r% r r r r r� � s r� c @ s e Zd ZdZdd� ZdS )�LenOnlyz:Dummy sequence class defining __len__ but not __getitem__.c C s dS )Nr; r )r r r r �__len__� s zLenOnly.__len__N)r r r �__doc__r� r r r r r� � s r� c @ s e Zd ZdZdd� ZdS )�GetOnlyz:Dummy sequence class defining __getitem__ but not __len__.c C s dS )Nr; r )r Zndxr r r � __getitem__� s zGetOnly.__getitem__N)r r r r� r� r r r r r� � s r� c @ s, e Zd ZdZdd� Ze Z Z Z ZZ dS )�CmpErrz;Dummy element that always raises an error during comparisonc C s t �d S )N)�ZeroDivisionError)r rx r r r �__eq__ s z
CmpErr.__eq__N)
r r r r� r� �__ne__ry r{ �__gt__�__ge__r r r r r� s r� c c s x| D ]
}|V qW dS )zRegular generatorNr )�seqnr, r r r �R s
r� c @ s e Zd ZdZdd� Zdd� ZdS )�GzSequence using __getitem__c C s
|| _ d S )N)r� )r r� r r r rw s z
G.__init__c C s
| j | S )N)r� )r r, r r r r� s z
G.__getitem__N)r r r r� rw r� r r r r r� s r� c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �Iz Sequence using iterator protocolc C s || _ d| _d S )Nr )r� r, )r r� r r r rw s z
I.__init__c C s | S )Nr )r r r r �__iter__ s z
I.__iter__c C s2 | j t| j�krt�| j| j }| j d7 _ |S )Nr0 )r, r< r� �
StopIteration)r �vr r r �__next__ s
z
I.__next__N)r r r r� rw r� r� r r r r r� s r� c @ s e Zd ZdZdd� Zdd� ZdS )�Igz9Sequence using iterator protocol defined with a generatorc C s || _ d| _d S )Nr )r� r, )r r� r r r rw " s z Ig.__init__c c s x| j D ]
}|V qW d S )N)r� )r �valr r r r� % s z Ig.__iter__N)r r r r� rw r� r r r r r� s r� c @ s e Zd ZdZdd� Zdd� ZdS )�Xz Missing __getitem__ and __iter__c C s || _ d| _d S )Nr )r� r, )r r� r r r rw + s z
X.__init__c C s2 | j t| j�krt�| j| j }| j d7 _ |S )Nr0 )r, r< r� r� )r r� r r r r� . s
z
X.__next__N)r r r r� rw r� r r r r r� ) s r� c @ s e Zd ZdZdd� Zdd� ZdS )�NzIterator missing __next__()c C s || _ d| _d S )Nr )r� r, )r r� r r r rw 6 s z
N.__init__c C s | S )Nr )r r r r r� 9 s z
N.__iter__N)r r r r� rw r� r r r r r� 4 s r� c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �Ez Test propagation of exceptionsc C s || _ d| _d S )Nr )r� r, )r r� r r r rw > s z
E.__init__c C s | S )Nr )r r r r r� A s z
E.__iter__c C s dd d S )N� r r )r r r r r� C s z
E.__next__N)r r r r� rw r� r� r r r r r� < s r� c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �SzTest immediate stopc C s d S )Nr )r r� r r r rw H s z
S.__init__c C s | S )Nr )r r r r r� J s z
S.__iter__c C s t �d S )N)r� )r r r r r� L s z
S.__next__N)r r r r� rw r� r� r r r r r� F s r� )rR c C s t tdd� ttt| �����S )z Test multiple tiers of iteratorsc S s | S )Nr )rB r r r rf R s zL.<locals>.<lambda>)rR �mapr� r� r� )r� r r r �LP s r� c @ s e Zd Zdd� Zdd� ZdS )� SideEffectLTc C s || _ || _d S )N)�valuer* )r r� r* r r r rw V s zSideEffectLT.__init__c C s g | j d d �< | j|jk S )N)r* r� )r rx r r r ry Z s zSideEffectLT.__lt__N)r r r rw ry r r r r r� U s r� c @ s\ e Zd Zdd� Zdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Zdd� Z dd� Z
dd� Z dS )�TestErrorHandlingc C sj x*| j j| j jfD ]}| jttf|d� qW x8| j j| j j| j j| j j fD ]}| jttf|dd� qJW d S )Nr; )
r% r r r' r( r) r
r rs rk )r rl r r r �test_non_sequencea s
z#TestErrorHandling.test_non_sequencec C s� x,| j j| j jfD ]}| jttf|t� � qW x.| j j| j jfD ]}| jttf|t� d� q@W x*| j j | j j
fD ]}| jt|dt� � qpW d S )Nr; rg ) r% r r r' r( r) r� r
r rs rk )r rl r r r �
test_len_onlyh s z TestErrorHandling.test_len_onlyc C s� x(| j j| j jfD ]}| jt|t� � qW x*| j j| j jfD ]}| jt|t� d� q<W x*| j j| j j fD ]}| jt|dt� � qhW d S )Nr; rg )
r% r r r' r( r� r
r rs rk )r rl r r r �
test_get_onlyp s z TestErrorHandling.test_get_onlyc C s� t � t � t � g}x&| jj| jjfD ]}| jt||� q"W x(| jj| jjfD ]}| jt||d� qJW x(| jj| jj fD ]}| jt|d|� qtW d S )Nr; rg )
r� r% r r r' r� r
r rs rk )r rX rl r r r r� x s c C sH xB| j j| j j| j j| j j| j j| j jfD ]}| jtt f|d� q*W d S )Nr; )
r% r r r
r rs rk r' r( r) )r rl r r r �test_arg_parsing� s z"TestErrorHandling.test_arg_parsingc C s� x�| j j| j jfD ]�}x�ddtd�d
tddd�fD ]�}x:tttttfD ](}| j t
|d ||���t
|d |��� qFW | j t
|d t |���g � | j t
|d t|�� | j t
|d t|�� | j t|d t|�� q2W qW d S ) NZ123� i� r0 �333333�?i� i� � rg )r0 r� )r% rs rk r" r� r� r� r� r� r r7 r� r' r( r� r� r� r� )r rl r[ �gr r r �test_iterable_args� s (z$TestErrorHandling.test_iterable_argsc sR g � � j � fdd�td�D �� | jttf�� | jj� td� �� W d Q R X d S )Nc 3 s | ]}t |� �V qd S )N)r� )r5 r, )r* r r � <genexpr>� s z@TestErrorHandling.test_heappush_mutating_heap.<locals>.<genexpr>�� r� )�extendr" r' r? � RuntimeErrorr% r
r� )r r )r* r �test_heappush_mutating_heap� s z-TestErrorHandling.test_heappush_mutating_heapc sJ g � � j � fdd�td�D �� | jttf�� | jj� � W d Q R X d S )Nc 3 s | ]}t |� �V qd S )N)r� )r5 r, )r* r r r� � s z?TestErrorHandling.test_heappop_mutating_heap.<locals>.<genexpr>r� )r� r" r' r? r� r% r )r r )r* r �test_heappop_mutating_heap� s z,TestErrorHandling.test_heappop_mutating_heapc sB G � fdd�dt �}g � | jj� |d�� | jt| jj� d� d S )Nc s e Zd Z� fdd�ZdS )zMTestErrorHandling.test_comparison_operator_modifiying_heap.<locals>.EvilClassc s � j � tS )N)�clear�NotImplemented)r �o)r* r r ry � s zTTestErrorHandling.test_comparison_operator_modifiying_heap.<locals>.EvilClass.__lt__N)r r r ry r )r* r r � EvilClass� s r� r r0 )rG r% r
r' r? r )r r� r )r* r �(test_comparison_operator_modifiying_heap� s z:TestErrorHandling.test_comparison_operator_modifiying_heapc s� G �fdd�dt �}G � fdd�dt �}g g � �| jj� |d�� | jj�|d�� | jttf| jj� |d�� | jttf| jj�|d�� d S )Nc s e Zd Z� fdd�ZdS )zOTestErrorHandling.test_comparison_operator_modifiying_heap_two_heaps.<locals>.hc s � j � tS )N)r� r� )r r� )�list2r r ry � s zVTestErrorHandling.test_comparison_operator_modifiying_heap_two_heaps.<locals>.h.__lt__N)r r r ry r )r� r r rI � s rI c s e Zd Z� fdd�ZdS )zOTestErrorHandling.test_comparison_operator_modifiying_heap_two_heaps.<locals>.gc s � j � tS )N)r� r� )r r� )�list1r r ry � s zVTestErrorHandling.test_comparison_operator_modifiying_heap_two_heaps.<locals>.g.__lt__N)r r r ry r )r� r r r� � s r� r r0 )rG r% r
r' r? r� )r rI r� r )r� r� r �2test_comparison_operator_modifiying_heap_two_heaps� s
zDTestErrorHandling.test_comparison_operator_modifiying_heap_two_heapsN) r r r r� r� r� r� r� r� r� r� r� r r r r r� _ s r� c @ s e Zd ZeZdS )�TestErrorHandlingPythonN)r r r r r% r r r r r� � s r� c @ s e Zd ZeZdS )�TestErrorHandlingCN)r r r r r% r r r r r� � s r� �__main__)%r� r# ZunittestZtestr r r �operatorr Zimport_fresh_moduler r r r r r r� r� r� r� r� r� r� r� r� r� r� r� � itertoolsrR r� r� r� r� r� r �mainr r r r �<module> sF
P
b