a 5dg @sddlZddlZddlmZmZmZmZddlmZmZm Z ddl Z ddl Z gdZ GdddZGdddZGd d d ZGd d d ZGd ddZGdddZGdddZGdddZGdddZGdddejZddZedkredS)N) run_unittestTESTFNunlink cpython_only)check_free_after_iterating ALWAYS_EQNEVER_EQ))rrr)rr)rr)rr r)rr r rr r )rr r)rr r )rr r )r rr)r rr )r rr )r r r)r r r )r r r )r r r)r r r )r r r )r rr)r rr )r rr )r r r)r r r )r r r )r r r)r r r )r r r c@s$eZdZddZddZddZdS)BasicIterClasscCs||_d|_dSNr)niselfrr*/usr/local/lib/python3.9/test/test_iter.py__init__szBasicIterClass.__init__cCs"|j}||jkrt|d|_|SNr )rr StopIterationrresrrr__next__s   zBasicIterClass.__next__cCs|SNrrrrr__iter__#szBasicIterClass.__iter__N__name__ __module__ __qualname__rrrrrrrr sr c@seZdZddZddZdS)IteratingSequenceClasscCs ||_dSrrrrrrr'szIteratingSequenceClass.__init__cCs t|jSr)r rrrrrr)szIteratingSequenceClass.__iter__Nrrr rrrrrrr!&sr!c@s$eZdZddZddZddZdS)IteratorProxyClasscCs ||_dSrrrrrrrr-szIteratorProxyClass.__init__cCs t|jSr)nextrrrrrr/szIteratorProxyClass.__next__cCs|Srrrrrrr1szIteratorProxyClass.__iter__Nrrrrrr$,sr$c@seZdZddZddZdS) SequenceClasscCs ||_dSrr"rrrrr5szSequenceClass.__init__cCs&d|kr|jkrnn|StdSr )r IndexErrorr&rrr __getitem__7szSequenceClass.__getitem__Nrrr rr*rrrrr(4sr(c@seZdZddZddZdS)SequenceProxyClasscCs ||_dSrs)rr.rrrr>szSequenceProxyClass.__init__cCs |j|Srr-r&rrrr*@szSequenceProxyClass.__getitem__Nr+rrrrr,=sr,c@seZdZddZdS)UnlimitedSequenceClasscCs|Srrr&rrrr*Dsz"UnlimitedSequenceClass.__getitem__Nrrr r*rrrrr/Csr/c@s eZdZdS)DefaultIterClassN)rrr rrrrr1Gsr1c@seZdZddZdZdS) NoIterClasscCs|Srrr&rrrr*KszNoIterClass.__getitem__N)rrr r*rrrrrr2Jsr2c@seZdZddZdS)BadIterableClasscCstdSr)ZeroDivisionErrorrrrrrPszBadIterableClass.__iter__Nrrr rrrrrr3Osr3c@seZdZduddZdvddZddZdd Zd d Zd d ZddZ ddZ ddZ ddZ ddZ ddZddZddZddZd d!Zd"d#Zd$d%Zd&d'Zd(d)Zd*d+Zd,d-Zd.d/Zd0d1Zd2d3Zd4d5Zd6d7Zd8d9Zd:d;Zdd?Z!d@dAZ"dBdCZ#dDdEZ$dFdGZ%dHdIZ&dJdKZ'dLdMZ(dNdOZ)dPdQZ*dRdSZ+e,dTdUZ-dVdWZ.dXdYZ/dZd[Z0d\d]Z1d^d_Z2d`daZ3dbdcZ4dddeZ5dfdgZ6dhdiZ7djdkZ8e,dldmZ9dndoZ:dpdqZ;drdsZr?rBrKrrrrtest_iter_basicszTestCase.test_iter_basiccCs.ttd}t|}t|}|||udSrN)rKrBr?rF)rr;r:Zit2rrrtest_iter_idempotencys zTestCase.test_iter_idempotencycCs |ttdttddSrNrAr?rBrKrrrrtest_iter_for_loopszTestCase.test_iter_for_loopcCsVtd}g}t|D]0}t|D]"}t|D]}||||fq,q q||tdSN)rBr?r8r9TRIPLETS)rr;rrjkrrrtest_iter_independences   zTestCase.test_iter_independencecs.tdfddtD}||tdS)NrUcs2g|]*}tD]}tD]}|||fqqqSr)r?.0rrWrXr;rr sz.)rBr?r9rVrrr\rtest_nested_comprehensions_iters  z(TestCase.test_nested_comprehensions_itercs*tdfddD}||tdS)NrUcs*g|]"}D]}D]}|||fqq qSrrrZr\rrr]z;TestCase.test_nested_comprehensions_for..)rBr9rVrrr\rtest_nested_comprehensions_forsz'TestCase.test_nested_comprehensions_forcCs|tdttddSrN)rAr!rKrBrrrrtest_iter_class_forszTestCase.test_iter_class_forcCs |ttdttddSrN)r>r?r!rKrBrrrrtest_iter_class_iterszTestCase.test_iter_class_itercCs|tdttddSrN)rAr(rKrBrrrrtest_seq_class_forszTestCase.test_seq_class_forcCs |ttdttddSrN)r>r?r(rKrBrrrrtest_seq_class_iterszTestCase.test_seq_class_itercCstd}ttjdD]v}t|}t||f|}t|\}}d|_|t |t || t |t tdt |t||f|}t|\}}d|_|t |t || t |t tddtddD] }t |qt||f|}t|\}}d|_|t |t || t |t tdd| tt |t||f|}t|\}}d|_|t|tjj| t |gqdS)Nr )r(rBr<rCr?rDrErZassertIstyper9rKr' assertRaisesrrFrGrHrIrJ)rZorigrLZitorigrMr:r;rrrr#test_mutating_seq_class_iter_pickles6 z,TestCase.test_mutating_seq_class_iter_picklecCsltd}t|}t|}|D] }t|qd|_|t|g|t|ddg|t|gddS)Nrerf)rr r rUrerj)r(r?r'rr9rK)raZexhitZempitxrrr&test_mutating_seq_class_exhausted_iters z/TestCase.test_mutating_seq_class_exhausted_itercCs$Gdddt}|tt|dS)Nc@seZdZddZdS)z5TestCase.test_new_style_iter_class..IterClasscSs|Srrrrrrrsz>TestCase.test_new_style_iter_class..IterClass.__iter__Nr5rrrr IterClasssro)objectrh TypeErrorr?)rrorrrtest_new_style_iter_classsz"TestCase.test_new_style_iter_classcCs2Gddd}|jt|dttddddS)Nc@seZdZddZddZdS)z&TestCase.test_iter_callable..CcSs d|_dSr r%rrrrrsz/TestCase.test_iter_callable..C.__init__cSs |j}|d|_|dkrt|S)Nr d)rr)r&rrr__call__s  z/TestCase.test_iter_callable..C.__call__N)rrr rrtrrrrCsrurOFr<r>r?rKrB)rrurrrtest_iter_callables zTestCase.test_iter_callablecCs0dgfdd}|jt|dttddddS)NrcSs|d}|d|d<|S)Nrr rstaterrrrspams z)TestCase.test_iter_function..spamrOFrvrwrr{rrrtest_iter_functionszTestCase.test_iter_functioncCs0dgfdd}|jt|dttddddS)NrcSs$|d}|dkrt|d|d<|S)NrrOr )rryrrrr{s  z.TestCase.test_iter_function_stop..spamrOFrvrwr|rrrtest_iter_function_stopsz TestCase.test_iter_function_stopc Csfdgfdd}g}zt|dD]}||qWn&tyV||ttdYn 0|ddS)NrcSs$|d}|d|d<|dkr t|S)Nrr rO) RuntimeErrorryrrrr{s  z.TestCase.test_exception_function..spamr~rOshould have raised RuntimeError)r?r8rr9rKrBfail)rr{rrmrrrtest_exception_function s z TestCase.test_exception_functionc CsfGdddt}g}z|dD]}||qWn&tyV||ttdYn 0|ddS)Nc@seZdZddZdS)z9TestCase.test_exception_sequence..MySequenceClasscSs|dkr tt||SrN)rr(r*r&rrrr* szETestCase.test_exception_sequence..MySequenceClass.__getitem__Nr0rrrrMySequenceClasssrr~rOr)r(r8rr9rKrBr)rrrrmrrrtest_exception_sequences  z TestCase.test_exception_sequencecCs0Gdddt}|j|dttddddS)Nc@seZdZddZdS)z4TestCase.test_stop_sequence..MySequenceClasscSs|dkr tt||SrN)rr(r*r&rrrr*0sz@TestCase.test_stop_sequence..MySequenceClass.__getitem__Nr0rrrrr/srr~rOFrv)r(rArKrB)rrrrrtest_stop_sequence.szTestCase.test_stop_sequencecCs |ttdttddS)Ni'rRrrrrtest_iter_big_range7szTestCase.test_iter_big_rangecCs|tggdSrrAr?rrrrtest_iter_empty;szTestCase.test_iter_emptycCs|tdttddS)N) rr r rUrkrerjrf rO)rAr?rKrBrrrrtest_iter_tuple?szTestCase.test_iter_tuplecCs |ttdttddSrNrRrrrrtest_iter_rangeCszTestCase.test_iter_rangecCs|tdgddSNZabcde)rlbcrMerrrrrtest_iter_stringGszTestCase.test_iter_stringcCs2i}tdD] }d||<q ||t|dSrN)rBrArKkeys)rdictrrrrtest_iter_dictKs  zTestCase.test_iter_dictcCsttd}z(tdD]}|d|qW|n |0ttd}zP|j|gddd|j|gddW|z ttWqtyYq0n*|z ttWntyYn00dS)Nwre%d rz0 1 2 3 4 Frv)openrrBwritecloserArOSError)rfrrrrtest_iter_fileRs&       zTestCase.test_iter_filecCs\|ttdttd|ttdg|tdgdddd}|t|t||ttt|ttdttd }z(tdD]}| d |qW| n | 0ttd }zb|t|gd | dd|t|gd W| z t tWnt y(Yn0n,| z t tWnt yTYn00dS) Nrerrr r rUonetwothree*rrrr)r9rKr(rBrrhrqrrrrseekrrrrMrrrrrtest_builtin_listes:        zTestCase.test_builtin_listcCsl|ttdd|ttdd|tgd|tdd|tddddd d }|t|t||ttt|ttd ttd }z(t dD]}| d |qW| n | 0ttd}zZ|t|d| dd|t|dW| z t tWnty8Yn0n,| z t tWntydYn00dS)Nrerr r rUrkrrrIrlrrr r rUrrrrrr)r9tupler(rrhrqrKrrrBrrrrrrrrrtest_builtin_tuples>        zTestCase.test_builtin_tuplecCs<|ttdtdttdd|ttdtdg|ttddg|ttddgddddd }|ttd|t||ttdt|ttdd Gd d d }|d }|d}Gddd}|||gd}|ttdd||gd|ttddt||gddS)Nrer rrrIrr rUrrc@seZdZddZddZdS)z-TestCase.test_builtin_filter..BooleancSs ||_dSrtruth)rrrrrrsz6TestCase.test_builtin_filter..Boolean.__init__cSs|jSrrrrrr__bool__sz6TestCase.test_builtin_filter..Boolean.__bool__N)rrr rrrrrrBooleansrTFc@seZdZddZddZdS)z)TestCase.test_builtin_filter..SeqcWs ||_dSrvals)rargsrrrrsz2TestCase.test_builtin_filter..Seq.__init__cSsGddd}||jS)Nc@s$eZdZddZddZddZdS)zCTestCase.test_builtin_filter..Seq.__iter__..SeqItercSs||_d|_dSr )rr)rrrrrrszLTestCase.test_builtin_filter..Seq.__iter__..SeqIter.__init__cSs|SrrrrrrrszLTestCase.test_builtin_filter..Seq.__iter__..SeqIter.__iter__cSs0|j}|d|_|t|jkr(|j|StdSr)rlenrrr&rrrrs   zLTestCase.test_builtin_filter..Seq.__iter__..SeqIter.__next__Nrrr rrrrrrrSeqItersrr)rrrrrrs z2TestCase.test_builtin_filter..Seq.__iter__Nr#rrrrSeqsrcSs| Srrrmrrrr_z.TestCase.test_builtin_filter..cSs| Srrrrrrrr_) r9rKfilterr(rBrrhrqr?)rrMrZbTrueZbFalserr;rrrtest_builtin_filters"   zTestCase.test_builtin_filtercCsz|ttdd|ttdd|tddd|tddddddd }|t|d |t|d |t|d|tt|dttd }z*|d |d|dW| n | 0ttd}zZ|t|d| dd|t|dW| z t tWnt yFYn0n,| z t tWnt yrYn00dS)Nrerkrrr r rUrrrrz medium line zxtra large line zitty-bitty line r) r9maxr(minvaluesr?rrrrrrr)rrMrrrrtest_builtin_max_mins<         zTestCase.test_builtin_max_mincsL|ttddtdttdddddd|ttfd dttfd d tdD}ttd }z,td D]}| d|dqW| n | 0ttd}zP|ttt |ttdddW| z t tWnt yYn0n,| z t tWnt yDYn00dS)NcSs|dSrrrrrrrr_z+TestCase.test_builtin_map..rer rjr rUrcSs |||fSrr)rXrMrrrrr_cs>g|]6}|tkr|pd||tkr4|p6dfqSr)rr[rrMZdkeysrrr]s z-TestCase.test_builtin_map..rrOZxy r)r9rKmapr(rBitemsrrrrrrrr)rexpectedrrrrrtest_builtin_maps:        $  zTestCase.test_builtin_mapc Cs|ttg|ttgg|ttddgddg|ttd|tttdd|tttdt|tttdgd|tttdgdd d dd }|t|tt|| Gd d d }t t d}z| dW| n | 0t t d}zR|tt|d||dgdW| z tt WntyrYn0n,| z tt WntyYn00|tttdddtdDGddd}Gddd|}Gddd|}dd}|t|d|t|d||||td||||td||||tdd dtdD}|||fD]0} |||fD]} ||| | |qqdS)!Nr r ab)r rl)r rrOrrU)r)r )r r r rc@s$eZdZddZddZddZdS)z+TestCase.test_builtin_zip..IntsFromcSs ||_dSrr%)rstartrrrr#sz4TestCase.test_builtin_zip..IntsFrom.__init__cSs|Srrrrrrr&sz4TestCase.test_builtin_zip..IntsFrom.__iter__cSs|j}|d|_|Srr%r&rrrr)s z4TestCase.test_builtin_zip..IntsFrom.__next__NrrrrrIntsFrom"srrz a bbb cc rr))ra r)r zbbb i)r zcc irecSsg|] }|fqSrrrrrrr]@r_z-TestCase.test_builtin_zip..c@seZdZddZdS)z.TestCase.test_builtin_zip..NoGuessLen5cSs|dkr t|SNre)r)r&rrrr*Dsz:TestCase.test_builtin_zip..NoGuessLen5.__getitem__Nr0rrrr NoGuessLen5Csrc@seZdZddZdS)z-TestCase.test_builtin_zip..Guess3Len5cSsdSrTrrrrr__len__Jsz5TestCase.test_builtin_zip..Guess3Len5.__len__Nrrr rrrrr Guess3Len5Isrc@seZdZddZdS)z.TestCase.test_builtin_zip..Guess30Len5cSsdS)NrrrrrrNsz6TestCase.test_builtin_zip..Guess30Len5.__len__Nrrrrr Guess30Len5MsrcWs tt|Sr)rKzip)rrrrlzipQsz'TestCase.test_builtin_zip..lziprcSsg|] }||fqSrrrrrrr]Zr_)r9rKrrhrqrBr!r(rrrrrrrrr) rrMrrrrrrrrmyrrrtest_builtin_zips` "     &zTestCase.test_builtin_zipcCsGddd}ttd}z|dW|n |0ttd}zFd||}||dW|z ttWqtyYq0n*|z ttWntyYn00dS)Nc@s$eZdZddZddZddZdS)z4TestCase.test_unicode_join_endcase..OhPhooeycSst||_d|_dSr )r?r:r)rr;rrrrds z=TestCase.test_unicode_join_endcase..OhPhooey.__init__cSs|Srrrrrrrhsz=TestCase.test_unicode_join_endcase..OhPhooey.__iter__cSs&|j}|d|_|dkrdSt|jS)Nr r z fooled you!)rr'r:r&rrrrks  z=TestCase.test_unicode_join_endcase..OhPhooey.__next__NrrrrrOhPhooeycsrra b c rz - za - b - fooled you! - c )rrrrjoinr9rr)rrrZgotrrrtest_unicode_join_endcase_s&       z"TestCase.test_unicode_join_endcasecCsxtdtdfD]J}tdD]}|||qdddddgddid|f D]}|||qHq|tttdg|ttdg|tttt g|ttt g|t tttg|t ttg| t dd | t d d | t d d dd d dd}|D]"}|||||| q| D]"}||| |||qB|D]2\}}|||f||||f|qnttd}z|dW|n |0ttd}zndD]:}|dd||||dd||d|qW|z ttWntyDYn0n,|z ttWntypYn00dS)NrerIrg(\5E@)rUrkr y@(cSsddvS)NrU rrrrrrr_z-TestCase.test_in_and_not_in..cSsdtvSrT)rrrrrrr_cSs dtvSrT)r3rrrrrr_r rU@rrr?rrrrr)r!r(rBZassertInZ assertNotInrr$r?r,rrhrqr4rrrrrrrrr)rZsc5rrMrXvrchunkrrrtest_in_and_not_insZ           zTestCase.test_in_and_not_incCsddlm}||gddd||ddd||ddd||ddd|t|d d |t|||dddd d }|D]}||||d q|||dd|||d d |||d dttd}z|dW| n | 0ttd}z`dD],\}}| dd||||d|q W| z t tWnt yxYn0n,| z t tWnt yYn00dS)Nr)countOfr r r rUr rer rU12232526rr rrrrza b c b r))rlr )rr )rr )rMrr) operatorrr9rhrqrrrrrrrr)rrrMrXrZlettercountrrr test_countOfs@        zTestCase.test_countOfcCsddlm}||gddd||ddd||ddd||ddd|t|dd|t|dd||d d d||d d d|t|d d |t|d d|t||||t|tdtt d}z| dW| n | 0tt d}z|t |}|||dd|||dd|||dd|t||dW| z t t WntyYn0n,| z t t WntyYn00td}tdD]}|||||q|t||ddS)Nr)indexOfrr r rUrerjrr5rrrz a b c d e rzb zd ze rr)rrr9rh ValueErrorrqr4r3rrrrr?rrr!rB)rrrZfiterZiclassrrrr test_indexOfsL       zTestCase.test_indexOfcsttd}z|t|jd|t|jd|ddg|d|ddi|iGdddGfd d d }||d d |tt}d dtdd D}|t||W|z t tWnt yYn0n,|z t tWnt yYn00dS)Nrrrr)rrz5 c@s$eZdZddZddZddZdS)z*TestCase.test_writelines..IteratorcSs||_||_|j|_dSr)rfinishrrrrrrrrsz3TestCase.test_writelines..Iterator.__init__cSs0|j|jkrtt|jd}|jd7_|S)Nrr )rrrstr)rresultrrrrs  z3TestCase.test_writelines..Iterator.__next__cSs|Srrrrrrr!sz3TestCase.test_writelines..Iterator.__iter__NrrrrrrJsrJcs eZdZddZfddZdS)z*TestCase.test_writelines..WhatevercSs||_||_dSrrrrrrrr%sz3TestCase.test_writelines..Whatever.__init__cs|j|jSrrrrJrrr)sz3TestCase.test_writelines..Whatever.__iter__Nr#rrrrWhatever$srrjicSsg|]}t|dqS)r)rrrrrr]0r_z,TestCase.test_writelines..r ) rrrhrq writelinesrrBr9rKrr)rrrrrrrtest_writeliness2      zTestCase.test_writelinescCsd\}}|||fdtd\}}}||||fdztd\}}WntyZYn 0|dztd\}}}WntyYn 0|dzt\}}}WntyYn 0|ddddd\}}}||||fd ttd }d }z |D]}| |qW| n | 0ttd }zJ|\}}}||||f|W| z t tWnt yvYn0n,| z t tWnt yYn00tddd i\}}\}||||fddS)NrrUr zshould have raised ValueErrorr zshould have raised TypeErrorr)r r rU)rrrr)rzbb zccc r)rr r) r9r!rrrrqrrrrrrr)rrlrrrlineslinerrrtest_unpack_iter<sV           zTestCase.test_unpack_itercCsGdddt}|}||jd~||jd|||g}||jdzt|\}}WntytYn0~||jddS)Nc@s eZdZdZddZddZdS)z.TestCase.test_ref_counting_behavior..CrcSs|jd7_t|Sr)rrp__new__)clsrrrrusz6TestCase.test_ref_counting_behavior..C.__new__cSs|j}|jd8_dSr) __class__r)rrrrr__del__xsz6TestCase.test_ref_counting_behavior..C.__del__N)rrr rrrrrrrrussrur rrU)rpr9rr?r)rrurmlrlrrrrtest_ref_counting_behaviorqs  z#TestCase.test_ref_counting_behaviorcCsPttd}t|}|t|ttd|tdd|t|gdSNrerO)rKrBr?r9extendrrlrrrrtest_sinkstate_lists  zTestCase.test_sinkstate_listcCs8d}t|}|t|ttd|t|gdS)Nrrer?r9rKrBr rrrtest_sinkstate_tupleszTestCase.test_sinkstate_tuplecCs4d}t|}|t|gd|t|gdSr)r?r9rKr rrrtest_sinkstate_stringszTestCase.test_sinkstate_stringcCsBtd}t|}|t|ttdd|_|t|gdSr )r(r?r9rKrBrr rrrtest_sinkstate_sequences z TestCase.test_sinkstate_sequencecCsDdgfdd}t|d}|t|ttd|t|gdS)NrcSs(|d}|d|d<|dkr$td|S)Nrr rOzshouldn't have gotten this far)AssertionErrorryrrrr{s  z.TestCase.test_sinkstate_callable..spamrer )rr{rrrrtest_sinkstate_callables z TestCase.test_sinkstate_callablecCsbdddddd}t||||fD]0}t|}|tt|d|t|gq,dS)Nr r rrkrU)r r rrkrUre)r?rrrr9rrKr rrrtest_sinkstate_dicts  zTestCase.test_sinkstate_dictcCs:dd}|}|t|ttd|t|gdS)NcsstdD] }|VqdSrrBr%rrrgens z*TestCase.test_sinkstate_yield..genre)r9rKrB)rrrrrrtest_sinkstate_yieldszTestCase.test_sinkstate_yieldcCs<td}t|}|t|ttd|t|gdSr)rBr?r9rKr rrrtest_sinkstate_rangeszTestCase.test_sinkstate_rangecCsNtd}t|}t|}|t|tttdtd|t|gdSr)rB enumerater?r9rKr)rrlrrrrrtest_sinkstate_enumerates "z!TestCase.test_sinkstate_enumeratecs<GfdddtzD]}qWnty6Yn0dS)Ncs eZdZddZfddZdS)z'TestCase.test_3720..BadIteratorcSs|Srrrrrrrsz0TestCase.test_3720..BadIterator.__iter__cs`dSr)rr BadIteratorrrrsz0TestCase.test_3720..BadIterator.__next__N)rrr rrrrrrrsr)rprqr&rrr test_3720s   zTestCase.test_3720cCsJdd}dgd}tdD]}|dq|||t|ddS)NcsstdD] }|VqdS)Nrr%rrrrs zITestCase.test_extending_list_with_iterator_does_not_segfault..genrri)rBpopr r9r)rrZlstrrrr3test_extending_list_with_iterator_does_not_segfaults     zrAr7rPrQrSrYr^r`rarbrcrdrirnrrrxr}rrrrrrrrrrrrrrrrrrrrrrrrrr rrrrrrrrrr r&r'r(r*rrrrr6Usv  $    !- O*1#*55    r6cCs ttdSr)rr6rrrr test_mainsr+__main__)r"ZunittestZ test.supportrrrrrrrr<collections.abcrHrVr r!r$r(r,r/r1r2r3r6r+rrrrrs4  A