a zeҁ@sddlZddlZddlmZmZmZddlmZmZmZddl Z ddl Z gdZ GdddZ GdddZGd d d ZGd d d ZGd ddZGdddZGdddZGdddZGdddZGdddejZedkredS)N)TESTFNunlink cpython_only)check_free_after_iterating ALWAYS_EQNEVER_EQ))rrr)rr)rr)rrr)rrrrrr )rr r)rr r)rr r )rrr)rrr)rrr )rrr)rrr)rrr )rr r)rr r)rr r )r rr)r rr)r rr )r rr)r rr)r rr )r r r)r r r)r r r c@s$eZdZddZddZddZdS)BasicIterClasscCs||_d|_dSNr)niselfr rB/opt/bitninja-python-dojo/embedded/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 ||_dSrr rrrrr'szIteratingSequenceClass.__init__cCs t|jSr)r r rrrrr)szIteratingSequenceClass.__iter__Nrrrrrrrrrr &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__Nrrrrr)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__Nrrrr)rrrrr.Csr.c@s eZdZdS)DefaultIterClassN)rrrrrrrr0Gsr0c@seZdZddZdZdS) NoIterClasscCs|Srrr%rrrr)KszNoIterClass.__getitem__N)rrrr)rrrrrr1Jsr1c@seZdZddZdS)BadIterableClasscCstdSr)ZeroDivisionErrorrrrrrPszBadIterableClass.__iter__Nrrrrrrrrr2Osr2c@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;drdsZrArJrrrrtest_iter_basicszTestCase.test_iter_basiccCs.ttd}t|}t|}|||udSrM)rJrAr>rE)rr:r9Zit2rrrtest_iter_idempotencys zTestCase.test_iter_idempotencycCs |ttdttddSrMr@r>rArJrrrrtest_iter_for_loopszTestCase.test_iter_for_loopcCsVtd}g}t|D]0}t|D]"}t|D]}||||fq,q q||tdSN)rAr>r7r8TRIPLETS)rr:rrjkrrrtest_iter_independences   zTestCase.test_iter_independencecs.tdfddtD}||tdS)NrTcs2g|]*}tD]}tD]}|||fqqqSr)r>.0rrVrWr:rr sz.)rAr>r8rUrrr[rtest_nested_comprehensions_iters  z(TestCase.test_nested_comprehensions_itercs*tdfddD}||tdS)NrTcs*g|]"}D]}D]}|||fqq qSrrrYr[rrr\z;TestCase.test_nested_comprehensions_for..)rAr8rUrrr[rtest_nested_comprehensions_forsz'TestCase.test_nested_comprehensions_forcCs|tdttddSrM)r@r rJrArrrrtest_iter_class_forszTestCase.test_iter_class_forcCs |ttdttddSrM)r=r>r rJrArrrrtest_iter_class_iterszTestCase.test_iter_class_itercCs|tdttddSrM)r@r'rJrArrrrtest_seq_class_forszTestCase.test_seq_class_forcCs |ttdttddSrM)r=r>r'rJrArrrrtest_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'rAr;rBr>rCrDr ZassertIstyper8rJr& assertRaisesrrErFrGrHrI)rZorigrKZitorigrLr9r: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)Nrdre)rrr rTrdri)r'r>r&r r8rJ)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__Nr4rrrr IterClasssrn)objectrg TypeErrorr>)rrnrrrtest_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)Nrd)rr(r%rrr__call__s  z/TestCase.test_iter_callable..C.__call__N)rrrrrsrrrrCsrtrNFr;r=r>rJrA)rrtrrrtest_iter_callables zTestCase.test_iter_callablecCs0dgfdd}|jt|dttddddS)NrcSs|d}|d|d<|S)Nrrrstaterrrrspams z)TestCase.test_iter_function..spamrNFrurvrrzrrrtest_iter_functionszTestCase.test_iter_functioncCs0dgfdd}|jt|dttddddS)NrcSs$|d}|dkrt|d|d<|S)NrrNr)rrxrrrrzs  z.TestCase.test_iter_function_stop..spamrNFrurvr{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)NrrrN) RuntimeErrorrxrrrrzs  z.TestCase.test_exception_function..spamr}rNshould have raised RuntimeError)r>r7rr8rJrAfail)rrzrrlrrrtest_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||SrM)rr'r)r%rrrr) szETestCase.test_exception_sequence..MySequenceClass.__getitem__Nr/rrrrMySequenceClasssrr}rNr)r'r7rr8rJrAr)rrrrlrrrtest_exception_sequences  z TestCase.test_exception_sequencecCs0Gdddt}|j|dttddddS)Nc@seZdZddZdS)z4TestCase.test_stop_sequence..MySequenceClasscSs|dkr tt||SrM)rr'r)r%rrrr)0sz@TestCase.test_stop_sequence..MySequenceClass.__getitem__Nr/rrrrr/srr}rNFru)r'r@rJrA)rrrrrtest_stop_sequence.szTestCase.test_stop_sequencecCs |ttdttddS)Ni'rQrrrrtest_iter_big_range7szTestCase.test_iter_big_rangecCs|tggdSrr@r>rrrrtest_iter_empty;szTestCase.test_iter_emptycCs|tdttddS)N) rrr rTrjrdrire rN)r@r>rJrArrrrtest_iter_tuple?szTestCase.test_iter_tuplecCs |ttdttddSrMrQrrrrtest_iter_rangeCszTestCase.test_iter_rangecCs|tdgddSNZabcde)rkbcrLerrrrrtest_iter_stringGszTestCase.test_iter_stringcCs2i}tdD] }d||<q ||t|dSrM)rAr@rJkeys)rdictrrrrtest_iter_dictKs  zTestCase.test_iter_dictcCsttd}z(tdD]}|d|qW|n |0ttd}zP|j|gddd|j|gddW|z ttWqtyYq0n*|z ttWntyYn00dS)Nwrd%d rz0 1 2 3 4 Fru)openrrAwritecloser@rOSError)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) Nrdrrrr rTonetwothree*rrrr)r8rJr'rArrgrprrrrseekrrrrLrrrrrtest_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)Nrdrrr rTrjrrrHrkrrrr rTrrrrrr)r8tupler'rrgrprJrrrArrrrrrrrrtest_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)NrdrrrrHrr rTrrc@seZdZddZddZdS)z-TestCase.test_builtin_filter..BooleancSs ||_dSrtruth)rrrrrrsz6TestCase.test_builtin_filter..Boolean.__init__cSs|jSrrrrrr__bool__sz6TestCase.test_builtin_filter..Boolean.__bool__N)rrrrrrrrrBooleansrTFc@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__NrrrrrrrrrrSeqItersrr)rrrrrrs z2TestCase.test_builtin_filter..Seq.__iter__Nr"rrrrSeqsrcSs| Srrrlrrrr^z.TestCase.test_builtin_filter..cSs| Srrrrrrrr^) r8rJfilterr'rArrgrpr>)rrLrZbTrueZbFalserr: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)Nrdrjrrrr rTrrrrz medium line zxtra large line zitty-bitty line r) r8maxr'minvaluesr>rrrrrrr)rrLrrrrtest_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..rdrrir rTrcSs |||fSrr)rWrLrrrrr^cs>g|]6}|tkr|pd||tkr4|p6dfqSr)rrZrrLZdkeysrrr\s z-TestCase.test_builtin_map..rrNZxy r)r8rJmapr'rAitemsrrrrrrrr)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)!Nrr ab)rrk)r rrNrrT)r)r)r rr 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)rzbbb i)r zcc irdcSsg|] }|fqSrrrrrrr\@r^z-TestCase.test_builtin_zip..c@seZdZddZdS)z.TestCase.test_builtin_zip..NoGuessLen5cSs|dkr t|SNrd)r(r%rrrr)Dsz:TestCase.test_builtin_zip..NoGuessLen5.__getitem__Nr/rrrr NoGuessLen5Csrc@seZdZddZdS)z-TestCase.test_builtin_zip..Guess3Len5cSsdSrSrrrrr__len__Jsz5TestCase.test_builtin_zip..Guess3Len5.__len__Nrrrrrrrr Guess3Len5Isrc@seZdZddZdS)z.TestCase.test_builtin_zip..Guess30Len5cSsdS)NrrrrrrNsz6TestCase.test_builtin_zip..Guess30Len5.__len__Nrrrrr Guess30Len5MsrcWs tt|Sr)rJzip)rrrrlzipQsz'TestCase.test_builtin_zip..lziprcSsg|] }||fqSrrrrrrr\Zr^)r8rJrrgrprAr r'rrrrrrrrr) rrLrrrrrrrrlyrrrtest_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>r9r)rr:rrrrds z=TestCase.test_unicode_join_endcase..OhPhooey.__init__cSs|Srrrrrrrhsz=TestCase.test_unicode_join_endcase..OhPhooey.__iter__cSs&|j}|d|_|dkrdSt|jS)Nrr z fooled you!)rr&r9r%rrrrks  z=TestCase.test_unicode_join_endcase..OhPhooey.__next__NrrrrrOhPhooeycsrra b c rz - za - b - fooled you! - c )rrrrjoinr8rr)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)NrdrHrg(\5E@)rTrjry@(cSsddvS)NrT rrrrrrr^z-TestCase.test_in_and_not_in..cSsdtvSrS)rrrrrrr^cSs dtvSrS)r2rrrrrr^r rT@rrr?rrrrr)r r'rAZassertInZ assertNotInrr#r>r+rrgrpr3rrrrrrrrr)rZsc5rrLrWvrchunkrrrtest_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)countOfrr r rTr rdr rT12232526rrrrrrza b c b r))rkr)rr )rr)rLrr) operatorrr8rgrprrrrrrrr)rrrLrWrZlettercountrrr 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)indexOfrrr rTrdrirr5rrrz a b c d e rzb zd ze rr)rrr8rg ValueErrorrpr3r2rrrrr>rrr rA)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__NrrrrrrIsrIcs eZdZddZfddZdS)z*TestCase.test_writelines..WhatevercSs||_||_dSrrrrrrrr%sz3TestCase.test_writelines..Whatever.__init__cs|j|jSrrrrIrrr)sz3TestCase.test_writelines..Whatever.__iter__Nr"rrrrWhatever$srriicSsg|]}t|dqS)r)rrrrrr\0r^z,TestCase.test_writelines..r) rrrgrp writelinesrrAr8rJrr)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)NrrTr zshould have raised ValueErrorr zshould have raised TypeErrorr)rr rT)rrrr)rzbb zccc r)rrr) r8r rrrrprrrrrrr)rrkrrrlineslinerrrtest_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)rro__new__)clsrrrrusz6TestCase.test_ref_counting_behavior..C.__new__cSs|j}|jd8_dSr) __class__r)rrrrr__del__xsz6TestCase.test_ref_counting_behavior..C.__del__N)rrrrrrrrrrrtssrtrrrT)ror8rr>r)rrtrllrkrrrrtest_ref_counting_behaviorqs  z#TestCase.test_ref_counting_behaviorcCsPttd}t|}|t|ttd|tdd|t|gdSNrdrN)rJrAr>r8extendrrkrrrrtest_sinkstate_lists  zTestCase.test_sinkstate_listcCs8d}t|}|t|ttd|t|gdS)Nrrdr>r8rJrAr rrrtest_sinkstate_tupleszTestCase.test_sinkstate_tuplecCs4d}t|}|t|gd|t|gdSr)r>r8rJr rrrtest_sinkstate_stringszTestCase.test_sinkstate_stringcCsBtd}t|}|t|ttdd|_|t|gdSr)r'r>r8rJrAr r rrrtest_sinkstate_sequences z TestCase.test_sinkstate_sequencecCsDdgfdd}t|d}|t|ttd|t|gdS)NrcSs(|d}|d|d<|dkr$td|S)NrrrNzshouldn't have gotten this far)AssertionErrorrxrrrrzs  z.TestCase.test_sinkstate_callable..spamrdr )rrzrrrrtest_sinkstate_callables z TestCase.test_sinkstate_callablecCsbdddddd}t||||fD]0}t|}|tt|d|t|gq,dS)Nrr rrjrT)rr rrjrTrd)r>rrrr8rrJr rrrtest_sinkstate_dicts  zTestCase.test_sinkstate_dictcCs:dd}|}|t|ttd|t|gdS)NcsstdD] }|VqdSrrAr$rrrgens z*TestCase.test_sinkstate_yield..genrd)r8rJrA)rrrrrrtest_sinkstate_yieldszTestCase.test_sinkstate_yieldcCs<td}t|}|t|ttd|t|gdSr)rAr>r8rJr rrrtest_sinkstate_rangeszTestCase.test_sinkstate_rangecCsNtd}t|}t|}|t|tttdtd|t|gdSr)rA enumerater>r8rJr)rrkrrrrrtest_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)rrrrrrrrrrsr)rorpr%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)rApopr r8r)rrZlstrrrr3test_extending_list_with_iterator_does_not_segfaults     zr. __setstate__sysmaxsizer8r&rg OverflowErrorrr9rrrtest_iter_overflows  & zTestCase.test_iter_overflowcCs8tt}|d|t|d|t|ddS)Nirr)r>r.r r8r&r$rrrtest_iter_neg_setstates  zTestCase.test_iter_neg_setstatecCst|ttddS)Nr)rr>r'rrrrtest_free_after_iteratingsz"TestCase.test_free_after_iteratingcCs2ttfD]}|tt|q|tttdSr)r0r1rgrpr>r3r2)rtyprrrtest_error_iter s zTestCase.test_error_iterN)T)T)=rrrr=r@r6rOrPrRrXr]r_r`rarbrcrhrmrqrwr|r~rrrrrrrrrrrrrrrrrrrrrrrrr r rrrrrrrrrr%r&r'r)rrrrr5Usv  $    !- O*1#*55    r5__main__)r!ZunittestZ test.supportrrrrrrr;collections.abcrGrUr r r#r'r+r.r0r1r2r5rmainrrrrs2  A