shell bypass 403
o
` b�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_max�_heapreplace_max� _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 �8/usr/local/python-3.10/lib/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�item�resultsZ 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 r7 �posr: � 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 �r0 )�.0Zdummyr r r �
<listcomp>W � z)TestHeap.test_heapify.<locals>.<listcomp>)�listr/ r2 r r. r4 r5 )r �sizer7 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 ��� �r0 � randrange�rF r9 r r r rG ^ � z-TestHeap.test_naive_nbest.<locals>.<listcomp>�� �
���)r/ r2 r �lenr
r3 r �sorted�r r8 r7 r: r r r �test_naive_nbest] s � zTestHeap.test_naive_nbestc c s* � z | j �|�V q ty Y d S w r" )r2 r
�
IndexError)r r7 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 rL rM rO rQ r r r rG u rR z'TestHeap.test_nbest.<locals>.<listcomp>rS rT r rU ) r/ r2 r r
r rI r[ rW r4 r5 rZ rX 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 rL rM rO rQ r r r rG � rR z4TestHeap.test_nbest_with_pushpop.<locals>.<listcomp>rS rT rU �x)r/ r2 r r r rI r[ rW rX 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 )NrT g $@r � � )r2 r r �type�int�float)r �hr] r r r �test_heappushpop� s zTestHeap.test_heappushpopc C s4 ddg}| � | j�|�d� | � | j�|�d� d S )N� � )r r2 r )r rd 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 rL )� rO rQ r r r rG � rR z*TestHeap.test_heapsort.<locals>.<listcomp>r> c s g | ]}�j �� ��qS r )r2 r
rQ �r7 r r r rG � s )r/ r0 rP r2 r r r rW )r ZtrialrJ r8 r: Z heap_sortedr rl 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 )
Nrk ri �ABCi ����� r r> )FT��key�reverse) r/ r0 rP �choicer1 r rW r �chainrI r2 r% )
r �inputsr9 �row�j�tuprq rr 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>�rq )r rI r2 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 )NrT � )rI r/ )�sr9 r r r �iterable� s
� �zCTestHeap.test_merge_does_not_suppress_index_error.<locals>.iterable)r4 rZ rI r2 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� rC � ro c S s g | ]}|j �qS r )�pairrQ r r r rG � s z1TestHeap.test_merge_stability.<locals>.<listcomp>) rb r/ r0 rP r� r1 r3 r2 r% r rW )r r� ru r9 �streamr] �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 rM rO rQ r r r rG � � z+TestHeap.test_nsmallest.<locals>.<listcomp>rS c S � | d d d S �Nr i# rN r �r] r r r r} � � z)TestHeap.test_nsmallest.<locals>.<lambda>� r r> rg rT ri i� i� rS iL r )r/ r rI r2 � nsmallestrW �r r8 �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� rM rO rQ r r r rG � r� z*TestHeap.test_nlargest.<locals>.<listcomp>rS c S r� r� r r� r r r r} � r� z(TestHeap.test_nlargest.<locals>.<lambda>r� T�rr r rp )r/ r rI r2 �nlargestrW 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 )rF r] )�compr r rG � rH zDTestHeap.test_comparison_operator.<locals>.hsort.<locals>.<listcomp>c s g | ] }�j �� �j�qS r )r2 r
r] rQ )r8 r r r rG � r� )r2 r r/ rV )r8 r� r� )r� r8 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 r] 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 rD r rE rQ r r r rG rH z5TestHeap.test_comparison_operator.<locals>.<listcomp>ri Tr� )r/ rW r r4 r5 )r r� r� r� r8 �targetr r� r �test_comparison_operator� s z!TestHeap.test_comparison_operatorN)r r r! r< r. rK rY r[ r\ r^ re rh rm rz r� r� r� r� r� r� r r r r r, 0 s$
r, c @ � e Zd ZeZdS )�TestHeapPythonN�r r r! r r2 r r r r r� � r� r c @ r� )� TestHeapCN�r r r! r r2 r r r r r� � r� c @ s e Zd ZdZdd� ZdS )�LenOnlyz:Dummy sequence class defining __len__ but not __getitem__.c C r{ �NrT 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 ��seqnr9 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 r9 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= �r9 rV 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 )Nrf 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� )rt c C s t tdd� ttt| �����S )z Test multiple tiers of iteratorsc S r� r" r r� r r r r} m r~ zL.<locals>.<lambda>)rt �mapr� r� r� r� r r r �Lk s r� c @ r� )� SideEffectLTc C s || _ || _d S r" )�valuer7 )r r� r7 r r r r� q r� zSideEffectLT.__init__c C s g | j d d �< | j|jk S r" )r7 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� )
r2 r r
r4 r5 r6 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 �NrT rg ) r2 r r
r4 r5 r6 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� r2 r r
r4 r� r r
r� r� )r ry 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� )
r2 r r
r r
r� r� r4 r5 r6 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� rS )r> g333333�?rN i� � rg )r2 r� r� r/ r� r� r� r� r� r rI r� r4 r5 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� rQ �r7 r r � <genexpr>� � � z@TestErrorHandling.test_heappush_mutating_heap.<locals>.<genexpr>�� r� )�extendr/ r4 rZ � RuntimeErrorr2 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� rQ r� r r r� � r� z?TestErrorHandling.test_heappop_mutating_heap.<locals>.<genexpr>r� )r� r/ r4 rZ r� r2 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> )rb r2 r r4 rZ 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 rd � r rd 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> )rb r2 r r4 rZ r� )r rd 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� r0 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� � itertoolsrt r� r� r� r r r �mainr r r r �<module> sL ]
Z �