
    tf                       d dl mZ d dlZd dlZd dlZd dlZd dlZd dl	m
Z
  e
ej                        Ze e
d      k\  Ze e
d      k\  Ze e
d      k\  Ze e
d      k\  Zej"                  dk\  Zej&                  d	k\  Zej&                  d
k\  Zej&                  dk\  Zej&                  dk\  Zej&                  dk\  Zej"                  dk\  Zd dlmZ d Zd Zd Zd Z d Z!d$dZ"d%dZ#d Z$d Z%ejL                  d&d'd       Z'ejL                  d        Z(ejL                  d        Z)ejL                  d        Z*ejL                  d        Z+ejL                  d        Z,ejL                  d(d       Z-ejL                  d        Z.ejL                  d         Z/ejL                  d!        Z0erejb                  jd                  Z2n ejf                  jh                  jd                  Z2d)d"Z5d)d#Z6y)*    )annotationsN)Versionz1.3.1z1.3.3z1.4.0z1.5.0   )r   r      )r   r   r   )r   r   r   )r   r   r   )r   r   r      c                    t        j                  | |g|i | t        | j                  t        j
                        s
J |  d       t        |j                  t        j
                        s
J | d       y )Nz is not categorical dtype)tmassert_extension_array_equal
isinstancedtypepdCategoricalDtype)leftrightargskwargss       `/var/www/html/software/conda/envs/higlass/lib/python3.12/site-packages/dask/dataframe/_compat.pyassert_categorical_equalr      s    ##D%A$A&A

B'' *
()*  R(( +
)*+     c                    t        j                  |       }t        j                  |      }t        j                  j	                  ||       | |    }||    }t        j                  j	                  ||       y N)r   isnanptestingassert_array_equal)r   r   left_naright_na
left_validright_valids         r   assert_numpy_array_equalr    %   s[    ggdmGwwu~HJJ!!'84whJ	"KJJ!!*k:r   c                     t         j                  j                  dd      } t        t        j
                        d d }t        j                  | |t        d            S )N      ABCDindexcolumns)r   randomrandnliststringascii_lettersr   	DataFrame)datar&   s     r   makeDataFramer/   /   sD    99??2q!D%%&s+E<<E4<@@r   c                 8    t               } t               | _        | S r   )r/   makeDateIndexr&   )r.   s    r   makeTimeDataFramer2   5   s    ?DDJKr   c                     t               d   S )NA)r2    r   r   makeTimeSeriesr6   ;   s    s##r   c                2    t        j                  d| |      S )N2000periodsfreq)r   
date_rangekr;   s     r   r1   r1   ?   s    ==66r   c                2    t        j                  d| |      S )Nz1 dayr9   )r   timedelta_ranger=   s     r   makeTimedeltaIndexrA   C   s    gqt<<r   c                     t               } | j                  }t        j                  |dkD  t        j                  |      }t        j                  || j                  | j                        S )Nr   r%   )	r/   valuesr   wherenanr   r-   r&   r'   )dfr.   s     r   makeMissingDataframerG   G   sG    	B99D88D1Hbffd+D<<BHHbjjAAr   c            	         t        j                  g dg dt        d      D  cg c]  } d|  	 c} t        j                  dd      d      }|S c c} w )N)        r   r   r   r#   )rI   r   r   r   r      fooz
2009-01-01)r:   )r4   BCD)r   r-   ranger<   )irF   s     r   makeMixedDataFramerQ   N   sM    	""%*1X.Cs).|Q7		

B I	 /s   Ac              #  |  K   g d}| |vrJt         rDt        s>t        j                         5  t        j                  ddt
               d  d d d        y |sH| |vrDt         s>t        j                         5  t        j                  ddt
               d  d d d        y d  y # 1 sw Y   y xY w# 1 sw Y   y xY ww)N)summedianprodminmaxstdvarquantileignorez!The default value of numeric_onlymessagecategoryz)Dropping of nuisance columns in DataFrame)PANDAS_GE_150PANDAS_GE_200warningscatch_warningsfilterwarningsFutureWarning)nameshow_nuisance_warningsupported_funcss      r   check_numeric_only_deprecationrh   Z   s     WO?"}]$$& 	##;&
 	 	 "d/&A-$$& 	##C&
 	 	 	%	 		 	s4   )B<!B$)B<5!B0B<$B-)B<0B95B<c               #     K   t         s?t        j                  d      5  t        j                  ddt               d  d d d        y d  y # 1 sw Y   y xY ww)NT)recordr[   zDropping of nuisance columns)r_   ra   rb   rc   rd   r5   r   r   check_nuisance_columns_warningrk   s   sP     $$D1 	##8- 		 	 		 	s   A AAAAc               #     K   t         r=t        j                         5  t        j                  ddt               d  d d d        y d  y # 1 sw Y   y xY ww)Nr[   zS.*Call without passing 'axis' instead|.*Operate on the un-grouped DataFrame insteadPANDAS_GE_210ra   rb   rc   rd   r5   r   r   check_groupby_axis_deprecationro      sO     $$& 	##e
 	 	 		 	s   A AAAAc               #     K   t         r>t        j                         5  t        j                  ddt               d  d d d        y d  y # 1 sw Y   y xY ww)Nr[   zThe default of observed=Falser\   rm   r5   r   r   check_observed_deprecationrq      sO     $$& 	##7&
 	 	 		 	   A!AAAAc               #     K   t         r>t        j                         5  t        j                  ddt               d  d d d        y d  y # 1 sw Y   y xY ww)Nr[   z#The 'axis' keyword|Support for axisr\   rm   r5   r   r   check_axis_keyword_deprecationrt      sO     $$& 	##=&
 	 	 		 	rr   c               #     K   t         r>t        j                         5  t        j                  ddt               d  d d d        y d  y # 1 sw Y   y xY ww)Nr[   zthe convert_dtype parameterr\   rm   r5   r   r   check_convert_dtype_deprecationrv      sO     $$& 	##5&
 	 	 		 	rr   c              #     K   t         r@| r>t        j                         5  t        j                  ddt               d  d d d        y d  y # 1 sw Y   y xY ww)Nr[   z0.*DatetimeProperties.to_pydatetime is deprecatedr\   rm   )catch_deprecation_warningss    r   check_to_pydatetime_deprecationry      sR     3$$& 	##J&
 	 	 		 	s   A!A AAAc               #     K   t         r>t        j                         5  t        j                  ddt               d  d d d        y d  y # 1 sw Y   y xY ww)Nr[   zReturning a DataFramer\   rm   r5   r   r   !check_apply_dataframe_deprecationr{      sO     $$& 	##/&
 	 	 		 	rr   c               #     K   t         r>t        j                         5  t        j                  ddt               d  d d d        y d  y # 1 sw Y   y xY ww)Nr[   z&DataFrame.applymap has been deprecatedr\   rm   r5   r   r   $check_applymap_dataframe_deprecationr}      sO     $$& 	##@&
 	 	 		 	rr   c               #     K   t         rDt        s>t        j                         5  t        j                  ddt
               d  d d d        y d  y # 1 sw Y   y xY ww)Nr[   zMinvalid value encountered in double_scalars|Degrees of freedom <= 0 for slicer\   )r`   PANDAS_GE_201ra   rb   rc   RuntimeWarningr5   r   r    check_reductions_runtime_warningr      sQ     ]$$& 	##g'
 	 	 		 	s   !A!AAAAc                    	 t         j                  j                  j                  |       S # t        $ r+ ddlm}m}m}  ||       xr  ||        xr	  ||        cY S w xY w)Nr   )is_bool_dtypeis_complex_dtypeis_numeric_dtype)	r   apitypesis_any_real_numeric_dtypeAttributeErrorpandas.api.typesr   r   r   )arr_or_dtyper   r   r   s       r   r   r      sf    
 vv||55lCC 
VV \* 0$\220!,//	

s   (+ 1AAc                    t        | d      r| j                  }n| }t        s*t        j                  j
                  j                  |d      S t        j                  j
                  j                  |      S )Nr   r+   )hasattrr   r`   r   r   r   is_dtype_equalis_string_dtype)r   r   s     r   r   r     sU     |W%""vv||**5(;;66<<''..r   )r"   rL   )r"   rN   )NF)rf   bool)rx   r   )returnr   )7
__future__r   
contextlibr+   ra   numpyr   pandasr   packaging.versionr   __version__PANDAS_VERSIONPANDAS_GE_131PANDAS_GE_133PANDAS_GE_140r_   majorr`   releaser   PANDAS_GE_202rn   PANDAS_GE_211PANDAS_GE_220PANDAS_GE_300pandas.testingr   r	   r   r    r/   r2   r6   r1   rA   rG   rQ   contextmanagerrh   rk   ro   rq   rt   rv   ry   r{   r}   r   errorsIndexingErrorcoreindexingr   r   r5   r   r   <module>r      s=   "      %(''"22''"22''"22''"22$$)&&)3&&)3&&)3&&)3&&)3$$) +;A$7=B	  0   
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 II++MGG$$22M

/r   