Your IP : 216.73.216.85
���]�m � @ sQ 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 dK e9 i Z: d dL dM � Z; e< dN k rMe; dO dP � d S)Q� N)�support)�*)�Decimal)�Fraction)�reduce� c G s t t | � � S)N)�list�zip)�args� r �3/opt/alt/python35/lib64/python3.5/test_itertools.py�lzip s r
c C s d | S)N� r )�xr r r �onearg s r c G s
t � d S)N)�
ValueError)r
r r r �errfunc s r c c s x d D] } | Vq Wd S)Nr r r )r r r r )�ir r r �gen3 s
r c C s | d d k S)Nr r r )r r r r �isEven" s r c C s | d d k S)Nr 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 d � Z d d � Z d S)�StopNowc 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__r r r r r r r 1 s r c C s t t | | � � S)N)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)Nr )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 r/ 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 )�.0r4 r r r �
<listcomp>M s r7 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 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/ d0 � Z d1 d2 � Z d3 d4 � Z d5 d6 � Z e
j
d � d7 d8 � � Z d9 d: � Z e
j
d � d; d< � � Z! d= d> � Z" d? d@ � Z# dA dB � Z$ e
j dC dD � � Z% e
j
d � 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[ d\ � Z1 d] d^ � Z2 d S)_� TestBasicOps� r Nc ss d � � f d d � � t j | | � } t j | � } | j t | � t | � � � | � � | � } } | j | | � | r� � | � }
| j | |
� t j | � } d } y/ x( t | � D] }
t | � | d 7} q� WWn t k
r� Yn Xt j | | � } t j | � } � | � � | � } } | j | | � | ro� | | d � � }
| j | |
� d S)Nr c sw � d k r t d � � 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 )r6 �e)�expandr r r r7 ^ 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 )r0 r2 r1 � assertEqual�typer �nextr )r �protocolr@ rB r% �compare�dumpZi2�a�b�cZi3Ztookr Zi4r )r; rB 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 QRXx7 t t j d � D]" } | j | t t d � � � q�Wd S)Nr. r r � � � � � �$ �- r( �abcrI �ab� � r r � � r9 � � i� i� )rC r �
accumulater �int�complexr r �map� assertRaisesr? �min�maxr& r' �chrr0 �HIGHEST_PROTOCOLrL )r �typr3 r4 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)Nr )� iterablesr@ �elementr r r �chain2� s
z'TestBasicOps.test_chain.<locals>.chain2rT �def�abcdef� r9 �abcdr rM )�chainrC 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 r9 rm r rM )rC 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 r9 rm r rM r rG )�copy�deepcopy�
picklecopiersrn rC r rE r% r` r? r r0 rd rL )r Zoperr@ r4 r r r �test_chain_reducible� s """.&z!TestBasicOps.test_chain_reduciblec C s9 | j t t � j f � | j t t � j g � | j t t � j d � | j t t � j g f � | j t t � j t g � g f � t � } | j t d d g � f � | j t | � d d d d d d g � t � } | j t d d g � t d
g � f � | j t | � d
d d d d d d g � d S) Nr rT rj rI rJ rK �dr: �fZghi)r` r? rn � __setstate__�iterrC r )r r@ r r r �test_chain_setstate� s % ( (z TestBasicOps.test_chain_setstatec
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 r5 � s z0TestBasicOps.test_combinations.<locals>.<lambda>� �ABCD�A�B�C�Dr9 rM r c 3 s� t | � � t � � } | | k r( d St t | � � } t � f d d � | D� � Vx� x7 t t | � � D] } | | | | | k rn Pqn Wd S| | d 7<x1 t | d | � D] } | | d d | | <q� Wt � f d d � | D� � Vq[ Wd S)Nc 3 s | ] } � | Vq d S)Nr )r6 r )�poolr r � <genexpr>� s zHTestBasicOps.test_combinations.<locals>.combinations1.<locals>.<genexpr>r c 3 s | ] } � | Vq d S)Nr )r6 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 sr t | � � t � � } xS t t | � | � D]< } t | � t | � k r. t � f d d � | D� � Vq. Wd S)Nc 3 s | ] } � | Vq d S)Nr )r6 r )r� r r r� � s zHTestBasicOps.test_combinations.<locals>.combinations2.<locals>.<genexpr>)r� r� � permutationsr �sortedr )r( r+ r r� r )r� r �
combinations2� s
z5TestBasicOps.test_combinations.<locals>.combinations2c 3 sr t | � � t � � } xS t t | � | � D]< } t t | � � | k r. t � f d d � | D� � Vq. Wd S)Nc 3 s | ] } � | Vq d S)Nr )r6 r )r� r r r� s zHTestBasicOps.test_combinations.<locals>.combinations3.<locals>.<genexpr>)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 )r6 r r r r r7 s z2TestBasicOps.test_combinations.<locals>.<listcomp>c 3 s | ] } | � k Vq d S)Nr )r6 r: )�valuesr r r� s z1TestBasicOps.test_combinations.<locals>.<genexpr>c s" g | ] } | � k r | � q Sr r )r6 r: )rK r r r7 s ���)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 rM )r r rM )r r rM )r r rM )r r rM )r r rM )r` r? � combinationsr rt rC r rE r r� r* r� r� �
assertTrue�allr0 rd rL )
r �op�testIntermediater� r� r� r r+ �resultr4 r )rK r� r �test_combinations� sP %
$
F"
& z TestBasicOps.test_combinationsc C s. | j t t f � � t d d � Wd QRXd 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 )rC 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 r5 * s zATestBasicOps.test_combinations_with_replacement.<locals>.<lambda>�ABCr} r~ r c 3 s� t | � � t � � } | r) | r) d Sd g | } t � f d d � | D� � Vx{ x3 t t | � � D] } | | | d k rj Pqj Wd S| | d g | | | | d � <t � f d d � | D� � VqW Wd S)Nr c 3 s | ] } � | Vq d S)Nr )r6 r )r� r r r� ; s zPTestBasicOps.test_combinations_with_replacement.<locals>.cwr1.<locals>.<genexpr>r c 3 s | ] } � | Vq d S)Nr )r6 r )r� r r r� C s )r� r� r� r )r( r+ r r� r r )r� r �cwr13 s
#z=TestBasicOps.test_combinations_with_replacement.<locals>.cwr1c 3 su t | � � t � � } xV t t | � d | �D]< } t | � t | � k r1 t � f d d � | D� � Vq1 Wd S)N�repeatc 3 s | ] } � | Vq d S)Nr )r6 r )r� r r r� K s zPTestBasicOps.test_combinations_with_replacement.<locals>.cwr2.<locals>.<genexpr>)r� r� �productr r� r )r( r+ r r� r )r� r �cwr2E 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 �numcombsM s zATestBasicOps.test_combinations_with_replacement.<locals>.numcombsrV c S s g | ] } d | d � q S)rW r� r )r6 r r r r r7 S s zCTestBasicOps.test_combinations_with_replacement.<locals>.<listcomp>r c S s g | ] \ } } | � q Sr r )r6 �k�vr r r r7 c s c 3 s | ] } | � k Vq d S)Nr )r6 r: )r� r r r� f s zBTestBasicOps.test_combinations_with_replacement.<locals>.<genexpr>c s" g | ] } | � k r | � q Sr r )r6 r: )rK r r r7 h s 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 rt rC r rE r r� r� r� r� r� �groupbyr� r0 rd rL )
r �cwrr� r� r� r� r� r r+ r� Z
regular_combsZnorunsr4 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 QRXd S)Nr� r � i @)r` r� r� r� )r r r r �+test_combinations_with_replacement_overflowo 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� rC r� r� r_ r� r� r )r r� r r r �.test_combinations_with_replacement_tuple_reuseu 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 � � � � x4 t t
j d � D] } | j | t � | � � q�WqWq� Wd S)NrT r r r{ r3 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)Nr c 3 s | ] } � | Vq d S)Nr )r6 r )r� r r r� � s zHTestBasicOps.test_permutations.<locals>.permutations1.<locals>.<genexpr>r c 3 s | ] } � | Vq d S)Nr )r6 r )r� r r r� � s ���)r� r� r r r� )r( r+ r r� Zcyclesr r� r )r� r �
permutations1� s& .( 0
(z5TestBasicOps.test_permutations.<locals>.permutations1c 3 s� t | � � t � � } | d k r* | n | } xV t t | � d | �D]< } t t | � � | k rI t � f d d � | D� � VqI Wd S)Nr� c 3 s | ] } � | Vq d S)Nr )r6 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 )r6 r r r r r7 � s z2TestBasicOps.test_permutations.<locals>.<listcomp>c 3 s | ] } | � k Vq d S)Nr )r6 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 rC r r r� r* r� r� r� r� r0 rd rL )r r� r� r r+ r� �pr4 r )r� r �test_permutations{ s8 <"
* z TestBasicOps.test_permutationsc C s. | j t t f � � t d d � Wd QRXd S)Nr} 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 )rC 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 )r6 �tr r r r7 � s z3TestBasicOps.test_combinatorics.<locals>.<listcomp>c s. g | ]$ } t t | � � � k r | � q Sr )r� r� )r6 r� )r+ r r r7 � s c S s. g | ]$ } t | � t | � k r | � q Sr )r� r )r6 r� r r r r7 � s c s. g | ]$ } t t | � � � k r | � q Sr )r� r� )r6 r� )r+ r r r7 � s )
r r r� r� r� r� rC r� r* r� r� �filter� __contains__)r r r3 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 � xzd d � d d � g t D]\} xSd 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�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 r5 � s z,TestBasicOps.test_compress.<locals>.<lambda>c S s
t j | � S)N)rr rs )rI r r r r5 � s ZCEFZBCDEFr )r r ) rC r �compressrn rp r� r r` r? rt rE )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 d t t d
� � t t t d
t d
� � � | j t d t t d
� � t t t d
t d
� � � | j t d t d � � d d
d g � | j t d t d/ � � d0 d1 d2 g � | j t d t 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 � g � d3 } | j t d t | � � | | d | d g � t d � } | j t | � d � t
| � | j t | � d � t d4 � } | j t | � d � t
| � | j t
| � d5 � | j t t d � � d � | j t t d � � d � | j t t
t d � � � t � xu t j d
t j d
d6 d7 d d t j d
t j d
f D]7 } t t | � � } d j | � } | j | | � q�Wx� d8 d t d
t d
f D]� } t | � } | j t
t j | � � | � | j t
t j | � � | � x1 t t j d � D] } | j | t | � � qaWq�Wt d t d
� t j � d S)9NrT rI r rJ r rK r rM r9 rW r. g
@g @g @y @z1.1z2.1z3.1rX i� zcount(3)zcount(4)rY z count(-9)g �$@z count(10.25)g $@z count(10.0)z count(%r))rI r )rJ r )rK r )rI rM )rJ r9 )rK rW )rI rM )rJ r9 r� r� )rI r� )rJ r ���r� )rI r� r� )rJ r� y
@ �y
@ �y @ �y @ �lC i����i����i����r� r� )rC r
�countr% r r` r? �maxsizer r r r �reprrE rD �float�sys�__mod__rr rs r0 rd rL �exc_info)r �BIGINTrK r �r1�r2�valuer4 r r r �
test_count sT %(... (( "!+0
"C! zTestBasicOps.test_countc C s | j t d t d d � � d2 d3 d4 g � | j t d t d d d
d � � d5 d6 d7 g � | j t d t d
d8 � � d9 d; d= g � | j t t d d � | j t d t d d � � d> d? d@ g � | j t d t d d � � dA dB dC g � | j t d t d d � � dD dE dF 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 t d � � t t d d d t d t d � � � | j t d t d d � � d d d g � | j t d t d dG � � d dH dI 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 � dJ } | j t d t d
| � � 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 dK d � } | j t | � d' � t | � | j t | � d' � t dL dM � } | 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. � | j t t d d- � � d/ � t d d- � } | j t
t | � � t � | j t
t | � � t � xt j d t j d dN dO d d t j d t j d f D]� } x� t j d t j d dP dQ d d d t j d t j d f D]� } t t | | � � } | d k r�d0 | } n d1 | | f } | j | | � x4 t t j d � D] } | j | t | | � � q�Wq}Wq7Wd S)RNrT r rM rI rJ rW rK rX �start�stepr r r9 � rO �d r. g �?g
@g @y @g @g !@y @z1.1z.1z1.2z1.3rV � rP i� g @g )@g .@z count(3, 5)z count(8, 5)rY z count(-9, 0)z
count(-9, -3)zcount(-12, -3)g %@zcount(10.5, 1.25)z count(10.5)g �?zcount(10.5, 1.0)zcount(10, 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 r9 )rI r )rJ rW )rK rX y
@ �y @ �y !@ �lC i����i����r� i����r� i����r� )rC r
r� r` r? r% r� r r r r r� rE rD r] r� r� r0 rd rL )r r� rK r r� r� r� r4 r r r �test_count_with_stride7 sf +!+++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)rC r% �cycler r` r? r# r rE rr rs r r0 rd r1 r2 rL )r rK r4 r r r �
test_cyclen s$ % F ('
'
zTestBasicOps.test_cyclec C s{ | j t t d � j f � | j t t d � j g � | j t t d � j d � | j t t d � j g f � | j t t d � j f d f � t d � } | j d d g d f � | j t t | d � � d d d d d d d d d d d d d d d g � t d � } | j d d g d
f � | j t t | d � � d d d d d d d d d d d d d d d g � d S) Nrl r rT �deZfgrO rI rJ rK r )r` r? r� rx rC r r# )r r@ r r r �test_cycle_setstate� s" " z TestBasicOps.test_cycle_setstatec
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 r5 � 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 r5 � s c S s g | ] \ } } | � q Sr r )r6 r� �gr r r r7 � s z-TestBasicOps.test_groupby.<locals>.<listcomp>c S s g | ] } | d � q S)r r )r6 r+ r r r r7 � s Z abracadabrac S s g | ] \ } } | � q Sr r )r6 r� r� r r r r7 � s rI rJ rK rv r+ c S s4 g | ]* \ } } t t | d d � � r | � q S)r r )r r# )r6 r� r� r r r r7 � s c S s. g | ]$ \ } } t t | � � | f � q Sr )r� r )r6 r� r� r r r r7 � s rW c S s. g | ]$ \ } } t t | � � | f � q Sr )r� r )r6 r� r� r r r r7 � 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 )r6 r� r� )�funcr r r7 � 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 �dst)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 rv )r r+ )rW rI )r r+ )r rJ )rC r r� r� r` r? �appendr r0 rd r1 r2 r, r- r� r� r� � Exceptionr� )r r3 �dupr� r� �elemr4 ZikZig�keysZ expectedkeysr+ r� r� r� r )r� r� r � test_groupby� 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 r9 r c S s | S)Nr )r r r r r5 s z*TestBasicOps.test_filter.<locals>.<lambda>c S s | S)Nr )r r r r r5 s rV rM )rC r r� r r �boolr% r� r` r? rE rr rs r0 rd r1 r2 rL )r �ansrK r4 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 r9 rV c S s | S)Nr )r r r r r5 ) s z/TestBasicOps.test_filterfalse.<locals>.<lambda>c S s | S)Nr )r r r r r5 * s )rC r � filterfalser r r� r% r� r` r? rE r0 rd rL )r r4 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 )r6 r �yr r r r7 2 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 )r6 �pairr r r r7 ; s rj c S s g | ] } | � q Sr r )r6 r� r r r r7 = s )rI r )rJ r )rK r ) r r� rC r r r
r% r` r? )r r� r r r �test_zip0 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 )r6 r r� r r r r7 H s z5TestBasicOps.test_zip_tuple_reuse.<locals>.<listcomp>rI r rJ r rK r c S s" g | ] \ } } | | f � q Sr r )r6 r r� r r r r7 K s c S s" g | ] \ } } | | f � q Sr r )r6 r r� r r r r7 O s c S s" g | ] \ } } | | f � q Sr r )r6 r r� r r r r7 U 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 rC ra rb r� �dict�fromkeysrr r� rs r r0 rd r1 r2 rE rL )r �idsr� r4 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� )r6 �arg)r r r r7 c s z;TestBasicOps.test_ziplongest.<locals>.<listcomp>.<listcomp>)r� )r6 )r
)r r r7 c 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 )r6 r: r r r r� g s z:TestBasicOps.test_ziplongest.<locals>.<listcomp>.<genexpr>)r� )r6 r� r r r r7 g 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 )r6 r� r r r r7 � s rj c S s g | ] } | � q Sr r )r6 r� r r r r7 � s )r rb r_ r� rC 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 rC ra rb r� r� r� )r r� r r r � test_zip_longest_tuple_reuse� 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 r0 rd rL r )r r4 r r r �test_zip_longest_pickling� 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__� s z5TestBasicOps.test_bug_7244.<locals>.Repeater.__init__c S s | S)Nr )r r r r r � 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 � s z5TestBasicOps.test_bug_7244.<locals>.Repeater.__next__N)r r! r" r r r r r r r �Repeater� s r
r rM r r9 c S sj g } x] t | | d d �D]F \ } } t j d � � t | | f � Wd QRX| j | | f � q W| S)Nr� r �stdout)r r Zcaptured_output�printr� )r� r� r� r r� r r r �run� s "z'TestBasicOps.test_bug_7244.<locals>.runr r� )r r )r r )r r )r r )r r )r r )r r )r rC r< r rE r` )r r
r� r� r
r@ r r r �
test_bug_7244� 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 r9 r� rV rN c s6 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� | | 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� )r6 r� r r r r� � s z>TestBasicOps.test_product.<locals>.product1.<locals>.<genexpr>c s s | ] \ } } | | Vq d S)Nr )r6 r� r r r r r� � s c s s | ] \ } } | | Vq d S)Nr )r6 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)Nr� r c s* g | ] } � D] } | | g � q q Sr r )r6 r r� )r� r r r7 � s z?TestBasicOps.test_product.<locals>.product2.<locals>.<listcomp>)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 )�random�choice)r6 r� )�argtypesr r r7 � s z-TestBasicOps.test_product.<locals>.<listcomp>rW )rI )rJ )r r )r r )r r )r r )r r )r r i�� )r rC r r� r� r� r` r? r� r� r � randranger) r_ ry )r r
r� r+ r r r Z expected_lenr )r r � test_product� s2 -$$4 '/ 3'("((zTestBasicOps.test_productc C s8 | j t t f � � t d g d d d � Wd QRXd S)NrU r rW r� � r{ i )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 )rC 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 rC r rr r� rs r0 rd rL )r r
r� r4 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� rx rC rE 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 r9 y zrepeat((1+0j))rW zrepeat((1+0j), 5)zrepeat((1+0j), 0)r. )r rI )r rI )r rI r� y �? y �? �aar" )rC r r� r
r r% r` r? r� rE rr rs r0 rd rL )r r+ rK r4 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� )rC r� r� )r r r r � test_repeat_with_negative_times2 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. r9 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 )rC r r_ r& �powr r r� r% r` r? �negrE r r r rr rs r0 rd rL )r r� rK r4 r r r �test_map8 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 r9 rW �extrar. r r rX r r rX i )r9 rW )r9 rW )r9 rW )r9 rW r r rX )rC r �starmapr& r% r r r% r� ry r` r? rE r r r rr rs r0 rd rL )r r� rK r4 r r r � test_starmapV 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 r9 rW r rI �2 c s s | ] } | Vq d S)Nr )r6 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/ r0 �r+ �r� )r1 r2 ���r3 r� r� )r. r� rM )r. rM r� )r. r� )r. rM )r� )r r )rC r r# r ry r` r? r r� r� r� rE rr rs r0 rd rL �weakref�refZassertIsNotNoner Z
gc_collectZ assertIsNone)r r
Ztgtargsr@ ZrarK r4 �wrr r r � test_isliceq 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 r9 rN rX r( r. r )r9 rW )r9 rW )r9 rW )rC r � takewhiler/ r` r? r& r% rE r r r� r rr rs r r0 rd rL )r r� r� r4 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 r9 rN rX r( r. )r9 rW )r9 rW )r9 rW )rC r � dropwhiler/ r` r? r& r% rE r r rr rs r r0 rd rL )r r� r4 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� �invalidc S s g | ] } t | � � q Sr )r )r6 r r r r r7 " s z)TestBasicOps.test_tee.<locals>.<listcomp>r. � __class__i' �< rG r� ) �teerC r r r
r rE r �shuffler� r` r? rD r r� r� r� r4 �proxy�getattr�ReferenceErrorrr r% rs r0 rd rL ry )r r rI rJ r r� �orderZlistsZitsr� rK r� rv �t1Zt2ZtnewZt3r� r� Zlong_ansr4 r r r �test_tee� s� ."""%
$ ".%%. F
))
))
))
)) zTestBasicOps.test_teec C sD t t d d � � \ } } y t | � ~ Wn ~ ~ � Yn Xd S)Ni -1)r@ r� r )r ZforwardZbackwardr r r �test_tee_del_backwardw 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 r5 � s z1TestBasicOps.test_StopIteration.<locals>.<lambda>c S s | S)Nr )r r r r r5 � s )r` r rE r rn r� r� r r# r@ r� r� r� r_ r8 r: r) )r rw r� �qr r r �test_StopIteration� s "z TestBasicOps.test_StopIteration)3r r! r" rL rf ro rq ru rz 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* r7 r9 r; rG rH rJ r r r r r8 P sZ )
P L ? ! " 0 7 n
* ( :
L � r8 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� TestExamplesc C s> | j t t d d d d d g � � d d d d d g � d S) Nr r rM r9 rW rN r. rO )rC r r\ )r r r r rf � 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 r9 rW rN r. rO ) r r0 rd r\ rC r r1 r2 rE rr rs )r r� Z accumulatedr4 r@ r r r �test_accumulate_reducible� s 59 )z&TestExamples.test_accumulate_reduciblec C s� t d d d g t j � } | j t | � d � xR t t j d � D]= } t j t j | | � � } | j t
| � d d g � qE W| j t
t j | � � d d g � | j t
t j | � � d d g � d S)Nr TF)
r\ r& �is_rC rE r r0 rd r1 r2 r rr rs )r r@ r4 Zit_copyr r r � test_accumulate_reducible_none� s %z+TestExamples.test_accumulate_reducible_nonec C s& | j d j t d d � � d � d S)Nrl r� �DEFr� )rC �joinrn )r r r r ro � s zTestExamples.test_chainc C s, | j d j t j d d g � � d � d S)Nrl r� rO r� )rC rP 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)Nr| r r} r~ r r� r9 rM r r )r} r~ )r} r )r} r� )r~ r )r~ r� )r r� )r r r )r r rM )r r rM )r r rM )rC 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 r} r~ r )r} r} )r} r~ )r} r )r~ r~ )r~ r )r r )rC 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� )rC 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� )rC r r# r� )r r r r r� � s zTestExamples.test_countc C s/ | j t t t d � d � � t d � � d S)Nr| r� Z ABCDABCDABCD)rC 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 r5 � s z-TestExamples.test_dropwhile.<locals>.<lambda>r r9 rN )rC r r: )r r r r r; � 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 )r6 r� r� r r r r7 � s z-TestExamples.test_groupby.<locals>.<listcomp>ZAAAABBBCCDAABBBZABCDABc S s" g | ] \ } } t | � � q Sr )r )r6 r� r� r r r r7 � s Z
AAAABBBCCDZAAAAZBBB�CCr� )rC 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 r5 � s z*TestExamples.test_filter.<locals>.<lambda>r. r rM rW rV rY )rC 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 r5 � s z/TestExamples.test_filterfalse.<locals>.<lambda>r. r r r9 rN rX )rC 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 r{ rY i� )r rM r. )rW r rM )rC 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 ZABr9 ZCDZCDEFGr ZACEG)rC r r# )r r r r r7 � s %((zTestExamples.test_islicec C s) | j t t d d � � d d g � d S) Nr| �xyr} r r~ r� )r} r )r~ r� )rC 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)Nr| rR r� �-r} r r~ r� r r� )r} r )r~ r� )r rS )r� rS )rC 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)
Nr| 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 )rC 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)Nr| rR 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 )rC r r� r_ r� rU 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 )rC 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. r{ rY i� )r rW )rM r )r. rM )rC 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 r5 s z-TestExamples.test_takewhile.<locals>.<lambda>r r9 rN )rC r r8 )r r r r r9 s zTestExamples.test_takewhileN)r r! r" rf rL rN ro rq r� r� r� r� r� r; r� r� r� r' r7 r� rT r� r r# rV r9 r r r r rK � s.
rK 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� rE )r �iterator� containerr r r � makecycle s
zTestGC.makecyclec C s, g } | j t d d | d g � | � d S)Nr r rM )rZ r\ )r rI r r r rf s zTestGC.test_accumulatec C s g } | j t | � | � d S)N)rZ rn )r rI r r r ro s zTestGC.test_chainc C s& g } | j t j | g � | � d S)N)rZ 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 )rZ 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 )rZ 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 )rZ 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 )rD r] r� rZ r� )r rI r[ r r r r� $ s zTestGC.test_countc C s' g } | j t | g d � | � d S)Nr )rZ r� )r rI r r r r� ) s zTestGC.test_cyclec C s, g } | j t t d | | g � | � d S)Nr )rZ r: r� )r rI r r r r; - 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 r5 3 s z%TestGC.test_groupby.<locals>.<lambda>)rZ r� )r rI r r r r� 1 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 r5 8 s z'TestGC.test_issue2246.<locals>.<lambda>r� r )r� r �__dict__�
setdefaultr� )r r r� r r� r r r �test_issue22465 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 r5 > s z$TestGC.test_filter.<locals>.<lambda>r )rZ 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 r5 B s z)TestGC.test_filterfalse.<locals>.<lambda>)rZ r� )r rI r r r r� @ s zTestGC.test_filterfalsec C s1 g } | j t | g d | g d � | � d S)Nr rM )rZ r )r rI r r r r� D 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� )rZ r )r rI rJ r r r rT H 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 r5 P s z!TestGC.test_map.<locals>.<lambda>r )rZ r_ )r rI r r r r' N s zTestGC.test_mapc C s* g } | j t | g d d � | � d S)Nr )rZ r# )r rI r r r r7 R s zTestGC.test_islicec C s/ g } | j t d d | d g d � | � d S)Nr r rM )rZ r� )r rI r r r r� V s zTestGC.test_permutationsc C s2 g } | j t d d | d g d d �| � d S)Nr r rM r� )rZ r� )r rI r r r r Z s zTestGC.test_productc C s g } | j t | � | � d S)N)rZ 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 r5 d s z%TestGC.test_starmap.<locals>.<lambda>r )rZ r) )r rI r r r r* b s zTestGC.test_starmapc C s/ g } | j t t d d | | g � | � d S)Nr r )rZ r8 r� )r rI r r r r9 f s zTestGC.test_takewhileN)r r! r" rZ rf ro rq r� r� r� r� r� r; r� r^ r� r� r� rT r' r7 r� r r# r* r9 r r r r rW s. rW c c s x | D] } | Vq Wd S)Nr )�seqnr r r r �Rj s
r` c @ s( e Z d Z d d � Z d d � Z d S)�Gc C s
| | _ d S)N)r_ )r r_ r r r r q s z
G.__init__c C s | j | S)N)r_ )r r r r r � __getitem__s s z
G.__getitem__N)r r! r" r rb r r r r ra o s ra c @ s4 e Z d Z d d � Z d d � Z d d � Z d S)�Ic C s | | _ d | _ d S)Nr )r_ r )r r_ r r r r x s z
I.__init__c C s | S)Nr )r r r r r { s z
I.__iter__c C sA | j t | j � k r t � | j | j } | j d 7_ | S)Nr )r r� r_ r )r r� r r r r } s
z
I.__next__N)r r! r" r r r r r r r rc v s rc c @ s( e Z d Z d d � Z d d � Z d S)�Igc C s | | _ d | _ d S)Nr )r_ r )r r_ r r r r � s z Ig.__init__c c s x | j D] } | Vq
Wd S)N)r_ )r �valr r r r � s z Ig.__iter__N)r r! r" r r r r r r rd � s rd c @ s( e Z d Z d d � Z d d � Z d S)r� c C s | | _ d | _ d S)Nr )r_ r )r r_ r r r r � s z
X.__init__c C sA | j t | j � k r t � | j | j } | j d 7_ | S)Nr )r r� r_ r )r r� r r r r � s
z
X.__next__N)r r! r" r r r r r r r� � s r� c @ s( e Z d Z d d � Z d d � Z d S)�Nc C s | | _ d | _ d S)Nr )r_ r )r r_ 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 rf � s rf c @ s4 e Z d Z d d � Z d d � Z d d � Z d S)�Ec C s | | _ d | _ d S)Nr )r_ r )r r_ r r r r � s z
E.__init__c C s | S)Nr )r r r r r � s z
E.__iter__c C s d d d S)NrM r r )r r r r r � s z
E.__next__N)r r! r" r r r r r r r rg � s rg c @ s4 e Z d Z d d � Z d d � Z d d � Z d S)�Sc C s d S)Nr )r r_ r r r r � s z
S.__init__c C s | S)Nr )r r r r r � s z
S.__iter__c C s
t � d S)N)r )r r r r r � s z
S.__next__N)r r! r" r r r r r r r rh � s rh c C s+ t t d d � t t t | � � � � � S)Nc S s | S)Nr )r r r r r5 � s zL.<locals>.<lambda>)rn r_ r` rd ra )r_ r r r �L� s ri 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 r9 rW rN r. rO )r� ra rc rd ri r` rC r r\ rh r` r? r� rf �ZeroDivisionErrorrg )r r3 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 )rm rn )r ra rc rd rh ri r` rC r rn r` r? r� rf rk rg )r r3 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) Nrl rl i� rm �333333�?i� i� rW r )rm ro )r r� ra rc rd rh ri r` rC r r� r� r` r? r� rf rk rg )r r3 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)
Nrl rl i� rm �333333�?i� i� rW )rm rp )r r` r? r� r� rf rk rg )r r3 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) Nrl rl i� rm �333333�?i� i� rW rM )rm rq )r ra rc rd rh ri r` r� r r# r� rC r` r? r� rf rk rg )r r3 r� ZtgtlenZexpected�actualr 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 )r6 r� Zsbr r r r7 � s z8TestVariousIteratorArgs.test_groupby.<locals>.<listcomp>)rV r� )r ra rc rd rh ri r` rC r� r r` r? r� rf rk rg )r r3 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 )r6 r r r r r7 � s z7TestVariousIteratorArgs.test_filter.<locals>.<listcomp>)rV r� )r ra rc rd rh ri r` rC r r� r r` r? r� rf rk rg )r r3 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 )r6 r r r r r7 � s z<TestVariousIteratorArgs.test_filterfalse.<locals>.<listcomp>)rV r� )r ra rc rd rh ri r` rC r r� r r` r? r� rf rk rg )r r3 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)
Nrl rl i� rm �333333�?i� i� rW )rm rs )r ra rc rd rh ri r` rC r r r
r` r? r� rf rk rg )r r3 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)
Nrl rl i� rm �333333�?i� i� rW )rm rt )r ra rc rd rh ri r` rC r r r r` r? r� rf rk rg )r r3 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 )r6 r r r r r7 s z4TestVariousIteratorArgs.test_map.<locals>.<listcomp>c S s g | ] } | | � q Sr r )r6 r r r r r7 s )rV r� )r ra rc rd rh ri r` rC r r_ r r& r% r` r? r� rf rk rg )r r3 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� rm �333333�?i� i� rW r r r. )rm ru )r ra rc rd rh ri r` rC r r# r` r? r� rf rk rg )r r3 r� r r r r7 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 )r6 r r r r r7 + s z8TestVariousIteratorArgs.test_starmap.<locals>.<listcomp>)rV r� )r ra rc rd rh ri r` r
rC r r) r& r% r` r? r� rf rk rg )r r3 r� �ssr 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]b } g } x. | | � D] } t | � s P| 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 ra rc rd rh ri r` r r� rC r r8 r` r? r� rf rk rg )r r3 r� �tgtr� r r r r9 0 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]k } g } x7 | | � D]) } | r� t | � r� ql | 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 ra rc rd rh ri r` r r� rC r r: r` r? r� rf rk rg )r r3 r� rw r� r r r r; < 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) Nrl rl i� rm �333333�?i� i� rW r )rm rx )r ra rc rd rh ri r` r@ rC r r` r? r� rf rk rg )r r3 r� Zit1Zit2r r r rG H s . "&z TestVariousIteratorArgs.test_teeN)r r! r" rf ro r� r r� r� r� r� r� r r' r7 r* r9 r; rG r r r r rj � s
rj 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� )rC r& � length_hintr� )r r r r r# T 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� )rC r& rz r� )r r r r r$ Y s ""%z2LengthTransparency.test_repeat_with_negative_timesN)r r! r" r# r$ r r r r ry R s ry c @ s4 e Z d Z d 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 � � � | S)N)rE )r� �first)rw �zr r r� h s
z@RegressionTests.test_sf_793826.<locals>.mutatingtuple.<locals>.g)r r_ r r� rE )Ztuple1rw Ztuple2r� �items�genr )rw r} r �
mutatingtupled s z5RegressionTests.test_sf_793826.<locals>.mutatingtuplec s | a t t � � d d � <d S)N)�Tr )r� )r| r r rw s s z)RegressionTests.test_sf_793826.<locals>.fr r rM r9 rW rN )r r rM )r9 rW rN )r r� rC )r r� rw �secondr )r| r �test_sf_793826a 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 �gen1� s
z,RegressionTests.test_sf_950057.<locals>.gen1c 3 s# � j d � d V� j d � d S)NrM r r9 )r� )r )r� r r �gen2� s
z,RegressionTests.test_sf_950057.<locals>.gen2Fr r T)r` r� r rn rC r� )r r� r� r )r� r �test_sf_950057~ s %% z RegressionTests.test_sf_950057c C sG t j d d � t d � D� � } | j t � � t | � Wd QRXd S)Nc s s | ] } f Vq d S)Nr )r6 Zunusedr r r r� � s zERegressionTests.test_long_chain_of_empty_iterables.<locals>.<genexpr>i��� )rn rp r r` r rE )r r@ r r r �"test_long_chain_of_empty_iterables� s "z2RegressionTests.test_long_chain_of_empty_iterablesN)r r! r" r� r� r� r r r r r{ _ s r{ 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 � s zKSubclassWithKwargsTest.test_keywords_in_subclass.<locals>.Subclass.__init__)r r! r" r r )r� r r �Subclass� s r� r� r z does not take keyword argumentsr )r� r r� r� rn r_ r) r# r8 r: r� r� r? Z assertNotInr
)r r� �errr )r� r �test_keywords_in_subclass� s z0SubclassWithKwargsTest.test_keywords_in_subclassN)r r! r" r� r r r r r� � s r� 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 )�struct�calcsize�ssize_t)r r r r �setUp� s zSizeofTest.setUpc C s[ t j d � } | j } | t d d � | d | j � | t d � | d | j � d S) NZ3PirU Z12r rT r. )rT )
rT rT rT rT rT rT rT rT rT rT )r � calcobjsize� check_sizeofr� r� )r �basesize�checkr r r �test_product_sizeof� 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. r9 )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. r9 )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 r9 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� � 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 all_equal(iterable):
... "Returns True if all the elements are equal to each other"
... g = groupby(iterable)
... return next(g, True) and not next(g, False)
>>> 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
>>> [all_equal(s) for s in ('', 'A', 'AAAA', 'AAAB', 'AAABA')]
[True, True, True, False, False]
>>> 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 | � t j t j t | � d S)N�gettotalrefcountr rW )r8 rj rW r{ ry r� rK 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 � itertoolsr4 �decimalr Z fractionsr r� r& r rr r0 � functoolsr r� ZMAX_Py_ssize_tr� Zminsizer
r r r r r r r r r% r) r* r, r- r/ r rd rt ZTestCaser8 rK rW r` ra rc rd r� rf rg rh ri rj ry r{ r� Z cpython_onlyr� r� Z__test__r� r r r r r �<module> sz
� � � � � Lne
�
C� "