
    >ie                    
   d dl mZ d dlZd dlZd dlZd dlmZ  ej        d          Z	d  e
d          D             Zdddd	d
dddddgdZd Zd Zd Zd Zej                            dg d          d             Zd ZdS )    )annotationsNfastavroc                f    g | ].}t          j        g d           t          j        dd          d/S ))fredwilmabarneybettyr   d   )namenumber)randomchoicerandint).0_s     8lib/python3.11/site-packages/dask/bag/tests/test_avro.py
<listcomp>r      sT       
 	
 BBBCC.C((         DescrRandomtestrecordr   stringr   typer   int)docr   	namespacer   fieldsc                   t          |           } t          j                            | d          }t	          |d          5 }t
                              |t          t                     d d d            n# 1 swxY w Y   t          j
        |d           }|j        dk    sJ |                                t          k    sJ d S )None.avrowbrecordsschema	blocksize   strospathjoinopenr   writerexpectedr&   db	read_avronpartitionscompute)tmpdirfnfbs       r   test_onefile_oneblockr:      s    [[F	fj	)	)B	b$ <18F;;;< < < < < < < < < < < < < < <
R4(((A=A99;;(""""""s    (A44A8;A8c                   t          |           } t          j                            | d          }t          j                            | d          }t	          |d          5 }t
                              |t          d d         t                     d d d            n# 1 swxY w Y   t	          |d          5 }t
                              |t          dd          t                     d d d            n# 1 swxY w Y   t          j
        t          j                            | d          d           }|j        dk    sJ |                                t          k    sJ d S )	Nr"   two.avror#     r$   *.avror'      r*   r6   fn1fn2r8   r9   s        r   test_twofile_oneblockrC   &   s   [[F
',,vz
*
*C
',,vz
*
*C	c4 BA8DSD>&AAAB B B B B B B B B B B B B B B	c4 BA8CDD>&AAAB B B B B B B B B B B B B B B
RW\\&(33tDDDA=A99;;(""""""s$    0BB #B 70C33C7:C7c                   t          |           } t          j                            | d          }t          j                            | d          }t	          |d          5 }t
                              |t          d d         t          d           d d d            n# 1 swxY w Y   t	          |d          5 }t
                              |t          dd          t          d           d d d            n# 1 swxY w Y   t          j
        t          j                            | d          d           }|j        d	k    sJ |                                t          k    sJ t          j
        t          j                            | d          d
          }|j        d	k    sJ |                                t          k    sJ d S )Nr"   r<   r#   r=   r
   )r%   r&   sync_intervalr>   r'   r?   r   r*   r@   s        r   test_twofile_multiblockrF   3   s   [[F
',,vz
*
*C
',,vz
*
*C	c4 UA8DSD>&PSTTTU U U U U U U U U U U U U U U	c4 UA8CDD>&PSTTTU U U U U U U U U U U U U U U
RW\\&(33tDDDA=A99;;(""""
RW\\&(33tDDDA=199;;(""""""s$    1BB!$B!81C55C9<C9c                   ddl m} t          |           } t          j                            | d          }t          j        d dD             d          }dd	d
ddgd}|                    ||d          }t          |d         |          sJ |                    ||          }t          |          dk    sJ t          j        |          }|                                |                                k    sJ d S )Nr   )Delayed	out*.avroc                    g | ]}d |iS )a )r   is     r   r   z)test_roundtrip_simple.<locals>.<listcomp>I   s    <<<q3(<<<r   )r)   r?            r?   r4   Testr   rK   r   r   r   r   r    F)r5   )dask.delayedrH   r+   r,   r-   r.   r2   from_sequenceto_avro
isinstancelenr3   r5   )r6   rH   r7   r9   r&   outb2s          r   test_roundtrip_simpler[   D   s   $$$$$$[[F	fk	*	*B
<<O<<<!LLLA//0 F
 ))B)
.
.Cc!fg&&&&&
))B

Cs88q====	b		B99;;"**,,&&&&&&r   codec)nulldeflatesnappyc                   t          |           } |dk    rt          j        d           t          j                            | d          }t          j        t          d          }|	                    |t          |           t          j        |          }|                                |                                k    sJ d S )Nr_   rI   rN   rQ   )r&   r\   )r+   pytestimportorskipr,   r-   r.   r2   rU   r1   rV   r&   r3   r5   )r6   r\   r7   r9   rZ   s        r   test_roundtriprc   W   s    [[FH%%%	fk	*	*B
q111AIIbuI---	b		B99;;"**,,&&&&&&r   c                D   t          |           } t          j        t          d          }t          j                            | d          }t          j        t                    5  |
                    |g            d d d            n# 1 swxY w Y   t          j        t                    5  |
                    |i            d d d            n# 1 swxY w Y   t          j        t                    5  |
                    |ddi           d d d            n# 1 swxY w Y   t          j        t                    5  |
                    |ddi           d d d            n# 1 swxY w Y   t          j        t                    5  |
                    |dd	d
           d d d            n# 1 swxY w Y   t          j        t                    5  |
                    |ddd
           d d d            n# 1 swxY w Y   t          j        t                    5  |
                    |ddd
           d d d            n# 1 swxY w Y   t          j        t                    5  |
                    |ddddigd           d d d            d S # 1 swxY w Y   d S )NrN   rQ   rI   )r&   r   unknownr   r   wrongr   r   rK   rS   )r+   r2   rU   r1   r,   r-   r.   ra   raisesAssertionErrorrV   )r6   r9   r7   s      r   test_invalid_schemari   c   s   [[F
q111A	fk	*	*B	~	&	& ! !			"R	   ! ! ! ! ! ! ! ! ! ! ! ! ! ! !	~	&	& ! !			"R	   ! ! ! ! ! ! ! ! ! ! ! ! ! ! !	~	&	& 1 1			"eY/	0001 1 1 1 1 1 1 1 1 1 1 1 1 1 1	~	&	& / /			"ff-	.../ / / / / / / / / / / / / / /	~	&	& @ @			"fg>>	???@ @ @ @ @ @ @ @ @ @ @ @ @ @ @	~	&	& A A			"fh??	@@@A A A A A A A A A A A A A A A	~	&	& A A			"fh??	@@@A A A A A A A A A A A A A A A	~	&	& 
 
			fc]OTT 	 	
 	
 	

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
s   $BBB,CCC4DD!D>E$$E(+E(F//F36F3G::G>G>II	I	)JJJ)
__future__r   r,   r   ra   dask.bagbagr2   rb   r   ranger1   r&   r:   rC   rF   r[   markparametrizerc   ri   rL   r   r   <module>rp      s;   " " " " " " 				        6z** 
 U4[[   11He3T3TU
 
# # #
# 
# 
## # #"' ' '& "?"?"?@@' ' A@'
 
 
 
 
r   