a zeb@sdZddlZddlmZddlmZddlmZdZGdddZ Gd d d e Z Gd d d e ej Z Gd dde ej Z Gddde ej ZGddde ej ZGddde ej ZGddde ej ZGddde ej ZGddde ej ZGddde ej ZGddde ej ZGdd d e ej ZGd!d"d"e ej ZGd#d$d$eZGd%d&d&eZGd'd(d(eZGd)d*d*ej Zed+kredS),a Test Iterator Length Transparency Some functions or methods which accept general iterable arguments have optional, more efficient code paths if they know how many items to expect. For instance, map(func, iterable), will pre-allocate the exact amount of space required whenever the iterable can report its length. The desired invariant is: len(it)==len(list(it)). A complication is that an iterable and iterator can be the same object. To maintain the invariant, an iterator needs to dynamically update its length. For instance, an iterable such as range(10) always reports its length as ten, but it=iter(range(10)) starts at ten, and then goes to nine after next(it). Having this capability means that map() can ignore the distinction between map(func, iterable) and map(func, iter(iterable)). When the iterable is immutable, the implementation can straight-forwardly report the original length minus the cumulative number of calls to next(). This is the case for tuples, range objects, and itertools.repeat(). Some containers become temporarily immutable during iteration. This includes dicts, sets, and collections.deque. Their implementation is equally simple though they need to permanently set their length to zero whenever there is an attempt to iterate after a length mutation. The situation slightly more involved whenever an object allows length mutation during iteration. Lists and sequence iterators are dynamically updatable. So, if a list is extended during iteration, the iterator will continue through the new items. If it shrinks to a point before the most recent iteration, then no further items are available and the length is reported at zero. Reversed objects can also be wrapped around mutable objects; however, any appends after the current position are ignored. Any other approach leads to confusion and possibly returning the same item more than once. The iterators not listed above, such as enumerate and the other itertools, are not length transparent because they have no way to distinguish between iterables that report static length and iterators whose length changes with each call (i.e. the difference between enumerate('abc') and enumerate(iter('abc')). N)repeat)deque) length_hint c@seZdZddZdS)TestInvariantWithoutMutationscCsh|j}ttdtdD]}|t||t|q|t|d|tt||t|ddSNr) itreversedrangen assertEqualrnext assertRaises StopIteration)selfr irE/opt/bitninja-python-dojo/embedded/lib/python3.9/test/test_iterlen.pytest_invariant6s z,TestInvariantWithoutMutations.test_invariantN)__name__ __module__ __qualname__rrrrrr4src@seZdZddZdS)TestTemporarilyImmutablecCs\|j}|t|tt||t|td||tt||t|ddSr)r r rr rmutater RuntimeError)rr rrrtest_immutable_during_iterationAsz8TestTemporarilyImmutable.test_immutable_during_iterationN)rrrrrrrrr?src@seZdZddZdS) TestRepeatcCstdt|_dSN)rr r rrrrsetUpQszTestRepeat.setUpNrrrr rrrrrOsrc@seZdZddZdS) TestXrangecCsttt|_dSriterr r r rrrrr VszTestXrange.setUpNr!rrrrr"Tsr"c@seZdZddZdS)TestXrangeCustomReversedcCsttt|_dSrr r r r rrrrr [szTestXrangeCustomReversed.setUpNr!rrrrr%Ysr%c@seZdZddZdS) TestTuplecCstttt|_dSr)r$tupler r r rrrrr `szTestTuple.setUpNr!rrrrr'^sr'c@seZdZddZdS) TestDequecCs"ttt}t||_|j|_dSr)rr r r$r poprrdrrrr gs  zTestDeque.setUpNr!rrrrr)esr)c@seZdZddZdS)TestDequeReversedcCs"ttt}t||_|j|_dSr)rr r r r r*rr+rrrr ns  zTestDequeReversed.setUpNr!rrrrr-lsr-c@seZdZddZdS) TestDictKeyscCs$ttt}t||_|j|_dSr)dictfromkeysr r r$r popitemrr+rrrr us zTestDictKeys.setUpNr!rrrrr.ssr.c@seZdZddZdS) TestDictItemscCs(ttt}t||_|j|_dSr) r/r0r r r$itemsr r1rr+rrrr |szTestDictItems.setUpNr!rrrrr2zsr2c@seZdZddZdS)TestDictValuescCs(ttt}t||_|j|_dSr) r/r0r r r$valuesr r1rr+rrrr szTestDictValues.setUpNr!rrrrr4sr4c@seZdZddZdS)TestSetcCs"ttt}t||_|j|_dSr)setr r r$r r*rr+rrrr s  z TestSet.setUpNr!rrrrr6sr6c@seZdZddZddZdS)TestListcCsttt|_dSrr#rrrrr szTestList.setUpcCsttt}t|}t|t||t|td|t|t|tdg|dd<|t|d|t|g|td|t|ddSNrr) listr r r$rr rappendextendrr,r rrr test_mutations   zTestList.test_mutationNrrrr r@rrrrr8sr8c@seZdZddZddZdS)TestListReversedcCsttt|_dSrr&rrrrr szTestListReversed.setUpcCsttt}t|}t|t||t|td|t|t|tdg|dd<|t|d|t|g|td|t|ddSr9) r<r r r rr rr=r>r?rrrr@s   zTestListReversed.test_mutationNrArrrrrBsrBc@seZdZddZddZdS)BadLencCs ttdSNrr$r rrrr__iter__szBadLen.__iter__cCs tddSNZhellorrrrr__len__szBadLen.__len__N)rrrrFrIrrrrrCsrCc@seZdZddZddZdS) BadLengthHintcCs ttdSrDrErrrrrFszBadLengthHint.__iter__cCs tddSrGrHrrrr__length_hint__szBadLengthHint.__length_hint__NrrrrFrKrrrrrJsrJc@seZdZddZddZdS)NoneLengthHintcCs ttdSrDrErrrrrFszNoneLengthHint.__iter__cCstSr)NotImplementedrrrrrKszNoneLengthHint.__length_hint__NrLrrrrrMsrMc@seZdZddZddZdS)TestLengthHintExceptionscCsx|ttt|ttt|tgjt|tgjtttd}|t|jt|t|jtdSrD)rrr<rCrJr> bytearrayr )rbrrrtest_issue1242657s z*TestLengthHintExceptions.test_issue1242657cCs|ttttddSrD)r r<rMr rrrrtest_invalid_hintsz*TestLengthHintExceptions.test_invalid_hintN)rrrrRrSrrrrrOs rO__main__)__doc__Zunittest itertoolsr collectionsroperatorrr rrZTestCaserr"r%r'r)r-r.r2r4r6r8rBobjectrCrJrMrOrmainrrrrs2+