python (3.12.0)
ΑeA d Z ddlZddlZddlZddlmZ ddlmZmZ ddlm Z ej ddg Z ej ddg Z g d
Z
G d d e Zd
Z G d d Z G d dee Z ee d 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. d/e!e Z" ee d G d0 d1e!e Z#e$d2k( r ejJ yy)3zUnittests for heapq. N)
import_helper)TestCase
skipUnless)
itemgetterheapq_heapq)blocked)fresh)heapifyheappopheappush heappushpop heapreplace _heappop_max_heapreplace_max _heapify_maxc 0 e Zd Zd Z eed d Zy) TestModulesc n t D ], }| j t t | j d . y )Nr )
func_names assertEqualgetattrpy_heapq
__module__selffnames C/BuggyBox/python/3.12.0/bootstrap/lib/python3.12/test/test_heapq.pytest_py_functionsz TestModules.test_py_functions s0 KE WXu 5 @ @' J K requires _heapqc n t D ], }| j t t | j d . y )Nr )r r r c_heapqr r s r test_c_functionsz TestModules.test_c_functions s0 KE WWe 4 ? ? J Kr N)__name__r __qualname__r r r# r$ r r r r s& K *+K ,Kr r c x G d d }|j t j t | |S )Nc e Zd Zd Zy)+load_tests.<locals>.HeapqMergeDocTestFinderc h t j }|j t j S N)doctest
DocTestFinderfindr merge)r argskwargsdtfs r r/ z0load_tests.<locals>.HeapqMergeDocTestFinder.find' s# '')C88HNN+ +r N)r% r r& r/ r' r r HeapqMergeDocTestFinderr* & s ,r r4 ) test_finder)addTestsr- DocTestSuiter )loadertestsignorer4 s r
load_testsr; s6 , ,
NN7''4K4MO P Lr c r e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d Z d
Z d Z
d Zd
Zd Zd Zd Zd Zy)TestHeapc g }g }| j | t d D ]T }t j }|j | | j j || | j | V g }|r@| j j
| }| j | |j | |r@|d d }|j | j || | j | | j t | j j
g | j t | j j
d d | j t | j j d y # t $ r Y y w xY w)N ) check_invariantrangerandomappendmoduler
r sortr assertRaises TypeErrorAttributeError)r heapdataiitemresults data_sorteds r
test_push_popzTestHeap.test_push_pop1 sE T"s 'A==?D KK KK t , & '
;;&&t,D & NN4 1g g. W% )T[[%9%92> i)=)=tT J i)<)<d C s AE8 8 FFc n t | D ]' \ }}|s |dz
dz }| j || |k ) y N ) enumerate
assertTrue)r rI posrL parentposs r r@ zTestHeap.check_invariantM s> "4 9IC UqL Y4 78 9r c J t t d dgz D ]X }t | D cg c] }t j }}| j j | | j | Z | j
t | j j d y c c}w )N N )listrA rB rD r r@ rF rG )r sizedummyrI s r test_heapifyzTestHeap.test_heapifyT s rOug- 'D16t=FMMO=D= KK % & '
)T[[%8%8$? >s B c ^ t d D cg c] }t j d }}g }|D ]H }| j j || t | dkD s.| j j
| J |j | j |t | dd y c c}w )N
)
rA rB randrangerD r
lenr rE r sortedr rK rJ rI rL s r test_naive_nbestzTestHeap.test_naive_nbest\ s 05d <1 &<< *D KK t ,4y2~ ##D) *
vd|CD 12 =s B*c # h K | j j | # t $ r Y y w xY wwr, )rD r
IndexErrorr rI s r heapiterzTestHeap.heapiterf s9 kk))$// s 2 # /2/2c t t d D cg c] }t j d }}|d d }| j j | |dd D ]' }||d kD s | j j || ) | j
t | j | t | dd | j t | j j
d | j t | j j
d d | j t | j j
g d y c c}w )Nr_ r` ra r rb )
rA rB rc rD r r r rZ rk re rF rG ri rf s r
test_nbestzTestHeap.test_nbestn s 16d <1 &<<CRy D!I 4Dd1g~ ''d3 4
dmmD 12F4L4FG )T[[%<%<dC )T[[%<%<dDI *dkk&=&=r4H =s D5c t d D cg c] }t j d }}|d d }| j j | |dd D ] }| j j || | j
t | j | t | dd | j
| j j g d d y c c}w )Nr_ r` ra rb x)
rA rB rc rD r r r rZ rk re rf s r test_nbest_with_pushpopz TestHeap.test_nbest_with_pushpop s 05d <1 &<<CRy D!I 0D KK # #D$ / 0 dmmD 12F4L4FG 00S93?
=s Cc 2 g }| j j |d }| j ||fg df dg}| j j |d }| j ||fdgdf | j t |d t | j t | t
dg}| j j |d }| j ||fdgdf dg}| j j |d }| j ||fdgdf y )Nra g $@r )rD r r typeintfloat)r hro s r test_heappushpopzTestHeap.test_heappushpop s KK # #Ar * !Q"b*
D KK # #At , !Q2$ . adS) a%(
D KK # #Aq ) !Q2$+
D KK # #Ar * !Q2$,r c ddg}| j | j j | d | j | j j | d y )N )r rD r )r rw s r test_heappop_maxzTestHeap.test_heappop_max sK F 11!4a8 11!4a8r c t d D ] }t j d }t | D cg c] }t j d }}|dz r!|d d }| j j | n%g }|D ] }| j j || t | D cg c] }| j j
| }}| j |t | y c c}w c c}w )Nd 2 rR ) rA rB rc rD r r
r r re )r trialr[ rK rJ rI rL heap_sorteds r
test_heapsortzTestHeap.test_heapsort s 3Z 8E##B'D27+>QF$$R(>D>qyAw ##D) 5DKK((t45>CDkJ4;;..t4JKJ [&, 7 8> Ks C)'"C.c
g }t t j d D ]t }g }t t j d D ]> }t j d t j dd f}|j | @ |j | v d t d t d t dd fD ] }dD ] }g }|D ] } |j t
| || ! | j t
t | || t | j j |||d | j t | j j g y )
Nr r~ ABCi r rR )FTkeyreverse) rA rB rc choicerC r re r chainrZ rD r0 )
r inputsrK rowjtupr r seqsseqs
r
test_mergezTestHeap.test_merge sI v''+, AC6++C01
mmE*F,<,<T3,GG
3
MM# *Q-A
1a8HI @C(
@! GCKKsW EFG vC!Q!%&7dkk&7&73PW&X!Y[ dkk&7&7&9!:B?
@ @r c | j t | j j g g g | j t | j j g g d g y )Nc y)N r' r' r r <lambda>z,TestHeap.test_empty_merges.<locals>.<lambda> s r r )r rZ rD r0 r s r test_empty_mergeszTestHeap.test_empty_merges sU
dkk / /B 78"= dkk / /BI / FGLr c d }| j t 5 t | j j | | d d d y # 1 sw Y y xY w)Nc 3 b K t t d } t d D ] }| | y w)Nra )rZ rA )srK s r iterablezCTestHeap.test_merge_does_not_suppress_index_error.<locals>.iterable s/ U2YA2Y
d
s -/)rF ri rZ rD r0 )r r s r (test_merge_does_not_suppress_index_errorz1TestHeap.test_merge_does_not_suppress_index_error sI
z
* < ""8:xz: ; < < <s 0AAc G d dt }g g g g g}t d D ]Q }t j d }t j d } || }||f|_ || j | S |D ] }|j
| j j | D cg c] }|j }}| j |t | y c c}w )Nc e Zd Zy)*TestHeap.test_merge_stability.<locals>.IntN)r% r r& r' r r Intr s r r rY r ) ru rA rB rc pairrC rE rD r0 r re )r r r rK streamro objresults r test_merge_stabilityz TestHeap.test_merge_stability s # b"b!u 'A%%a(F %Aa&C6{CH 6N ! !# & ' F KKM "3$++"3"3V"<=Q!&&== 0 >s 'Cc t d D cg c] }t j d |f }}d d fD ] }dD ] }| j t | j
j
|| t | d | | j t | j
j
||| t || d | y c c}w )Nr_ r` c | d dz dz S Nr i# r` r' ro s r r z)TestHeap.test_nsmallest.<locals>.<lambda> 1Q4#:#4 r r rR r{ ra r~ i i r_ iL r )rA rB rc r rZ rD nsmallestre r rK rJ fns r test_nsmallestzTestHeap.test_nsmallest s 5:4[A!!$'+AA45 :A=
: dkk&;&;At&D!E!' bq!13 dkk&;&;At&;&K!L!'!!4Ra!8:
: : Bs Cc t d D cg c] }t j d |f }}d d fD ] }dD ] }| j t | j
j
|| t |d d | | j t | j
j
||| t ||d d | y c c}w ) Nr_ r` c | d dz dz S r r' r s r r z(TestHeap.test_nlargest.<locals>.<lambda> r r r Tr r r )rA rB rc r rZ rD nlargestre r s r
test_nlargestzTestHeap.test_nlargest s 5:4[A!!$'+AA45 HA=
H dkk&:&:1d&C!D!'d!;BQ!?A dkk&:&:1d&:&J!K!'!T!B2A!FH
H H Bs C c fd} G d d } G d d }t d D cg c] }t j }}t |d } j ||| | j t
|| y c c}w ) Nc | D cg c]
} || } }j j | t t | D cg c]' }j j | j
) c}S c c}w c c}w r, )rD r rA rd r ro )rJ compro rK r s r hsortz0TestHeap.test_comparison_operator.<locals>.hsort sc %)*DG*D* KK %9>s4y9IJADKK''-//J J +Js A7,A<c e Zd Zd Zd Zy)-TestHeap.test_comparison_operator.<locals>.LTc || _ y r, r r ro s r __init__z6TestHeap.test_comparison_operator.<locals>.LT.__init__ r c 4 | j |j kD S r, r r others r __lt__z4TestHeap.test_comparison_operator.<locals>.LT.__lt__ s vv''r Nr% r r& r r r' r r LTr s
(r r c e Zd Zd Zd Zy)-TestHeap.test_comparison_operator.<locals>.LEc || _ y r, r r s r r z6TestHeap.test_comparison_operator.<locals>.LE.__init__ r r c 4 | j |j k\ S r, r r s r __le__z4TestHeap.test_comparison_operator.<locals>.LE.__le__ s vv((r N)r% r r& r r r' r r LEr s
)r r r~ Tr )rA rB re r rF rG )r r r r rK rJ targets ` r test_comparison_operatorz!TestHeap.test_comparison_operator su K ( (
) )
*/s4A
44d+ tR &1 )T2. 5s BN)r% r r& rO r@ r] rg rk rm rp rx r| r r r r r r r r r' r r r= r= / s] 89@3I$@-&9
8 @$M <
1 :H/r r= c e Zd ZeZy)TestHeapPythonNr% r r& r rD r' r r r r
Fr r r! c e Zd ZeZy) TestHeapCNr% r r& r# rD r' r r r r
Fr r c e Zd ZdZd Zy)LenOnlyz:Dummy sequence class defining __len__ but not __getitem__.c yNra r' r s r __len__zLenOnly.__len__ s r N)r% r r& __doc__r r' r r r r s
@r r c * e Zd ZdZd ZexZxZxZxZZ y)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# A 177F7V7f7vr r c # " K | D ] }| yw)zRegular generatorNr' seqnrK s r Rr ! s
s
c e Zd ZdZd Zd Zy)GzSequence using __getitem__c || _ y r, r r r s r r z
G.__init__( s r c | j | S r, r )r rK s r __getitem__z
G.__getitem__* s yy|r N)r% r r& r r r r' r r r r & s r r c " e Zd ZdZd Zd Zd Zy)Iz Sequence using iterator protocolc || _ d| _ y Nr r r s r r z
I.__init__/ r c | S r, r' r s r __iter__z
I.__iter__2 r c | j t | j k\ rt | j | j }| xj dz
c_ |S rQ rK rd r
StopIterationr vs r __next__z
I.__next__4 ? 66S^ #=%8 IIdff ! r Nr% r r& r r r r r' r r r r - s &r r c e Zd ZdZd Zd Zy)Igz9Sequence using iterator protocol defined with a generatorc || _ d| _ y r r r s r r z Ig.__init__< r r c # 6 K | j D ] }| y wr, r )r vals r r z Ig.__iter__? s 99 CI s Nr% r r& r r r r' r r r r : s ?r r c e Zd ZdZd Zd Zy)Xz Missing __getitem__ and __iter__c || _ d| _ y r r r s r r z
X.__init__E r r c | j t | j k\ rt | j | j }| xj dz
c_ |S rQ r r s r r z
X.__next__H r r N)r% r r& r r r r' r r r
r
C s &r r
c e Zd ZdZd Zd Zy)NzIterator missing __next__()c || _ d| _ y r r r s r r z
N.__init__P r r c | S r, r' r s r r z
N.__iter__S r r Nr r' r r r r N s !r r c " e Zd ZdZd Zd Zd Zy)Ez Test propagation of exceptionsc || _ d| _ y r r r s r r z
E.__init__X r r c | S r, r' r s r r z
E.__iter__[ r r c ddz y )Nrz r r' r s r r z
E.__next__] s Qr Nr r' r r r r V s $r r c " e Zd ZdZd Zd Zd Zy)SzTest immediate stopc y r, r' r s r r z
S.__init__b s r c | S r, r' r s r r z
S.__iter__d r r c t r, )r r s r r z
S.__next__f s r Nr r' r r r r ` s
r r )r c d t t d t t t | S )z Test multiple tiers of iteratorsc | S r, r' r s r r zL.<locals>.<lambda>l s a r )r mapr r r r s r Lr j s Z2ag; 0 11r c e Zd Zd Zd Zy) SideEffectLTc || _ || _ y r, )valuerI )r r" rI s r r zSideEffectLT.__init__p s
r c R g | j d d | j |j k S r, )rI r" r s r r zSideEffectLT.__lt__t s" ! zzEKK''r Nr r' r r r r o s (r r c B e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d Z y
) TestErrorHandlingc | j j | j j fD ] }| j t t
f|d ! | j j | j j | j j | j j fD ] }| j t t
f|dd " y r )
rD r r rF rG rH r
r r r r r s r test_non_sequencez#TestErrorHandling.test_non_sequence{ s ++%%t{{':':; BA y. 91b A B++&& (?(?++&& (=(=? FA y. 91b" E Fr c | j j | j j fD ]' }| j t t
f|t
) | j j | j j fD ]( }| j t t
f|t
d * | j j | j j fD ]" }| j t |dt
$ y Nra r{ ) rD r r rF rG rH r r
r r r r' s r
test_len_onlyz TestErrorHandling.test_len_only s ++%%t{{':':; IA y. 91gi H I++&& (?(?@ MA y. 91gi L M++&& (=(=> :A iAwy 9 :r c t t t g}| j j | j j fD ] }| j t
|| | j j | j j fD ] }| j t
||d | j j | j j fD ] }| j t
|d| y r* )
r rD r r rF r r
r r r )r r r s r test_cmp_errz TestErrorHandling.test_cmp_err s x68,++%%t{{':':; 9A /C 8 9++&& (?(?@ =A /C < =++&& (=(=> <A /As ; <r c H | j j | j j | j j | j j | j j
| j j fD ] }| j t t f|d ! y r )
rD r r r
r r r rF rG rH r' s r test_arg_parsingz"TestErrorHandling.test_arg_parsing sy ++%%t{{':':++&& (?(?++&& (=(=? BA
y. 91b A Br c
v | j j | j j fD ] }ddt d dt ddd fD ] }t t
t t t fD ]: }| j t |d || t |d| < | j t |dt | g | j t |dt | | j t |dt | | j t |dt# | y ) N123 r_ )rR g333333?r` i r{ )rD r r rA r r r r r r rZ r rF rG r
r r r )r r r gs r test_iterable_argsz$TestErrorHandling.test_iterable_args s ++&& (=(=> AARthd48JK
AQAq) EA$$T!Aqt*%5tAaF|DE a1Q4j!126!!)Q1Q48!!)Q1Q48!!"3Q1Q4@
A Ar c g j fdt d D | j t t f 5 | j
j
t d d d d y # 1 sw Y y xY w)Nc 3 6 K | ] }t | y wr, r .0rK rI s r <genexpr>z@TestErrorHandling.test_heappush_mutating_heap.<locals>.<genexpr> >aLD)> r3 )extendrA rF ri RuntimeErrorrD r
r rj s @r test_heappush_mutating_heapz-TestErrorHandling.test_heappush_mutating_heap sa >5:>>
L 9
: > KK |At'< = > > >s 'A22A;c g j fdt d D | j t t f 5 | j
j
d d d y # 1 sw Y y xY w)Nc 3 6 K | ] }t | y wr, r8 r9 s r r; z?TestErrorHandling.test_heappop_mutating_heap.<locals>.<genexpr> r< r= r> )r? rA rF ri r@ rD r rj s @r test_heappop_mutating_heapz,TestErrorHandling.test_heappop_mutating_heap sX >5:>>
L 9
: & KK % & & &s A''A0c G fddt }g | j j |d | j t | j j
d y )Nc e Zd Z fdZy)MTestErrorHandling.test_comparison_operator_modifiying_heap.<locals>.EvilClassc 0 j t S r, clearNotImplemented)r orI s r r zTTestErrorHandling.test_comparison_operator_modifiying_heap.<locals>.EvilClass.__lt__ s
%%r Nr% r r& r )rI s r EvilClassrG
&r rN r rR )ru rD r
rF ri r )r rN rI s @r (test_comparison_operator_modifiying_heapz:TestErrorHandling.test_comparison_operator_modifiying_heap sL & &
T9Q<0 *dkk&=&=tQGr c G fddt } G fddt }g g c| j j |d | j j |d | j t t
f| j j |d | j t t
f| j j |d y )Nc e Zd Z fdZy)OTestErrorHandling.test_comparison_operator_modifiying_heap_two_heaps.<locals>.hc 0 j t S r, rI )r rL list2s r r zVTestErrorHandling.test_comparison_operator_modifiying_heap_two_heaps.<locals>.h.__lt__
%%r NrM )rU s r rw rS rO r rw c e Zd Z fdZy)OTestErrorHandling.test_comparison_operator_modifiying_heap_two_heaps.<locals>.gc 0 j t S r, rI )r rL list1s r r zVTestErrorHandling.test_comparison_operator_modifiying_heap_two_heaps.<locals>.g.__lt__ rV r NrM )rZ s r r4 rX rO r r4 r rR )ru rD r
rF ri r@ )r rw r4 rZ rU s @@r 2test_comparison_operator_modifiying_heap_two_heapszDTestErrorHandling.test_comparison_operator_modifiying_heap_two_heaps s & &
& &
2 u UAaD) UAaD) :|4dkk6J6JESTUVSWX :|4dkk6J6JESTUVSWXr N) r% r r& r( r+ r- r/ r5 rA rD rP r[ r' r r r% r% y s4 F:<B A>&
HYr r% c e Zd ZeZy)TestErrorHandlingPythonNr r' r r r] r] r r r] c e Zd ZeZy)TestErrorHandlingCNr r' r r r_ r_ r r r_ __main__)&r rB unittestr- 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 r r r% r] r_ r% mainr' r r <module>rg sr
& ) ,= , ,Wxj I
+-
+
+GH:
FB
K( K"Y/ Y/xXx G&'( (
8 8
2
( (XY XYt / G&'*H ( zHMMO r 
Browse
Usage
Versions