Your IP : 216.73.216.196
o
0�i�A � @ s� d Z ddlZddlZddlZddlmZ ddlmZ ddlmZm Z ddl
m Z ej ddgd �Z
ej ddgd
�Zg d �ZG d d
� d
e�Zdd� ZG dd� d�ZG dd� dee�Ze ed�G dd� dee��ZG dd� d�ZG dd� d�Zdd � ZG d d � d �ZG d d � d �ZG d!d"� d"�ZG d#d$� d$�ZG d%d&� d&�Z G d'd(� d(�Z G d)d*� d*�Z dd+l m Z d,d-� Z!G d.d/� d/�Z"G d0d1� d1�Z#G d2d3� d3e#e�Z$e ed�G d4d5� d5e#e��Z%e&d6kr�e�'� dS dS )7zUnittests for heapq.� N)�support)�
import_helper)�TestCase�
skipUnless)�
itemgetter�heapq�_heapq)Zblocked)Zfresh)�heapify�heappop�heappush� heappushpop� heapreplace� _heappop_maxZ_heapreplace_maxZ _heapify_maxc @ s& e Zd Zdd� Zeed�dd� �ZdS )� TestModulesc C �" t D ] }| �tt|�jd� qd S )Nr )�
func_names� assertEqual�getattr�py_heapq�
__module__��selfZfname� r �6/opt/alt/python310/lib64/python3.10/test/test_heapq.py�test_py_functions s �z TestModules.test_py_functions�requires _heapqc C r )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�||� | � |� q
g }|r;| j�|�}| � |� |�|� |s)|d d � }|�� | �||� | � |� | � t
| jjg � z| � t
| jjd d � | � t
| jjd � W d S t yy Y d S w )N� ) �check_invariant�range�random�append�moduler r
�sortr � assertRaises� TypeError�AttributeError)r �heap�data�i�itemZresultsZ data_sortedr r r �
test_push_pop2 s2
�
�zTestHeap.test_push_popc C s8 t |�D ]\}}|r|d d? }| �|| |k� qd S �N� )� enumerateZ
assertTrue)r r5 �posr8 Z parentposr r r r, N s ��zTestHeap.check_invariantc C sV t td��dg D ]}dd� t|�D �}| j�|� | �|� q | �t| jjd � d S )N� � N c S � g | ]}t � � �qS r �r. )�.0Zdummyr r r �
<listcomp>W � z)TestHeap.test_heapify.<locals>.<listcomp>)�listr- r0 r r, r2 r3 )r �sizer5 r r r � test_heapifyU s
zTestHeap.test_heapifyc C sj dd� t d�D �}g }|D ]}| j�||� t|�dkr"| j�|� q
|�� | �|t|�dd � � d S )Nc S � g | ]}t �d ��qS ��� �r. � randrange�rB r7 r r r rC ^ � 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 q ty Y d S w r )r0 r
�
IndexError)r r5 r r r �heapiterg s �� �zTestHeap.heapiterc C s� dd� t d�D �}|d d� }| j�|� |dd � D ]}||d kr*| j�||� q| �t| �|��t|�dd � � | �t | jjd � | �t | jjd d � | �t
| jjg d � d S )Nc S rH rI rK rM r r r rC u rN z'TestHeap.test_nbest.<locals>.<listcomp>rO rP r rQ ) r- r0 r r
r rE rW rS r2 r3 rV rT r r r �
test_nbesto s �"zTestHeap.test_nbestc C s� dd� t d�D �}|d d� }| j�|� |dd � D ] }| j�||� q| �t| �|��t|�dd � � | �| j�g d�d� d S )Nc S rH rI rK rM r r r rC � rN z4TestHeap.test_nbest_with_pushpop.<locals>.<listcomp>rO rP rQ �x)r- r0 r r r rE rW rS rT 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 )NrP g $@r � � )r0 r r �type�int�float)r �hrY 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 ]C}t�d�}dd� t |�D �}|d@ r%|d d � � �j�� � ng � |D ] }�j�� |� q)� �fdd�t |�D �}��|t|�� qd S )N�d �2 c S rH )� rK rM r r r rC � rN z*TestHeap.test_heapsort.<locals>.<listcomp>r; c s g | ]}�j �� ��qS r )r0 r
rM �r5 r r r rC � s )r- r. rL r0 r r r rS )r ZtrialrF r6 r8 Z heap_sortedr rh r �
test_heapsort� s
�zTestHeap.test_heapsortc
C s� g }t t�d��D ]#}g }t t�d��D ]}t�d�t�dd�f}|�|� q|�|� q d td�td�tdd�fD ]:}dD ]5}g }|D ] } |�t| ||d �� qD| �tt|� ||d �t | j
j |||d ���� | �t | j
� � �g � q>q:d S )
Nrg re �ABCi ����� r r; )FT��key�reverse) r- r. rL Zchoicer/ r rS r �chainrE r0 r# )
r �inputsr7 �row�j�tuprm rn 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 � dS )N� r r r r r �<lambda>� � z,TestHeap.test_empty_merges.<locals>.<lambda>�rm )r rE r0 r# �r r r r �test_empty_merges� s &zTestHeap.test_empty_mergesc C sN dd� }| � t�� t| j�|� |� �� W d � d S 1 s w Y d S )Nc s s* � t td��} td�D ]}| | V q d S )NrP � )rE r- )�sr7 r r r �iterable� s
� �zCTestHeap.test_merge_does_not_suppress_index_error.<locals>.iterable)r2 rV 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 ] }t�d�}t�d�}||�}||f|_|| �|� q|D ]}|�� q1dd� | 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? � rk c S s g | ]}|j �qS r )�pairrM r r r rC � s z1TestHeap.test_merge_stability.<locals>.<listcomp>) r^ r- r. rL r� r/ r1 r0 r# r rS )r r� rp r7 �streamrY �obj�resultr r r �test_merge_stability� s
z TestHeap.test_merge_stabilityc
C s� dd� t d�D �}d dd� fD ]1}dD ],}| �t| j�||��t|�d |� � | �t| jj|||d��t||d�d |� � qqd S )Nc S � g | ] }t �d �|f�qS rI rK rM r r r rC � � z+TestHeap.test_nsmallest.<locals>.<listcomp>rO c S � | d d d S �Nr i# rJ r �rY r r r rx � � z)TestHeap.test_nsmallest.<locals>.<lambda>� r r; rc rP re i� i� rO iL rz )r- r rE r0 � nsmallestrS �r r6 �f�nr r r �test_nsmallest� s ����zTestHeap.test_nsmallestc
C s� dd� t d�D �}d dd� fD ]4}dD ]/}| �t| j�||��t|dd�d |� � | �t| jj|||d ��t||dd
�d |� � qqd S ) Nc S r� rI rK rM r r r rC � r� z*TestHeap.test_nlargest.<locals>.<listcomp>rO c S r� r� r r� r r r rx � r� z(TestHeap.test_nlargest.<locals>.<lambda>r� T�rn rz rl )r- r rE r0 �nlargestrS 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 rY )�compr r rC � rD zDTestHeap.test_comparison_operator.<locals>.hsort.<locals>.<listcomp>c s g | ] }�j �� �j�qS r )r0 r
rY rM )r6 r r r rC � r� )r0 r r- rR )r6 r� r{ )r� r6 r �hsort� s z0TestHeap.test_comparison_operator.<locals>.hsortc @ � e Zd Zdd� Zdd� ZdS )z-TestHeap.test_comparison_operator.<locals>.LTc S �
|| _ d S r r� �r rY r r r �__init__� �
z6TestHeap.test_comparison_operator.<locals>.LT.__init__c S s | j |j kS r r� �r �otherr r r �__lt__� � z4TestHeap.test_comparison_operator.<locals>.LT.__lt__N�r r r r� r� r r r r �LT� � r� c @ r� )z-TestHeap.test_comparison_operator.<locals>.LEc S r� r r� r� r r r r� r� z6TestHeap.test_comparison_operator.<locals>.LE.__init__c S s | j |j kS r r� r� r r r �__le__ r� z4TestHeap.test_comparison_operator.<locals>.LE.__le__N)r r r r� r� r r r r �LE r� r� c S r@ r rA rM r r r rC rD z5TestHeap.test_comparison_operator.<locals>.<listcomp>re Tr� )r- rS r r2 r3 )r r� r� r� r6 �targetr r{ r �test_comparison_operator� s z!TestHeap.test_comparison_operatorN)r r r r9 r, rG rU rW rX rZ ra rd ri ru r| r� r� r� r� r� r r r r r* 0 s$
r* c @ � e Zd ZeZdS )�TestHeapPythonN�r r r r r0 r r r r r� � r� r c @ r� )� TestHeapCN�r r r r r0 r r r r r� � r� c @ s e Zd ZdZdd� ZdS )�LenOnlyz:Dummy sequence class defining __len__ but not __getitem__.c C rv �NrP r r{ r r r �__len__ � zLenOnly.__len__N)r r r �__doc__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 � t �r )�ZeroDivisionErrorr� r r r �__eq__ r� z
CmpErr.__eq__N)
r r r r� r� �__ne__r� r� �__gt__�__ge__r r r r r� s r� c c s � | D ]}|V qdS )zRegular generatorNr ��seqnr7 r r r �R" s ��r� c @ � e Zd ZdZdd� Zdd� ZdS )�GzSequence using __getitem__c C r� r �r� �r r� r r r r� ) r� z
G.__init__c C s
| j | S r r� )r r7 r r r � __getitem__+ r� z
G.__getitem__N)r r r r� r� r� r r r r r� ' s r� c @ �( e Zd ZdZdd� Zdd� Zdd� ZdS ) �Iz Sequence using iterator protocolc C � || _ d| _d S �Nr r� r� r r r r� 0 �
z
I.__init__c C � | S r r r{ r r r �__iter__3 r� z
I.__iter__c C �2 | j t| j�kr
t�| j| j }| j d7 _ |S r: �r7 rR r� �
StopIteration�r �vr r r �__next__5 � z
I.__next__N�r r r r� r� r� r� r r r r r� . �
r� c @ r� )�Igz9Sequence using iterator protocol defined with a generatorc C r� r� r� r� r r r r� = r� 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 r� ; � r� c @ r� )�Xz Missing __getitem__ and __iter__c C r� r� r� r� r r r r� F r� z
X.__init__c C r� r: r� r� r r r r� I r� z
X.__next__N)r r r r� r� r� r r r r r� D r� r� c @ r� )�NzIterator missing __next__()c C r� r� r� r� r r r r� Q r� z
N.__init__c C r� r r r{ r r r r� T r� z
N.__iter__Nr� r r r r r� O r� r� c @ r� ) �Ez Test propagation of exceptionsc C r� r� r� r� r r r r� Y r� z
E.__init__c C r� r r r{ r r r r� \ r� z
E.__iter__c C s dd d S )Nrb r r r{ r r r r� ^ r� z
E.__next__Nr� r r r r r� W r� r� c @ r� ) �SzTest immediate stopc C s d S r r r� r r r r� c r� z
S.__init__c C r� r r r{ r r r r� e r� z
S.__iter__c C r� r )r� r{ r r r r� g r� z
S.__next__Nr� r r r r r� a s
r� )ro c C s t tdd� ttt| �����S )z Test multiple tiers of iteratorsc S r� r r r� r r r rx m ry zL.<locals>.<lambda>)ro �mapr� r� r� r� r r r �Lk s r� c @ r� )� SideEffectLTc C s || _ || _d S r )�valuer5 )r r� r5 r r r r� q r� zSideEffectLT.__init__c C s g | j d d �< | j|jk S r )r5 r� r� r r r r� u s zSideEffectLT.__lt__Nr� r r r r r� p 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� q"d 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� � q3d S �NrP rc ) 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� q"| jj| jj fD ]
}| �t|d|� q5d 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� qd 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 ]Y}ddtd�dtddd�fD ]I}tttttfD ]}| � t
|d||���t
|d|��� q | � t
|dt |���g � | � t
|dt|�� | � t
|dt|�� | � t|dt|�� qqd S ) NZ123� rO )r; g333333�?rJ i� � rc )r0 r� r� r- r� r� r� r� r� r rE r� r2 r3 r� r� r� r� )r r� r~ �gr r r �test_iterable_args� s &��z$TestErrorHandling.test_iterable_argsc sh g � � � � fdd�td�D �� | �ttf�� | j�� td� �� W d � d S 1 s-w Y d S )Nc 3 � � | ]}t |� �V qd S r �r� rM �r5 r r � <genexpr>� � � z@TestErrorHandling.test_heappush_mutating_heap.<locals>.<genexpr>�� r� )�extendr- r2 rV � RuntimeErrorr0 r r� r{ r r� r �test_heappush_mutating_heap� s
"�z-TestErrorHandling.test_heappush_mutating_heapc s` g � � � � fdd�td�D �� | �ttf�� | j�� � W d � d S 1 s)w Y d S )Nc 3 r� r r� rM r� r r r� � r� z?TestErrorHandling.test_heappop_mutating_heap.<locals>.<genexpr>r� )r� r- r2 rV r� r0 r
r{ 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 � e Zd Z� fdd�ZdS )zMTestErrorHandling.test_comparison_operator_modifiying_heap.<locals>.EvilClassc � � � � tS r ��clear�NotImplemented�r �or� r r r� � � zTTestErrorHandling.test_comparison_operator_modifiying_heap.<locals>.EvilClass.__lt__N�r r r r� r r� r r � EvilClass� � r� r r; )r^ r0 r r2 rV 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 r� )zOTestErrorHandling.test_comparison_operator_modifiying_heap_two_heaps.<locals>.hc r� r r� r� ��list2r r r� � r� zVTestErrorHandling.test_comparison_operator_modifiying_heap_two_heaps.<locals>.h.__lt__Nr� r r r r r` � r� r` c r� )zOTestErrorHandling.test_comparison_operator_modifiying_heap_two_heaps.<locals>.gc r� r r� r� ��list1r r r� � r� zVTestErrorHandling.test_comparison_operator_modifiying_heap_two_heaps.<locals>.g.__lt__Nr� r r r r r� � r� r� r r; )r^ r0 r r2 rV 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� z s r� c @ r� )�TestErrorHandlingPythonNr� r r r r r � r� r c @ r� )�TestErrorHandlingCNr� r r r r r � r� r �__main__)(r� r. Zunittestr! �testr Z test.supportr 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 �