Your IP : 216.73.216.213
a
XC?h�A � @ s� d dl Z d dlZd dlZd dlmZ d dlmZmZ d dlmZ ej ddgd�Z
ej ddgd�Z g d �Z G d
d � d e�Z
d d
� ZG dd� d�ZG dd� dee�Zee d�G dd� dee��ZG dd� d�ZG dd� d�Zdd� ZG dd � 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 d)l m Z d*d+� Z G d,d-� d-�Z G d.d/� d/�Z G d0d1� d1e e�Z!ee d�G d2d3� d3e e��Z"e#d4k�r�e�$� dS )5� 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 �4/opt/alt/python39/lib64/python3.9/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 y� Y n0 d S )N� ) �check_invariant�range�random�append�moduler
r �sortr � assertRaises� TypeError�AttributeError)r �heap�data�i�itemZresultsZ 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. )�.0Zdummyr r r �
<listcomp>V � 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�rB r7 r r r rC ] rD 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 ty& Y n0 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 rH rJ rL r r r rC t rD z'TestHeap.test_nbest.<locals>.<listcomp>rM rN r rO ) r- r0 r r r rE rU rQ r2 r3 rT rR 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 rH rJ rL r r r rC � rD z4TestHeap.test_nbest_with_pushpop.<locals>.<listcomp>rM rN rO �x)r- r0 r r r rE rU rQ rR 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 )NrN g $@r � � )r0 r r �type�int�float)r �hrW 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 )� rJ rL r r r rC � rD z*TestHeap.test_heapsort.<locals>.<listcomp>r; c s g | ]}�j �� ��qS r )r0 r rL �r5 r r r rC � rD )r- r. rK r0 r r
r rQ )r ZtrialrF r6 r8 Z heap_sortedr rf 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 )
Nre rc �ABCi ����� r r; )FT��key�reverse) r- r. rK Zchoicer/ r rQ r �chainrE r0 r# )
r �inputsr7 �row�j�tuprk rl 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>� rD z,TestHeap.test_empty_merges.<locals>.<lambda>�rk )r rE r0 r# �r r r r �test_empty_merges� s zTestHeap.test_empty_mergesc C sL dd� }| � t��& t| j�|� |� �� W d � n1 s>0 Y d S )Nc s s( t td��} td�D ]}| | V qd S )NrN � )rE r- )�sr7 r r r �iterable� s zCTestHeap.test_merge_does_not_suppress_index_error.<locals>.iterable)r2 rT rE 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} r@ � ri c S s g | ]
}|j �qS r )�pairrL r r r rC � rD z1TestHeap.test_merge_stability.<locals>.<listcomp>) r\ r- r. rK r r/ r1 r0 r# r rQ )r r} rn r7 �streamrW �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 rH rJ rL r r r rC � rD z+TestHeap.test_nsmallest.<locals>.<listcomp>rM c S s | d d d S �Nr i# rI r �rW r r r ru � rD z)TestHeap.test_nsmallest.<locals>.<lambda>� r r; ra rN rc i� i� rM iL rv )r- r rE r0 � nsmallestrQ �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 rH rJ rL r r r rC � rD z*TestHeap.test_nlargest.<locals>.<listcomp>rM c S s | d d d S r� r r� r r r ru � rD z(TestHeap.test_nlargest.<locals>.<lambda>r� T�rl rv rj )r- r rE r0 �nlargestrQ 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 )rB rW )�compr r rC � rD zDTestHeap.test_comparison_operator.<locals>.hsort.<locals>.<listcomp>c s g | ]}�j �� �j�qS r )r0 r rW rL )r6 r r r rC � rD )r0 r r- rP )r6 r� rw )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 rW 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 rA rL r r r rC rD z5TestHeap.test_comparison_operator.<locals>.<listcomp>rc Tr� )r- rQ r r2 r3 )r r� r� r� r6 �targetr rw r �test_comparison_operator� s z!TestHeap.test_comparison_operatorN)r r r r9 r, rG rS rU rV rX r_ rb rg rs rx 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 �NrN r rw 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 rw r r r �__iter__2 s z
I.__iter__c C s2 | j t| j�krt�| j| j }| j d7 _ |S r: �r7 rP 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 rw 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 rw r r r r� [ s z
E.__iter__c C s dd d S )Nr` r r rw 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 rw r r r r� d s z
S.__iter__c C s t �d S r )r� rw r r r r� f s z
S.__next__Nr� r r r r r� ` s r� )rm c C s t tdd� ttt| �����S )Nc S s | S r r r� r r r ru l rD zL.<locals>.<lambda>)rm �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 �NrN ra ) 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 rr 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� rM )r; g333333�?rI i� � ra )r0 r� r� r- r� r� r� r� r� r rE r� r2 r3 r� r� r� r� )r r� rz �gr r r �test_iterable_args� s &z$TestErrorHandling.test_iterable_argsc sf g � � � � fdd�td�D �� | �ttf��$ | j�� td� �� W d � n1 sX0 Y d S )Nc 3 s | ]}t |� �V qd S r �r� rL �r5 r r � <genexpr>� rD z@TestErrorHandling.test_heappush_mutating_heap.<locals>.<genexpr>�� r� )�extendr- r2 rT � RuntimeErrorr0 r
r� rw r r� r �test_heappush_mutating_heap� s z-TestErrorHandling.test_heappush_mutating_heapc s^ g � � � � fdd�td�D �� | �ttf�� | j�� � W d � n1 sP0 Y d S )Nc 3 s | ]}t |� �V qd S r r� rL r� r r r� � rD z?TestErrorHandling.test_heappop_mutating_heap.<locals>.<genexpr>r� )r� r- r2 rT r� r0 r rw 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 rT 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 rT 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� � itertoolsrm r� r� r� r� r� r �mainr r r r �<module> sD ]
Z