
    ?f                         d dl mZ d dlmZ d dlmZ ddlmZ dee	         fdZ
dee	         fdZ G d	 d
e          ZdS )    )AsyncIterable)TestCaseN   )
async_testreturnc                 8   K   t          d          D ]} | W V  d S )N   )rangeis    p/home/panchajanya/Development/work/volatility3/lib/python3.11/site-packages/aioitertools/tests/more_itertools.py_genr      s2      1XX       c                 
   K   d S N r   r   r   _emptyr      s      
Fr   c                   .   e Zd Zedd            Zedd            Zedd            Zedd            Zedd            Zedd            Z	edd	            Z
edd
            Zedd            Zedd            Zedd            Zedd            ZdS )MoreItertoolsTestr   Nc                    K   |                      t          j        dt                                 d {V ddg           |                      t          j        dt	          d                     d {V ddg           d S )N   r   r   r	   )assertEqualmittaker   r
   selfs    r   	test_takezMoreItertoolsTest.test_take   s      sx46622222222QF;;;sx58844444444q!f=====r   c                    K   |                      t          j        dt                                 d {V g            d S )Nr   )r   r   r   r   r   s    r   test_take_zeroz MoreItertoolsTest.test_take_zero   sD      sx46622222222B77777r   c                    K   |                      t                    5  t          j        dt	                                 d {V  d d d            d S # 1 swxY w Y   d S )N)assertRaises
ValueErrorr   r   r   r   s    r   test_take_negativez$MoreItertoolsTest.test_take_negative   s      z** 	' 	'(2tvv&&&&&&&&&	' 	' 	' 	' 	' 	' 	' 	' 	' 	' 	' 	' 	' 	' 	' 	' 	' 	's   (AAAc                    K   |                      t          j        dt                                 d {V t	          t          d                               d S )N
   r	   )r   r   r   r   listr
   r   s    r   test_take_more_than_iterablez.MoreItertoolsTest.test_take_more_than_iterable$   sP      sxDFF33333333T%((^^DDDDDr   c                 r  K   t                      }|                     t          t          j        d|           d {V           d           |                     t          j        d|           d {V g            |                     t          j        dt                                 d {V g            d S )Nr	   r   )r   r   lenr   r   r   )r   its     r   test_take_emptyz!MoreItertoolsTest.test_take_empty(   s      VV38Ar??22222233Q777sx2......333sx68844444444b99999r   c                 @  K   |                      d t          j        t                      d          2              d {V ddgddgdgg           |                      d t          j        t	          d          d          2              d {V ddgddgdgg           d S )	Nc                 "   K   g | 3 d {V }|
6 S r   r   .0chunks     r   
<listcomp>z2MoreItertoolsTest.test_chunked.<locals>.<listcomp>2   s.      ========UU====   r   r   r         c                 "   K   g | 3 d {V }|
6 S r   r   r/   s     r   r2   z2MoreItertoolsTest.test_chunked.<locals>.<listcomp>5   s.      ????????UU????r3   r	   )r   r   chunkedr   r
   r   s    r   test_chunkedzMoreItertoolsTest.test_chunked/   s      ==ck$&&!&<&<=========AAQRPS?T	
 	
 	
 	??ck%((A&>&>?????????1a&1a&STRUAV	
 	
 	
 	
 	
r   c                    K   |                      g d t          j        t                      d          2              d {V            d S )Nc                 "   K   g | 3 d {V }|
6 S r   r   r/   s     r   r2   z8MoreItertoolsTest.test_chunked_empty.<locals>.<listcomp>:   s.      PPPPPPPPeePPPPr3   r   )r   r   r7   r   r   s    r   test_chunked_emptyz$MoreItertoolsTest.test_chunked_empty8   sS      PPs{688Q7O7OPPPPPPPPPQQQQQr   c                   K   t                      }t          j        d |           d {V \  }}|                     d |2              d {V g d           |                     d |2              d {V ddg           d S )Nc                     | dk    S Nr   r   r   s    r   <lambda>z?MoreItertoolsTest.test_before_and_after_split.<locals>.<lambda>?   s
    Q!V r   c                 "   K   g | 3 d {V }|
6 S r   r   r0   elms     r   r2   zAMoreItertoolsTest.test_before_and_after_split.<locals>.<listcomp>@   .      66666666#6666r3   r   r   r   c                 "   K   g | 3 d {V }|
6 S r   r   rA   s     r   r2   zAMoreItertoolsTest.test_before_and_after_split.<locals>.<listcomp>A   .      55555555#5555r3   r4   r5   r   r   before_and_afterr   r   r+   beforeafters       r   test_before_and_after_splitz-MoreItertoolsTest.test_before_and_after_split<   s      VV!23C3CRHHHHHHHH66v666666666			BBB55u5555555551v>>>>>r   c                    K   t                      }t          j        d |           d {V \  }}|                     d |2              d {V g d           |                     d |2              d {V g            d S )Nc                     dS NTr   r   s    r   r?   zEMoreItertoolsTest.test_before_and_after_before_only.<locals>.<lambda>F       T r   c                 "   K   g | 3 d {V }|
6 S r   r   rA   s     r   r2   zGMoreItertoolsTest.test_before_and_after_before_only.<locals>.<listcomp>G   rC   r3   r   r   r   r4   r5   c                 "   K   g | 3 d {V }|
6 S r   r   rA   s     r   r2   zGMoreItertoolsTest.test_before_and_after_before_only.<locals>.<listcomp>H   rF   r3   rG   rI   s       r   !test_before_and_after_before_onlyz3MoreItertoolsTest.test_before_and_after_before_onlyC   s      VV!2>>2FFFFFFFF66v666666666HHH55u555555555r:::::r   c                    K   t                      }t          j        d |           d {V \  }}|                     d |2              d {V g            |                     d |2              d {V g d           d S )Nc                     dS )NFr   r   s    r   r?   zDMoreItertoolsTest.test_before_and_after_after_only.<locals>.<lambda>M   s    U r   c                 "   K   g | 3 d {V }|
6 S r   r   rA   s     r   r2   zFMoreItertoolsTest.test_before_and_after_after_only.<locals>.<listcomp>N   rC   r3   c                 "   K   g | 3 d {V }|
6 S r   r   rA   s     r   r2   zFMoreItertoolsTest.test_before_and_after_after_only.<locals>.<listcomp>O   rF   r3   rR   rG   rI   s       r    test_before_and_after_after_onlyz2MoreItertoolsTest.test_before_and_after_after_onlyJ   s      VV!2??BGGGGGGGG66v666666666;;;55u555555555GGGGGr   c                 &  K   dt           dt          fd}t                      }t          j        ||           d {V \  }}|                     d |2              d {V g d           |                     d |2              d {V ddg           d S )	NrB   r   c                    K   | dk    S r>   r   )rB   s    r   	predicatezJMoreItertoolsTest.test_before_and_after_async_predicate.<locals>.predicateS   s      !8Or   c                 "   K   g | 3 d {V }|
6 S r   r   rA   s     r   r2   zKMoreItertoolsTest.test_before_and_after_async_predicate.<locals>.<listcomp>X   rC   r3   rD   c                 "   K   g | 3 d {V }|
6 S r   r   rA   s     r   r2   zKMoreItertoolsTest.test_before_and_after_async_predicate.<locals>.<listcomp>Y   rF   r3   r4   r5   )intboolr   r   rH   r   )r   r\   r+   rJ   rK   s        r   %test_before_and_after_async_predicatez7MoreItertoolsTest.test_before_and_after_async_predicateQ   s      	 	 	 	 	 	 VV!29bAAAAAAAA66v666666666			BBB55u5555555551v>>>>>r   c                    K   t                      }t          j        d |           d {V \  }}|                     d |2              d {V g            |                     d |2              d {V g            d S )Nc                     dS rO   r   r   s    r   r?   z?MoreItertoolsTest.test_before_and_after_empty.<locals>.<lambda>^   rP   r   c                 "   K   g | 3 d {V }|
6 S r   r   rA   s     r   r2   zAMoreItertoolsTest.test_before_and_after_empty.<locals>.<listcomp>_   rC   r3   c                 "   K   g | 3 d {V }|
6 S r   r   rA   s     r   r2   zAMoreItertoolsTest.test_before_and_after_empty.<locals>.<listcomp>`   rF   r3   )r   r   rH   r   rI   s       r   test_before_and_after_emptyz-MoreItertoolsTest.test_before_and_after_empty[   s      XX!2>>2FFFFFFFF66v666666666;;;55u555555555r:::::r   )r   N)__name__
__module____qualname__r   r   r   r$   r(   r,   r8   r;   rL   rT   rY   ra   rf   r   r   r   r   r      s       > > > Z> 8 8 8 Z8 ' ' ' Z' E E E ZE : : : Z: 
 
 
 Z
 R R R ZR ? ? ? Z? ; ; ; Z; H H H ZH ? ? ? Z? ; ; ; Z; ; ;r   r   )typingr   unittestr   aioitertools.more_itertoolsmore_itertoolsr   helpersr   r_   r   r   r   r   r   r   <module>ro      s    !                 ) ) ) ) ) )      M#&    
mC(    
K; K; K; K; K; K; K; K; K; K;r   