File: //opt/cppython/lib/python3.8/test/__pycache__/test_heapq.cpython-38.opt-2.pyc
U
.D�g�A � @ s� d dl Z d dlZd dlZd dlmZ d dlmZmZ d dlmZ ej ddgd�Z
ej ddgd�Z d d
d d d
dddgZ G dd� de�Z
dd� ZG dd� d�ZG dd� dee�Zee d�G dd� dee��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.d/� d/�Zd d0l m Z d1d2� Z G d3d4� d4�Z G d5d6� d6�Z G d7d8� d8e e�Z!ee d�G d9d:� d:e e��Z"e#d;k�r�e�$� dS )<� 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" t D ]}| �tt|�jd� qd S )Nr )�
func_names� assertEqual�getattr�py_heapq�
__module__��selfZfname� r �./opt/cppython/lib/python3.8/test/test_heapq.py�test_py_functions s z TestModules.test_py_functions�requires _heapqc C s" t D ]}| �tt|�jd� qd S )Nr )r r r �c_heapqr 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 C s( G dd� d�}|� tjt|� d�� |S )Nc @ s e Zd Zdd� ZdS )z+load_tests.<locals>.HeapqMergeDocTestFinderc _ s t �� }|�tj�S �N)�doctestZ
DocTestFinder�findr �merge)r �args�kwargsZdtfr r r r" ' s z0load_tests.<locals>.HeapqMergeDocTestFinder.findN)r r r r" r r r r �HeapqMergeDocTestFinder& s r& )Z test_finder)ZaddTestsr! Z DocTestSuiter )�loaderZtests�ignorer& r r r �
load_tests 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 d � Zd!d"� Zd#S )$�TestHeapc C s� g }g }| � |� td�D ].}t�� }|�|� | j�||� | � |� qg }|rt| j�|�}| � |� |�|� qN|d d � }|�� | �||� | � |� | � t
| jjg � z*| � t
| jjd d � | � 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_pop1 s.
zTestHeap.test_push_popc C s8 t |�D ]*\}}|r|d d? }| �|| |k� qd S �N� )� enumerateZ
assertTrue)r r5 �posr8 � parentposr r r r, M s zTestHeap.check_invariantc C sV t td��dg D ],}dd� t|�D �}| j�|� | �|� q| �t| jjd � d S )N� � N c S s g | ] }t � � �qS r �r. )�.0�dummyr r r �
<listcomp>V s z)TestHeap.test_heapify.<locals>.<listcomp>)�listr- r0 r r, r2 r3 )r �sizer5 r r r � test_heapifyT s
zTestHeap.test_heapifyc C sj dd� t d�D �}g }|D ]*}| j�||� t|�dkr| j�|� q|�� | �|t|�dd � � d S )Nc S s g | ]}t �d ��qS ��� �r. � randrange�rC r7 r r r rE ] s z-TestHeap.test_naive_nbest.<locals>.<listcomp>�� �
���)r- r0 r
�lenr r1 r �sorted�r r6 r5 r8 r r r �test_naive_nbest\ s zTestHeap.test_naive_nbestc c s. z| j �|�V qW n tk
r( Y nX d S r )r0 r �
IndexError)r r5 r r r �heapiterf s zTestHeap.heapiterc C s� dd� t d�D �}|d d� }| j�|� |dd � D ] }||d kr6| j�||� q6| �t| �|��t|�dd � � | �t | jjd � | �t | jjd d � | �t
| jjg d � d S )Nc S s g | ]}t �d ��qS rI rK rM r r r rE t s z'TestHeap.test_nbest.<locals>.<listcomp>rN rO r rP ) r- r0 r r r rF rV rR r2 r3 rU rS r r r �
test_nbestn s "zTestHeap.test_nbestc C s� dd� t d�D �}|d d� }| j�|� |dd � D ]}| j�||� q6| �t| �|��t|�dd � � | �| j�g d�d� d S )Nc S s g | ]}t �d ��qS rI rK rM r r r rE � s z4TestHeap.test_nbest_with_pushpop.<locals>.<listcomp>rN rO rP �x)r- r0 r r r rF rV rR rS r r r �test_nbest_with_pushpop� s "z TestHeap.test_nbest_with_pushpopc C s� g }| j �|d�}| �||fg df� dg}| j �|d�}| �||fdgdf� | �t|d �t� | �t|�t� dg}| j �|d�}| �||fdgdf� dg}| j �|d�}| �||fdgdf� d S )NrO g $@r � � )r0 r r �type�int�float)r �hrX r r r �test_heappushpop� s zTestHeap.test_heappushpopc C s4 ddg}| � | j�|�d� | � | j�|�d� d S )N� � )r r0 r
)r r_ r r r �test_heappop_max� s zTestHeap.test_heappop_maxc s� t d�D ]�}t�d�}dd� t |�D �}|d@ rJ|d d � � �j�� � n g � |D ]}�j�� |� qR� �fdd�t |�D �}��|t|�� qd S )N�d �2 c S s g | ]}t �d ��qS )� rK rM r r r rE � s z*TestHeap.test_heapsort.<locals>.<listcomp>r< c s g | ]}�j �� ��qS r )r0 r rM �r5 r r r rE � s )r- r. rL r0 r r
r rR )r ZtrialrG r6 r8 Z heap_sortedr rg r �
test_heapsort� s
zTestHeap.test_heapsortc
C s� g }t t�d��D ]F}g }t t�d��D ]$}t�d�t�dd�f}|�|� q(|�|� qd td�td�tdd�fD ]t}dD ]j}g }|D ]} |�t| ||d �� q�| �tt|� ||d �t | j
j |||d ���� | �t | j
� � �g � q|qtd S )
Nrf rd �ABCi ����� r r< )FT��key�reverse) r- r. rL �choicer/ r rR r �chainrF r0 r# )
r �inputsr7 �row�j�tuprl rm Zseqs�seqr r r �
test_merge� s �zTestHeap.test_mergec C s@ | � t| j�g g ��g � | � t| jjg g dd� d��g � d S )Nc S s dS )N� r r r r r �<lambda>� � z,TestHeap.test_empty_merges.<locals>.<lambda>�rl )r rF r0 r# �r r r r �test_empty_merges� s zTestHeap.test_empty_mergesc C s8 dd� }| � t�� t| j�|� |� �� W 5 Q R X d S )Nc s s( t td��} td�D ]}| | V qd S )NrO � )rF r- )�sr7 r r r �iterable� s zCTestHeap.test_merge_does_not_suppress_index_error.<locals>.iterable)r2 rU rF r0 r# )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}td�D ]8}t�d�}t�d�}||�}||f|_|| �|� q$|D ] }|�� qbdd� | jj|� D �}| � |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� rA � rj c S s g | ]
}|j �qS r )�pairrM r r r rE � s z1TestHeap.test_merge_stability.<locals>.<listcomp>) r] r- r. rL r� r/ r1 r0 r# r rR )r r� rp r7 �streamrX �obj�resultr r r �test_merge_stability� s
z TestHeap.test_merge_stabilityc
C s� dd� t d�D �}d dd� fD ]b}dD ]X}| �t| j�||��t|�d |� � | �t| jj|||d��t||d�d |� � q&q d S )Nc S s g | ]}t �d �|f�qS rI rK rM r r r rE � s z+TestHeap.test_nsmallest.<locals>.<listcomp>rN c S s | d d d S �Nr i# rJ r �rX r r r rw � rx z)TestHeap.test_nsmallest.<locals>.<lambda>� r r< rb rO rd i� i� rN iL ry )r- r rF r0 � nsmallestrR �r r6 �f�nr r r �test_nsmallest� s ��zTestHeap.test_nsmallestc
C s� dd� t d�D �}d dd� fD ]h}dD ]^}| �t| j�||��t|dd�d |� � | �t| jj|||d ��t||dd
�d |� � q&q d S ) Nc S s g | ]}t �d �|f�qS rI rK rM r r r rE � s z*TestHeap.test_nlargest.<locals>.<listcomp>rN c S s | d d d S r� r r� r r r rw � rx z(TestHeap.test_nlargest.<locals>.<lambda>r� T�rm ry rk )r- r rF r0 �nlargestrR r� r r r �
test_nlargest� s ��zTestHeap.test_nlargestc sj � fdd�}G dd� d�}G dd� d�}dd� t d �D �}t|d
d �}� �|||�|� � �t||� d S ) Nc s: � fdd��D ���j ��� ��fdd�tt���D �S )Nc s g | ] }� |��qS r r )rC rX )�compr r rE � s zDTestHeap.test_comparison_operator.<locals>.hsort.<locals>.<listcomp>c s g | ]}�j �� �j�qS r )r0 r rX rM )r6 r r r rE � s )r0 r r- rQ )r6 r� rz )r� r6 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 r r� �r rX r r r �__init__� s z6TestHeap.test_comparison_operator.<locals>.LT.__init__c S s | j |j kS r r� �r �otherr r r �__lt__� s z4TestHeap.test_comparison_operator.<locals>.LT.__lt__N�r r r r� r� r r r r �LT� s r� c @ s e Zd Zdd� Zdd� ZdS )z-TestHeap.test_comparison_operator.<locals>.LEc S s
|| _ d S r r� r� r r r r� s z6TestHeap.test_comparison_operator.<locals>.LE.__init__c S s | j |j kS r r� r� r r r �__le__ s z4TestHeap.test_comparison_operator.<locals>.LE.__le__N)r r r r� r� r r r r �LE s r� c S s g | ] }t � � �qS r rB rM r r r rE s z5TestHeap.test_comparison_operator.<locals>.<listcomp>rd Tr� )r- rR r r2 r3 )r r� r� r� r6 �targetr rz r �test_comparison_operator� s z!TestHeap.test_comparison_operatorN)r r r r: r, rH rT rV rW rY r` rc rh ru r{ r r� r� r� r� r r r r r* / s"
r* c @ s e Zd ZeZdS )�TestHeapPythonN�r r r r r0 r r r r r� s r� r c @ s e Zd ZeZdS )� TestHeapCN�r r r r r0 r r r r r� s r� c @ s e Zd Zdd� ZdS )�LenOnlyc C s dS �NrO r rz r r r �__len__ s zLenOnly.__len__N)r r r r� r r r r r� s r� c @ s( e Zd Zdd� Ze Z Z Z ZZdS )�CmpErrc C s t �d S r )�ZeroDivisionErrorr� r r r �__eq__ s z
CmpErr.__eq__N) r r r r� �__ne__r� r� �__gt__�__ge__r r r r r� s r� c c s | D ]
}|V qd S r r ��seqnr7 r r r �R! s r� c @ s e Zd Zdd� Zdd� ZdS )�Gc C s
|| _ d S r �r� �r r� r r r r� ( s z
G.__init__c C s
| j | S r r� )r r7 r r r � __getitem__* s z
G.__getitem__N)r r r r� r� r r r r r� & s r� c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�Ic C s || _ d| _d S �Nr r� r� r r r r� / s z
I.__init__c C s | S r r rz r r r �__iter__2 s z
I.__iter__c C s2 | j t| j�krt�| j| j }| j d7 _ |S r; �r7 rQ r� �
StopIteration�r �vr r r �__next__4 s
z
I.__next__N�r r r r� r� r� r r r r r� - s r� c @ s e Zd Zdd� Zdd� ZdS )�Igc C s || _ d| _d S r� r� r� r r r r� < s z Ig.__init__c c s | j D ]
}|V qd S r r� )r �valr r r r� ? s
z Ig.__iter__N�r r r r� r� r r r r r� : s r� c @ s e Zd Zdd� Zdd� ZdS )�Xc C s || _ d| _d S r� r� r� r r r r� E s z
X.__init__c C s2 | j t| j�krt�| j| j }| j d7 _ |S r; r� r� r r r r� H s
z
X.__next__N)r r r r� r� r r r r r� C s r� c @ s e Zd Zdd� Zdd� ZdS )�Nc C s || _ d| _d S r� r� r� r r r r� P s z
N.__init__c C s | S r r rz r r r r� S s z
N.__iter__Nr� r r r r r� N s r� c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�Ec C s || _ d| _d S r� r� r� r r r r� X s z
E.__init__c C s | S r r rz r r r r� [ s z
E.__iter__c C s dd d S )Nra r r rz r r r r� ] s z
E.__next__Nr� r r r r r� V s r� c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�Sc C s d S r r r� r r r r� b s z
S.__init__c C s | S r r rz r r r r� d s z
S.__iter__c C s t �d S r )r� rz r r r r� f s z
S.__next__Nr� r r r r r� ` s r� )ro c C s t tdd� ttt| �����S )Nc S s | S r r r� r r r rw l rx zL.<locals>.<lambda>)ro �mapr� r� r� r� r r r �Lj s r� c @ s e Zd Zdd� Zdd� ZdS )� SideEffectLTc C s || _ || _d S r )�valuer5 )r r� r5 r r r r� p s zSideEffectLT.__init__c C s g | j d d �< | j|jk S r )r5 r� r� r r r r� t s zSideEffectLT.__lt__Nr� r r r r r� o s r� c @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dS )�TestErrorHandlingc C sb | j j| j jfD ]}| �ttf|d� q| j j| j j| j j| j j fD ]}| �ttf|dd� qDd S r� )
r0 r r r2 r3 r4 r
r r� r� �r r� r r r �test_non_sequence{ s �z#TestErrorHandling.test_non_sequencec C s� | j j| j jfD ]}| �ttf|t� � q| j j| j jfD ]}| �ttf|t� d� q:| j j | j j
fD ]}| �t|dt� � qfd S �NrO rb ) r0 r r r2 r3 r4 r� r
r r� r� r� r r r �
test_len_only� s z TestErrorHandling.test_len_onlyc C s� t � t � t � g}| jj| jjfD ]}| �t||� q | jj| jjfD ]}| �t||d� qD| jj| jj fD ]}| �t|d|� qjd S r� )
r� r0 r r r2 r� r
r r� r� )r rt r� r r r � test_cmp_err� s z TestErrorHandling.test_cmp_errc C sD | j j| j j| j j| j j| j j| j jfD ]}| �tt f|d� q(d S r� )
r0 r r r
r r� r� r2 r3 r4 r� r r r �test_arg_parsing� s �z"TestErrorHandling.test_arg_parsingc
C s� | j j| j jfD ]�}ddtd�dtddd�fD ]�}tttttfD ](}| � t
|d||���t
|d|��� q@| � t
|dt |���g � | � t
|dt|�� | � t
|dt|�� | � t|dt|�� q.qd S ) NZ123� rN )r< g333333�?rJ i� � rb )r0 r� r� r- r� r� r� r� r� r rF r� r2 r3 r� r� r� r� )r r� r} �gr r r �test_iterable_args� s &z$TestErrorHandling.test_iterable_argsc sR g � � � � fdd�td�D �� | �ttf�� | j�� td� �� W 5 Q R X d S )Nc 3 s | ]}t |� �V qd S r �r� rM �r5 r r � <genexpr>� s z@TestErrorHandling.test_heappush_mutating_heap.<locals>.<genexpr>�� r� )�extendr- r2 rU � RuntimeErrorr0 r
r� rz r r� r �test_heappush_mutating_heap� s z-TestErrorHandling.test_heappush_mutating_heapc sJ g � � � � fdd�td�D �� | �ttf�� | j�� � W 5 Q R X d S )Nc 3 s | ]}t |� �V qd S r r� rM r� r r r� � s z?TestErrorHandling.test_heappop_mutating_heap.<locals>.<genexpr>r� )r� r- r2 rU r� r0 r rz r r� r �test_heappop_mutating_heap� s z,TestErrorHandling.test_heappop_mutating_heapc sB G � fdd�dt �}g � | j�� |d�� | �t| jj� d� d S )Nc s e Zd Z� fdd�ZdS )zMTestErrorHandling.test_comparison_operator_modifiying_heap.<locals>.EvilClassc s � � � tS r ��clear�NotImplemented�r �or� r r r� � s zTTestErrorHandling.test_comparison_operator_modifiying_heap.<locals>.EvilClass.__lt__N�r r r r� r r� r r � EvilClass� s r� r r< )r] r0 r
r2 rU 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 � �| j�� |d�� | j��|d�� | �ttf| jj� |d�� | �ttf| jj�|d�� d S )Nc s e Zd Z� fdd�ZdS )zOTestErrorHandling.test_comparison_operator_modifiying_heap_two_heaps.<locals>.hc s � � � tS r r� r� ��list2r r r� � s zVTestErrorHandling.test_comparison_operator_modifiying_heap_two_heaps.<locals>.h.__lt__Nr� r r� r r r_ � s r_ c s e Zd Z� fdd�ZdS )zOTestErrorHandling.test_comparison_operator_modifiying_heap_two_heaps.<locals>.gc s � � � tS r r� r� ��list1r r r� � s zVTestErrorHandling.test_comparison_operator_modifiying_heap_two_heaps.<locals>.g.__lt__Nr� r r� r r r� � s r� r r< )r] r0 r
r2 rU r� )r r_ 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� y s r� c @ s e Zd ZeZdS )�TestErrorHandlingPythonNr� r r r r r� � s r� c @ s e Zd ZeZdS )�TestErrorHandlingCNr� r r r r r� � s r� �__main__)%r. Zunittestr! �testr r r �operatorr Zimport_fresh_moduler r r r r) r* r� r� r� r� r� r� r� r� r� r� r� r� � itertoolsro r� r� r� r� r� r �mainr r r r �<module> sL
� ]
Z