
    >ieQ                    "   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ej                            d ej        g ddz  g ddz   ej        dej        	           d
           ej        e                    d dd          e                    d          e                    d dd          d
           ej         ej        e                    d dd          d           ej        e                    d          d           ej        e                    d dd          d          d
           ej        de                    d dd          i           ej        e                    d dd          g dd           ej         ej        d           ej        d           ej        d          dz    ej        d          d           ej        dg ddz  i           ej        dej
        
                    d          i           ej        g ddz  g ddz  g ddz   ej        dej        	           d           ej        g ddz             ej        ej
        
                    d                     ej        ej
        
                    d           ej        d                     ej        ej
        
                    d          ej
        
                    d                    g          ej                            dddg          d                         Zej                            dg d           ej                            dddg          d!                         Zd" ZdS )#    )annotationsN`   df         r	   )g333333?g333333@gffffff@	   )dtype)xyzi@B )'  r   i )  r   r   )   )abetisr   taxonbs)r   r   )i N  iLI )wr   r   r   )y      ?       @y      @      @y      @      @)r   bcd)indexnpartitionsr      c                *   t          j        | |          }|                                                    d          }t	          |                                           }t          ||z
            dk    st          ||z
            |z  dk     sJ d S d S )Nr   sync	schedulerr   皙?)ddfrom_pandasnunique_approxcomputelendrop_duplicatesabs)r   r   ddfapproxexacts        Elib/python3.11/site-packages/dask/dataframe/tests/test_hyperloglog.py
test_basicr0      s    r .
5
5
5C!!))F);;F""$$%%Ev~!##s6E>':':U'BT'I'I'I'I##'I'I    split_every)Nr   
   c                `   t          j        g ddz            }t          j        ||          }|                    |                               d          }t          |                                          }t          ||z
            dk    st          ||z
            |z  dk     sJ d S d S )	Nr   r   r    )r2   r!   r"   r   r$   )	pdSeriesr%   r&   r'   r(   r)   r*   r+   )r2   r   r   r,   r-   r.   s         r/   test_split_everyr7   L   s     
999t#	$	$B
.
5
5
5CK88@@6@RRF""$$%%Ev~!##s6E>':':U'BT'I'I'I'I##'I'Ir1   c                     t           j                            ddt          t          dddd          } |                                                                 dk    sJ d S )	Nz
2000-01-01z
2000-04-01)valueid10s1Dr   )freqpartition_freqseedr   )r%   demomake_timeseriesfloatintr'   r(   )r   s    r/   test_larger_datarD   W   si    		 	 s## 
! 
 
B &&((4//////r1   )
__future__r   numpynppandasr5   pytestdask.dataframe	dataframer%   randomRandomStatersmarkparametrize	DataFramearangeint8randintrandnuniformrepeatzerosr6   onesr0   r7   rD    r1   r/   <module>r[      s7   " " " " " "               Y2 YY]$__q( ry"'2223 	
 	
 	ZZ7H55XXe__ZZ7H55 	
 	
 	RYrzz!Wg>>BBRYrxx~~q11RYrzz!Wg>>BB 	
 	
 	c2::a(;;<==ZZ7D11??? 	
 	
 	RXh''RXh''RXh'''1RXh''	 	
 	
 	c999t+,--c29++D11233YY]$__q(---1 ry"'2223	 	
 	
 		)))d"##	")""4(())	")""4((>>>	")""4((	0@0@0F0FGGGe36 6n B00J J 10o6 6pJ 66B00J J 10 76J	0 	0 	0 	0 	0r1   