
    biQ                    L   d dl mZ d dlZd dlZd dlZd dlmZ	 ej                  j                  d      Zej                  j                  d ej                  g ddz  g ddz   ej                   dej"                  	       d
       ej                  ej%                  d dd      ej'                  d      ej)                  d dd      d
       ej                   ej*                  ej%                  d dd      d       ej*                  ej'                  d      d       ej*                  ej)                  d dd      d      d
       ej                  dej%                  d dd      i       ej                  ej%                  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                  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                  j                  d             ej.                  ej                  j                  d       ej0                  d             ej.                  ej                  j                  d      ej                  j                  d            g      ej                  j                  dddg      d               Zej                  j                  dg d       ej                  j                  dddg      d!               Zd" Zy)#    )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                  | |      }|j                         j                  d      }t	        | j                               }t        ||z
        dk  st        ||z
        |z  dk  sJ y y )Nr   sync	schedulerr   皙?)ddfrom_pandasnunique_approxcomputelendrop_duplicatesabs)r   r   ddfapproxexacts        `/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/dask/dataframe/tests/test_hyperloglog.py
test_basicr0      sv    r ..
5C!))F);F""$%Ev~!#s6E>':U'BT'III'I#    split_every)Nr   
   c                ,   t        j                  g ddz        }t        j                  ||      }|j	                  |       j                  d      }t        |j                               }t        ||z
        dk  st        ||z
        |z  dk  sJ y y )	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     
9t#	$B
..
5CK8@@6@RF""$%Ev~!#s6E>':U'BT'III'I#r1   c                     t         j                  j                  ddt        t        dddd      } | j                         j                         dkD  sJ y )	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   sT    		 	 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[      sA   "    YY2 ]$q( ryy"''23	
 	ZZ7H5XXe_ZZ7H5	
 	RYYrzz!Wg>BRYYrxx~q1RYYrzz!Wg>B	
 	c2::a(;<=ZZ7D1?	
 	RXXh'RXXh'RXXh''1RXXh'		
 	c9t+,-c299++D123]$q(-1 ryy"''23		
 			)d"#		"))""4()		"))""4(>		"))""4(		0@0@0FGe36n B0J 1o6pJ 6B0J 1 7J	0r1   