OwlCyberSecurity - MANAGER
Edit File: test_more.cpython-38.pyc
U ���Z�� � @ s� d dl mZmZmZ d dlmZ d dlmZ d dlm Z d dl m Z m Z d dl mZ d dlmZ d dlmZmZmZmZmZmZmZ d d lmZmZm Z d d l m Z d d l m Z m!Z!m"Z"m#Z# d d l$Z%d d� Z&G dd� de �Z'G dd� de �Z(G dd� de �Z)G dd� de �Z*G dd� de �Z+G dd� de �Z,G dd � d e �Z-G d d � d e �Z.G d d � d e �Z/G d!d"� d"e �Z0G d#d$� d$e �Z1G d%d&� d&e �Z2G d'd(� d(e �Z3G d)d*� d*e �Z4G d+d,� d,e �Z5G d-d.� d.e �Z6G d/d0� d0e �Z7G d1d2� d2e �Z8G d3d4� d4e �Z9G d5d6� d6e �Z:G d7d8� d8e �Z;G d9d:� d:e �Z<G d;d<� d<e �Z=G d=d>� d>e �Z>G d?d@� d@e �Z?G dAdB� dBe �Z@G dCdD� dDe �ZAG dEdF� dFe �ZBG dGdH� dHe �ZCG dIdJ� dJe �ZDG dKdL� dLe �ZEG dMdN� dNe �ZFG dOdP� dPe �ZGG dQdR� dRe �ZHG dSdT� dTe �ZIG dUdV� dVe �ZJG dWdX� dXe �ZKG dYdZ� dZe �ZLG d[d\� d\e �ZMG d]d^� d^e �ZNG d_d`� d`e �ZOG dadb� dbe �ZPG dcdd� dde �ZQG dedf� dfe �ZRG dgdh� dhe �ZSG didj� dje �ZTd S )k� )�division�print_function�unicode_literals)�Decimal)� DocTestSuite)�Fraction)�partial�reduce)�merge)�StringIO)�chain�count�groupby�islice� permutations�product�repeat)�add�mul� itemgetter)�TestCase)�filter�map�range�zipNc C s |� td�� |S )Nzmore_itertools.more)ZaddTestsr )�loaderZtests�ignore� r �@/usr/lib/python3/dist-packages/more_itertools/tests/test_more.py� load_tests s r c @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dS )� CollateTestsz Unit tests for ``collate()``c C sH t d�t d�t dd�g}| �tttjdd� |D ���ttj|� �� dS )z%Test with the default `key` function.� � � � c S s g | ] }t |��qS r ��list��.0�itr r r � <listcomp>( s z-CollateTests.test_default.<locals>.<listcomp>N)r � assertEqual�sortedr r& �__add__�mi�collate��self� iterablesr r r � test_default$ s �zCollateTests.test_defaultc C s^ t ddd�t ddd�g}tttjdd� |D ��dd�}ttj|d d d � i��}| �||� d S ) z#Test using a custom `key` function.� r ���r! c S s g | ] }t |��qS r r% r' r r r r* 0 s z)CollateTests.test_key.<locals>.<listcomp>T��reverse�keyc S s | S �Nr ��xr r r �<lambda>2 � z'CollateTests.test_key.<locals>.<lambda>N�r r, r r& r- r. r/ r+ �r1 r2 �actual�expectedr r r �test_key, s �zCollateTests.test_keyc C s | � g tt�� �� dS )z-Be nice if passed an empty list of iterables.N)r+ r& r. r/ �r1 r r r � test_empty5 s zCollateTests.test_emptyc C s" | � ddgtt�td���� dS )z$Work when only 1 iterable is passed.r � � N)r+ r& r. r/ r rC r r r �test_one9 s zCollateTests.test_onec C sd t ddd�t ddd�t ddd�g}tttjdd� |D ��d d �}ttj|d d i��}| �||� d S ) zTest the `reverse` kwarg.r! r r5 r" r# r$ c S s g | ] }t |��qS r r% r' r r r r* B s z-CollateTests.test_reverse.<locals>.<listcomp>Tr6 r7 Nr> r? r r r � test_reverse= s " �zCollateTests.test_reversec C s( | � tjtjj� | � tjtjj� d S r9 )�assertNotEqualr �__doc__r. r/ r rC r r r � test_aliasG s zCollateTests.test_aliasN) �__name__� __module__� __qualname__rJ r3 rB rD rG rH rK r r r r r s r c @ s e Zd ZdZdd� Zdd� ZdS )� ChunkedTestszTests for ``chunked()``c C s, | � tt�dd��dddgdddgg� d S ) z?Test when ``n`` divides evenly into the length of the iterable.ZABCDEFr# �A�B�C�D�E�FN�r+ r& r. ZchunkedrC r r r � test_evenO s �zChunkedTests.test_evenc C s* | � tt�dd��dddgddgg� dS ) zYTest when ``n`` does not divide evenly into the length of the iterable. ZABCDEr# rP rQ rR rS rT NrV rC r r r �test_oddU s �zChunkedTests.test_oddN)rL rM rN rJ rW rX r r r r rO L s rO c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd S ) � FirstTestszTests for ``first()``c C s$ | � t�dd� td�D ��d� dS )z*Test that it works on many-item iterables.c s s | ] }|V qd S r9 r �r( r; r r r � <genexpr>f s z'FirstTests.test_many.<locals>.<genexpr>r! r N)r+ r. �firstr rC r r r � test_manyb s zFirstTests.test_manyc C s | � t�dg�d� dS )z5Test that it doesn't raise StopIteration prematurely.r# N�r+ r. r\ rC r r r rG h s zFirstTests.test_onec C s | � tdd� � dS )z2It should raise StopIteration for empty iterables.c S s t �g �S r9 )r. r\ r r r r r< n r= z6FirstTests.test_empty_stop_iteration.<locals>.<lambda>N�� assertRaises� ValueErrorrC r r r �test_empty_stop_iterationl s z$FirstTests.test_empty_stop_iterationc C s | � t�g d�d� dS )z>It should return the provided default arg for empty iterables.ZbooNr^ rC r r r r3 p s zFirstTests.test_defaultN)rL rM rN rJ r] rG rb r3 r r r r rY _ s rY c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(S ))� PeekableTestsz\Tests for ``peekable()`` behavor not incidentally covered by testing ``collate()`` c C s t �g �}| �|�d�d� dS )z2Make sure passing a default into ``peek()`` works.r" N)r. �peekabler+ �peek�r1 �pr r r �test_peek_defaultz s z PeekableTests.test_peek_defaultc C s0 t �g �}| �|� t �td��}| �|� dS )zdMake sure a ``peekable`` tests true iff there are items remaining in the iterable. r# N)r. rd � assertFalser � assertTruerf r r r �test_truthiness s z PeekableTests.test_truthinessc C sB t �td��}| �t|�d� | �|�� d� | �t|�d� dS )zgMake sure ``next`` and ``peek`` advance and don't advance the iterator, respectively. � r rE N)r. rd r r+ �nextre rf r r r �test_simple_peeking� s z!PeekableTests.test_simple_peekingc s� t �d�� | �� d d� | �t� �d� | �� d d� | �t� �d� | �� d d� | �� d d � | �t� �d� | �� d d � | �� d d � | �� d d� | �t� fdd�� dS )zL Indexing into the peekable shouldn't advance the iterator. � abcdefghijklr �arF �d�b�c� �l� ����k���c s � d S )N���r r �rg r r r< � r= z-PeekableTests.test_indexing.<locals>.<lambda>N)r. rd r+ rm r` � IndexErrorrC r r{ r � test_indexing� s zPeekableTests.test_indexingc C s" t d�}t�|�}| �|dd� |dd� � | �t|�d� | �|dd� |dd� dd� � | �|dd� |dd� dd� � | �|dd� |dd� dd� � | �|dd� |dd� dd� � | �|ddd� |dd� ddd� � | �|ddd � |dd� ddd � � dS ) z4Slicing the peekable shouldn't advance the iterator.ro rE r! rp Nr4 �d rF r5 )r& r. rd r+ rm )r1 �seqrg r r r � test_slicing� s $$$$(zPeekableTests.test_slicingc C s� ddddddg}t tdt|�d ��d g }d d d dd dddg}t|||�D ]L}t|�}t�|�}t|� t|� }|| }|d d � | } | � || |� qLd S )z*Test slicing on a fresh iterable each time�0�1�2�3�4�5���r! NrE rF r# r5 rw ���) r& r �lenr �iterr. rd rm �slicer+ ) r1 �iterable�indexes�steps� slice_argsr) rg �indexr@ rA r r r �test_slicing_reset� s z PeekableTests.test_slicing_resetc C sr d}t �t|��}|�� t|j�}| �t�� |ddd� W 5 Q R X | �|t|j�� | �t|�t|�� d S )NZ01234567rE r5 r ) r. rd r� re r& �_cacher` ra r+ )r1 r� rg Z old_cacher r r �test_slicing_error� s z PeekableTests.test_slicing_errorc C s, dddddg}t t�|��}| �||� dS )z�Iterating a peekable without using ``peek()`` or ``prepend()`` should just give the underlying iterable's elements (a trivial test but useful to set a baseline in case something goes wrong)rE rF r# r! r4 N)r& r. rd r+ �r1 rA r@ r r r �test_passthrough� s z PeekableTests.test_passthroughc C s� t �td��}g }|�d� |t|�t|�g7 }|�d� |t|�t|�g7 }|�d� |t|�g7 }dddddg}| �||� dS )z6Tests intersperesed ``prepend()`` and ``next()`` callsrF rl � � r rE N)r. rd r �prependrm r+ �r1 r) r@ rA r r r � test_prepend� s zPeekableTests.test_prependc C sl t �td��}t|�t|�g}|�ddd� |�dd� |t|�7 }dddddddd d d g }| �||� d S ) z@Tests prepending multiple items and getting them in proper orderr4 rl r� r� � � r rE rF r# r! N)r. rd r rm r� r& r+ r� r r r �test_multi_prepend s z PeekableTests.test_multi_prependc C s2 t �g �}|�d� t|�}dg}| �||� dS )z.Tests prepending in front of an empty iterablerl N)r. rd r� r& r+ r� r r r rD s zPeekableTests.test_emptyc C sv t �td��}| �|� t|�}| �|� |�d� | �|� |t|�g7 }| �|� ddddddg}| �||� dS ) zXTests that ``__bool__()`` or ``__nonzero__()`` works properly with ``prepend()``r4 rl r rE rF r# r! N) r. rd r rj r& ri r� rm r+ r� r r r �test_prepend_truthiness s z%PeekableTests.test_prepend_truthinessc C s� t �td��}t|�t|�g}| �|�� d� |�ddd� | �|�� d� |�dd� | �|�� d� |t|�7 }| �|� dd ddddddd d g }| �||� d S ) zZTests prepending multiple elements and getting them in reverse order while peekingr4 rF rl r� r� r� r� r rE r# r! N) r. rd r rm r+ re r� r& ri r� r r r �test_multi_prepend_peek! s z%PeekableTests.test_multi_prepend_peekc sj t �td��� | �t� �dddg� | �t� fdd�� � �d� | �t� �d� | �t� fdd�� d S ) z3Test resuming iteration after a previous exhaustionr# r rE rF c s t � �S r9 �rm r �r) r r r< 4 r= z7PeekableTests.test_prepend_after_stop.<locals>.<lambda>rl c s t � �S r9 r� r r� r r r< 7 r= N) r. rd r r+ r& r` � StopIterationr� rm rC r r� r �test_prepend_after_stop0 s z%PeekableTests.test_prepend_after_stopc C s t td��}t�|�}|�ddd� dddg| }| �|d d� | �|dd� |dd� � | �|dd� |dd� � | �|dd � |dd � � | �|dd� |dd� � | �|dd � |dd � � | �|ddd � |ddd � � | �|ddd � |ddd � � dS ) z0Tests interaction between prepending and slicingr� � �( �2 r rE rv Nr4 r~ rF r5 )r& r r. rd r� r+ )r1 r rg Zpseqr r r �test_prepend_slicing9 s z"PeekableTests.test_prepend_slicingc s� t td��}t�|�� � �ddd� | �� d d� | �t� �d� | �� d d� | �t� �d� | �� d d� | �� d d� | �t� �d� | �� d d� | �� d d � | �� d d � | �t� fd d�� dS )z1Tests interaction between prepending and indexingr� r� r� r� r rF rt rv rw � ry r� c s � d S )Ni��r r r{ r r r< \ r= z5PeekableTests.test_prepend_indexing.<locals>.<lambda>N) r& r r. rd r� r+ rm r` r| �r1 r r r{ r �test_prepend_indexingK s z#PeekableTests.test_prepend_indexingc C sT t �td��}|jdd� td�D �� t|�}tttd�td���}| �||� dS )z!Tests prepending from an iterabler4 c s s | ] }|V qd S r9 r rZ r r r r[ c s z6PeekableTests.test_prepend_iterable.<locals>.<genexpr>N�r. rd r r� r& r r+ r� r r r �test_prepend_iterable^ s z#PeekableTests.test_prepend_iterablec C sT t �td��}|jdd� td�D �� t|�}tttd�td���}| �||� dS )z5Tests that prepending a huge number of elements worksr4 c s s | ] }|V qd S r9 r rZ r r r r[ m s z2PeekableTests.test_prepend_many.<locals>.<genexpr>i N Nr� r� r r r �test_prepend_manyh s z PeekableTests.test_prepend_manyc C sD t �td��}|jtd�� t|�}ddddddg}| �||� d S ) z)Tests prepending from a reversed iterabler# )rl r� r� r� r� rl r rE rF N)r. rd r r� �reversedr& r+ r� r r r �test_prepend_reversedr s z#PeekableTests.test_prepend_reversedN)rL rM rN rJ rh rk rn r} r� r� r� r� r� r� rD r� r� r� r� r� r� r� r� r r r r rc u s( rc c @ s e Zd ZdZdd� ZdS )� ConsumerTestszTests for ``consumer()``c C s" t jdd� �}|� }|�d� d S )Nc s s d V } q d S r9 r r: r r r �eater s z*ConsumerTests.test_consumer.<locals>.eater�hi)r. Zconsumer�send)r1 r� �er r r � test_consumer~ s zConsumerTests.test_consumerN)rL rM rN rJ r� r r r r r� { s r� c @ s e Zd Zdd� Zdd� ZdS )�DistinctPermutationsTestsc C s@ dddddddg}t t�|��}t tt|���}| �||� dS )zlMake sure the output for ``distinct_permutations()`` is the same as set(permutations(it)). �zrp �q�yN)r, r. �distinct_permutations�setr r+ �r1 r� Z test_outputZ ref_outputr r r �test_distinct_permutations� s z4DistinctPermutationsTests.test_distinct_permutationsc C s� dd� dD �}t t�|��}dd� dD �}t tt|���}| �||� tdddddddg�}t t�|��}tdddddddg�}t tt|���}| �||� d S ) z^Make sure ``distinct_permutations()`` accepts a different type of iterables. c s s | ] }|V qd S r9 r �r( rs r r r r[ � s zADistinctPermutationsTests.test_other_iterables.<locals>.<genexpr>)r� rp rp r� r� r� r� c s s | ] }|V qd S r9 r r� r r r r[ � s r� rp r� r� N)r, r. r� r� r r+ r� r� r r r �test_other_iterables� s z.DistinctPermutationsTests.test_other_iterablesN)rL rM rN r� r� r r r r r� � s r� c @ s e Zd Zdd� ZdS )� IlenTestsc C s^ | � t�tdd� td���d� | � t�dd� td�D ��d� | � t�ttd���d� d S ) z Sanity-checks for ``ilen()``.c S s | d dkS )Nrl r r r: r r r r< � r= z%IlenTests.test_ilen.<locals>.<lambda>�e r� c s s | ] }|V qd S r9 r rZ r r r r[ � s z&IlenTests.test_ilen.<locals>.<genexpr>r r$ N)r+ r. Zilenr r r& rC r r r � test_ilen� s � zIlenTests.test_ilenN)rL rM rN r� r r r r r� � s r� c @ s e Zd Zdd� ZdS )� WithIterTestsc C s> t d�}dd� t�|�D �}| �|ddg� | �|jd� d S )NzOne fish Two fishc S s g | ]}|� � d �qS �r )�split)r( �liner r r r* � s z0WithIterTests.test_with_iter.<locals>.<listcomp>ZOneZTwoT)r r. Z with_iterr+ �closed)r1 �sZ initial_wordsr r r �test_with_iter� s z WithIterTests.test_with_iterN)rL rM rN r� r r r r r� � s r� c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�OneTestsc C s t dg�}| �t�|�d� d S )N�item)r� r+ r. �one)r1 r) r r r � test_basic� s zOneTests.test_basicc s4 t g �� | �t� fdd�� | �t� fdd�� d S )Nc s t �� �S r9 �r. r� r r� r r r< � r= z)OneTests.test_too_short.<locals>.<lambda>c s t j� td�S )N)Z too_short)r. r� r| r r� r r r< � r= )r� r` ra r| rC r r� r �test_too_short� s zOneTests.test_too_shortc sB t � � | �t� fdd�� | �t� �d� | �t� fdd�� d S )Nc s t �� �S r9 r� r r� r r r< � r= z(OneTests.test_too_long.<locals>.<lambda>rF c s t j� td�S )N)Ztoo_long)r. r� � OverflowErrorr r� r r r< � r= )r r` ra r+ rm r� rC r r� r � test_too_long� s �zOneTests.test_too_longN)rL rM rN r� r� r� r r r r r� � s r� c @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dS )�IntersperseTestz Tests for intersperse() c C s0 dd� dD �}| � tt�d |��dd dg� d S )Nc s s | ] }|V qd S r9 r rZ r r r r[ � s z,IntersperseTest.test_even.<locals>.<genexpr>Z01r� r� �r+ r& r. � intersperse�r1 r� r r r rW � s �zIntersperseTest.test_evenc C s4 dd� dD �}| � tt�d |��dd dd dg� d S )Nc s s | ] }|V qd S r9 r rZ r r r r[ � s z+IntersperseTest.test_odd.<locals>.<genexpr>�012r� r� r� r� r� r r r rX � s �zIntersperseTest.test_oddc C s@ d}dd� dD �}t t�||��}dddddg}| �||� d S )N�rp rr c s s | ] }|V qd S r9 r rZ r r r r[ � s z.IntersperseTest.test_nested.<locals>.<genexpr>r� r� r� r� �r& r. r� r+ )r1 �elementr� r@ rA r r r � test_nested� s zIntersperseTest.test_nestedc C s | � tdd� � d S )Nc S s t �dd�S )Nr; rE )r. r� r r r r r< � r= z3IntersperseTest.test_not_iterable.<locals>.<lambda>�r` � TypeErrorrC r r r �test_not_iterable� s z!IntersperseTest.test_not_iterablec C s dddddddddddddg fd dddddddddgfd ddddddddgfd ddddddddgfd ddddddddgfd dddddddgfddddddddgfd ddgdddddgdddgffD ]8\}}}dd� dD �}t tj|||d��}| �||� q�d S )NrE �_r� r� r� r� r� r� rF r# r! r4 r$ r" rp rr c s s | ] }|V qd S r9 r rZ r r r r[ � s z)IntersperseTest.test_n.<locals>.<genexpr>Z012345��nr� )r1 r� r� rA r� r@ r r r �test_n� s � zIntersperseTest.test_nc C s | � tdd� � d S )Nc S s t tjdddd��S )Nr; r� r r� )r& r. r� r r r r r< r= z-IntersperseTest.test_n_zero.<locals>.<lambda>r_ rC r r r � test_n_zero� s �zIntersperseTest.test_n_zeroN) rL rM rN rJ rW rX r� r� r� r� r r r r r� � s r� c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �UniqueToEachTestsz Tests for ``unique_to_each()``c C s0 ddgdddgdddgg}| � tj|� |� d S ) zRWhen all the input iterables are unique the output should match the input.rE rF r# r! r4 r$ r" rv N�r+ r. �unique_to_eachr0 r r r �test_all_unique s z!UniqueToEachTests.test_all_uniquec C s, ddg}| � tj|� ddgdddgg� dS )z|When there are duplicates in any of the input iterables that aren't in the rest, those duplicates should be emitted.Z mississippiZmissourirg �o�u�rNr� r0 r r r �test_duplicates s �z!UniqueToEachTests.test_duplicatesc C sF ddd� t d�D �dddgt� g}| �tj|� dgdgdgg g� dS ) zbWhen the input iterables contain different types the function should still behave properlyr; c s s | ] }|V qd S r9 r �r( �ir r r r[ s z/UniqueToEachTests.test_mixed.<locals>.<genexpr>r# rE rF r N)r �tupler+ r. r� r0 r r r � test_mixed s "z UniqueToEachTests.test_mixedN)rL rM rN rJ r� r� r� r r r r r� s r� c @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dS )� WindowedTestszTests for ``windowed()``c C s4 t t�dddddgd��}dddg}| �||� d S ) NrE rF r# r! r4 �rE rF r# �rF r# r! �r# r! r4 �r& r. �windowedr+ �r1 r@ rA r r r r� s zWindowedTests.test_basicc C s0 t t�dddddgd��}dg}| �||� dS ) z� When the window size is larger than the iterable, and no fill value is given,``None`` should be filled in. rE rF r# r! r4 r$ )rE rF r# r! r4 NNr� r� r r r �test_large_size$ s z WindowedTests.test_large_sizec C sd dddddg}di dgfdddid d gffD ]2\}}}t tj||fd d i|���}| �||� q,d S )zU When sizes don't match evenly, the given fill value should be used. rE rF r# r! r4 r$ )rE rF r# r! r4 �!�stepr� )r! r4 r� � fillvaluer� Nr� )r1 r� r� �kwargsrA r@ r r r �test_fillvalue- s � z WindowedTests.test_fillvaluec C s2 t t�dddddgd��}t� g}| �||� dS )z?When the window size is zero, an empty tuple should be emitted.rE rF r# r! r4 r N)r& r. r� r� r+ r� r r r � test_zero: s zWindowedTests.test_zeroc C s4 | � t�� tt�dddddgd�� W 5 Q R X dS )z>When the window size is negative, ValueError should be raised.rE rF r# r! r4 r5 N)r` ra r& r. r� rC r r r � test_negative@ s zWindowedTests.test_negativec C s� dddddddg}dddd d gfdddd d gfdddd gfdddd gfdddd gfdddgfdddgffD ]*\}}}t tj|||d��}| �||� qj| �t�� t tj|ddd�� W 5 Q R X dS )z9The window should advance by the number of steps providedrE rF r# r! r4 r$ r" r� r� )r4 r$ r" )r! r4 r$ )r" NN)r$ r" Nrv )rE rF r# r! r4 r$ r" )r� r N)r& r. r� r+ r` ra )r1 r� r� r� rA r@ r r r � test_stepE s � zWindowedTests.test_stepN) rL rM rN rJ r� r� r� r� r� r� r r r r r� s r� c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) � BucketTestszTests for ``bucket()``c C s� ddddddddd d g }t j|d d � d �}| �t|d �dddg� | �t|d �ddd g� | �t|d �dddd g� | �t|d �g � d S )Nrl r� r� r� r� � r� � � �! c S s d| d S �Nrl r r: r r r r< ^ r= z(BucketTests.test_basic.<locals>.<lambda>�r8 r� )r. �bucketr+ r& �r1 r� rS r r r r� \ s zBucketTests.test_basicc C sz ddddddddd d g }t j|d d � d �}| �d|k� | �d|k� | �d|k� | �d|k� | �t|d �d� d S )Nrl r� r� r� r� r r� r r r c S s d| d S r r r: r r r r< k r= z%BucketTests.test_in.<locals>.<lambda>r r� )r. r rj ri r+ rm r r r r �test_ini s zBucketTests.test_inc C sx t d�}dd� }dd� }tj|||d�}| �t�d|d �ddd g� | �d|� | �d|j� | �t|d �g � d S ) Nr c S s t t| �d �S �Nr ��int�strr: r r r r< w r= z,BucketTests.test_validator.<locals>.<lambda>c S s d| k odk S S )Nr rl r r: r r r r< x r= )� validatorr# rE rl r� )r r. r r+ �take� assertNotInr� r& )r1 r� r8 r rS r r r �test_validatoru s zBucketTests.test_validatorN)rL rM rN rJ r� r r r r r r r� Y s r� c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd S ) �SpyTestszTests for ``spy()``c C sF t d�}t�|�\}}| �|dg� | �t|�dddddddg� d S ) N�abcdefgrp rr rs rq r� �f�g�r� r. Zspyr+ r& �r1 �original_iterable�head� new_iterabler r r r� � s �zSpyTests.test_basicc C sd t d�}t�|d�\\}}}}| �|d� | �|d� | �|d� | �t|�ddddddd g� d S ) Nr r# rp rr rs rq r� r r r )r1 r r\ �secondZthirdr r r r �test_unpacking� s �zSpyTests.test_unpackingc C sD t d�}t�|d�\}}| �|dddg� | �t|�dddg� d S )N�abcr! rp rr rs r r r r r � test_too_many� s zSpyTests.test_too_manyc C s> t d�}t�|d�\}}| �|g � | �t|�dddg� d S )Nr r rp rr rs r r r r r r� � s zSpyTests.test_zeroN)rL rM rN rJ r� r r r� r r r r r � s r c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�InterleaveTestsc C sJ t t�dddgdddgddd g��}ddddddddd g }| �||� d S � NrE r! r" rF r4 rv r# r$ rt �r& r. � interleaver+ r� r r r rW � s $zInterleaveTests.test_evenc C sB t t�ddgdddgdddg��}ddddddg}| �||� d S � NrE r! rF r4 r" r# r$ rv r r� r r r � test_short� s "zInterleaveTests.test_shortc C sT ddddg}d}t � }tt�|||��}dddddd dd d dd d g }| �||� d S )Nrp rr rs rq �12345r� r r� rE r� rF r� r# )r r& r. r! r+ )r1 �it_list�it_strZit_infr@ rA r r r �test_mixed_types� s z InterleaveTests.test_mixed_typesN�rL rM rN rW r# r' r r r r r � s r c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�InterleaveLongestTestsc C sJ t t�dddgdddgddd g��}ddddddddd g }| �||� d S r �r& r. �interleave_longestr+ r� r r r rW � s $z InterleaveLongestTests.test_evenc C sF t t�ddgdddgdddg��}ddddddddg}| �||� d S r" r* r� r r r r# � s "z!InterleaveLongestTests.test_shortc C s` ddddg}d}dd� t d�D �}tt�|||��}dd d dd d dd ddddg }| �||� d S )Nrp rr rs rq r$ c s s | ] }|V qd S r9 r rZ r r r r[ � s z:InterleaveLongestTests.test_mixed_types.<locals>.<genexpr>r# r� r r� rE r� rF r� r� )r r& r. r+ r+ )r1 r% r& Zit_genr@ rA r r r r' � s z'InterleaveLongestTests.test_mixed_typesNr( r r r r r) � s r) c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd S ) � TestCollapsezTests for ``collapse()``c C s> dgddgdgdgggg}| � tt�|��dddddg� d S )NrE rF r# r! r4 �r+ r& r. �collapse�r1 ru r r r � test_collapse� s zTestCollapse.test_collapsec C s> dgddgdgdgggg}| � tt�|��dddddg� d S )N�s1�s2Zs3Zs4Zs5r- r/ r r r �test_collapse_to_string� s z$TestCollapse.test_collapse_to_stringc C sD dgdgdgdgdgggg}| � ttj|dd��tt�|��� d S �NrE rF r# r! r4 )Zlevels)r+ r& r. r. Zflattenr/ r r r �test_collapse_flatten� s z"TestCollapse.test_collapse_flattenc C sv dgddgdgdgggg}| � ttj|dd��dddddgg� | � ttjtj|dd�dd��ttj|dd��� d S r4 r- r/ r r r �test_collapse_to_level� s &�z#TestCollapse.test_collapse_to_levelc C sL ddgdddgdff}t tj|t d��}ddgdddgdg}| �||� d S )NrE rF r# r! �r4 Zab�Z base_type)r& r. r. r+ )r1 ru r@ rA r r r �test_collapse_to_list� s z"TestCollapse.test_collapse_to_listN) rL rM rN rJ r0 r3 r5 r6 r9 r r r r r, � s r, c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd S ) �SideEffectTestszTests for ``side_effect()``c sN dg� � fdd�}t t�|td���}| �|t td��� | �� d d� d S )Nr c s � d d7 < d S �Nr rE r ��arg�Zcounterr r �func� s z-SideEffectTests.test_individual.<locals>.funcrl �r& r. � side_effectr r+ �r1 r? �resultr r> r �test_individual� s z SideEffectTests.test_individualc sP dg� � fdd�}t t�|td�d��}| �|t td��� | �� d d� d S )Nr c s � d d7 < d S r; r r< r> r r r? � s z*SideEffectTests.test_chunked.<locals>.funcrl rF r4 r@ rB r r> r � test_chunked� s z SideEffectTests.test_chunkedc s� t � �g � � �fdd�}dd� }�fdd�}�j}z t�tj||� ||d�� W n tk rd Y nX | �� dd g� | ��j� d S ) Nc s t | �d� � ���� � d S �N��file)�print�append�getvalue�r� �Z collectorr r r r? s z/SideEffectTests.test_before_after.<locals>.funcc s s dV dV t d��d S )Nrp rr Zkaboom�� RuntimeErrorr r r r r) s z-SideEffectTests.test_before_after.<locals>.itc s t d� d�S )NZHEADERrG �rI r �r r r r< r= z3SideEffectTests.test_before_after.<locals>.<lambda>��before�afterz HEADER a z HEADER a b ) r �closer. �consumerA rO r+ rj r� )r1 r? r) rS rT r rM r �test_before_after s z!SideEffectTests.test_before_afterc s^ t � � � fdd�}dd� }z t�tj|d|� jd�� W n tk rL Y nX | �� j� d S )Nc s t | � d�S rF rP r: rQ r r r< r= z3SideEffectTests.test_before_fails.<locals>.<lambda>c S s t d��d S )NZouchrN r r r r rS s z1SideEffectTests.test_before_fails.<locals>.beforer rR )r r. rV rA rU rO rj r� )r1 r? rS r rQ r �test_before_fails s �z!SideEffectTests.test_before_failsN)rL rM rN rJ rD rE rW rX r r r r r: � s r: c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) � SlicedTestszTests for ``sliced()``c C s& d}| � tt�|d��dddg� dS )z8Test when the length of the sequence is divisible by *n*� ABCDEFGHIr# �ABCZDEFZGHIN�r+ r& r. �slicedr� r r r rW 2 s zSlicedTests.test_evenc C s& d}| � tt�|d��dddg� dS )z<Test when the length of the sequence is not divisible by *n*rZ r! ZABCDZEFGH�INr\ r� r r r rX 7 s zSlicedTests.test_oddc C s8 dd� dD �}| � t�� tt�|d�� W 5 Q R X d S )Nc s s | ] }|V qd S r9 r rZ r r r r[ = s z1SlicedTests.test_not_sliceable.<locals>.<genexpr>rZ r# )r` r� r& r. r] r� r r r �test_not_sliceable< s z SlicedTests.test_not_sliceableN)rL rM rN rJ rW rX r_ r r r r rY / s rY c @ s e Zd ZdZdd� Zdd� ZdS )� SplitAtTestszTests for ``split()``c s> � fdd�}t tdjt�||���}|�� �}| �||� d S )Nc s | � kS r9 r �rs ��delimr r r< G r= z2SplitAtTests.comp_with_str_split.<locals>.<lambda>� )r& r �joinr. Zsplit_atr� r+ )r1 Z str_to_splitrc �predr@ rA r rb r �comp_with_str_splitF s z SplitAtTests.comp_with_str_splitc C s0 ddddg}t |d�D ]\}}| �||� qd S )Nrd ZabcbaZ aaabbbcccdddr� �abcd)r rg )r1 Z test_strsr� rc r r r �test_seperatorsL s z SplitAtTests.test_seperatorsN)rL rM rN rJ rg ri r r r r r` C s r` c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �SplitBeforeTestz Tests for ``split_before()``c C s8 t t�ddd� ��}dddgdddgg}| �||� d S )N�xooxooc S s | dkS �Nr; r ra r r r r< V r= z6SplitBeforeTest.test_starts_with_sep.<locals>.<lambda>r; r� �r& r. Z split_beforer+ r� r r r �test_starts_with_sepU s z$SplitBeforeTest.test_starts_with_sepc C s: t t�ddd� ��}ddgdddgdgg}| �||� d S )N�ooxooxc S s | dkS rl r ra r r r r< [ r= z4SplitBeforeTest.test_ends_with_sep.<locals>.<lambda>r� r; rm r� r r r �test_ends_with_sepZ s z"SplitBeforeTest.test_ends_with_sepc C s0 t t�ddd� ��}dddgg}| �||� d S )N�oooc S s | dkS rl r ra r r r r< ` r= z-SplitBeforeTest.test_no_sep.<locals>.<lambda>r� rm r� r r r � test_no_sep_ s zSplitBeforeTest.test_no_sepN�rL rM rN rJ rn rp rr r r r r rj R s rj c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �SplitAfterTestzTests for ``split_after()``c C s: t t�ddd� ��}dgdddgddgg}| �||� d S )Nrk c S s | dkS rl r ra r r r r< i r= z5SplitAfterTest.test_starts_with_sep.<locals>.<lambda>r; r� �r& r. Z split_afterr+ r� r r r rn h s z#SplitAfterTest.test_starts_with_sepc C s8 t t�ddd� ��}dddgdddgg}| �||� d S )Nro c S s | dkS rl r ra r r r r< n r= z3SplitAfterTest.test_ends_with_sep.<locals>.<lambda>r� r; ru r� r r r rp m s z!SplitAfterTest.test_ends_with_sepc C s0 t t�ddd� ��}dddgg}| �||� d S )Nrq c S s | dkS rl r ra r r r r< s r= z,SplitAfterTest.test_no_sep.<locals>.<lambda>r� ru r� r r r rr r s zSplitAfterTest.test_no_sepNrs r r r r rt e s rt c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd S ) � PaddedTestzTests for ``padded()``c C sZ dddg}| � t�dt�|��dddd d g� | � t�dtj|dd��dddddg� d S )NrE rF r# r4 rd )r� )r+ r. r �paddedr� r r r � test_no_n{ s $ �zPaddedTest.test_no_nc C s$ | � tdd� � | � tdd� � d S )Nc S s t tjdddgdd��S )NrE rF r# r5 r� �r& r. rw r r r r r< � r= z+PaddedTest.test_invalid_n.<locals>.<lambda>c S s t tjdddgdd��S )NrE rF r# r r� ry r r r r r< � r= r_ rC r r r �test_invalid_n� s zPaddedTest.test_invalid_nc C s� dddddg}| � ttj|dd��dddddg� | � ttj|dd��dddddg� | � ttj|dd��dddddd d g� | � ttj|ddd ��dddddddg� d S ) NrE rF r# r! r4 r� r" rd )r� r� �r+ r& r. rw r� r r r � test_valid_n� s $$ � �zPaddedTest.test_valid_nc C s� ddddddg}| � ttj|ddd��ddddddg� | � ttj|d dd��ddddddd d g� | � ttj|ddd��ddddddg� | � ttj|ddd��ddddddd d g� | � ttj|d ddd ��ddddddd d g� d S ) NrE rF r# r! r4 r$ T)r� � next_multiplerv rd )r� r� r} r{ r� r r r �test_next_multiple� s* �� ���z PaddedTest.test_next_multipleN)rL rM rN rJ rx rz r| r~ r r r r rv x s rv c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �DistributeTestzTests for distribute()c C s$ | � tdd� � | � tdd� � d S )Nc S s t �ddddg�S �Nr5 rE rF r# �r. � distributer r r r r< � r= z/DistributeTest.test_invalid_n.<locals>.<lambda>c S s t �ddddg�S �Nr rE rF r# r� r r r r r< � r= r_ rC r r r rz � s z DistributeTest.test_invalid_nc C s� ddddddddd d g }d|gfdddddd gddddd ggfddddd gdddgddd ggfd d d � t dd �D �ffD ]&\}}| �dd � t�||�D �|� qxd S )NrE rF r# r! r4 r$ r" rv rt rl c S s g | ] }|g�qS r r �r( r� r r r r* � s z-DistributeTest.test_basic.<locals>.<listcomp>r� c S s g | ] }t |��qS r r% rZ r r r r* � s )r r+ r. r� �r1 r� r� rA r r r r� � s � �zDistributeTest.test_basicc C sB ddddg}| � dd� t�d|�D �dgdgdgdgg g g� d S )NrE rF r# r! c S s g | ] }t |��qS r r% rZ r r r r* � s z/DistributeTest.test_large_n.<locals>.<listcomp>r$ )r+ r. r� r� r r r � test_large_n� s �zDistributeTest.test_large_nN�rL rM rN rJ rz r� r� r r r r r � s r c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) � StaggerTestzTests for ``stagger()``c C s4 ddddg}t t�|��}dddg}| �||� d S )Nr rE rF r# r; �r rE rF r� )r& r. �staggerr+ �r1 r� r@ rA r r r r3 � s zStaggerTest.test_defaultc C sb ddddg}dddgfdd d d d d gfd d d gffD ](\}}t j||dd�}| �t|�|� q4d S )Nr rE rF r# )rw r rF )rd r rF )rd rE r# �rw r5 �rd rd �rd r �r rE �rE rF �rF r# rd ��offsetsr� �r. r� r+ r& �r1 r� r� rA Z all_groupsr r r � test_offsets� s � zStaggerTest.test_offsetsc C sn ddddg}ddddd d gfd d d ddddgfddddgffD ]*\}}t j||ddd�}| �t|�|� q>d S )Nr rE rF r# �r5 r rE �rd r rE r� r� )rF r# rd )r# rd rd r� r� r� r� r� r� )r# rd rd T)r� r� �longestr� r� r r r � test_longest� s � � �zStaggerTest.test_longestN)rL rM rN rJ r3 r� r� r r r r r� � s r� c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) � ZipOffsetTestzTests for ``zip_offset()``c C sf ddddg}ddddddg}ddddddddg}t tj|||d d d ��}d d dddg}| �||� d S )Nr rE rF r# r! r4 r$ r" r� rd r� r� r� r� r� r� �r& r. � zip_offsetr+ �r1 Za_1Za_2Za_3r@ rA r r r � test_shortest� s �zZipOffsetTest.test_shortestc C sj ddddg}ddddddg}ddddddddg}t tj|||d d d ��}d d dddddg}| �||� d S )Nr rE rF r# r! r4 r$ r" r� T)r� r� r; r� r� r� r� )Nr4 r$ )NNr" r� r� r r r r� s �� zZipOffsetTest.test_longestc s2 dddgdddgf� d�| � t� �fdd�� d S ) Nr rE rF r# r! r� c s t tj� d�i��S )Nr� )r& r. r� r �r2 r� r r r< r= z-ZipOffsetTest.test_mismatch.<locals>.<lambda>r_ rC r r� r � test_mismatch s �zZipOffsetTest.test_mismatchN)rL rM rN rJ r� r� r� r r r r r� � s r� c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd S ) �SortTogetherTestzTests for sort_together()c C s� ddddddgddddddgddd d d dgg}| � t�|�d d d g� | � tj|dd�d ddg� | � tj|dd�d ddg� | � tj|dd�dddg� dS )z@tests `key_list` including default, iterables include duplicates�GA�CT�May�Aug.�June�July�a r� r~ �F �r� r� r� r� r� r� �r� r� r� r� r� r� �r� r~ r� r� r� r~ r� ��key_list)r� r� r� r� r� r� )r~ r� r� r� r� r~ r� )r� r~ r� r� r� r~ )rF )r� r� r� r� r� r� )r� r� r� r� r� r� )r� r� r� r� r~ r~ N�r+ r. � sort_togetherr0 r r r � test_key_list% s@ ��� �� �� ��z SortTogetherTest.test_key_listc sF ddddddgddddddgddd d d dgg� | � t� fd d �� d S )z9tests `key_list` for indexes not available in `iterables`r� r� r� r� r� r� r� r� r~ r� c s t j� dd�S )Nr7 r� )r. r� r �r2 r r r< Z r= z8SortTogetherTest.test_invalid_key_list.<locals>.<lambda>N)r` r| rC r r� r �test_invalid_key_listQ s � �z&SortTogetherTest.test_invalid_key_listc C sP ddddddgddddddgddd d d dgg}| � tj|d d d �dddg� dS )zAtests `reverse` to ensure a reverse sort for `key_list` iterablesr� r� r� r� r� r� r� r� r~ r� r� T)r� r7 )r� r� r� r� r� r� )r� r� r� r� r� r� )r~ r� r� r� r~ r� Nr� r0 r r r rH ] s ���z SortTogetherTest.test_reversec C sN dddddddgddddddgdd d d d d d gg}| � t�|�d ddg� dS )zAtests trimming of iterables to the shortest length before sortingr� r� ZMAr� r� r� r� r� r� r~ r� r r� r� r� Nr� r0 r r r �test_uneven_iterablesl s ���z&SortTogetherTest.test_uneven_iterablesN)rL rM rN rJ r� r� rH r� r r r r r� " s , r� c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) � DivideTestzTests for divide()c C s$ | � tdd� � | � tdd� � d S )Nc S s t �ddddg�S r� �r. �divider r r r r< � r= z+DivideTest.test_invalid_n.<locals>.<lambda>c S s t �ddddg�S r� r� r r r r r< � r= r_ rC r r r rz s zDivideTest.test_invalid_nc C s� ddddddddd d g }d|gfddddddgdddd d ggfdddddgdddgdd d ggfd d d � t dd �D �ffD ]&\}}| �dd � t�||�D �|� qxd S )NrE rF r# r! r4 r$ r" rv rt rl c S s g | ] }|g�qS r r r� r r r r* � s z)DivideTest.test_basic.<locals>.<listcomp>r� c S s g | ] }t |��qS r r% rZ r r r r* � s )r r+ r. r� r� r r r r� � s � �zDivideTest.test_basicc C sB ddddg}| � dd� t�d|�D �dgdgdgdgg g g� d S )NrE rF r# r! c S s g | ] }t |��qS r r% rZ r r r r* � s z+DivideTest.test_large_n.<locals>.<listcomp>r$ )r+ r. r� r� r r r r� � s �zDivideTest.test_large_nNr� r r r r r� | s r� c @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dS )�TestAlwaysIterablezTests for always_iterable()c C s | � tt�d��dg� d S �NrE �r+ r& r. �always_iterablerC r r r � test_single� s z TestAlwaysIterable.test_singlec C s. dD ]$}t t�|��}|g}| �||� qd S )N)�foos barZbaz)r& r. r� r+ )r1 �objr@ rA r r r � test_strings� s z TestAlwaysIterable.test_stringsc C s~ ddd�}d}t t�|��}t |�}| �||� t tj|td��}|g}| �||� t tj|d d��}t |�}| �||� d S )NrE rF r� Z123r8 )r& r. r� r+ �dict) r1 Zdict_objZstr_objZdefault_actualZdefault_expectedZ custom_actualZcustom_expectedZ str_actualZ str_expectedr r r �test_base_type� s z!TestAlwaysIterable.test_base_typec C s| | � tt�ddg��ddg� | � ttjddgtd��ddgg� | � tt�td���dddg� | � tt�g ��g � d S )Nr rE r8 r� r r� )r+ r& r. r� r� rC r r r �test_iterables� s � �z!TestAlwaysIterable.test_iterablesc C s | � tt�d ��g � d S r9 r� rC r r r � test_none� s z TestAlwaysIterable.test_nonec C s( dd� }| � tt�|� ��ddg� d S )Nc s s dV dV d S r; r r r r r �_gen� s z/TestAlwaysIterable.test_generator.<locals>._genr rE r� )r1 r� r r r �test_generator� s z!TestAlwaysIterable.test_generatorN) rL rM rN rJ r� r� r� r� r� r� r r r r r� � s r� c @ s\ e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dS )� AdjacentTestsc C s@ t t�dd� td���}dddddd d d d d g }| �||� d S )Nc S s | d dkS �Nr4 r r r: r r r r< � r= z,AdjacentTests.test_typical.<locals>.<lambda>rl �Tr �TrE )FrF �Fr# �Tr! �Tr4 �Tr$ )Fr" �Frv �Frt �r& r. �adjacentr r+ r� r r r � test_typical� s �zAdjacentTests.test_typicalc C s( t t�dd� g ��}g }| �||� d S )Nc S s | d dkS r� r r: r r r r< � r= z3AdjacentTests.test_empty_iterable.<locals>.<lambda>�r& r. r� r+ r� r r r �test_empty_iterable� s z!AdjacentTests.test_empty_iterablec C sT t t�dd� dg��}dg}| �||� t t�dd� dg��}dg}| �||� d S )Nc S s | d dkS r� r r: r r r r< � r= z/AdjacentTests.test_length_one.<locals>.<lambda>r r� c S s | d dkS r� r r: r r r r< � r= rE )FrE r� r� r r r �test_length_one� s z AdjacentTests.test_length_onec C s@ t t�dd� td���}dddddd d d d d g }| �||� dS )zsTest that when the predicate matches multiple consecutive elements it doesn't repeat elements in the outputc S s | d dk S )Nr4 rF r r: r r r r< � r= z5AdjacentTests.test_consecutive_true.<locals>.<lambda>rl r� r� �TrF r� r� r� r� �Tr" r� r� Nr� r� r r r �test_consecutive_true� s �z#AdjacentTests.test_consecutive_truec C s� t tjdd� td�dd��}dddd d d d d ddg }| �||� t tjdd� td�dd��}dddd d d d d ddg }| �||� d S )Nc S s | d dkS r� r r: r r r r< � r= z-AdjacentTests.test_distance.<locals>.<lambda>rl rF �Zdistancer� r� r� )Tr# r� r� r� r� r� r� c S s | d dkS r� r r: r r r r< � r= r# )Trv r� r� r r r � test_distance� s$ � �zAdjacentTests.test_distancec C sx t d�}ttjdd� |dd��}tttd�|��}| �||� ttjdd� |dd��}tttd�|��}| �||� d S ) z4Test distance larger than the length of the iterablerl c S s | d dkS )Nr4 r! r r: r r r r< � r= z3AdjacentTests.test_large_distance.<locals>.<lambda>r� r� Tc S s dS )NFr r: r r r r< � r= FN)r r& r. r� r r r+ r� r r r �test_large_distance� s z!AdjacentTests.test_large_distancec C sP t d�}dd� }t�||d�}tt||�|�}| �tdd� t||�D ��� dS )z:Test that adjacent() reduces to zip+map when distance is 0�� c S s | d dkS )Nr! rF r r: r r r r< r= z2AdjacentTests.test_zero_distance.<locals>.<lambda>r c s s | ]\}}||kV qd S r9 r �r( rp r� r r r r[ s z3AdjacentTests.test_zero_distance.<locals>.<genexpr>N)r r. r� r r rj �all�r1 r� � predicater@ rA r r r �test_zero_distance� s z AdjacentTests.test_zero_distancec s4 dd� � | � t� fdd�� | � t� fdd�� dS )z;Test that adjacent() raises an error with negative distancec S s | S r9 r r: r r r r< r= z6AdjacentTests.test_negative_distance.<locals>.<lambda>c s t �� td�d�S )Nr� r5 �r. r� r r �rf r r r< r= c s t �� td�d�S )Nrl rz r� r r� r r r< r= Nr_ rC r r� r �test_negative_distance s � �z$AdjacentTests.test_negative_distancec C sx t �dd� td��}t �|td�td��}dd� |D �}dddgfd d d gfdd d dgfd dddgfg}| �||� dS )z7Test interaction of adjacent() with groupby_transform()c S s | d dkS r� r r: r r r r< r= z-AdjacentTests.test_grouping.<locals>.<lambda>rl r rE c S s g | ]\}}|t |�f�qS r r% �r( rx r r r r r* s z/AdjacentTests.test_grouping.<locals>.<listcomp>TFrF r# r! r4 r$ r" rv rt N)r. r� r �groupby_transformr r+ )r1 r� �grouperr@ rA r r r � test_grouping s �zAdjacentTests.test_groupingc sJ t � � td�}� �fdd�}tt�||��}dd� |D �}��||� dS )z5Test that the predicate is only called once per item.rl c s �� | � � � �| � dS )NT)r r rL �Z already_seenr1 r r r� " s z/AdjacentTests.test_call_once.<locals>.predicatec S s g | ] }d |f�qS )Tr rZ r r r r* ( s z0AdjacentTests.test_call_once.<locals>.<listcomp>N)r� r r& r. r� r+ r� r r� r �test_call_once s z AdjacentTests.test_call_onceN) rL rM rN r� r� r� r� r� r� r� r� r� r� r r r r r� � s r� c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) �GroupByTransformTestsc sp t � ��D ]8\}}|\}}|\}}| �||� | �t|�t|�� q | �t� fdd�� | �t�fdd�� dS )z?Compare two groupby objects for equality, both keys and groups.c s t � �S r9 r� r )�groupby1r r r< 4 r= z<GroupByTransformTests.assertAllGroupsEqual.<locals>.<lambda>c s t � �S r9 r� r )�groupby2r r r< 5 r= N)r r+ ZassertListEqualr& r` r� ) r1 r� r� rp rr Zkey1Zgroup1Zkey2Zgroup2r )r� r� r �assertAllGroupsEqual- s z*GroupByTransformTests.assertAllGroupsEqualc C s4 dd� t d�D �}t�|�}t|�}| �||� dS )z@Test that groupby_transform() with default args mimics groupby()c S s g | ]}|d |f�qS r7 r rZ r r r r* 9 s z<GroupByTransformTests.test_default_funcs.<locals>.<listcomp>r� N�r r. r� r r� r� r r r �test_default_funcs7 s z(GroupByTransformTests.test_default_funcsc C s dd� t d�D �}tj|td�td�d�}dd� |D �}dddd d d gfdd d dddgfg}| �||� tj|td�td�d�}dd� |D �}dddd gfdd d d gfd d ddgfd dgfg}| �||� ttt d�d��}tjt d�dd� |jd�}dd� |D �}ddg}| �||� d S )Nc S s& g | ] }t |d �t |d �|f�qS )r4 r# �r rZ r r r r* ? s z8GroupByTransformTests.test_valuefunc.<locals>.<listcomp>rl r r5 )�keyfunc� valuefuncc S s g | ]\}}|t |�f�qS r r% r� r r r r* E s rE rF r# r! r4 r$ r" rv rt c S s g | ]\}}|t |�f�qS r r% r� r r r r* L s Z abcdefghijc S s | d S �Nr4 r r: r r r r< S r= z6GroupByTransformTests.test_valuefunc.<locals>.<lambda>c S s g | ]\}}|d � |�f�qS )rd )re r� r r r r* U s )r Zabcde)rE Zfghij)r r. r� r r+ r� r �get)r1 r� r� r@ rA rq r r r �test_valuefunc> s4 �$ �0 �z$GroupByTransformTests.test_valuefuncc C s\ t d�}dd� }tj||d d�}t||�}| �||� t�||�}t||�}| �||� d S )Nr� c S s | d S r� r r: r r r r8 \ s z4GroupByTransformTests.test_no_valuefunc.<locals>.key)r� r� )r1 r� r8 r@ rA r r r �test_no_valuefuncY s z'GroupByTransformTests.test_no_valuefuncN)rL rM rN r� r� r� r� r r r r r� , s r� c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�NumericRangeTestsc C sV dddddgfdddd d gfd dd d gfd dddgfd dd d dgfdddddgfdddddgfdddddgfddd dd gfd dd d d d!gfd"dd#d$d%d&gfddt dd�ft dd�t dd�gfd'g fd(g fd)g fd*g ft dd�ft dd�t dd�gftd+�ftd,�td-�gffD ]B\}}ttj|� �}| �||� | �td.d/� t||�D ��� �qd S )0N)r! r rE rF r# )� @� � �?g @� @)r� r! )rE r� )r� r4 r� )r r� r4 r4 rl � )r r� � @r� g $@g .@)r rl r# r$ rt )r rl r� g @g "@)r ���r5 r5 rw r� r� )r� r r5 g �g �g �g �r� )r� )rE r )r� r� z2.0z0.0z1.0c s s" | ]\}}t |�t |�kV qd S r9 )�typer� r r r r[ � s z/NumericRangeTests.test_basic.<locals>.<genexpr>) r r r& r. � numeric_ranger+ rj r� r )r1 �argsrA r@ r r r r� i s0 " � �z NumericRangeTests.test_basicc C s$ | � tdd� � | � tdd� � d S )Nc S s t t�� �S r9 �r& r. r r r r r r< � r= z2NumericRangeTests.test_arg_count.<locals>.<lambda>c S s t t�dddd��S r� r r r r r r< � r= r� rC r r r �test_arg_count� s �z NumericRangeTests.test_arg_countc C s | � tdd� � d S )Nc S s t t�ddd��S )NrE rF r r r r r r r< � r= z2NumericRangeTests.test_zero_step.<locals>.<lambda>r_ rC r r r �test_zero_step� s �z NumericRangeTests.test_zero_stepN)rL rM rN r� r r r r r r r� h s r� c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�CountCycleTestsc C sN ddddddddd g }t �d t �d ��tt �d d ��fD ]}| �||� q8d S ) N)r rp )r rr )r rs )rE rp )rE rr )rE rs )rF rp )rF rr )rF rs rt r r# )r. r � count_cycler& r+ r� r r r r� � s ��zCountCycleTests.test_basicc C s2 | � tt�d��g � | � tt�dd��g � d S )Nrd rF �r+ r& r. r rC r r r rD � s zCountCycleTests.test_emptyc C s | � tt�dd��g � d S )Nr r� r rC r r r r� � s z CountCycleTests.test_negativeN)rL rM rN r� rD r� r r r r r � s r c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� LocateTestsc C s: dddddddg}t t�|��}dddg}| �||� d S )Nr rE rF r! �r& r. Zlocater+ r� r r r �test_default_pred� s z LocateTests.test_default_predc C s, dddg}t t�|��}g }| �||� d S r r r� r r r �test_no_matches� s zLocateTests.test_no_matchesc C sF dddddddg}dd� }t t�||��}ddddg}| �||� d S ) Nr� rE c S s | dkS )Nr� r r: r r r r< � r= z.LocateTests.test_custom_pred.<locals>.<lambda>r r# r4 r$ r )r1 r� rf r@ rA r r r �test_custom_pred� s z LocateTests.test_custom_predN)rL rM rN r r r r r r r r � s r c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�StripFunctionTestsc C sh t d�}dd� }| �t t�||��t d�� | �t t�||��t d�� | �t t�||��t d�� d S )Nzwww.example.comc S s | t d�kS )Nzcmowz.)r� r: r r r r< � r= z2StripFunctionTests.test_hashable.<locals>.<lambda>z example.comz www.exampleZexample�r& r+ r. �lstrip�rstrip�strip�r1 r� rf r r r � test_hashable� s z StripFunctionTests.test_hashablec C s� t d�t d�t d�t d�g}dd� }| �t t�||��|dd � � | �t t�||��|d d� � | �t t�||��|dd� � d S ) N�http://�wwwz.example�.comc S s | t d�t d�t d�fkS )Nr r r r% r: r r r r< � r= z6StripFunctionTests.test_not_hashable.<locals>.<lambda>rF r# r r r r r �test_not_hashable� s � z$StripFunctionTests.test_not_hashablec C s� dddddddddddg }dd� }| � tt�||��|dd � � | � tt�||��|d d� � | � tt�||��|dd� � d S )Nr rE rF r# c S s | dkS �NrF r r: r r r r< � r= z.StripFunctionTests.test_math.<locals>.<lambda>r� )r+ r& r. r r r r r r r � test_math� s z StripFunctionTests.test_mathN)rL rM rN r r r r r r r r � s r c @ s e Zd Zdd� Zdd� ZdS )�IsliceExtendedTestsc C s� ddddddg}t tdt|�d ��d g }d d d dd d ddg}t|||�D ]h}zt tj|f|�� �}W n0 tk r� } z| �||f� W 5 d }~X Y nX |t|� }| � |||� qLd S )Nr� r� r� r� r� r� r� r! rE rF r# r5 rw r� ) r& r r� r r. �islice_extended� ExceptionZfailr� r+ )r1 r� r� r� r� r@ r� rA r r r �test_all� s z IsliceExtendedTests.test_allc C s4 | � t�� tt�dddgddd�� W 5 Q R X d S )NrE rF r# r )r` ra r& r. r rC r r r r � s z"IsliceExtendedTests.test_zero_stepN)rL rM rN r r r r r r r � s r c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�ConsecutiveGroupsTestc C s` ddddddddd d g }d d � t �|�D �}dgdddgddgddgd gd gg}| �||� d S ) Nrz i����i����i����rE rF r! r4 r5 r" c S s g | ] }t |��qS r r% �r( r r r r r* � s z6ConsecutiveGroupsTest.test_numbers.<locals>.<listcomp>�r. �consecutive_groupsr+ r� r r r � test_numbers� s $z"ConsecutiveGroupsTest.test_numbersc C s^ ddddddddg}d d � }d d � t �||�D �}dgddgdddgddgg}| �||� d S ) Nr� Z10Z11Z20Z21Z22Z30Z31c S s t | �S r9 r� r: r r r r< � r= z<ConsecutiveGroupsTest.test_custom_ordering.<locals>.<lambda>c S s g | ] }t |��qS r r% r! r r r r* � s z>ConsecutiveGroupsTest.test_custom_ordering.<locals>.<listcomp>r" �r1 r� Zorderingr@ rA r r r �test_custom_ordering� s z*ConsecutiveGroupsTest.test_custom_orderingc C sZ ddddddg}t td��j}dd � t�||�D �}dgdddgddgg}| �||� d S ) N)rp rr rs rq )rp rs rr rq )rp rs rq rr )rp rq rr rs )rq rr rs rp )rq rs rp rr rh c S s g | ] }t |��qS r r% r! r r r r* s z>ConsecutiveGroupsTest.test_exotic_ordering.<locals>.<listcomp>)r& r r� r. r# r+ r% r r r �test_exotic_ordering� s ��z*ConsecutiveGroupsTest.test_exotic_orderingN)rL rM rN r$ r& r' r r r r r � s r c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd S ) �DifferenceTestc C s: dddddg}t t�|��}dddddg}| �||� d S )Nrl r� r� r� r� )r& r. � differencer+ r� r r r � test_normal s zDifferenceTest.test_normalc C s< dddddg}t t�|t��}dddddg}| �||� d S )Nrl r� r� r� r� r� �Z )r& r. r) r r+ r� r r r � test_custom s zDifferenceTest.test_customc C s4 t td��}t�|�}t t�|��}| �||� d S )Nr~ )r& r r. � accumulater) r+ )r1 ZoriginalZ accumulatedr@ r r r �test_roundtrip s z DifferenceTest.test_roundtripc C s | � tt�dg��dg� d S r �r+ r& r. r) rC r r r rG s zDifferenceTest.test_onec C s | � tt�g ��g � d S r9 r/ rC r r r rD s zDifferenceTest.test_emptyN)rL rM rN r* r, r. rG rD r r r r r( s r( c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd S ) � SeekableTestc C sZ dd� t d�D �}t�|�}| �t|�|� | �t|�g � |�d� | �t|�|� d S )Nc S s g | ] }t |��qS r �r r� r r r r* & s z6SeekableTest.test_exhaustion_reset.<locals>.<listcomp>rl r )r r. �seekabler+ r& �seek�r1 r� r� r r r �test_exhaustion_reset% s z"SeekableTest.test_exhaustion_resetc C s^ dd� t d�D �}t�|�}| �t�d|�|d d� � |�d� | �t|�|dd � � d S )Nc S s g | ] }t |��qS r r1 r� r r r r* 0 s z3SeekableTest.test_partial_reset.<locals>.<listcomp>rl r4 rE �r r. r2 r+ r r3 r& r4 r r r �test_partial_reset/ s z SeekableTest.test_partial_resetc C sx dd� t d�D �}t�|�}| �t�d|�|d d� � |�d� | �t|�|dd � � |�d� | �t|�|� d S )Nc S s g | ] }t |��qS r r1 r� r r r r* 9 s z-SeekableTest.test_forward.<locals>.<listcomp>rl rE r# r r6 r4 r r r � test_forward8 s zSeekableTest.test_forwardc C sp dd� t d�D �}t�|�}| �t�d|�|d d� � |�d� | �t|�g � |�d� | �t|�|� d S )Nc S s g | ] }t |��qS r r1 r� r r r r* E s z.SeekableTest.test_past_end.<locals>.<listcomp>rl rE r� r r6 r4 r r r � test_past_endD s zSeekableTest.test_past_endc s� t tt� �}t�|�}t�d|� |�� � | �� fdd�td�D �dd� td�D �� | �t � �d� t�d|� | �t � �dd� td�D �� d S )Nrl c s g | ] }� | �qS r r r� ��elementsr r r* X s z.SeekableTest.test_elements.<locals>.<listcomp>c S s g | ] }t |��qS r r1 r� r r r r* X s c S s g | ] }t |��qS r r1 r� r r r r* ] s r� ) r r r r. r2 r r; r+ r r� r& r4 r r: r � test_elementsP s � zSeekableTest.test_elementsN)rL rM rN r5 r7 r8 r9 r< r r r r r0 $ s r0 c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd S ) �SequenceViewTestsc C s. t �d�}| �t|�d� | �tdd� � d S )Nr� zSequenceView((1, 2, 3))c S s t �i �S r9 )r. � SequenceViewr r r r r< d r= z-SequenceViewTests.test_init.<locals>.<lambda>)r. r> r+ �reprr` r� )r1 �viewr r r � test_inita s zSequenceViewTests.test_initc C s` dddg}t �|�}| �t|�d� | �t|�d� |�� | �t|�d� | �t|�d� d S )NrE rF r# zSequenceView([1, 2, 3])zSequenceView([1, 2]))r. r> r+ r� r? �pop�r1 r r@ r r r � test_updatef s z SequenceViewTests.test_updatec C s@ d}t �|�}tt|� t|��D ]}| �|| || � q"d S )N�rp rr rs rq r� r )r. r> r r� r+ )r1 r r@ r� r r r r} p s z SequenceViewTests.test_indexingc C s� d}t �|�}t|�}tt| d |d ��d g }tt| |d ��}|�d� t|||�D ] }t|� }| �|| || � q^d S )NrE rE r ) r. r> r� r& r �remover r� r+ )r1 r r@ r� r� r� r� r� r r r r� v s z SequenceViewTests.test_slicingc C s� d}t �|�}| �d|� | �d|� | �tt|��t|�� | �tt|��tt|��� | �|�d�d� | �|� d�d� d S )N)rp rr rs rq r� r r rr r rE r rF ) r. r> ZassertInr r+ r& r� r� r� r rC r r r �test_abc_methods� s z"SequenceViewTests.test_abc_methodsN)rL rM rN rA rD r} r� rG r r r r r= ` s r= c @ s e Zd Zdd� Zdd� ZdS )� RunLengthTestc C sB dd� t d�D �}t�dtj�|��}ddddg}| �||� d S ) Nc s s | ]}t t|�d �V qdS )r Nr r� r r r r[ � s z,RunLengthTest.test_encode.<locals>.<genexpr>i r! )rv r~ )rt r~ )rE r� )rF r� )r r. r � run_length�encoder+ r� r r r � test_encode� s zRunLengthTest.test_encodec C s2 ddddg}d� tj�|��}d}| �||� d S )N)rq r! )rs r# )rr rF )rp rE rd Z ddddcccbba)re r. rI �decoder+ r� r r r � test_decode� s zRunLengthTest.test_decodeN)rL rM rN rK rM r r r r rH � s rH c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) � ExactlyNTestszTests for ``exactly_n()``c C sh | � t�dddgd�� | � t�ddddgd�� | � t�ddgd�� | � t�td�dd d � �� d S ) z$Iterable has ``n`` ``True`` elementsTFrF rE r r# r~ rl c S s | dk S r r r: r r r r< � r= z)ExactlyNTests.test_true.<locals>.<lambda>N)rj r. � exactly_nr rC r r r � test_true� s zExactlyNTests.test_truec C sr | � t�dddgd�� | � t�dddgd�� | � t�dgd�� | � t�dgd�� | � t�td�d�� dS )z.Iterable does not have ``n`` ``True`` elementsTFrF rE r5 r~ N)ri r. rO r rC r r r � test_false� s zExactlyNTests.test_falsec C s( | � t�g d�� | �t�g d�� dS )z7Return ``True`` if the iterable is empty and ``n`` is 0r rE N)rj r. rO ri rC r r r rD � s zExactlyNTests.test_emptyN)rL rM rN rJ rP rQ rD r r r r rN � s rN c @ s e Zd ZdZdd� Zdd� ZdS )�AlwaysReversibleTestsz!Tests for ``always_reversible()``c C sz | � tttd���tt�td���� | � ttdddg��tt�dddg��� | � tdddg�jt�dddg�j� d S )Nrl rE rF r# )r+ r& r� r r. �always_reversible� __class__rC r r r �test_regular_reversed� s ���z+AlwaysReversibleTests.test_regular_reversedc C s� | � t�� tdd� td�D �� W 5 Q R X | �tttd���tt�dd� td�D ���� | �ttdddg��tt�dd� d D ���� | �td �j t�d d� d D ��j � d S ) Nc s s | ] }|V qd S r9 r rZ r r r r[ � s z=AlwaysReversibleTests.test_nonseq_reversed.<locals>.<genexpr>rl c s s | ] }|V qd S r9 r rZ r r r r[ � s rE rF r# c s s | ] }|V qd S r9 r rZ r r r r[ � s r� r� c s s | ] }|V qd S r9 r rZ r r r r[ � s ) r` r� r� r r+ r& r. rS rI rT rC r r r �test_nonseq_reversed� s �� �z*AlwaysReversibleTests.test_nonseq_reversedN)rL rM rN rJ rU rV r r r r rR � s rR c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�CircularShiftsTestsc C s | � tt�g ��g � d S r9 )r+ r& r. �circular_shiftsrC r r r rD � s z CircularShiftsTests.test_emptyc C s" | � t�td��ddddg� d S )Nr! )r rE rF r# )rE rF r# r )rF r# r rE )r# r rE rF )r+ r. rX r rC r r r �test_simple_circular_shifts� s �z/CircularShiftsTests.test_simple_circular_shiftsc C s& | � t�ddddg�ddddg� d S )Nr rE )r rE r rE )rE r rE r )r+ r. rX rC r r r r� � s �z#CircularShiftsTests.test_duplicatesN)rL rM rN rD rY r� r r r r rW � s rW c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�MakeDecoratorTestsc sX t �t�}|ddd�d� fdd� �}|dddd �}t|�}d d d d dg}� �||� d S )NrE rl rF c s0 � � | d� � � |d� � � |d� ttt� �S )N�arg_1�arg_2�kwarg_1)r+ r r r )r[ r\ r] rC r r � user_function� s z4MakeDecoratorTests.test_basic.<locals>.user_functionr[ r\ r] �r] r� r� r� �7�9)N)r. �make_decoratorr r& r+ )r1 Zslicerr^ r) r@ rA r rC r r� � s z MakeDecoratorTests.test_basicc sb � fdd�}t j|dd�}|dddd�d d � �}|d�}t �d |�}d d dddg}� �||� d S )Nc sB � � | d d� | d }� � | d d� � � |d d� tt|�S )Nr �arg_0rE rF r\ r] )r+ r r )r r� r� rC r r � stringify� s z7MakeDecoratorTests.test_result_index.<locals>.stringifyrE )Z result_indexrc r\ r] r_ c S s t | �S r9 )r r� r r r r^ s z;MakeDecoratorTests.test_result_index.<locals>.user_functionr4 r� r� r� r� r� )r. rb r r+ )r1 rd Z stringifierr^ r) r@ rA r rC r �test_result_index� s z$MakeDecoratorTests.test_result_indexc C sd t �t j�}|� dd� �}|d�}| �t|�dddddg� |�d � | �t|�dddddg� d S ) Nc S s t tt| ��S r9 )r r r r� r r r r^ s z9MakeDecoratorTests.test_wrap_class.<locals>.user_functionr4 r� r� r� r� r� r )r. rb r2 r+ r& r3 )r1 Zseekerr^ r) r r r �test_wrap_class s z"MakeDecoratorTests.test_wrap_classN)rL rM rN r� re rf r r r r rZ � s rZ c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) �MapReduceTestsc C s^ dd� t d�D �}dd� }tt�||��� �}dddgfd d d gfd d gfg}| �||� d S )Nc s s | ]}t |�V qd S r9 r1 rZ r r r r[ s z.MapReduceTests.test_default.<locals>.<genexpr>r4 c S s t | �d S r r� r: r r r r< r= z-MapReduceTests.test_default.<locals>.<lambda>r r� r� rE r� r� rF r� )r r, r. � map_reduce�itemsr+ )r1 r� r� r@ rA r r r r3 s zMapReduceTests.test_defaultc C sd dd� t d�D �}dd� }t}tt�|||��� �}dddgfddd gfdd gfg}| �||� d S ) Nc s s | ]}t |�V qd S r9 r1 rZ r r r r[ # s z0MapReduceTests.test_valuefunc.<locals>.<genexpr>r4 c S s t | �d S r r� r: r r r r< $ r= z/MapReduceTests.test_valuefunc.<locals>.<lambda>r rE rF r# r! �r r r, r. rh ri r+ )r1 r� r� r� r@ rA r r r r� " s z MapReduceTests.test_valuefuncc C sX dd� t d�D �}dd� }t}dd� }tt�||||��� �}ddd g}| �||� d S ) Nc s s | ]}t |�V qd S r9 r1 rZ r r r r[ + s z1MapReduceTests.test_reducefunc.<locals>.<genexpr>r4 c S s t | �d S r r� r: r r r r< , r= z0MapReduceTests.test_reducefunc.<locals>.<lambda>c S s t t| d�S r� )r r )Z value_listr r r r< . r= )r r )rE r$ )rF r! rj )r1 r� r� r� Z reducefuncr@ rA r r r �test_reducefunc* s � z MapReduceTests.test_reducefuncc sN t �ddddddgt�� | �� dddgdddgd�� | �t� fdd�� d S )NrE r rF )FTc s � d � d�S r� )rJ r �rq r r r< 8 r= z)MapReduceTests.test_ret.<locals>.<lambda>)r. rh �boolr+ r` �KeyErrorrC r rl r �test_ret5 s zMapReduceTests.test_retN)rL rM rN r3 r� rk ro r r r r rg s rg )UZ __future__r r r Zdecimalr Zdoctestr Z fractionsr � functoolsr r �heapqr �ior � itertoolsr r r r r r r �operatorr r r Zunittestr Z six.movesr r r r Zmore_itertoolsr. r r rO rY rc r� r� r� r� r� r� r� r� r� r r r) r, r: rY r` rj rt rv r r� r� r� r� r� r� r� r� r r r r r r( r0 r= rH rN rR rW rZ rg r r r r �<module> sx $ , ! .='" GC #(Z 3a<( !<71