Your IP : 216.73.216.213
�
f f7_ � @ sU d d l Z d d l m Z d d l Td d l Z d d l m Z d d l m Z d d l Z d d l
Z
d d l Z d d l Z d d l
Z
d d l m Z d d l Z d d l Z e j Z e d Z d d � Z d
d � Z d d
� Z d d � Z d d � Z d d � Z d d � Z d d � Z Gd d � d � Z d d � Z d d � Z d d � Z d d! � Z d" d# � Z! d$ d% � Z" d& d' � e# e
j$ d � D� Z% Gd( d) � d) e j& � Z' Gd* d+ � d+ e j& � Z( Gd, d- � d- e j& � Z) d. d/ � Z* Gd0 d1 � d1 � Z+ Gd2 d3 � d3 � Z, Gd4 d5 � d5 � Z- Gd6 d7 � d7 � Z. Gd8 d9 � d9 � Z/ Gd: d; � d; � Z0 Gd< d= � d= � Z1 d> d? � Z2 Gd@ dA � dA e j& � Z3 GdB dC � dC e j& � Z4 GdD dE � dE e j& � Z5 GdF dG � dG e j& � Z6 e j7 GdH dI � dI e j& � � Z8 dJ Z9 i e9 dK 6Z: d dL dM � Z; e< dN k rQe; dO dP � n d S)Q� N)�support)�*)�Decimal)�Fraction)�reduce� c G s t t | � � S)N)�list�zip)�args� r �8/opt/alt/python34/lib64/python3.4/test/test_itertools.py�lzip s r
c C s d | S)z Test function of one argument� r )�xr r r �onearg s r c G s
t � d S)z"Test function that raises an errorN)�
ValueError)r
r r r �errfunc s r c c s x d D] } | Vq Wd S)z Non-restartable source sequencer r r N)r r r r )�ir r r �gen3 s
r c C s | d d k S)zTest predicater r r )r r r r �isEven" s r c C s | d d k S)zTest predicater r r )r r r r �isOdd& s r c G s | S)Nr )r
r r r �tupleize* s r c c s x t | � D] } | Vq
Wd S)N)�range)�nr r r r �irange- s r c @ s. e Z d Z d Z d d � Z d d � Z d S)�StopNowz"Class emulating an empty iterable.c C s | S)Nr )�selfr r r �__iter__3 s zStopNow.__iter__c C s
t � d S)N)�
StopIteration)r r r r �__next__5 s zStopNow.__next__N)�__name__�
__module__� __qualname__�__doc__r r r r r r r 1 s r c C s t t | | � � S)zHConvenience function for partially consuming a long of infinite iterable)r �islice)r �seqr r r �take8 s r& c C s t t j | d � S)Nr )r �operator�mul)�iterabler r r �prod< s r* c C s t t d | d � � S)Z Factorialr )r* r )r r r r �fact? s r+ c C s | d S)Nr r )�rr r r �testRD s r- c C s | d S)Nr r )r, r r r �testR2G s r. c C s
| d k S)N�
r )r r r r �undertenJ s r0 c C s g | ] } | d d � � q S)c S s t j t j | | � � S)N)�pickle�loads�dumps)�s�protor r r �<lambda>M s z<listcomp>.<lambda>r )�.0r5 r r r �
<listcomp>M s r8 c @ s� e Z d Z d d d d d � Z d d � Z d d � Z d
d � Z d d
� Z d d � Z e j
d d � � Z e j d � d d � � Z
d d � Z e j
d d � � Z e j d � d d � � Z d d � Z e j
d d � � Z e j d � d d � � Z d! d" � Z d# d$ � Z d% d& � Z d' d( � Z d) d* � Z d+ d, � Z d- d. � Z d/ d0 � Z d1 d2 � Z e j d � d3 d4 � � Z d5 d6 � Z e j d � d7 d8 � � Z d9 d: � Z d; d< � Z! d= d> � Z" e j
d? d@ � � Z# e j d � dA dB � � Z$ dC dD � Z% dE dF � Z& dG dH � Z' dI dJ � Z( dK dL � Z) dM dN � Z* dO dP � Z+ dQ dR � Z, dS dT � Z- dU dV � Z. dW dX � Z/ dY dZ � Z0 d S)[� TestBasicOps� r Nc sy d � � f d d � � t j | | � } t j | � } | j t | � t | � � � | � � | � } } | j | | � | r� � | � }
| j | |
� n t j | � } d } y/ x( t | � D] }
t | � | d 7} q� WWn t k
r� Yn Xt j | | � } t j | � } � | � � | � } } | j | | � | ru� | | d � � }
| j | |
� n d S)zITest that an iterator is the same after pickling, also when part-consumedr c sz � d k r t d � � n t | t � r. | Sy t t | � � � } Wn t k
r_ | SYn X� � f d d � | D� S)Nr/ z infinite recursion encounteredc s# g | ] } � | � d � � q S)r r )r7 �e)�expandr r r r8 ^ s z;TestBasicOps.pickletest.<locals>.expand.<locals>.<listcomp>)� RuntimeError�
isinstance�strr r$ � TypeError)�itr �l)r< �stop)r r r< T s
z'TestBasicOps.pickletest.<locals>.expandr N)r1 r3 r2 � assertEqual�typer �nextr )r ZprotocolrA rC r& �compare�dumpZi2�a�b�cZi3Ztookr Zi4r )r< rC r �
pickletestR s0
zTestBasicOps.pickletestc C s
| j t t t d � � � d d d d d d d d d d
g
� | j t t d t d � � � d d d d d d d d d d
g
� xr t t t t f D]^ } | j t t t | t d � � � � t t | d d d d d d d d d d
g
� � � q� W| j t t d � � d
d d g � | j t t g � � g � | j t t d g � � d g � | j t
t t d � d d � | j t
t � | j t
t d t d � �| j t
t t d g g � � d d d d d d d d d d g
} | j t t | t � � d d d d d d d d d d g
� | j t t | t � � d d d d d d d d d d g
� | j t t | t
j � � d d d d d d d d d d g
� | j t
� � t t | t � � Wd QXx7 t t j d � D]" } | j | t t d � � � q�Wd S)Nr/ r r � � � � � �$ �- r) �abcrI �ab� � r r � � r: � � i� i� )rD r �
accumulater �int�complexr r �map� assertRaisesr@ �min�maxr' r( �chrr1 �HIGHEST_PROTOCOLrL )r �typr4 r5 r r r �test_accumulate{ s4 % % 8% " $%%%z TestBasicOps.test_accumulatec C s� d d � } x� t | f D]� } | j t | d d � � t d � � | j t | d � � t d � � | j t | d � � g � | j t d | d d � � t d � � | j t t | d d
� � q Wd S) Nc w s+ x$ | D] } x | D] } | Vq Wq Wd S)z Pure python version in the docsNr )� iterablesrA �elementr r r �chain2� s
z'TestBasicOps.test_chain.<locals>.chain2rT �def�abcdef� r: �abcdr rM )�chainrD r r& r` r@ )r ri rK r r r �
test_chain� s %" (zTestBasicOps.test_chainc C s� | j t t j d d g � � t d � � | j t t j d g � � t d � � | j t t j d g � � g � | j t d t j d d g � � t d � � | j t t t j d d g � � d S) NrT rj rk rl r: rm r rM )rD r rn �
from_iterabler& r` r@ )r r r r �test_chain_from_iterable� s
+(".z%TestBasicOps.test_chain_from_iterablec C s= x� t j g t D]� } t d d � } | j t | | � � t d � � | j t | � d � | j t | | � � t d � � | j t | t d � � � g � | j t d | t d d � � � t d � � | j t t | t d d
� � � q Wx@ t
t j d � D]+ } | j
| t d d � d t d � �q
Wd S)
NrT rj rk rI Zbcdefrl r: rm r rM r rG )�copy�deepcopy�
picklecopiersrn rD r rF r& r` r@ r r1 rd rL )r ZoperrA r5 r r r �test_chain_reducible� s """.&z!TestBasicOps.test_chain_reduciblec
s� | j t t d � | j t t d d d � | j t t d � | j t t d d � x6d d � g t D]!} | j t | t d d � � � g � | j t | t d d � � � d d d d d d! g � t d d � } t | � | j t | | � � d" d# d$ d% d& g � | j t | t t d � d
� � � d' d( d) d* g � t t d � d
� } t | � | j t | | � � d+ d, d- g � qi Wd d � } d d � } d d � } x t d � D]} d d � t | � D� � x�t | d � D]�} t t � | � � } | j t | � | | k r+d n! t
| � t
| � t
| | � � | j t | � t t | � � � | j | t | � � x� | D]� � | j t � � | � | j t t � � � | � | j t � � t � � � | j
t � f d d � � D� � � | j t � � � f d d � � D� � q�W| j | t | � | � � � | j | t | � | � � � | j | t | � | � � � x4 t t j d � D] } | j | t � | � � q�Wq�Wq�Wd S).NrT r r c S s | S)Nr )rI r r r r6 � s z0TestBasicOps.test_combinations.<locals>.<lambda>� �ABCD�A�B�C�Dr: rM r c 3 s t | � � t � � } | | k r( d St t | � � } t � f d d � | D� � Vx� x: t t | � � D]" } | | | | | k rn Pqn qn Wd S| | d 7<x1 t | d | � D] } | | d d | | <q� Wt � f d d � | D� � Vq[ Wd S)z%Pure python version shown in the docsNc 3 s | ] } � | Vq d S)Nr )r7 r )�poolr r � <genexpr>� s zHTestBasicOps.test_combinations.<locals>.combinations1.<locals>.<genexpr>r c 3 s | ] } � | Vq d S)Nr )r7 r )r| r r r} � s )�tuple�lenr r �reversed)r) r, r �indicesr �jr )r| r �
combinations1� s z5TestBasicOps.test_combinations.<locals>.combinations1c 3 su t | � � t � � } xV t t | � | � D]? } t | � t | � k r. t � f d d � | D� � Vq. q. Wd S)z%Pure python version shown in the docsc 3 s | ] } � | Vq d S)Nr )r7 r )r| r r r} � s zHTestBasicOps.test_combinations.<locals>.combinations2.<locals>.<genexpr>N)r~ r � permutationsr �sortedr )r) r, r r� r )r| r �
combinations2� s
z5TestBasicOps.test_combinations.<locals>.combinations2c 3 su t | � � t � � } xV t t | � | � D]? } t t | � � | k r. t � f d d � | D� � Vq. q. Wd S)z Pure python version from cwr()c 3 s | ] } � | Vq d S)Nr )r7 r )r| r r r} � s zHTestBasicOps.test_combinations.<locals>.combinations3.<locals>.<genexpr>N)r~ r � combinations_with_replacementr �set)r) r, r r� r )r| r �
combinations3� s
z5TestBasicOps.test_combinations.<locals>.combinations3rV c S s g | ] } d | d � q S)rW � r )r7 r r r r r8 � s z2TestBasicOps.test_combinations.<locals>.<listcomp>c 3 s | ] } | � k Vq d S)Nr )r7 r; )�valuesr r r} s z1TestBasicOps.test_combinations.<locals>.<genexpr>c s" g | ] } | � k r | � q Sr r )r7 r; )rK r r r8 s ���)rx ry )rx rz )rx r{ )ry rz )ry r{ )rz r{ )rx rz )rx r{ )ry rz )ry r{ )rz r{ )r r r )r r rM )r r rM )r r rM )r r rM )r r rM )r r rM )r` r@ � combinationsr rt rD r rF r r r+ r� r� �
assertTrue�allr1 rd rL )
r �op�testIntermediater� r� r� r r, �resultr5 r )rK r� r �test_combinations� sP %
$
F"
& z TestBasicOps.test_combinationsc C s- | j t t f � � t d d � Wd QXd S)N�AAr � i )r` �
OverflowError� MemoryErrorr� )r r r r �test_combinations_overflow s z'TestBasicOps.test_combinations_overflowz"tuple reuse is specific to CPythonc C sf | j t t t t t d d � � � � d � | j t t t t t t d d � � � � � d � d S)N�abcderM r )rD r r� r_ �idr� �assertNotEqualr )r r r r � test_combinations_tuple_reuse s .z*TestBasicOps.test_combinations_tuple_reusec
sn t } | j t | d � | j t | d d d � | j t | d � | j t | d d � x� d d � g t D]� } | j t | | d d � � � d d d d d d g � | d d � } t | � | j t | | � � d d d! d" d# g � qo Wd
d � } d d
� } d d � } xOt d � D]A} d d � t | � D� � x t | d � D]
} t | � | � � } | j t | � | | | � � | j t | � t t
| � � � | j | t | � � t t � | � � }
| d k s�| d k r| j | |
� n | j
t
| � t
|
� k � x� | D]� � | j t � � | � d d � t � � D� } | j t | � t t
| � � � | j t � � t � � � | j
t � f d d � � D� � � | j | � f d d � � D� � q-W| j | t | � | � � � | j | t | � | � � � x4 t t j d � D] } | j | | � | � � q?WqUWq%Wd S)$NrT r r c S s | S)Nr )rI r r r r6 s zATestBasicOps.test_combinations_with_replacement.<locals>.<lambda>�ABCrx ry rz c 3 s� t | � � t � � } | r) | r) d Sd g | } t � f d d � | D� � Vx~ x6 t t | � � D] } | | | d k rj Pqj qj Wd S| | d g | | | | d � <t � f d d � | D� � VqW Wd S)z%Pure python version shown in the docsNr c 3 s | ] } � | Vq d S)Nr )r7 r )r| r r r} . s zPTestBasicOps.test_combinations_with_replacement.<locals>.cwr1.<locals>.<genexpr>r c 3 s | ] } � | Vq d S)Nr )r7 r )r| r r r} 6 s )r~ r r� r )r) r, r r� r r )r| r �cwr1& s
#z=TestBasicOps.test_combinations_with_replacement.<locals>.cwr1c 3 sx t | � � t � � } xY t t | � d | �D]? } t | � t | � k r1 t � f d d � | D� � Vq1 q1 Wd S)z%Pure python version shown in the docs�repeatc 3 s | ] } � | Vq d S)Nr )r7 r )r| r r r} > s zPTestBasicOps.test_combinations_with_replacement.<locals>.cwr2.<locals>.<genexpr>N)r~ r �productr r� r )r) r, r r� r )r| r �cwr28 s
z=TestBasicOps.test_combinations_with_replacement.<locals>.cwr2c S s> | s | r d Sd St | | d � t | � t | d � S)Nr r )r+ )r r, r r r �numcombs@ s zATestBasicOps.test_combinations_with_replacement.<locals>.numcombsrV c S s g | ] } d | d � q S)rW r� r )r7 r r r r r8 F s zCTestBasicOps.test_combinations_with_replacement.<locals>.<listcomp>r c S s g | ] \ } } | � q Sr r )r7 �k�vr r r r8 V s c 3 s | ] } | � k Vq d S)Nr )r7 r; )r� r r r} Y s zBTestBasicOps.test_combinations_with_replacement.<locals>.<genexpr>c s" g | ] } | � k r | � q Sr r )r7 r; )rK r r r8 [ s r� )rx rx )rx ry )rx rz )ry ry )ry rz )rz rz )rx ry )rx rz )ry ry )ry rz )rz rz )r� r` r@ r rt rD r rF r r r� r� r� r� �groupbyr� r1 rd rL )
r �cwrr� r� r� r� r� r r, r� Z
regular_combsZnorunsr5 r )rK r� r �"test_combinations_with_replacement sL
"
" & z/TestBasicOps.test_combinations_with_replacementc C s- | j t t f � � t d d � Wd QXd S)Nr� r � i @)r` r� r� r� )r r r r �+test_combinations_with_replacement_overflowb s z8TestBasicOps.test_combinations_with_replacement_overflowc C sl t } | j t t t t | d d � � � � d � | j t t t t t | d d � � � � � d � d S)Nr� rM r )r� rD r r� r_ r� r� r )r r� r r r �.test_combinations_with_replacement_tuple_reuseh s .z;TestBasicOps.test_combinations_with_replacement_tuple_reusec s� | j t t � | j t t d d d � | j t t d � | j t t d d � | j t t d d � � g � | j t t d d � | j t t t d � d � � d d d d d d g � d d d � } d d
d � } x�t d � D]�} d
d � t | � D� � x�t | d � D]�} t t � | � � } | j t | � | | k rUd n t | � t | | � � | j t | � t t | � � � | j | t
| � � xf | D]^ } | j t | � | � | j t t | � � | � | j t � f d d � | D� � � q�W| j | t | � | � � � | j | t | � | � � � | | k r�| j | t t � d � � � | j | t t � � � � n x4 t t
j d � D] } | j | t � | � � q�WqWq� Wd S)NrT r r rv r4 rM r c 3 s� t | � � t � � } | d k r* | n | } | | k r@ d St t | � � } t t | | d | d � � d d d � } t � f d d � | d | � D� � Vx� | r�x� t t | � � D]� } | | d 8<| | d k r)| | d d � | | | d � | | d � <| | | | <q� | | } | | | | | | <| | <t � f d d � | d | � D� � VPq� Wd Sq� Wd S)z%Pure python version shown in the docsNr c 3 s | ] } � | Vq d S)Nr )r7 r )r| r r r} � s zHTestBasicOps.test_permutations.<locals>.permutations1.<locals>.<genexpr>r c 3 s | ] } � | Vq d S)Nr )r7 r )r| r r r} � s ���)r~ r r r r� )r) r, r r� Zcyclesr r� r )r| r �
permutations1x s& .( 0
(z5TestBasicOps.test_permutations.<locals>.permutations1c 3 s� t | � � t � � } | d k r* | n | } xY t t | � d | �D]? } t t | � � | k rI t � f d d � | D� � VqI qI Wd S)z%Pure python version shown in the docsNr� c 3 s | ] } � | Vq d S)Nr )r7 r )r| r r r} � s zHTestBasicOps.test_permutations.<locals>.permutations2.<locals>.<genexpr>)r~ r r� r r� )r) r, r r� r )r| r �
permutations2� s z5TestBasicOps.test_permutations.<locals>.permutations2rV c S s g | ] } d | d � q S)rW r� r )r7 r r r r r8 � s z2TestBasicOps.test_permutations.<locals>.<listcomp>c 3 s | ] } | � k Vq d S)Nr )r7 r; )r� r r r} � s z1TestBasicOps.test_permutations.<locals>.<genexpr>r� )r r )r r )r r )r r )r r )r r )r` r@ r� r rD r r r r+ r� r� r� r� r1 rd rL )r r� r� r r, r� �pr5 r )r� r �test_permutationsn s8 <"
* z TestBasicOps.test_permutationsc C s- | j t t f � � t d d � Wd QXd S)Nrx r r� i @)r` r� r� r� )r r r r �test_permutations_overflow� s z'TestBasicOps.test_permutations_overflowc C sf | j t t t t t d d � � � � d � | j t t t t t t d d � � � � � d � d S)Nr� rM r )rD r r� r_ r� r� r� r )r r r r � test_permutations_tuple_reuse� s .z*TestBasicOps.test_permutations_tuple_reusec s� x�t d � D]�} d d | � } x�t d � D]�� t t | d � �� } t t | � � � } t t | � � � } t t | � � � } | j t | � | � � | j t | � | r� t | � d � t � � t | d � n � � | j t | � � | k rd n t | � t | � � � | j t | � � | k rMd n! t | � t � � t | � � � | j | t t
| � � � | j | t t
| � � � | j | t t
| � � � | j | t t
| � � � | j | d d � | D� � | j | � f d d � | D� � | j | d
d � | D� � | j | � f d d � | D� � | j | t t t
| � j | � � � | j | t t t
| � j | � � � | j | t t
| � t
| � @� � q0 Wq
Wd S) NrN �ABCDEFGrX r� r r c S s. g | ]$ } t | � t | � k r | � q Sr )r� r )r7 �tr r r r8 � s z3TestBasicOps.test_combinatorics.<locals>.<listcomp>c s. g | ]$ } t t | � � � k r | � q Sr )r r� )r7 r� )r, r r r8 � s c S s. g | ]$ } t | � t | � k r | � q Sr )r� r )r7 r� r r r r8 � s c s. g | ]$ } t t | � � � k r | � q Sr )r r� )r7 r� )r, r r r8 � s )
r r r� r� r� r� rD r r+ r� r� �filter� __contains__)r r r4 r* r� ZpermZcombr )r, r �test_combinatorics� s, I<F # #((z TestBasicOps.test_combinatoricsc C s� | j t t d d d d d d d d d g � � t d � � | j t t d d d d d d d g � � t d � � | j t t d d d d d d d g � � t d � � | j t t d d d d d d d g � � t d � � | j t t d d d d g � � t d � � | j t t d d d d d d d g � � t d
� � d } t j t t d � | � � } t j t d � � } | j t t | | � � d d
d g | � | j t t d t d � � | j t t t d � d � | j t t t d � � | j t t t d � d � x}d d � d d � g t D]_} xVd d d d d d d g d d f d d d d d d d g d d f d d d d d d d g d d f d d d d g d d f d d d d d d d g d
d f g D]� \ } } } } | j t | t d | d | � � � t | � � | j t | t | | � � � t | � � t | | � } | r�t
| � | j t | | � � t | � � q�q�Wq6Wd S)N�data�ABCDEF� selectorsr r �ACEFrl ZACr� ZBCi' rN rM rW c S s
t j | � S)N)rr )rI r r r r6 � s z,TestBasicOps.test_compress.<locals>.<lambda>c S s
t j | � S)N)rr rs )rI r r r r6 � s ZCEFZBCDEFrz )r r ) rD r �compressrn rp r� r r` r@ rt rF )r r r� r� r� Zresult1Zresult2r� r r r �
test_compress� s6 =777.7 , #!!!71+
zTestBasicOps.test_compressc C s� | j t d t � � d d d g � | j t d t d � � d d d g � | j t d t d t d � � � d d g � | j t d t d t d � � � d d g � | j t d t d t d � � � d" d$ g � | j t t d d d � | j t t d � | j t t t t d
� d � � t t
t d
t d
� � � | j t t t t d
� d � � t t
t d
t d
� � � | j t t t d t d
� d � � t t
d d d t d
t d
� � � t d � } | j t | � d � t | � | j t | � d
� t d% � } | j t | � d � t | � | j t t d � � d � | j t | � d& � xu t
j d
t
j d
d' d( d d t
j d
t
j d
f D]7 } t t | � � } d j | � } | j | | � q�Wx� d) d t d
t d
f D]� } t | � } | j t t j | � � | � | j t t j | � � | � x1 t
t j d � D] } | j | t | � � qkWqWt d t d
� t
j � d S)*NrT rI r rJ r rK r rM r: rW r/ zcount(3)zcount(4)rY z count(-9)g �$@z count(10.25)rX z count(%r))rI r )rJ r )rK r )rI rM )rJ r: )rK rW )rI rM )rJ r: r� r� )rI r� )rJ r ���r� )rI r� r� )rJ r� i����i����i����r� r� )rD r
�countr& r r` r@ r r$ �maxsizer �reprrF �sys�__mod__rr rs r1 rd rL �exc_info)r rK r �r1�r2�valuer5 r r r �
test_count� sD %(..." # %)
C! zTestBasicOps.test_countc C s� | j t d t d d � � d/ d0 d1 g � | j t d t d d d
d � � d2 d3 d4 g � | j t d t d
d5 � � d6 d8 d: g � | j t d t d d � � d; d< d= g � | j t d t d d � � d> d? d@ g � | j t d t d d � � dA dB dC g � | j t d t t d d � � t d t t d t d d � � � | j t d t t d d � � t d t t d t d d � � � | j t d t d dD � � d dE dF g � | j t d t t d � t d � � � t d � t d � t d � g � | j t d t t d d � t d d � � � t d d � t d d � t d d � g � | j t t d t d d � � � t d d d g � � t d d � } | j t | � d! � t | � | j t | � d" � t dG d � } | j t | � d$ � t | � | j t | � d$ � t dH dI � } | j t | � d% � t | � | j t | � d& � | j t | � d& � | j t t d' d( � � d) � | j t t d' d � � d* � | j t t d' d+ � � d, � xt
j d t
j d dJ dK d d t
j d t
j d f D]� } x� t
j d t
j d dL dM d d d t
j d t
j d f D]� } t t | | � � } | d k r�d- | } n d. | | f } | j | | � x4 t t j d � D] } | j
| t | | � � q�WqYWqWd S)NNrT r rM rI rJ rW rK rX �start�stepr r r: � rO �d g
@y @g @g !@y @z1.1z.1z1.2z1.3rV � rP r/ g @g )@g .@z count(3, 5)z count(8, 5)rY z count(-9, 0)z
count(-9, -3)zcount(-12, -3)g %@g �?zcount(10.5, 1.25)z count(10.5)g �?zcount(10.5, 1.0)z count(%r)z
count(%r, %r))rI r )rJ rW )rK rX )rI r )rJ rW )rK rX r� )rI r r� )rJ r� r� )rK r� )rI r )rJ r )rK r )rI r )rJ rM )rK r: )rI r )rJ rW )rK rX y
@ �y @ �y !@ �i����i����r� i����r� i����r� )rD r
r� r& r� r r r r� rF r� r1 rd rL )r rK r r� r� r� r5 r r r �test_count_with_stride" sR +!+++CF+'"-+7
CF
z#TestBasicOps.test_count_with_stridec C s� | j t d t d � � t d � � | j t t d � � g � | j t t � | j t t d � | j t t t t � � d � � d d d d d d d d d d g
� t d � } | j t | � d � | j t d t j
| � � t d
� � x� t t j
d � D]� } | j t d t j t j | | � � � t d
� � t | � | j t d t j t j | | � � � t d � � t | � t | � qWx1 t t j
d � D] } | j | t d � � q�Wd S) Nr/ rT Z
abcabcabcarl rW r r r rI Z
bcabcabcabZ
cabcabcabc)rD r& �cycler r` r@ r$ r rF rr rs r r1 rd r2 r3 rL )r rK r5 r r r �
test_cycleO s$ % F ('
'
zTestBasicOps.test_cyclec
s� | j g t t g � � � | j g t t g d t �� � | j t t t d g � � | j t t d � | j t t d d d � d � d1 d2 d3 d4 d5 d6 d7 d8 g } g } xU t | d d � � D]> \ } } x/ | D]' } | j | | d � | j | � q� Wq� W| j | | � x� t t j d � D]� } g } xd t j
t j t | t � | � � D]> \ } } x/ | D]' } | j | | d � | j | � qpWq]W| j | | � q,Wg } x� t | t � D]r \ } } xc t | t
� D]R \ } } xC | D]; } | j | | d � | j | | d � | j | � q�Wq�Wq�W| j | | � x� t t j d � D]� } g } x� t j
t j t | t � | � � D]� \ } } xx t j
t j t | t
� | � � D]R \ } } xC | D]; } | j | | d � | j | | d � | j | � q�Wq�Wq�W| j | | � qcWd d � t | t � D� } t d d � | D� � }
| j t | � |
� | j t | � t |
� � d } d d � t t | � � D� } | j | d d d d d g � d d � t t | � � D� } | j | d d d g � d! d � t t | � � D� } | j | d9 d: d; d< d= g � t d# d � t t | � � D� d$ d% �d d � } | j | d> d? d@ g � Gd& d' � d' t � � d � f d( d) � } d t d* d+ � }
| j � |
| d � � | j � |
| d � � G� f d, d- � d- � } | � | � d g } | j � |
| d. t �| j � |
| � � � f d/ d0 � � d � _ | j � |
d g � � d � _ | j � |
d d g � � d S)AN�keyrT c S s | S)Nr )r r r r r6 n s z+TestBasicOps.test_groupby.<locals>.<lambda>r/ r r� � rP r� r �
� � r rO rM rZ � r� c S s | d S)Nr r )r, r r r r6 t s c S s g | ] \ } } | � q Sr r )r7 r� �gr r r r8 � s z-TestBasicOps.test_groupby.<locals>.<listcomp>c S s g | ] } | d � q S)r r )r7 r, r r r r8 � s Z abracadabrac S s g | ] \ } } | � q Sr r )r7 r� r� r r r r8 � s rI rJ rK �dr, c S s4 g | ]* \ } } t t | d d � � r | � q S)r r )r r$ )r7 r� r� r r r r8 � s c S s. g | ]$ \ } } t t | � � | f � q Sr )r r )r7 r� r� r r r r8 � s rW c S s. g | ]$ \ } } t t | � � | f � q Sr )r r )r7 r� r� r r r r8 � s �reverseTc @ s e Z d Z d S)z0TestBasicOps.test_groupby.<locals>.ExpectedErrorN)r r! r" r r r r �
ExpectedError� s r� c 3 s& x t | � D] } d Vq
W� � d S)NZyo)r )r r )r� r r �
delayed_raise� s z0TestBasicOps.test_groupby.<locals>.delayed_raisec s � f d d � t | | � D� S)Nc s" g | ] \ } } � | � � q Sr r )r7 r� r� )�funcr r r8 � s z;TestBasicOps.test_groupby.<locals>.gulp.<locals>.<listcomp>)r� )r) Zkeypr� r )r� r �gulp� s z'TestBasicOps.test_groupby.<locals>.gulpc s" e Z d Z � f d d � Z d S)z+TestBasicOps.test_groupby.<locals>.DummyCmpc s
� � d S)Nr )r Zdst)r� r r �__eq__� s z2TestBasicOps.test_groupby.<locals>.DummyCmp.__eq__N)r r! r" r� r )r� r r �DummyCmp� s r� r� c s, � j d k r" � j d 8_ | S� � d S)Nr r )�skip)�obj)r� �keyfuncr r r� � s z*TestBasicOps.test_groupby.<locals>.keyfunc)r r/ r� )r r� rP )r r� rP )r r� rP )r r� r� )r rO r� )rM rZ r� )rM r� r� )rW rI )r rJ )r rK )r r� )r r, )rW rI )r r, )r rJ )rD r r� r� r` r@ �appendr r1 rd r2 r3 r- r. r� r r� � Exceptionr� )r r4 �dupr� r� �elemr5 ZikZig�keysZ expectedkeysr, r� r� r� r )r� r� r � test_groupbyh s� " "
1
11
5 zTestBasicOps.test_groupbyc C s� | j t t t t d � � � d d d g � | j t t d d d d d d g � � d d g � | j t t t d d d d d g � � d d g � | j t d t t t � � � d d d d g � | j t t � | j t t d d � � | j t t d d � t d � d � | j t t t d
� | j t t
t t d � t d � � � d d d g } t t t d � � } | j t t j | � � | � t t t d � � } | j t t j | � � | � x� t t
j d � D]� } t t t d � � } | j t t
j t
j | | � � � | � t
| � | j t t
j t
j | | � � � | d d � � q�Wx@ t t
j d � D]+ } t t t d � � } | j | | � q{Wd S) NrN r r r: r c S s | S)Nr )r r r r r6 � s z*TestBasicOps.test_filter.<locals>.<lambda>c S s | S)Nr )r r r r r6 � s rV rM )rD r r� r r �boolr& r� r` r@ rF rr rs r1 rd r2 r3 rL )r �ansrK r5 r r r � test_filter� s, .441%( +
9zTestBasicOps.test_filterc C s� | j t t t t d � � � d d d g � | j t t d d d d d d g � � d d d g � | j t t t d d d d d g � � d d d g � | j t d t t t � � � d d d d g � | j t t � | j t t d d
� � | j t t d d
� t d � d � | j t t t d � | j t t
t t d � t d � � � x: t t j d � D]% } | j
| t t t d � � � qmWd S) NrN r rM rW r r r: rV c S s | S)Nr )r r r r r6 � s z/TestBasicOps.test_filterfalse.<locals>.<lambda>c S s | S)Nr )r r r r r6 � s )rD r � filterfalser r r� r& r� r` r@ rF r1 rd rL )r r5 r r r �test_filterfalse� s .771%(z TestBasicOps.test_filterfalsec C s� d d � t d t � � D� } | j | d d d g � | j t t d t d
� � � t d t d
� � � | j t t d t d � � � t d t d � � � | j t d t d t � � � t d t d � � � | j t t d � � t d � � | j t t � � t � � | j t t d � | j t t t d � d � | j d
d � t d d � D� t d d � � | j d d � t d d � D� t d d � � d S)Nc S s" g | ] \ } } | | f � q Sr r )r7 r �yr r r r8 s z)TestBasicOps.test_zip.<locals>.<listcomp>rT rI r rJ r rK r rN rk rM c S s" g | ] } t t | � � � q Sr )r~ r )r7 �pairr r r r8 s rj c S s g | ] } | � q Sr r )r7 r� r r r r8 s )rI r )rJ r )rK r ) r r� rD r r r
r& r` r@ )r r� r r r �test_zip� s 444" zTestBasicOps.test_zipc C s& t t t t d d � � � } | j t | � t | � � t t t t t d d � � � � } | j t t j | � � t | � � d d � t
j
t d t � � � D� } | j | d d d g � d d � t
j t d t � � � D� } | j | d d d g � xh t
t j d � D]S } d d � t j t j t d t � � | � � D� } | j | d d d g � qWxu t
t j d � D]` } t d t � � } t | � d
d � t j t j | | � � D� } | j | d d g � q�Wx7 t
t j d � D]" } | j | t d t � � � q�Wd S)NrT rj c S s" g | ] \ } } | | f � q Sr r )r7 r r� r r r r8 s z5TestBasicOps.test_zip_tuple_reuse.<locals>.<listcomp>rI r rJ r rK r c S s" g | ] \ } } | | f � q Sr r )r7 r r� r r r r8 s c S s" g | ] \ } } | | f � q Sr r )r7 r r� r r r r8 s c S s" g | ] \ } } | | f � q Sr r )r7 r r� r r r r8 # s )rI r )rJ r )rK r )rI r )rJ r )rK r )rI r )rJ r )rK r )rJ r )rK r )r r_ r� r rD ra rb r �dict�fromkeysrr r� rs r r1 rd r2 r3 rF rL )r �idsr� r5 r� r r r �test_zip_tuple_reuse s$ $%((4
(z!TestBasicOps.test_zip_tuple_reusec sD xcd t d � g t d � d g t d � t d d � t d d � g t d � t d � t d d � t d � t d
� g t d � t d � t d d � t d � t d
� t d � g g D]� � � f d d � t t t t � � � � D� } | j t t � � � | � | j t t � i � � | � d
d � | D� } | j t t � t d d � � � | � q� W| j t d t d t � � � t t
d t d � � � � | j t t � � t t
� � � | j t t g � � t t
g � � � | j t t d � � t t
d � � � | j t t d d i � � t t
t d � d g d � � � | j t t d � | j t t t d � d � xT d d g D]F } y t
| t � t � � Wn t k
r�Yq�X| j d | � q�W| j d d � t d d � D� t t
d d � � � | j d d � t d d � D� t t
d d � � � d S)NrT rN i� i� i4 i� i� r i� i� c s/ g | ]% � t � f d d � � D� � � q S)c s2 g | ]( } � t | � k r( | � n d � q S)N)r )r7 �arg)r r r r8 1 s z;TestBasicOps.test_ziplongest.<locals>.<listcomp>.<listcomp>)r~ )r7 )r
)r r r8 1 s z0TestBasicOps.test_ziplongest.<locals>.<listcomp>c S s) g | ] } t d d � | D� � � q S)c s s' | ] } | d k r d p | Vq d S)N�Xr )r7 r; r r r r} 5 s z:TestBasicOps.test_ziplongest.<locals>.<listcomp>.<genexpr>)r~ )r7 r� r r r r8 5 s � fillvaluer� rM rk Zdefgzzip_longest('abc', fv=1)z3zip_longest('abc', fillvalue=1, bogus_keyword=None)zDid not raise Type in: c S s" g | ] } t t | � � � q Sr )r~ r )r7 r� r r r r8 N s rj c S s g | ] } | � q Sr r )r7 r� r r r r8 P s )r rb r_ r rD r � zip_longestr� r& r� r r` r@ �eval�globals�localsZfail)r �targetZstmtr )r
r �test_ziplongest) s@ $3F ,:"((#
z TestBasicOps.test_ziplongestc C s� t t t t d d � � � } | j t | � t | � � t t t t t d d � � � � } | j t t j | � � t | � � d S)NrT rj )
r r_ r� r� rD ra rb r r� r� )r r� r r r � test_zip_longest_tuple_reuseS s $z)TestBasicOps.test_zip_longest_tuple_reusec C s� x� t t j d � D]p } | j | t d d � � | j | t d d � � | j | t d d d d �� | j | t d d � � q Wd S)Nr rT rj Zdefghr� rl )r r1 rd rL r� )r r5 r r r �test_zip_longest_picklingZ s
z&TestBasicOps.test_zip_longest_picklingc C s� Gd d � d � } | d d t � } | d d t � } d d � } | j | | | � d d d
d g � | d d t � } | d d t � } t | | d
d �} | j t | � d � | j t | � d � | j t | � d � | j t t | � d S)Nc @ s4 e Z d Z d d � Z d d � Z d d � Z d S)z,TestBasicOps.test_bug_7244.<locals>.Repeaterc S s% | | _ t | � | _ | | _ d S)N)�or] r� r; )r r� r� r; r r r �__init__e s z5TestBasicOps.test_bug_7244.<locals>.Repeater.__init__c S s | S)Nr )r r r r r i s z5TestBasicOps.test_bug_7244.<locals>.Repeater.__iter__c S s2 | j d k r% | j d 8_ | j S| j � d S)Nr r )r� r� r; )r r r r r k s z5TestBasicOps.test_bug_7244.<locals>.Repeater.__next__N)r r! r" r r r r r r r �Repeaterc s r r rM r r: c S si g } x\ t | | d d �D]E \ } } t j d � � t | | f � Wd QX| j | | f � q W| S)Nr� r �stdout)r� r Zcaptured_output�printr� )r� r� r� r r� r r r �runv s "z'TestBasicOps.test_bug_7244.<locals>.runr r� )r r )r r )r r )r r )r r )r r )r r )r rD r= r� rF r` )r r r� r� r rA r r r �
test_bug_7244a s %zTestBasicOps.test_bug_7244c s� x6g f g f d g d d g f t d � t d � g d d d d d d! g f t d � t d � t d � g g f t d � t d � t d � g g f t d � t d � t d � g g f g D]w \ } } | j t t | � � | � xL t d � D]> } | j t t | | � � t t | t d | � � � � q� Wq� W| j t t t t d
� g d � � � d" � | j t t t d � d � d d
� } d d � } d d d t d � t d � t d d d d d d � t d � t d � t t d � � g � x� t d � D]� } � f d d � t t
j d � � D� } t t
t | � � } | j t t t | � � � | � | j t t | � � t | | � � � | j t t | � � t | | � � � t
t | � } | j t t t | � � � | � qWd S)#NrU rI rJ r rM r r r: r� rV rN c s9 t t t | � � | j d d � } t | � } | d k rF f Vd St d d � | D� � rc d Sd g | } t d d � t | | � D� � Vx� x� t t | � � D]� } | | t | | � d k r� q� n | | d 7<x% t | d | � D] } d | | <q� Wt d d � t | | � D� � VPq� Wd Sq� Wd S)Nr� r r c s s! | ] } t | � d k Vq d S)r N)r )r7 r| r r r r} � s z>TestBasicOps.test_product.<locals>.product1.<locals>.<genexpr>c s s | ] \ } } | | Vq d S)Nr )r7 r| r r r r r} � s c s s | ] \ } } | | Vq d S)Nr )r7 r| r r r r r} � s ) r r_ r~ �getr �anyr r� r )r
�kwds�poolsr r� r r� r r r �product1� s& %
! !z+TestBasicOps.test_product.<locals>.product1c ? sx t t t | � � | j d d � } g g } x' | D] � � f d d � | D� } q5 Wx | D] } t | � Vq_ Wd S)z Pure python version used in docsr� r c s* g | ] } � D] } | | g � q q Sr r )r7 r r� )r| r r r8 � s z?TestBasicOps.test_product.<locals>.product2.<locals>.<listcomp>N)r r_ r~ r )r
r r r� r* r )r| r �product2� s %
z+TestBasicOps.test_product.<locals>.product2rl rT rK Zabcdefgr� r� r� c s g | ] } t j � � � q Sr )�randomZchoice)r7 r� )�argtypesr r r8 � s z-TestBasicOps.test_product.<locals>.<listcomp>rW )rI )rJ )r r )r r )r r )r r )r r )r r i�� )r rD r r� r� r r` r@ r� r~ r Z randranger* r_ �iter)r r
r� r, r
r r Z expected_lenr )r
r � test_product� s2 -$$4 '/ 3'("((zTestBasicOps.test_productc
C s7 | j t t f � � t d d d g d � Wd QXd S)Nr� r � rU rW i rv )r` r� r� r� )r r r r �test_product_overflow� s z"TestBasicOps.test_product_overflowc C sf | j t t t t t d d � � � � d � | j t t t t t t d d � � � � � d � d S)NrT rj r )rD r r� r_ r� r� r� r )r r r r �test_product_tuple_reuse� s .z%TestBasicOps.test_product_tuple_reusec
C sP xIg f g f d g d d g f t d � t d � g d
d d d
d d g f t d � t d � t d � g g f t d � t d � t d � g g f t d � t d � t d � g g f g D]� \ } } | j t t j t | � � � | � | j t t j t | � � � | � x1 t t j d � D] } | j | t | � � q(Wq� Wd S)NrU rI rJ r rM r r )rI )rJ )r r )r r )r r )r r )r r )r r ) r rD r rr r� rs r1 rd rL )r r
r� r5 r r r �test_product_pickling� s -$$4%%z"TestBasicOps.test_product_picklingc C sh t d d � } | j d � | j t | � d � t d
f d � } | j d � | j t t | � d S)
Nr r rM r � )r r )rM )r r )r rM )r r )rM )r r r )r� � __setstate__rD rF r` r )r r� r r r �test_product_issue_25021� s
z%TestBasicOps.test_product_issue_25021c C sL | j t t d d d d � � d d d g � | j t t d � t d � � d d d g � | j t t d d � � d d d g � | j t d t d � � d d d g � | j t t d d � � g � | j t t d d � � g � | j t t � | j t t d d d � | j t t d d � t d � } | j t | � d
� t d d � } | j t | � d � t | � | j t | � d
� t d d d d � } | j t | � d � | j t d t
j
| � � t d � � | j t d t
j | � � t d � � x: t t j
d � D]% } | j | t d d d d � � q Wd S)N�objectrI �timesrM r r r r: y zrepeat((1+0j))rW zrepeat((1+0j), 5)zrepeat((1+0j), 0)r/ )r rI )r rI )r rI r� y �? y �? �aar )rD r r� r
r r& r` r@ r� rF rr rs r1 rd rL )r r, rK r5 r r r � test_repeat� s, . ((
((zTestBasicOps.test_repeatc C s� | j t t d d � � d � | j t t d d � � d � | j t t d d d �� d � | j t t d d d �� d � d S)
NrI r zrepeat('a', 0)r r r� r� r� r� )rD r� r� )r r r r � test_repeat_with_negative_times s "z,TestBasicOps.test_repeat_with_negative_timesc C sg | j t t t j t d � t d d � � � d
d d g � | j t t t d t d � � � d d d g � | j t t t d t � � � d d d g � | j t d t t d t � � � d d g � | j t t t j g � � g � | j t
t � | j t
t t d t d � t d � � � | j t
t t j � | j t
t t d t d � � � | j t
t t t d g d g � � | j t
t t t d g d g � � d d d g } t t d t � � } | j t t j | � � | � t t d t � � } | j t t j | � � | � x@ t t j d � D]+ } t t d t � � } | j | | � q4Wd S)NrM r rV r r rT rW rI rJ rK r/ r: r r rX )rI r )rJ r )rK r )rI r )rJ r )rK r )rI r )rJ r )rI r )rJ r )rK r )rD r r_ r' �powr r r� r& r` r@ �negrF r r r rr rs r1 rd rL )r r� rK r5 r r r �test_map s. -! !
"+"%% zTestBasicOps.test_mapc C s� | j t t t j t t d � t d d � � � � d
d d g � | j t d t t j t t � t d � � � � d
d d g � | j t t t j g � � g � | j t t t j t d d g � g � � d g � | j
t t t t j d g � � | j
t t � | j
t t t j d g d � | j
t t t d d g � � | j
t
t t t d g � � | j
t t t t d g � � d d d g } t t j t t d � t d d � � � } | j t t j | � � | � t t j t t d � t d d � � � } | j t t j | � � | � xU t t j d � D]@ } t t j t t d � t d d � � � } | j | | � q<Wd S)NrM r rV r r r: rW �extrar/ r r rX r r rX i )r: rW )r: rW )r: rW )r: rW r r rX )rD r �starmapr' r r r r& r� r r` r@ rF r r r rr rs r1 rd rL )r r� rK r5 r r r � test_starmap$ s( 30"4" * * *zTestBasicOps.test_starmapc C s� xN d d d d d g D]7 } | j t t t d � | � � t t | � � � q WxN d d d g D]= \ } } | j t t t d � | � � t t | � � � qa W| j t t t d � d � � t t d � � � | j t t t d � d d � � t t d � � � | j t t t d � d d d � � t t d � � � | j t t t d � d d � � t t d d � � � | j t t t d � d d d � � t t d d d � � � t t d � � } | j t t | d � � t t d � � � | j t | � t t d d � � � t d � } | j t t | � | j t t | d d d d � | j t t | d d d � | j t t | d d d � | j t t | d d d � | j t t | d d d
� | j t t | d � | j t t | d d � | j t t | d d � | j t t | d d d � | j t t | d d d � | j t t t t � d d t
� � � d � t � } | j t t | d d d � � d g � | j t | � d � x� d! d" d# d$ d% g D]� } | j t t j t t d � | � � � t t | � � � | j t t j
t t d � | � � � t t | � � � x: t t j d � D]% } | j | t t d � | � � q[Wq�Wd
d � d& D� } t j | � } t | d � } | j | � � t | � t j � | j | � � d S)'Nr/ r� rM r� �n r r r: rW r rI �2 c s s | ] } | Vq d S)Nr )r7 r r r r r} � s z+TestBasicOps.test_islice.<locals>.<genexpr>)r/ r� rM )r/ rM r� )r/ r� )r/ rM )r� �r/ r" rM �r/ r� rM )r$ r% �r/ r" �r/ r� )r& r' �r" �r� )r( r) ���r* r� r� )r/ r� rM )r/ rM r� )r/ r� )r/ rM )r� )r r )rD r r$ r r r` r@ r r r� r� rF rr rs r1 rd rL �weakref�refZassertIsNotNoner Z
gc_collectZ assertIsNone)r r
ZtgtargsrA ZrarK r5 �wrr r r � test_islice? sr
1477=+% . (
'''
zTestBasicOps.test_islicec C s� d d d d d d d d g } | j t t t | � � d d d g � | j t t t g � � g � | j t t � | j t t t j � | j t t t j d g d � | j t t t d
d
g � � | j t t t t
d g � � t t d d d d d d g � } | j t | � d d d g � | j t t | � | j t t
j
t t | � � � d d d g � | j t t
j t t | � � � d d d g � x4 t t j d � D] } | j | t t | � � q�Wd S)Nr rM rW r� r r: rN rX r r/ r )r: rW )r: rW )r: rW )rD r � takewhiler0 r` r@ r' r rF r r r� r rr rs r r1 rd rL )r r� r� r5 r r r �test_takewhile� s ( ! 1!zTestBasicOps.test_takewhilec C s� d d d d d d d d g } | j t t t | � � d d d d d g � | j t t t g � � g � | j t t � | j t t t j � | j t t t j d g d � | j t t t d
d g � � | j t t t t
d
g � � | j t t j t t | � � � d d d d d g � | j t t j t t | � � � d d d d d g � x4 t
t j d � D] } | j | t t | � � qpWd S)Nr rM rW r� r r: rN rX r r/ )r: rW )r: rW )r: rW )rD r � dropwhiler0 r` r@ r' r rF r r rr rs r r1 rd rL )r r� r5 r r r �test_dropwhile� s . 7!zTestBasicOps.test_dropwhilec C s% d } t g � \ } } | j t | � g � | j t | � g � t t | � � \ } } | j t | | � t t | � t | � � � t t | � � \ } } | j t | � t t | � � � | j t | � t t | � � � t t | � � \ } } x* t d � D] } | j t | � | � q W~ | j t | � t t | � � � t t | � � \ } } x* t d � D] } | j t | � | � quW~ | j t | � t t d | � � � x� t d � D]� } d g | d g | } t j | � g g f } t t | � � } x/ | D]' } t | | � } | | j | � qW| j | d t t | � � � | j | d t t | � � � q�W| j
t t � | j
t t d � | j
t t d d g d � | j
t t d d g d d � t d � \ } } t | � d
� }
| j t |
� t d
� � t t d � d � \ } } }
x* t d � D] } | j t | � | � qRW| j t | � t t d � � � | j t |
� t |
� g t t d � � � | j t | � t t d d � � � | j t |
� t t d d � � � | j
t t d d � | j
t
t g d � xy t d � D]k } t d | � } | j t | � t � | j t | � | � | j d
d � | D� t d � g | � qEWt d � \ } } t | � \ }
} | j | |
k � t d � \ }
} t |
� } | j
t | � | j
t | d � | |
� } | j t |
� t | � k oxt | � k oxt d � k n � t t d � � \ } } t j | � } | j t | d � t | � � ~ | j
t t | d � t d � } t t d � � } t d � \ } } | j t t j | � � | � | j t t j | � � | � t t t d � � � \ } } | j t t j | � � | � | j t t j | � � | � t d � \ } } t d | � t d | � | j t t j | � � | d d � � | j t t j | � � | d d � � | j t | � | d d � � | j t | � | d d � � t t d � � \ } } t d | � t d | � | j t t j | � � | d d � � | j t t j | � � | d d � � | j t | � | d d � � | j t | � | d d � � t d � \ } } | j t t j | � � | � | j t t j | � � | � | j t | � | � | j t | � | � t t d � � \ } } | j t t j | � � | � | j t t j | � � | � | j t | � | � | j t | � | � t d � \ } } t d | � t d | � | j t t j | � � | d d � � | j t t j | � � | d d � � | j t | � | d d � � | j t | � | d d � � t t d � � \ } } t d | � t d | � | j t t j | � � | d d � � | j t t j | � � | d d � � | j t | � | d d � � | j t | � | d d � � xu t t j d � D]` } | j | t t d � � � t d � \ } } | j | | d | �| j | | d | �q�
Wd S)N�� r� rW r r rM r r rT rj i� Zinvalidc S s g | ] } t | � � q Sr )r )r7 r r r r r8 � s z)TestBasicOps.test_tee.<locals>.<listcomp>r/ � __class__i' �< rG r� ) �teerD r r r
r rF r Zshuffler� r` r@ rE r r~ r r� r+ �proxy�getattr�ReferenceErrorrr r& rs r1 rd rL r )r r rI rJ r r� ZorderZlists�itsr� rK r� r� Zt1Zt2ZtnewZt3r� r� Zlong_ansr5 r r r �test_tee� s� ."""%
$ ".%%. F
))
))
))
)) zTestBasicOps.test_teec C sD t t d d � � \ } } y t | � ~ Wn ~ ~ � Yn Xd S)Ni -1)r6 r� r )r ZforwardZbackwardr r r �test_tee_del_backwardE s
z"TestBasicOps.test_tee_del_backwardc C s� | j t t t � � xO t t t t f D]; } | j t t | g � � | j t t | t � � � q) W| j t t t g d � � | j t t t t � d � � t g � \ } } | j t t | � | j t t | � t t � � \ } } | j t t | � | j t t | � | j t t t
d d � � xg t t t
t t t f D]M } | j t t | d d � g � � | j t t | d d � t � � � qKWd S)Nr c S s | S)Nr )r r r r r6 b s z1TestBasicOps.test_StopIteration.<locals>.<lambda>c S s | S)Nr )r r r r r6 c s )r` r rF r rn r� r� r r$ r6 r� r� r� r_ r/ r1 r )r �fr� �qr r r �test_StopIterationN s "z TestBasicOps.test_StopIteration)1r r! r" rL rf ro rq ru r� r Zbigaddrspacetestr� Z impl_detailr� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r r r r r r! r. r0 r2 r; r< r? r r r r r9 P sV )
P L ? ! " ( - n
* ( :
L � r9 c @ s e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
d � Z d d � Z
d d � Z d d � Z d d � Z
d d � Z d d � Z d d � Z d d � Z d d � Z d! d" � Z d# d$ � Z d% d& � Z d' d( � Z d) d* � Z d+ d, � Z d- S).� TestExamplesc C s> | j t t d d d d d g � � d d d d d g � d S) Nr r rM r: rW rN r/ rO )rD r r\ )r r r r rf g s z TestExamples.test_accumulatec C sL d d d d d g } d d d d d g } x� t t j d � D]� } t | � } | j t t j t j | | � � � | d d � � | j t | � d � | j t t j t j | | � � � | d d � � q> Wt | � } | j t | � d � | j t t j
| � � | d d � � | j t t j | � � | d d � � d S) Nr r rM r: rW rN r/ rO ) r r1 rd r\ rD r r2 r3 rF rr rs )r r� Z accumulatedr5 rA r r r �test_accumulate_reduciblej s 59 )z&TestExamples.test_accumulate_reduciblec C s& | j d j t d d � � d � d S)Nrl r� �DEFr� )rD �joinrn )r r r r ro y s zTestExamples.test_chainc C s, | j d j t j d d g � � d � d S)Nrl r� rB r� )rD rC rn rp )r r r r rq | s z%TestExamples.test_chain_from_iterablec C sf | j t t d d � � d d d
d d d g � | j t t t d � d � � d d d d g � d S)Nrw r rx ry rz r{ r: rM r r )rx ry )rx rz )rx r{ )ry rz )ry r{ )rz r{ )r r r )r r rM )r r rM )r r rM )rD r r� r )r r r r r� s z TestExamples.test_combinationsc C s5 | j t t d d � � d d d d d
d g � d S) Nr� r rx ry rz )rx rx )rx ry )rx rz )ry ry )ry rz )rz rz )rD r r� )r r r r r� � s z/TestExamples.test_combinations_with_replacementc
C s; | j t t d d d d d d d g � � t d � � d S)Nr� r r r� )rD r r� )r r r r r� � s zTestExamples.test_compressc C s8 | j t t t d � d � � d d d d d g � d S)Nr/ rW r� r� r� r� )rD r r$ r� )r r r r r� � s zTestExamples.test_countc C s/ | j t t t d � d � � t d � � d S)Nrw r� Z ABCDABCDABCD)rD r r$ r� )r r r r r� � s zTestExamples.test_cyclec C sA | j t t d d � d d d d d g � � d d d g � d S)Nc S s
| d k S)NrW r )r r r r r6 � s z-TestExamples.test_dropwhile.<locals>.<lambda>r r: rN )rD r r1 )r r r r r2 � s zTestExamples.test_dropwhilec C st | j d d � t d � D� t d � � | j d d � t d � D� t d � t d � t d � t d
� g � d S) Nc S s g | ] \ } } | � q Sr r )r7 r� r� r r r r8 � s z-TestExamples.test_groupby.<locals>.<listcomp>ZAAAABBBCCDAABBBZABCDABc S s" g | ] \ } } t | � � q Sr )r )r7 r� r� r r r r8 � s Z
AAAABBBCCDZAAAAZBBB�CCr{ )rD r� r )r r r r r� � s
zTestExamples.test_groupbyc C s> | j t t d d � t d � � � d d d d d g � d S) Nc S s | d S)Nr r )r r r r r6 � s z*TestExamples.test_filter.<locals>.<lambda>r/ r rM rW rV rY )rD r r� r )r r r r r� � s zTestExamples.test_filterc C s> | j t t d d � t d � � � d d d d d g � d S) Nc S s | d S)Nr r )r r r r r6 � s z/TestExamples.test_filterfalse.<locals>.<lambda>r/ r r r: rN rX )rD r r� r )r r r r r� � s z TestExamples.test_filterfalsec C s/ | j t t t d d � � d d d g � d S)
Nr rM r/ rW rv rY i� )r rM r/ )rW r rM )rD r r_ r )r r r r r � s zTestExamples.test_mapc C s� | j t t d d � � t d � � | j t t d d d � � t d � � | j t t d d d � � t d � � | j t t d d d d � � t d � � d S) Nr� r ZABr: ZCDZCDEFGr ZACEG)rD r r$ )r r r r r. � s %((zTestExamples.test_islicec C s) | j t t d d � � d d g � d S) Nrw �xyrx r ry r� )rx r )ry r� )rD r r )r r r r r� � s zTestExamples.test_zipc C s5 | j t t d d d d �� d d d
d g � d S)Nrw rE r� �-rx r ry r� rz r{ )rx r )ry r� )rz rF )r{ rF )rD r r� )r r r r �test_zip_longest� s z TestExamples.test_zip_longestc
C sl | j t t d d � � t t t d j � � � � | j t t t d � � � d d d d
d d g � d S)
Nrw r z#AB AC AD BA BC BD CA CB CD DA DB DCrM r r )r r r )r r r )r r r )r r r )r r r )r r r )rD r r� r_ r~ �splitr )r r r r r� � s z TestExamples.test_permutationsc C sx | j t t d d � � t t t d j � � � � | j t t t d � d d �� d d
d d d
d d d g � d S)Nrw rE zAx Ay Bx By Cx Cy Dx Dyr r� rM r r )r r r )r r r )r r r )r r r )r r r )r r r )r r r )r r r )rD r r� r_ r~ rH r )r r r r r � s
! zTestExamples.test_productc C s, | j t t d d � � d d d g � d S)Nr/ rM )rD r r� )r r r r r � s zTestExamples.test_repeatc C s5 | j t t t d d d
g � � d d d g � d S) Nr rW rM r/ rv rY i� )r rW )rM r )r/ rM )rD r r r )r r r r � test_stapmap� s !zTestExamples.test_stapmapc C s> | j t t d d � d d d d d g � � d d g � d S)Nc S s
| d k S)NrW r )r r r r r6 � s z-TestExamples.test_takewhile.<locals>.<lambda>r r: rN )rD r r/ )r r r r r0 � s zTestExamples.test_takewhileN)r r! r" rf rA ro rq r� r� r� r� r� r2 r� r� r� r r. r� rG r� r r rI r0 r r r r r@ e s, r@ c @ s$ e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
d � Z d d � Z
d d � Z d d � Z d d � Z
d d � Z d d � Z d d � Z d d � Z d d � Z d! d" � Z d# d$ � Z d% d& � Z d' d( � Z d) d* � Z d+ d, � Z d- d. � Z d/ S)0�TestGCc C s! | j | � t | � ~ ~ d S)N)r� rF )r �iteratorZ containerr r r � makecycle� s
zTestGC.makecyclec C s, g } | j t d d | d g � | � d S)Nr r rM )rL r\ )r rI r r r rf � s zTestGC.test_accumulatec C s g } | j t | � | � d S)N)rL rn )r rI r r r ro � s zTestGC.test_chainc C s& g } | j t j | g � | � d S)N)rL rn rp )r rI r r r rq � s z TestGC.test_chain_from_iterablec C s/ g } | j t d d | d g d � | � d S)Nr r rM )rL r� )r rI r r r r� � s zTestGC.test_combinationsc C s/ g } | j t d d | d g d � | � d S)Nr r rM )rL r� )r rI r r r r� � s z)TestGC.test_combinations_with_replacementc C s5 g } | j t d d d d d d d g � | � d S)Nr� r r )rL r� )r rI r r r r� � s zTestGC.test_compressc C sM g } t d t f t d | � � } | j t | d � | d � � | � d S)N�Intr r r )rE r] r� rL r� )r rI rM r r r r� � s zTestGC.test_countc C s' g } | j t | g d � | � d S)Nr )rL r� )r rI r r r r� � s zTestGC.test_cyclec C s, g } | j t t d | | g � | � d S)Nr )rL r1 r� )r rI r r r r2 � s zTestGC.test_dropwhilec C s0 g } | j t | g d d d � � | � d S)Nr c S s | S)Nr )r r r r r6 � s z%TestGC.test_groupby.<locals>.<lambda>)rL r� )r rI r r r r� � s zTestGC.test_groupbyc C s[ d } d d � } xB t t | � d | �D]( \ } } | j j d g � j | � q+ Wd S)Nr/ c S s | S)Nr )r r r r r6 � s z'TestGC.test_issue2246.<locals>.<lambda>r� r )r� r �__dict__�
setdefaultr� )r r r� r r� r r r �test_issue2246� s %zTestGC.test_issue2246c C s0 g } | j t d d � | g d � | � d S)Nc S s d S)NTr )r r r r r6 s z$TestGC.test_filter.<locals>.<lambda>r )rL r� )r rI r r r r� s zTestGC.test_filterc C s) g } | j t d d � | � | � d S)Nc S s d S)NFr )r r r r r6 s z)TestGC.test_filterfalse.<locals>.<lambda>)rL r� )r rI r r r r� s zTestGC.test_filterfalsec C s1 g } | j t | g d | g d � | � d S)Nr rM )rL r )r rI r r r r� s zTestGC.test_zipc C sj g } | j t | g d | g d � | � | d g } | j t | g d | g d d | �| � d S)Nr rM r� )rL r� )r rI rJ r r r rG s ' zTestGC.test_zip_longestc C s0 g } | j t d d � | g d � | � d S)Nc S s | S)Nr )r r r r r6 s z!TestGC.test_map.<locals>.<lambda>r )rL r_ )r rI r r r r s zTestGC.test_mapc C s* g } | j t | g d d � | � d S)Nr )rL r$ )r rI r r r r. s zTestGC.test_islicec C s/ g } | j t d d | d g d � | � d S)Nr r rM )rL r� )r rI r r r r� s zTestGC.test_permutationsc C s2 g } | j t d d | d g d d �| � d S)Nr r rM r� )rL r� )r rI r r r r s zTestGC.test_productc C s g } | j t | � | � d S)N)rL r� )r rI r r r r " s zTestGC.test_repeatc C s6 g } | j t d d � | | f g d � | � d S)Nc W s | S)Nr )r� r r r r6 ( s z%TestGC.test_starmap.<locals>.<lambda>r )rL r )r rI r r r r! & s zTestGC.test_starmapc C s/ g } | j t t d d | | g � | � d S)Nr r )rL r/ r� )r rI r r r r0 * s zTestGC.test_takewhileN)r r! r" rL rf ro rq r� r� r� r� r� r2 r� rP r� r� r� rG r r. r� r r r! r0 r r r r rJ � s. rJ c c s x | D] } | Vq Wd S)zRegular generatorNr )�seqnr r r r �R. s
rR c @ s. e Z d Z d Z d d � Z d d � Z d S)�GzSequence using __getitem__c C s
| | _ d S)N)rQ )r rQ r r r r 5 s z
G.__init__c C s | j | S)N)rQ )r r r r r � __getitem__7 s z
G.__getitem__N)r r! r" r# r rT r r r r rS 3 s rS c @ s: e Z d Z d Z d d � Z d d � Z d d � Z d S) �Iz Sequence using iterator protocolc C s | | _ d | _ d S)Nr )rQ r )r rQ r r r r < s z
I.__init__c C s | S)Nr )r r r r r ? s z
I.__iter__c C sD | j t | j � k r! t � n | j | j } | j d 7_ | S)Nr )r r rQ r )r r� r r r r A s
z
I.__next__N)r r! r" r# r r r r r r r rU : s rU c @ s. e Z d Z d Z d d � Z d d � Z d S)�Igz9Sequence using iterator protocol defined with a generatorc C s | | _ d | _ d S)Nr )rQ r )r rQ r r r r I s z Ig.__init__c c s x | j D] } | Vq
Wd S)N)rQ )r �valr r r r L s z Ig.__iter__N)r r! r" r# r r r r r r rV G s rV c @ s. e Z d Z d Z d d � Z d d � Z d S)r� z Missing __getitem__ and __iter__c C s | | _ d | _ d S)Nr )rQ r )r rQ r r r r R s z
X.__init__c C sD | j t | j � k r! t � n | j | j } | j d 7_ | S)Nr )r r rQ r )r r� r r r r U s
z
X.__next__N)r r! r" r# r r r r r r r� P s r� c @ s. e Z d Z d Z d d � Z d d � Z d S)�NzIterator missing __next__()c C s | | _ d | _ d S)Nr )rQ r )r rQ r r r r ] s z
N.__init__c C s | S)Nr )r r r r r ` s z
N.__iter__N)r r! r" r# r r r r r r rX [ s rX c @ s: e Z d Z d Z d d � Z d d � Z d d � Z d S) �Ez Test propagation of exceptionsc C s | | _ d | _ d S)Nr )rQ r )r rQ r r r r e s z
E.__init__c C s | S)Nr )r r r r r h s z
E.__iter__c C s d d d S)NrM r r )r r r r r j s z
E.__next__N)r r! r" r# r r r r r r r rY c s rY c @ s: e Z d Z d Z d d � Z d d � Z d d � Z d S) �SzTest immediate stopc C s d S)Nr )r rQ r r r r o s z
S.__init__c C s | S)Nr )r r r r r q s z
S.__iter__c C s
t � d S)N)r )r r r r r s s z
S.__next__N)r r! r" r# r r r r r r r rZ m s rZ c C s+ t t d d � t t t | � � � � � S)z Test multiple tiers of iteratorsc S s | S)Nr )r r r r r6 x s zL.<locals>.<lambda>)rn r_ rR rV rS )rQ r r r �Lv s r[ c @ s� e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
d � Z d d � Z
d d � Z d d � Z d d � Z
d d � Z d d � Z d d � Z d d � Z d d � Z d! S)"�TestVariousIteratorArgsc C s� d d d d d g } d d d d d g } t | � } x? t t t t t f D]( } | j t t | | � � � | � qL W| j t t t | � � � g � | j
t t t | � � | j
t t t
| � � | j
t t t t | � � � d S) Nr r rM r: rW rN r/ rO )r rS rU rV r[ rR rD r r\ rZ r` r@ r� rX �ZeroDivisionErrorrY )r r4 r, r r� r r r rf } s &"z'TestVariousIteratorArgs.test_accumulatec C s+ x$d d t d � d t d d d � f D]� } x� t t t t t t f D]{ } | j t t | | � � � t | | � � � | j t t | | � | | � � � t | | � � t | | � � � qG W| j
t t t t | � � � | j
t t t t
| � � � | j
t t t t | � � � q( Wd S)
N�123rl i� �do�333333�?i� i� rW )r_ r` )r rS rU rV rZ r[ rR rD r rn r` r@ r� rX r] rY )r r4 r� r r r ro � s . .K z"TestVariousIteratorArgs.test_chainc C s xd d t d � d
t d d d � f D]� } t | � } xW t t t t t t f D]= } | j t t
| | � t d � � � t | | � � � qS W| j t
t
t | � t d � � | j t
t
t | � t d � � | j t t t
t | � t d � � � q( Wd S) Nr^ rl i� r_ �333333�?i� i� rW r )zdora )r r rS rU rV rZ r[ rR rD r r� r� r` r@ r� rX r] rY )r r4 r r� r r r r� � s . ;""z%TestVariousIteratorArgs.test_compressc C s� xz d d t d � d t d d d � f D]Q } | j t t t | � � | j t t t | � � | j t t t | � � q( Wd S)
Nr^ rl i� r_ �333333�?i� i� rW )zdorb )r r` r@ r� r� rX r] rY )r r4 r r r r � s .z$TestVariousIteratorArgs.test_productc C s x� d d t d � d
t d d d � f D]� } xw t t t t t t f D]] } t | � d } t | | � � d } t t t
| | � � | � � } | j | | � qG W| j t
t
t | � � | j t
t
t | � � | j t t t
t | � � � q( Wd S) Nr^ rl i� r_ �333333�?i� i� rW rM )zdorc )r rS rU rV rZ r[ rR r r r$ r� rD r` r@ r� rX r] rY )r r4 r� ZtgtlenZexpectedZactualr r r r� � s . !z"TestVariousIteratorArgs.test_cyclec C s� x� t d � t d � t d � d t d d d � f D]� } xU t t t t t t f D]; } | j d d
� t | | � � D� t | | � � � qS W| j
t t t | � � | j
t t t
| � � | j
t t t t | � � � q4 Wd S) Nr/ r i� rV r� i� i� rW c S s g | ] \ } } | � q Sr r )r7 r� Zsbr r r r8 � s z8TestVariousIteratorArgs.test_groupby.<locals>.<listcomp>)rV r� )r rS rU rV rZ r[ rR rD r� r r` r@ r� rX r] rY )r r4 r� r r r r� � s : 9z$TestVariousIteratorArgs.test_groupbyc C s� x� t d � t d � t d � d t d d d � f D]� } xX t t t t t t f D]> } | j t t t
| | � � � d d
� | | � D� � qS W| j t t t
t
| � � | j t t t
t | � � | j t t t t
t | � � � q4 Wd S) Nr/ r i� rV r� i� i� rW c S s" g | ] } t | � r | � q Sr )r )r7 r r r r r8 � s z7TestVariousIteratorArgs.test_filter.<locals>.<listcomp>)rV r� )r rS rU rV rZ r[ rR rD r r� r r` r@ r� rX r] rY )r r4 r� r r r r� � s : z#TestVariousIteratorArgs.test_filterc C s� x� t d � t d � t d � d t d d d � f D]� } xX t t t t t t f D]> } | j t t t
| | � � � d d
� | | � D� � qS W| j t t t
t
| � � | j t t t
t | � � | j t t t t
t | � � � q4 Wd S) Nr/ r i� rV r� i� i� rW c S s" g | ] } t | � r | � q Sr )r )r7 r r r r r8 � s z<TestVariousIteratorArgs.test_filterfalse.<locals>.<listcomp>)rV r� )r rS rU rV rZ r[ rR rD r r� r r` r@ r� rX r] rY )r r4 r� r r r r� � s : z(TestVariousIteratorArgs.test_filterfalsec C s xd d t d � d t d d d � f D]� } x� t t t t t t f D]t } | j t t | | � � � t
| | � � � | j t t | | � | | � � � t
| | � | | � � � qG W| j t t t
| � � | j t t t | � � | j t t t t | � � � q( Wd S)
Nr^ rl i� r_ �333333�?i� i� rW )zdord )r rS rU rV rZ r[ rR rD r r r
r` r@ r� rX r] rY )r r4 r� r r r r� � s . .Dz TestVariousIteratorArgs.test_zipc C s$ x d d t d � d t d d d � f D]� } x� t t t t t t f D]� } | j t t | | � � � t t
| | � � � � | j t t | | � | | � � � t t
| | � | | � � � � qG W| j t t t
| � � | j t t t | � � | j t t t t | � � � q( Wd S)
Nr^ rl i� r_ �333333�?i� i� rW )zdore )r rS rU rV rZ r[ rR rD r r� r r` r@ r� rX r] rY )r r4 r� r r r r� � s . 4Jz'TestVariousIteratorArgs.test_ziplongestc C s; x4t d � t d � t d � d t d d d � f D]� } x� t t t t t t f D]� } | j t t t
| | � � � d d
� | | � D� � | j t t t j | | � | | � � � d d
� | | � D� � qS W| j
t t t
t | � � | j
t t t
t | � � | j
t t t t
t | � � � q4 Wd S)
Nr/ r r� rV r� r� r# rW c S s g | ] } t | � � q Sr )r )r7 r r r r r8 � s z4TestVariousIteratorArgs.test_map.<locals>.<listcomp>c S s g | ] } | | � q Sr r )r7 r r r r r8 � s )rV r� )r rS rU rV rZ r[ rR rD r r_ r r' r r` r@ r� rX r] rY )r r4 r� r r r r � s : * z TestVariousIteratorArgs.test_mapc C s� x� d d t d � d t d d d � f D]� } xd t t t t t t f D]J } | j t t | | � d d d
� � t | | � � d d d
� � qG W| j
t t t | � d � | j
t t t
| � d � | j
t t t t | � d � � q( Wd S)
NZ12345rl i� r_ �333333�?i� i� rW r r r/ )zdorf )r rS rU rV rZ r[ rR rD r r$ r` r@ r� rX r] rY )r r4 r� r r r r. � s . H z#TestVariousIteratorArgs.test_islicec C s x t d � t d � t d � d t d d d � f D]� } xj t t t t t t f D]P } t | | � } | j t t
t j | | � � � d d
� | | � D� � qS W| j
t t
t j t | � � | j
t t
t j t | � � | j
t t t
t j t | � � � q4 Wd S) Nr/ r r� rV r� r� r# rW c S s g | ] } | | � q Sr r )r7 r r r r r8 � s z8TestVariousIteratorArgs.test_starmap.<locals>.<listcomp>)rV r� )r rS rU rV rZ r[ rR r
rD r r r' r r` r@ r� rX r] rY )r r4 r� Zssr r r r! � s : ! z$TestVariousIteratorArgs.test_starmapc C s xt d � t d � t d � d t d d d � f D]� } x t t t t t t f D]e } g } x1 | | � D]# } t | � s� Pn | j | � ql W| j t
t t | | � � � | � qS W| j t
t t t | � � | j t
t t t | � � | j t t
t t t | � � � q4 Wd S)
Nr/ r i� rV r� i� i� rW )rV r� )r rS rU rV rZ r[ rR r r� rD r r/ r` r@ r� rX r] rY )r r4 r� �tgtr� r r r r0 � s : ) z&TestVariousIteratorArgs.test_takewhilec C s' x t d � t d � t d � d t d d d � f D]� } x� t t t t t t f D]n } g } x: | | � D], } | r� t | � r� ql n | j | � ql W| j t
t t | | � � � | � qS W| j t
t t t | � � | j t
t t t | � � | j t t
t t t | � � � q4 Wd S)
Nr/ r i� rV r� i� i� rW )rV r� )r rS rU rV rZ r[ rR r r� rD r r1 r` r@ r� rX r] rY )r r4 r� rg r� r r r r2 s : ) z&TestVariousIteratorArgs.test_dropwhilec C s
xd d t d � d
t d d d � f D]� } x| t t t t t t f D]b } t | | � � \ } } | j t | � t | | � � � | j t | � t | | � � � qG W| j
t t t | � � | j
t t t
| � � | j
t t t t | � � d � q( Wd S) Nr^ rl i� r_ �333333�?i� i� rW r )zdorh )r rS rU rV rZ r[ rR r6 rD r r` r@ r� rX r] rY )r r4 r� Zit1Zit2r r r r; s . "&z TestVariousIteratorArgs.test_teeN)r r! r" rf ro r� r r� r� r� r� r� r� r r. r! r0 r2 r; r r r r r\ { s
r\ c @ s( e Z d Z d d � Z d d � Z d S)�LengthTransparencyc C sj | j t j t d d � � d � | j t j t d d � � d � | j t j t d � d � d � d S)Nr# r r� )rD r' � length_hintr� )r r r r r s ""z LengthTransparency.test_repeatc C s� | j t j t d d � � d � | j t j t d d � � d � | j t j t d d d �� d � | j t j t d d d �� d � d S) Nr r r r r� r� r� r� )rD r' rj r� )r r r r r s ""%z2LengthTransparency.test_repeat_with_negative_timesN)r r! r" r r r r r r ri s ri c @ s( e Z d Z d d � Z d d � Z d S)�RegressionTestsc sT d d � } � f d d � } g � | d | d � t t � } | j � | � d S)
Nc ss d g � � f d d � } t | � } t | � | d d � <t | | � } t | g t | � � � t � � d S)Nr c s* | r& | d d � =� t � � � n | S)N)rF )r� �first)r= �zr r r� , s
z@RegressionTests.test_sf_793826.<locals>.mutatingtuple.<locals>.g)r r_ r r rF )Ztuple1r= Ztuple2r� �items�genr )r= rm r �
mutatingtuple( s z5RegressionTests.test_sf_793826.<locals>.mutatingtuplec s | a t t � � d d � <d S)N)�Tr )r� )rl r r r= 7 s z)RegressionTests.test_sf_793826.<locals>.fr r rM r: rW rN )r r rM )r: rW rN )r rq rD )r rp r= �secondr )rl r �test_sf_793826% s z RegressionTests.test_sf_793826c s� � f d d � } � f d d � } g � | j t t t | � | d � � � | j � d d g � g � | j t t t | � | d � � � | j � d d g � g � | j t t t | � � � | j � d d g � d S) Nc 3 s6 � j d � d V� j d � t � � j d � d S)Nr r r )r� �AssertionErrorr )�histr r �gen1F s
z,RegressionTests.test_sf_950057.<locals>.gen1c 3 s2 � j d � d V� j d � | r. t � n d S)NrM r r: )r� r )r )ru r r �gen2M s
z,RegressionTests.test_sf_950057.<locals>.gen2Fr r T)r` rt r rn rD r� )r rv rw r )ru r �test_sf_950057B s %% z RegressionTests.test_sf_950057N)r r! r" rs rx r r r r rk # s rk c @ s e Z d Z d d � Z d S)�SubclassWithKwargsTestc s� x� t t t t t t t t t t t
t f D]r � G� f d d � d � � } y | d d � Wq+ t k
r� } z | j
d | j d � WYd d } ~ Xq+ Xq+ Wd S)Nc s% e Z d Z d � f d d � Z d S)zBSubclassWithKwargsTest.test_keywords_in_subclass.<locals>.SubclassNc s � j | | � d S)N)r )r �newargr
)�clsr r r f s zKSubclassWithKwargsTest.test_keywords_in_subclass.<locals>.Subclass.__init__)r r! r" r r )r{ r r �Subclasse s r| rz r z does not take keyword argumentsr )r� r r� r� rn r_ r r$ r/ r1 r� r� r@ Z assertNotInr
)r r| �errr )r{ r �test_keywords_in_subclassa s z0SubclassWithKwargsTest.test_keywords_in_subclassN)r r! r" r~ r r r r ry ` s ry c @ sU e Z d Z d d � Z e j Z d d � Z d d � Z d d � Z d d
� Z d S) �
SizeofTestc C s t j d � | _ d S)Nr )�structZcalcsize�ssize_t)r r r r �setUpp s zSizeofTest.setUpc C s[ t j d � } | j } | t d d � | d | j � | t d � | d | j � d S) NZ3PirU Z12r rT r/ )�abc)
r� r� r� r� r� r� r� r� r� r� )r � calcobjsize� check_sizeofr� r� )r �basesize�checkr r r �test_product_sizeofu s !z SizeofTest.test_product_sizeofc C sd t j d � } | j } | t d d � | d | j � | t t d � d � | d | j � d S)N�3Pnirm rM r/ r: )r r� r� r� r� r )r r� r� r r r �test_combinations_sizeof{ s !z#SizeofTest.test_combinations_sizeofc C sj t } t j d � } | j } | | d d � | d | j � | | t d � d � | d | j � d S)Nr� rm rM r/ r: )r� r r� r� r� r )r r� r� r� r r r �)test_combinations_with_replacement_sizeof� s
!z4SizeofTest.test_combinations_with_replacement_sizeofc C s� t j d � } | j } | t d � | d | j d | j � | t d d � | d | j d | j � | t d d � | d | j d | j � | t t d � d � | d | j d | j � d S)NZ4Pnirm r: rM r� rW r/ )r r� r� r� r� r )r r� r� r r r �test_permutations_sizeof� s z#SizeofTest.test_permutations_sizeofN)
r r! r" r� r r� r� r� r� r� r r r r r n s r a� Doctest for examples in the library reference: libitertools.tex
>>> amounts = [120.15, 764.05, 823.14]
>>> for checknum, amount in zip(count(1200), amounts):
... print('Check %d is for $%.2f' % (checknum, amount))
...
Check 1200 is for $120.15
Check 1201 is for $764.05
Check 1202 is for $823.14
>>> import operator
>>> for cube in map(operator.pow, range(1,4), repeat(3)):
... print(cube)
...
1
8
27
>>> reportlines = ['EuroPython', 'Roster', '', 'alex', '', 'laura', '', 'martin', '', 'walter', '', 'samuele']
>>> for name in islice(reportlines, 3, None, 2):
... print(name.title())
...
Alex
Laura
Martin
Walter
Samuele
>>> from operator import itemgetter
>>> d = dict(a=1, b=2, c=1, d=2, e=1, f=2, g=3)
>>> di = sorted(sorted(d.items()), key=itemgetter(1))
>>> for k, g in groupby(di, itemgetter(1)):
... print(k, list(map(itemgetter(0), g)))
...
1 ['a', 'c', 'e']
2 ['b', 'd', 'f']
3 ['g']
# Find runs of consecutive numbers using groupby. The key to the solution
# is differencing with a range so that consecutive numbers all appear in
# same group.
>>> data = [ 1, 4,5,6, 10, 15,16,17,18, 22, 25,26,27,28]
>>> for k, g in groupby(enumerate(data), lambda t:t[0]-t[1]):
... print(list(map(operator.itemgetter(1), g)))
...
[1]
[4, 5, 6]
[10]
[15, 16, 17, 18]
[22]
[25, 26, 27, 28]
>>> def take(n, iterable):
... "Return first n items of the iterable as a list"
... return list(islice(iterable, n))
>>> def enumerate(iterable, start=0):
... return zip(count(start), iterable)
>>> def tabulate(function, start=0):
... "Return function(0), function(1), ..."
... return map(function, count(start))
>>> def nth(iterable, n, default=None):
... "Returns the nth item or a default value"
... return next(islice(iterable, n, None), default)
>>> def quantify(iterable, pred=bool):
... "Count how many times the predicate is true"
... return sum(map(pred, iterable))
>>> def padnone(iterable):
... "Returns the sequence elements and then returns None indefinitely"
... return chain(iterable, repeat(None))
>>> def ncycles(iterable, n):
... "Returns the sequence elements n times"
... return chain(*repeat(iterable, n))
>>> def dotproduct(vec1, vec2):
... return sum(map(operator.mul, vec1, vec2))
>>> def flatten(listOfLists):
... return list(chain.from_iterable(listOfLists))
>>> def repeatfunc(func, times=None, *args):
... "Repeat calls to func with specified arguments."
... " Example: repeatfunc(random.random)"
... if times is None:
... return starmap(func, repeat(args))
... else:
... return starmap(func, repeat(args, times))
>>> def pairwise(iterable):
... "s -> (s0,s1), (s1,s2), (s2, s3), ..."
... a, b = tee(iterable)
... try:
... next(b)
... except StopIteration:
... pass
... return zip(a, b)
>>> def grouper(n, iterable, fillvalue=None):
... "grouper(3, 'ABCDEFG', 'x') --> ABC DEF Gxx"
... args = [iter(iterable)] * n
... return zip_longest(*args, fillvalue=fillvalue)
>>> def roundrobin(*iterables):
... "roundrobin('ABC', 'D', 'EF') --> A D E B F C"
... # Recipe credited to George Sakkis
... pending = len(iterables)
... nexts = cycle(iter(it).__next__ for it in iterables)
... while pending:
... try:
... for next in nexts:
... yield next()
... except StopIteration:
... pending -= 1
... nexts = cycle(islice(nexts, pending))
>>> def powerset(iterable):
... "powerset([1,2,3]) --> () (1,) (2,) (3,) (1,2) (1,3) (2,3) (1,2,3)"
... s = list(iterable)
... return chain.from_iterable(combinations(s, r) for r in range(len(s)+1))
>>> def unique_everseen(iterable, key=None):
... "List unique elements, preserving order. Remember all elements ever seen."
... # unique_everseen('AAAABBBCCDAABBB') --> A B C D
... # unique_everseen('ABBCcAD', str.lower) --> A B C D
... seen = set()
... seen_add = seen.add
... if key is None:
... for element in iterable:
... if element not in seen:
... seen_add(element)
... yield element
... else:
... for element in iterable:
... k = key(element)
... if k not in seen:
... seen_add(k)
... yield element
>>> def unique_justseen(iterable, key=None):
... "List unique elements, preserving order. Remember only the element just seen."
... # unique_justseen('AAAABBBCCDAABBB') --> A B C D A B
... # unique_justseen('ABBCcAD', str.lower) --> A B C A D
... return map(next, map(itemgetter(1), groupby(iterable, key)))
>>> def first_true(iterable, default=False, pred=None):
... '''Returns the first true value in the iterable.
...
... If no true value is found, returns *default*
...
... If *pred* is not None, returns the first item
... for which pred(item) is true.
...
... '''
... # first_true([a,b,c], x) --> a or b or c or x
... # first_true([a,b], x, f) --> a if f(a) else b if f(b) else x
... return next(filter(pred, iterable), default)
This is not part of the examples but it tests to make sure the definitions
perform as purported.
>>> take(10, count())
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
>>> list(enumerate('abc'))
[(0, 'a'), (1, 'b'), (2, 'c')]
>>> list(islice(tabulate(lambda x: 2*x), 4))
[0, 2, 4, 6]
>>> nth('abcde', 3)
'd'
>>> nth('abcde', 9) is None
True
>>> quantify(range(99), lambda x: x%2==0)
50
>>> a = [[1, 2, 3], [4, 5, 6]]
>>> flatten(a)
[1, 2, 3, 4, 5, 6]
>>> list(repeatfunc(pow, 5, 2, 3))
[8, 8, 8, 8, 8]
>>> import random
>>> take(5, map(int, repeatfunc(random.random)))
[0, 0, 0, 0, 0]
>>> list(pairwise('abcd'))
[('a', 'b'), ('b', 'c'), ('c', 'd')]
>>> list(pairwise([]))
[]
>>> list(pairwise('a'))
[]
>>> list(islice(padnone('abc'), 0, 6))
['a', 'b', 'c', None, None, None]
>>> list(ncycles('abc', 3))
['a', 'b', 'c', 'a', 'b', 'c', 'a', 'b', 'c']
>>> dotproduct([1,2,3], [4,5,6])
32
>>> list(grouper(3, 'abcdefg', 'x'))
[('a', 'b', 'c'), ('d', 'e', 'f'), ('g', 'x', 'x')]
>>> list(roundrobin('abc', 'd', 'ef'))
['a', 'd', 'e', 'b', 'f', 'c']
>>> list(powerset([1,2,3]))
[(), (1,), (2,), (3,), (1, 2), (1, 3), (2, 3), (1, 2, 3)]
>>> all(len(list(powerset(range(n)))) == 2**n for n in range(18))
True
>>> list(powerset('abcde')) == sorted(sorted(set(powerset('abcde'))), key=len)
True
>>> list(unique_everseen('AAAABBBCCDAABBB'))
['A', 'B', 'C', 'D']
>>> list(unique_everseen('ABBCcAD', str.lower))
['A', 'B', 'C', 'D']
>>> list(unique_justseen('AAAABBBCCDAABBB'))
['A', 'B', 'C', 'D', 'A', 'B']
>>> list(unique_justseen('ABBCcAD', str.lower))
['A', 'B', 'C', 'A', 'D']
>>> first_true('ABC0DEF1', '9', str.isdigit)
'0'
�
libreftestc C s� t t t t t t t t f } t j | � | r� t
t d � r� d d l } d g d } xA t
t | � � D]- } t j | � | j � t j � | | <ql Wt | � n t j t j t | � d S)N�gettotalrefcountr rW )r9 r\ rJ rk ri ry r@ r r Z run_unittest�hasattrr� �gcr r Zcollectr� r Z run_doctest�modulesr )�verboseZ test_classesr� Zcountsr r r r � test_main� s
r� �__main__r� T)=ZunittestZtestr � itertoolsr+ �decimalr Z fractionsr r� r' r rr r1 � functoolsr r� ZMAX_Py_ssize_tr� Zminsizer
r r r r r r r r r& r* r+ r- r. r0 r rd rt ZTestCaser9 r@ rJ rR rS rU rV r� rX rY rZ r[ r\ ri rk ry Z cpython_onlyr r� Z__test__r� r r r r r �<module> sz
� � � � � de
�
=�