python (3.11.7)
eA
d Z ddlZddlZddlZddlmZ ddlmZ ddlmZm Z ddl
m Z ej ddg Z
ej ddg
Zg d Z G d d
e Zd Z G d d Z G d dee Z e ed G d dee Z G d d Z G d d Zd Z G d d Z G d d Z G d d Z G d! d" Z G d# d$ Z G d% d& Z G d' d( Z dd)l m Z d* Z! G d+ d, Z" G d- d. Z# G d/ d0e#e Z$ e ed G d1 d2e#e Z%e&d3k r ej' dS dS )4zUnittests for heapq. N)support)
import_helper)TestCase
skipUnless)
itemgetterheapq_heapq)blocked)fresh)heapifyheappopheappush heappushpop heapreplace _heappop_max_heapreplace_max _heapify_maxc > e Zd Zd Z eed d ZdS ) TestModulesc v t D ]0}| t t | j d 1d S )Nr )
func_names assertEqualgetattrpy_heapq
__module__selffnames C/BuggyBox/python/3.11.7/bootstrap/lib/python3.11/test/test_heapq.pytest_py_functionsz TestModules.test_py_functions sE K KE WXu 5 5 @' J J J J K K requires _heapqc v t D ]0}| t t | j d 1d S )Nr )r r r c_heapqr r s r test_c_functionsz TestModules.test_c_functions sE K KE WWe 4 4 ? J J J J K Kr! N)__name__r __qualname__r r r$ r% r! r r r sT K K K Z*++K K ,+K K Kr! r c G d d }| t j t | |S )Nc e Zd Zd ZdS )+load_tests.<locals>.HeapqMergeDocTestFinderc f t j }| t j S N)doctest
DocTestFinderfindr merge)r argskwargsdtfs r r0 z0load_tests.<locals>.HeapqMergeDocTestFinder.find( s$ '))C88HN++ +r! N)r& r r' r0 r( r! r HeapqMergeDocTestFinderr+ ' s# , , , , ,r! r5 ) test_finder)addTestsr. DocTestSuiter )loadertestsignorer5 s r
load_testsr< so , , , , , , , ,
NN7'4K4K4M4MO O O P P P Lr! c t e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d Z d
Z d Z
d Zd
Zd Zd Zd Zd ZdS )TestHeapc < g }g }| | t d D ]Z}t j }| | | j || | | [g }|rF| j | }| | | | |F|d d }| | || | | | t | j j g | t | j j d d | t | j j d d S # t $ r Y d S w xY w)N ) check_invariantrangerandomappendmoduler r
sortr assertRaises TypeErrorAttributeError)r heapdataiitemresults data_sorteds r
test_push_popzTestHeap.test_push_pop2 s T"""s ' 'A=??D KK K t , , , & & & & !;&&t,,D & & & NN4 ! 111g g... W%%% )T[%92>>> i)=tT J J J i)<d C C C C C DD s >A
F
FFc t | D ].\ }}|r'|dz
dz }| || |k /d S N ) enumerate
assertTrue)r rJ posrM parentposs r rA zTestHeap.check_invariantN sV "4 9 9IC
9 UqL Y4 7888 9 9r! c ( t t d dgz D ]J}d t | D }| j | | | K| t | j j d d S )N N c 4 g | ]}t j S r( rC ).0dummys r
<listcomp>z)TestHeap.test_heapify.<locals>.<listcomp>W s ===FMOO===r! )listrB rE r rA rG rH )r sizerJ s r test_heapifyzTestHeap.test_heapifyU s rOOug- ' 'D==t===D K % % % & & & & )T[%8$?????r! c T d t d D }g }|D ]J}| j || t | dk r| j | K| | |t | dd d S )Nc 6 g | ]}t j d S rC randranger] rL s r r_ z-TestHeap.test_naive_nbest.<locals>.<listcomp>^ # <<<1 &&<<<r!
)rB rE r lenr
rF r sortedr rK rJ rM s r test_naive_nbestzTestHeap.test_naive_nbest] s <<d <<< * *D K t , , ,4yy2~~ ##D))) vd||CDD 122222r! c # f K | j | V # t $ r Y d S w xY wr- )rE r
IndexErrorr rJ s r heapiterzTestHeap.heapiterg sS
0k))$/////
0 DD s "
00c f d t d D }|d d }| j | |dd D ])}||d k r| j || *| t | | t | dd | t | j j d | t | j j d d | t | j j g d d S )Nc 6 g | ]}t j d S re rg ri s r r_ z'TestHeap.test_nbest.<locals>.<listcomp>u rj r! rk rl r rm ) rB rE r r r r` ru ro rG rH rs rp s r
test_nbestzTestHeap.test_nbesto s =<d <<<CRCy D!!!I 4 4Dd1g~~ ''d333 dmmD 1 122F4LL4FGGG )T[%<dCCC )T[%<dDIII *dk&=r4HHHHHr! c d t d D }|d d }| j | |dd D ] }| j || | t | | t | dd | | j g d d d S )Nc 6 g | ]}t j d S re rg ri s r r_ z4TestHeap.test_nbest_with_pushpop.<locals>.<listcomp> rj r! rk rl rm x)rB rE r r r r` ru ro rp s r test_nbest_with_pushpopz TestHeap.test_nbest_with_pushpop s <<d <<<CRCy D!!!I 0 0D K # #D$ / / / / dmmD 1 122F4LL4FGGG 00S993?????r! c v g }| j |d }| ||fg df dg}| j |d }| ||fdgdf | t |d t | t | t
dg}| j |d }| ||fdgdf dg}| j |d }| ||fdgdf d S )Nrl g $@r )rE r r typeintfloat)r hr{ s r test_heappushpopzTestHeap.test_heappushpop s8 K # #Ar * * !Q"b***
D K # #At , , !Q2$ ... adS))) a%(((
D K # #Aq ) ) !Q2$+++
D K # #Ar * * !Q2$,,,,,r! c ddg}| | j | d | | j | d d S )N )r rE r )r r s r test_heappop_maxzTestHeap.test_heappop_max s_ F 11!44a888 11!44a88888r! c t d D ]}t j d }d t | D }|dz r%|d d j n"g |D ] } j | fdt | D } |t | d S )Nd 2 c 6 g | ]}t j d S ) rg ri s r r_ z*TestHeap.test_heapsort.<locals>.<listcomp> s# >>>QF$R((>>>r! rS c D g | ] }j S r( )rE r
)r] rL rJ r s r r_ z*TestHeap.test_heapsort.<locals>.<listcomp> s) JJJ4;..t44JJJr! )rB rC rh rE r r r ro )r trialra rK rM heap_sortedrJ s ` @r
test_heapsortzTestHeap.test_heapsort s 3ZZ 8 8E#B''D>>%++>>>Dqy
5AAAw ##D)))) 5 5DK((t4444JJJJJeDkkJJJK [&,, 7 7 7 7 8 8r! c
g }t t j d D ]{}g }t t j d D ]@}t j d t j dd f}| | A| | |d t d t d t dd fD ]}dD ]}g }|D ]'} | t
| || (| t
t | || t | j
j |||d | t | j
g d S )
Nr r ABCi r rS )FTkeyreverse) rB rC rh choicerD r ro r chainr` rE r1 )
r inputsrL rowjtupr r seqsseqs
r
test_mergezTestHeap.test_merge s v'++,, AC6+C0011
mE**F,<T3,G,GG
3 MM# *Q--A
1a8H8HI @ @C(
@
@! G GCKKsW E E EFFFF vC!Q!Q!Q!%&7dk&73PW&X&X&X!Y!Y[ [ [ dk&7&7&9&9!:!:B????
@ @ @r! c | t | j g g g | t | j g g d g d S )Nc dS )N r( r( r! r <lambda>z,TestHeap.test_empty_merges.<locals>.<lambda> s A r! r )r r` rE r1 r s r test_empty_mergeszTestHeap.test_empty_merges sq
dk / /B 7 788"=== dk / /BII / F FGGLLLLLr! c d }| t 5 t | j | | d d d d S # 1 swxY w Y d S )Nc 3 z K t t d } t d D ] }| | V
d S )Nrl )r` rB )srL s r iterablezCTestHeap.test_merge_does_not_suppress_index_error.<locals>.iterable sE U2YYA2YY
d
r! )rG rs r` rE r1 )r r s r (test_merge_does_not_suppress_index_errorz1TestHeap.test_merge_does_not_suppress_index_error s
z
*
* < < ""88::xxzz:: ; ; ; < < < < < < < < < < < < < < < < < <s 9A$$A(+A(c G d dt }g g g g g}t d D ]Y}t j d }t j d } || }||f|_ || | Z|D ]}| d | j j | D }| |t | d S )Nc e Zd ZdS )*TestHeap.test_merge_stability.<locals>.IntN)r& r r' r( r! r Intr s Dr! r rZ r c g | ] }|j
S r( )pairri s r r_ z1TestHeap.test_merge_stability.<locals>.<listcomp> s ===Q!&===r! ) r rB rC rh r rD rF rE r1 r ro )r r r rL streamr{ objresults r test_merge_stabilityz TestHeap.test_merge_stability s # b"b!u ' 'A%a((F %%A#a&&C6{CH 6N ! !# & & & & F KKMMMM=="3$+"3V"<=== 00000r! c d t d D }d d fD ]}dD ]}| t | j || t | d | | t | j ||| t || d | d S )Nc : g | ]}t j d |fS re rg ri s r r_ z+TestHeap.test_nsmallest.<locals>.<listcomp> ( AAA!$''+AAAr! rk c | d dz dz S Nr i# rf r( r{ s r r z)TestHeap.test_nsmallest.<locals>.<lambda> 1Q4#:#4 r! r rS r rl r i i rk iL r )rB r r` rE nsmallestro r rK fns r test_nsmallestzTestHeap.test_nsmallest s AAU4[[AAA445 : :A=
:
: dk&;&;At&D&D!E!E!' bqb!13 3 3 dk&;&;At&;&K&K!L!L!'!!4!4!4RaR!8: : : :
: : :r! c d t d D }d d fD ]}dD ]}| t | j || t |d d | | t | j ||| t ||d d | d S ) Nc : g | ]}t j d |fS re rg ri s r r_ z*TestHeap.test_nlargest.<locals>.<listcomp> r r! rk c | d dz dz S r r( r s r r z(TestHeap.test_nlargest.<locals>.<lambda> r r! r Tr r r )rB r r` rE nlargestro r s r
test_nlargestzTestHeap.test_nlargest s AAU4[[AAA445 H HA=
H
H dk&:&:1d&C&C!D!D!'d!;!;!;BQB!?A A A dk&:&:1d&:&J&J!K!K!'!T!B!B!B2A2!FH H H H
H H Hr! c fd} G d d } G d d }d t d D }t |d } ||| | t || d S )
Nc fd D j fdt t D S )Nc & g | ]
} | S r( r( )r] r{ comps r r_ zDTestHeap.test_comparison_operator.<locals>.hsort.<locals>.<listcomp> s! ***DDGG***r! c N g | ]!}j j "S r( )rE r
r{ )r] rL rK r s r r_ zDTestHeap.test_comparison_operator.<locals>.hsort.<locals>.<listcomp> s, JJJADK''--/JJJr! )rE r rB rn )rK r r s ``r hsortz0TestHeap.test_comparison_operator.<locals>.hsort s_ ****T***D K % % %JJJJJs4yy9I9IJJJ Jr! c e Zd Zd Zd ZdS )-TestHeap.test_comparison_operator.<locals>.LTc || _ d S r- r r r{ s r __init__z6TestHeap.test_comparison_operator.<locals>.LT.__init__
r! c " | j |j k S r- r r others r __lt__z4TestHeap.test_comparison_operator.<locals>.LT.__lt__ s v''r! Nr& r r' r r r( r! r LTr s2
(
(
(
(
(r! r c e Zd Zd Zd ZdS )-TestHeap.test_comparison_operator.<locals>.LEc || _ d S r- r r s r r z6TestHeap.test_comparison_operator.<locals>.LE.__init__ r r! c " | j |j k S r- r r s r __le__z4TestHeap.test_comparison_operator.<locals>.LE.__le__ s v((r! N)r& r r' r r r( r! r LEr s2
)
)
)
)
)r! r c 4 g | ]}t j S r( r\ ri s r r_ z5TestHeap.test_comparison_operator.<locals>.<listcomp> s 444A
444r! r Tr )rB ro r rG rH )r r r r rK targets ` r test_comparison_operatorz!TestHeap.test_comparison_operator s K K K K K ( ( ( ( ( ( ( (
) ) ) ) ) ) ) )
54s444d+++ tR &111 )T2.....r! N)r& r r' rP rA rb rq ru rx r| r r r r r r r r r r r( r! r r> r> 0 s% 89 9 9@ @ @3 3 3 I I I$@ @ @- - -&9 9 9
8
8
8 @ @ @$M M M < < <
1
1
1 : : :H H H/ / / / /r! r> c e Zd ZeZdS )TestHeapPythonNr& r r' r rE r( r! r r r
FFFr! r r" c e Zd ZeZdS ) TestHeapCNr& r r' r$ rE r( r! r r r
FFFr! r c e Zd ZdZd ZdS )LenOnlyz:Dummy sequence class defining __len__ but not __getitem__.c dS Nrl r( r s r __len__zLenOnly.__len__ s rr! N)r& r r' __doc__r r( r! r r r s) @@ r! r c , e Zd ZdZd ZexZxZxZxZZ dS )CmpErrz;Dummy element that always raises an error during comparisonc t r- )ZeroDivisionErrorr s r __eq__z
CmpErr.__eq__ s r! N)
r& r r' r r __ne__r r __gt____ge__r( r! r r r s< AA 177F7V7f7vr! r c # K | D ]}|V dS )zRegular generatorNr( seqnrL s r Rr " s*
r! c e Zd ZdZd Zd ZdS )GzSequence using __getitem__c || _ d S r- r r r s r r z
G.__init__) s
r! c | j | S r- r )r rL s r __getitem__z
G.__getitem__+ s y|r! N)r& r r' r r r r( r! r r r ' s8 r! r c $ e Zd ZdZd Zd Zd ZdS )Iz Sequence using iterator protocolc " || _ d| _ d S Nr r r s r r z
I.__init__0 r! c | S r- r( r s r __iter__z
I.__iter__3 r! c | j t | j k rt | j | j }| xj dz
c_ |S rR rL rn r
StopIterationr vs r __next__z
I.__next__5 > 6S^^ # #=%8 Idf ! r! Nr& r r' r r r r r( r! r r r . sG && r! r c e Zd ZdZd Zd ZdS )Igz9Sequence using iterator protocol defined with a generatorc " || _ d| _ d S r r r s r r z Ig.__init__= r r! c # & K | j D ]}|V d S r- r )r vals r r z Ig.__iter__@ s, 9 CIIII r! Nr& r r' r r r r( r! r r r ; s8 ?? r! r c e Zd ZdZd Zd ZdS )Xz Missing __getitem__ and __iter__c " || _ d| _ d S r r r s r r z
X.__init__F r r! c | j t | j k rt | j | j }| xj dz
c_ |S rR r r s r r z
X.__next__I r r! N)r& r r' r r r r( r! r r r D s8 && r! r c e Zd ZdZd Zd ZdS )NzIterator missing __next__()c " || _ d| _ d S r r r s r r z
N.__init__Q r r! c | S r- r( r s r r z
N.__iter__T r r! Nr r( r! r r# r# O s8 !! r! r# c $ e Zd ZdZd Zd Zd ZdS )Ez Test propagation of exceptionsc " || _ d| _ d S r r r s r r z
E.__init__Y r r! c | S r- r( r s r r z
E.__iter__\ r r! c ddz d S )Nr r r( r s r r z
E.__next__^ s Qr! Nr r( r! r r' r' W sG $$ r! r' c $ e Zd ZdZd Zd Zd ZdS )SzTest immediate stopc d S r- r( r s r r z
S.__init__c s r! c | S r- r( r s r r z
S.__iter__e r r! c t r- )r r s r r z
S.__next__g s r! Nr r( r! r r, r, a sG
r! r, )r c t t d t t t | S )z Test multiple tiers of iteratorsc | S r- r( r s r r zL.<locals>.<lambda>m s a r! )r mapr r r r s r Lr3 k s. ZZ2agg;; 00 1 11r! c e Zd Zd Zd ZdS ) SideEffectLTc " || _ || _ d S r- )valuerJ )r r7 rJ s r r zSideEffectLT.__init__q s
r! c : g | j d d < | j |j k S r- )rJ r7 r s r r zSideEffectLT.__lt__u s !!! zEK''r! Nr r( r! r r5 r5 p s2 ( ( ( ( (r! r5 c D e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d Z d
S ) TestErrorHandlingc , | j j | j j fD ]%}| t t
f|d &| j j | j j | j j | j j fD ]&}| t t
f|dd 'd S r )
rE r r
rG rH rI r r r r r r s r test_non_sequencez#TestErrorHandling.test_non_sequence| s +%t{':; B BA y. 91b A A A A+& (?+& (=? F FA y. 91b" E E E E F Fr! c | j j | j j fD ]1}| t t
f|t
2| j j | j j fD ]2}| t t
f|t
d 3| j j | j j
fD ]+}| t |dt
,d S Nrl r ) rE r r
rG rH rI r r r r r r< s r
test_len_onlyz TestErrorHandling.test_len_only s +%t{':; I IA y. 91gii H H H H+& (?@ M MA y. 91gii L L L L+& (=> : :A iAwyy 9 9 9 9 : :r! c t t t g}| j j | j j fD ] }| t
|| | j j | j j fD ] }| t
||d | j j | j j fD ] }| t
|d| d S r? )
r rE r r
rG r r r r r )r r r s r test_cmp_errz TestErrorHandling.test_cmp_err s xx688,+%t{':; 9 9A /C 8 8 8 8+& (?@ = =A /C < < < <+& (=> < <A /As ; ; ; ; < <r! c | j j | j j | j j | j j | j j | j j fD ]%}| t t f|d &d S r )
rE r r
r r r r rG rH rI r< s r test_arg_parsingz"TestErrorHandling.test_arg_parsing sp +%t{':+& (?+& (=? B BA
y. 91b A A A A B Br! c
| j j | j j fD ]S}ddt d dt ddd fD ]+}t t
t t t fD ]O}| t |d || t |d| P| t |dt | g | t |dt | | t |dt | | t |dt# | -Ud S ) N123 rk )rS g333333?rf i r )rE r r rB r r r r3 r r r` r, rG rH r r# r r' )r r r gs r test_iterable_argsz$TestErrorHandling.test_iterable_args sS +& (=> A AARthd48J8JK
A
AQAq) E EA$$T!!Aqqtt**%5%5tAAaFF||DDDD aa1Q44jj!1!12666!!)Q1Q44888!!)Q1Q44888!!"3Q1Q44@@@@
A A Ar! c . g fdt d D | t t f 5 | j t d d d d d S # 1 swxY w Y d S )Nc 3 8 K | ]}t | V d S r- r5 r] rL rJ s r <genexpr>z@TestErrorHandling.test_heappush_mutating_heap.<locals>.<genexpr> - >>aLD))>>>>>>r! rH )extendrB rG rs RuntimeErrorrE r r5 rt s @r test_heappush_mutating_heapz-TestErrorHandling.test_heappush_mutating_heap s >>>>5::>>>>>>
L 9
:
: > > K |At'<'< = = = > > > > > > > > > > > > > > > > > >s *B
BBc g fdt d D | t t f 5 | j d d d d S # 1 swxY w Y d S )Nc 3 8 K | ]}t | V d S r- rM rN s r rO z?TestErrorHandling.test_heappop_mutating_heap.<locals>.<genexpr> rP r! rQ )rR rB rG rs rS rE r
rt s @r test_heappop_mutating_heapz,TestErrorHandling.test_heappop_mutating_heap s >>>>5::>>>>>>
L 9
:
: & & K % % % & & & & & & & & & & & & & & & & & &s A;;A?A?c G fddt }g | j |d | t | j j d d S )Nc e Zd Z fdZdS )MTestErrorHandling.test_comparison_operator_modifiying_heap.<locals>.EvilClassc : t S r- clearNotImplemented)r orJ s r r zTTestErrorHandling.test_comparison_operator_modifiying_heap.<locals>.EvilClass.__lt__ s
%%r! Nr& r r' r )rJ s r EvilClassrZ .
&
&
&
&
&
&
&r! ra r rS )r rE r rG rs r )r ra rJ s @r (test_comparison_operator_modifiying_heapz:TestErrorHandling.test_comparison_operator_modifiying_heap s & & & & & & & & & &
T99Q<<000 *dk&=tQGGGGGr! c G fddt } G fddt }g g c| j |d | j |d | t t
f| j j |d | t t
f| j j |d d S )Nc e Zd Z fdZdS )OTestErrorHandling.test_comparison_operator_modifiying_heap_two_heaps.<locals>.hc : t S r- r\ )r r_ list2s r r zVTestErrorHandling.test_comparison_operator_modifiying_heap_two_heaps.<locals>.h.__lt__
%%r! Nr` )rh s r r rf rb r! r c e Zd Z fdZdS )OTestErrorHandling.test_comparison_operator_modifiying_heap_two_heaps.<locals>.gc : t S r- r\ )r r_ list1s r r zVTestErrorHandling.test_comparison_operator_modifiying_heap_two_heaps.<locals>.g.__lt__ ri r! Nr` )rm s r rI rk rb r! rI r rS )r rE r rG rs rS )r r rI rm rh s @@r 2test_comparison_operator_modifiying_heap_two_heapszDTestErrorHandling.test_comparison_operator_modifiying_heap_two_heaps s! & & & & & & & & & &
& & & & & & & & & &
2 u UAAaDD))) UAAaDD))) :|4dk6JESTSTUVSWSWXXX :|4dk6JESTSTUVSWSWXXXXXr! N) r& r r' r= r@ rB rD rJ rT rW rc rn r( r! r r: r: z s F F F: : :< < <B B B A A A> > >& & &
H
H
HY Y Y Y Yr! r: c e Zd ZeZdS )TestErrorHandlingPythonNr r( r! r rp rp r r! rp c e Zd ZeZdS )TestErrorHandlingCNr r( r! r rr rr r r! rr __main__)(r rC unittestr. testr test.supportr r r operatorr import_fresh_moduler r$ r r r< r> r r r r r r r r r r# r' r, itertoolsr r3 r5 r: rp rr r& mainr( r! r <module>r{ s3
& & & & & & ) ) ) ) ) ) ) ) ,= ,Wxj I I I
+-
+GH:
F
F
FB B B
K K K K K( K K K "Y/ Y/ Y/ Y/ Y/ Y/ Y/ Y/x Xx G&'' ( ('
8 8 8 8 8 8 8 8
2 2 2
( ( ( ( ( ( ( (XY XY XY XY XY XY XY XYt / G&'' *H (' zHMOOOOO r! 
Browse
Usage
Versions