
    Dh                     n    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y)    )AsyncIterable)TestCaseN   )
async_testreturnc                 6   K   t        d      D ]  } |  	 y w)N   )rangeis    /var/www/fastuser/data/www/generator.snapmosaic.io/flask_app/venv/lib/python3.12/site-packages/aioitertools/tests/more_itertools.py_genr      s     1X s   c                    K   y w)N r       r   _emptyr      s	     
s   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y)MoreItertoolsTestNc                    K   | j                  t        j                  dt                      d {   ddg       | j                  t        j                  dt	        d             d {   ddg       y 7 E7 w)N   r   r   r	   )assertEqualmittaker   r
   selfs    r   	test_takezMoreItertoolsTest.test_take   sY     sxx4622QF;sxx5844q!f= 34s!   -A9A58A9(A7)A97A9c                 |   K   | j                  t        j                  dt                      d {   g        y 7 
w)Nr   )r   r   r   r   r   s    r   test_take_zeroz MoreItertoolsTest.test_take_zero   s)     sxx4622B72s   -<:<c                    K   | j                  t              5  t        j                  dt	                      d {    d d d        y 7 # 1 sw Y   y xY ww)N)assertRaises
ValueErrorr   r   r   r   s    r   test_take_negativez$MoreItertoolsTest.test_take_negative   sG     z* 	'((2tv&&&	' 	'&	' 	's+   A"A
AA
	AA

AAc                    K   | j                  t        j                  dt                      d {   t	        t        d                   y 7 w)N
   r	   )r   r   r   r   listr
   r   s    r   test_take_more_than_iterablez.MoreItertoolsTest.test_take_more_than_iterable$   s1     sxxDF33T%(^D3s   -AAAc                 b  K   t               }| j                  t        t        j                  d|       d {         d       | j                  t        j                  d|       d {   g        | j                  t        j                  dt                      d {   g        y 7 r7 B7 w)Nr	   r   )r   r   lenr   r   r   )r   its     r   test_take_emptyz!MoreItertoolsTest.test_take_empty(   sy     V388Ar?23Q7sxx2.3sxx6844b9 3.4s3   4B/B)1B/(B+)5B/B-B/+B/-B/c                 T  K   | j                  t        j                  t               d      2 cg c3 d {   }|7 6 nc c}w c}ddgddgdgg       | j                  t        j                  t	        d      d      2 cg c3 d {  7  }|6 nc c}w c}ddgddgdgg       y w)Nr   r   r         r	   )r   r   chunkedr   r
   r   chunks     r   test_chunkedzMoreItertoolsTest.test_chunked/   s     &)kk$&!&<==UU===AAQRPS?T	
 	&)kk%(A&>??UU??1a&1a&STRUAV	
sN   ,B(A >
<>
A >
A AB(BB
B
B
BB(c           	         K   | j                  g t        j                  t               d      2 cg c3 d {   }|7 6 nc c}w c}       y wNr   )r   r   r/   r   r0   s     r   test_chunked_emptyz$MoreItertoolsTest.test_chunked_empty8   s8     s{{68Q7OPPeePPPQs*   -AA
?=?A
?A
 Ac                 &  K   t               }t        j                  d |       d {   \  }}| j                  |2 cg c3 d {   }|7 $7 	6 nc c}w c}g d       | j                  |2 cg c3 d {  7  }|6 nc c}w c}ddg       y w)Nc                     | dk  S r4   r   r   s    r   <lambda>z?MoreItertoolsTest.test_before_and_after_split.<locals>.<lambda>?   s
    Q!V r   r   r   r   r-   r.   r   r   before_and_afterr   r   r*   beforeafterelms        r   test_before_and_after_splitz-MoreItertoolsTest.test_before_and_after_split<   sx     V!223CRHHv66# I666	Bu55#551v>sh   %BABA A
AA
ABA
AB0B2A?
6A97A?
;B Bc                 "  K   t               }t        j                  d |       d {   \  }}| j                  |2 cg c3 d {   }|7 $7 	6 nc c}w c}g d       | j                  |2 cg c3 d {  7  }|6 nc c}w c}g        y w)Nc                      yNTr   r   s    r   r8   zEMoreItertoolsTest.test_before_and_after_before_only.<locals>.<lambda>F       r   r   r   r   r-   r.   r:   r<   s        r   !test_before_and_after_before_onlyz3MoreItertoolsTest.test_before_and_after_before_onlyC   ss     V!22>2FFv66# G666Hu55#55r:sh   %BABA A
AA
ABA
AB0B2A?
6A97A?
;B Bc                 "  K   t               }t        j                  d |       d {   \  }}| j                  |2 cg c3 d {   }|7 $7 	6 nc c}w c}g        | j                  |2 cg c3 d {  7  }|6 nc c}w c}g d       y w)Nc                      y)NFr   r   s    r   r8   zDMoreItertoolsTest.test_before_and_after_after_only.<locals>.<lambda>M   rD   r   rE   r:   r<   s        r    test_before_and_after_after_onlyz2MoreItertoolsTest.test_before_and_after_after_onlyJ   ss     V!22?BGGv66# H666;u55#55Gsh   %BABA A
AA
ABA
AB.A?0A=
4A75A=
9A?>Bc                 D  K   dt         dt        fd}t               }t        j                  ||       d {   \  }}| j                  |2 cg c3 d {   }|7 $7 	6 nc c}w c}g d       | j                  |2 cg c3 d {  7  }|6 nc c}w c}ddg       y w)Nr?   r   c                    K   | dk  S wr4   r   )r?   s    r   	predicatezJMoreItertoolsTest.test_before_and_after_async_predicate.<locals>.predicateS   s     !8Os   r9   r-   r.   )intboolr   r   r;   r   )r   rL   r*   r=   r>   r?   s         r   %test_before_and_after_async_predicatez7MoreItertoolsTest.test_before_and_after_async_predicateQ   s     	 	 	 V!229bAAv66# B666	Bu55#551v>si   4B AB A!A
AA
A!B A
A! B ?BB
BB

BB c                   K   t               }t        j                  d |       d {   \  }}| j                  |2 cg c3 d {   }|7 $7 	6 nc c}w c}g        | j                  |2 cg c3 d {  7  }|6 nc c}w c}g        y w)Nc                      yrC   r   r   s    r   r8   z?MoreItertoolsTest.test_before_and_after_empty.<locals>.<lambda>^   rD   r   )r   r   r;   r   r<   s        r   test_before_and_after_emptyz-MoreItertoolsTest.test_before_and_after_empty[   ss     X!22>2FFv66# G666;u55#55r:sh   %BABA A
AA
ABA
AB.A?0A=
4A75A=
9A?>B)r   N)__name__
__module____qualname__r   r   r   r#   r'   r+   r2   r5   r@   rF   rI   rO   rR   r   r   r   r   r      s    > > 8 8 ' ' E E : : 
 
 R R ? ? ; ; H H ? ? ; ;r   r   )typingr   unittestr   aioitertools.more_itertoolsmore_itertoolsr   helpersr   rM   r   r   r   r   r   r   <module>r[      sC    !  ) M#& 
mC( 
K; K;r   