o
    ҷhFq                     @   s  d 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jjddZejjddgddggdd	gd
ZddgifZdg diefZg ejejdgddeddfejdgfedg fejdgfeddgfejdgfeddfejdgfedfejdgfejdddfejdgfedfejddgfedfejddgfedfejddgfedfejddgfeddd fejddgfeddfejddgfedddgfejddgfed ddgfejddgfed!dgfejedejgfed"dfejddgfed#ddifejddgfed$fejddgfed%ddgfejddgfed&ddfejddgfed'fejddgfed(fejddgfed)fejddgfed*fejdgfed+edgfejdgejd,dd-fed.fejdgejd,dd-fed/fejejeed0ejdgd1fejjd2dd3ejeed4fejeeddfejeeddgfejeededgfejd5dgifeddgfejeed6d7fejeejd8d9d:d;fejeejd<d=d>fejeejd?dd@fejeeddgfejeed ddgfejeejd!dgdAfejeejd!dgd1fejeejdddidAfejeejddBd d1fejeed"dfejeejd"dCdDfejeedEdfejeedfejeed'fejeed(fejeed)fejeed*fejeedFfejeedfejeedfejeejdGddHfejeedIfejeedJddfejeedKddfejeedLfejeed+eje fejeedMeje ejfejeedNeje fejejeedOeje fed3ejeejdPddAfejdgdgdQfejdRddAfejdgdgdQfejdRddSdTgdUfejeedVfejeedWdfejeedXfejg dYg dZg d[d\fejd]dgd	gd^fejeed_d`d fejejeedaeddgifed3ejeeddfejeedbfejejeedcfejddgd3ejeedeeje fejeedffejeedgfejeedhfejeedifejeedjfejdgedjfejeedkfejeejdlddmfejeejdldndogddpfejdejddqejddqgifejdldrdmfejdedsedtgifejdlddmfejddgied,dugfed.fejddgie d,gfejd/dudvfejeed%dgfejeed%edgfejeed%eddgifejeedwdfejeedxdfejeejdydzd{fejddgfedyfejddgfejd|dd1fejeejd|ddAfejeej!fejdgej!fejeej"fejdgej"fejeej#fejdgej#fejee$fejdge$fejee%fejdge%fejeedddgfejeed}dfejdefed}dfejeed~dfejeededg difejeedeg dfejeeddfejeeddfejdddgfeddfejdddgfeddfejddgfedGfejdgd fedfejddgd ifedfejdgd fedfejddgd ifedfejddgfejddddfejefejddddfejddgfede&fejeede&fejddgfedfejeedfejddgde'fedfejdejddge'difedfejddgfedfejeedfejg dfedfejdg difedfejddgfejdddfejeejdddfejdejd,dd-feddfejdg diejd,dd-feddfejdejd,dd-feddfejdg diejd,dd-feddfejdejd,dd-fedddfejdg diejd,dd-fedddfejdejd,dd-feddfejdg diejd,dd-feddfejddgfedfejeedfejddgfededdrgfejeededggfejddgfededdrgfejeededggfejddgfejdddfejeejdddfejdejd,dddfeddfejdg diejd,dddfeddfejdejd,dd-feddfejdg diejd,dd-feddfejddgfedfejeedfejddgfedfejeedfejdgfeddd fejeeddd fejdgfeddd fejeeddd fejdgfedfejeedfejdgfedfejeedfejdgfedfejeedfejdgfedfejeedfejeedfejeedfejeedfejeedfejeedTfejeedfejeedSfejeedfejeedfejeedfejeedfZ(dd Z)ej*e(dd ddd Z+ejddáddń Z,ej-dedejd,dd-edg diejd,dd-gddȄ Z.ej-dedejd,dd-edg diejd,dd-gddʄ Z/edd̄ Z0ej-dg d΢ejj-ddedgfdeddgifedgdfeddgidfedgedgfeddgieddgifedgeddgifeddgiedgfgdd dэddӄ Z1ejj-dg edաed֡eddgeddedfdeddڡeddeddݡejdddrdލeddedded~deddededdedededdeddeddeddedded#ddeddeddeddeddededdejddddeddedededddiededdeddedededededed edededede)dэdd Z2ejj-ded/eddededdedded	ded
dededg	e)dэdd Z3ej-dg ddd Z4ej-dg ddd Z5ej-dedgdd Z6ej-deddgededdd eddeded dd	d!ed"ddged#dd	gg	ed$d% Z7ej-d&eddgeddgddgdQgej-dedTd'd d(d d)d d*d gd+d, Z8ej-d&eddgeddgddgdQgej-dԐd-d d.d d/d d0d d1d d2d ged3d4 Z9d5d6 Z:dS (7  zG
An exhaustive list of pandas methods exercising NDFrame.__finalize__.
    Nznot implementedreasonab   AB)names)r            float64dtypeviewint64take__getitem__Trepeatr
   reset_index)dropto_framedrop_duplicates
duplicatedroundrenamec                 C   s   | d S )Nr    xr   r   U/var/www/html/venv/lib/python3.10/site-packages/pandas/tests/generic/test_finalize.py<lambda>0   s    r    nameset_axisreindexr   fillnareplaceshiftisinbetweenisnaisnullnotnanotnulladd2000periodsto_timestamp	to_perioddot)indexzImplement binary finalize)marks	transpose)r   r   queryzA == 1evalA + 1pythonengineselect_dtypesint)includeassign)r   )columnsc                 C      | S Nr   r   r   r   r   r    `       ffill)method	set_indexdropnasort_values)by
sort_indexnlargest	nsmallest	swaplevelcombinecombine_firstupdatepivot)r   r   pivot_tablemeansum)rA   aggfuncstackexplodeunstack)r   r   c)r   r      )r
   r      )r   r   Cmelt)id_vars
value_varsmapc                 C   rB   rC   r   r   r   r   r   r       rD   mergecorrcovzignore::RuntimeWarningcorrwithcountnuniqueidxminidxmaxmodemedianquantile)numeric_onlyg      ?g      ?)qrm   )daysFz
2022-01-01z
2022-01-02D)freq	droplevelpopsqueezerA   )axisrename_axisxsgetreindex_like)r   r
   r   )r   r   r
   
add_prefix_
add_suffixr   
   headtailsample)nr%   astypecopyinfer_objectsconvert_dtypes)r   Nr   interpolateclip)lowerr   asfreqH)r   r   r   r   at_timez12:00between_timez13:00last3Drankwheremasktruncate)beforeUTC)r0   tz
tz_convertCETtz_localizedescribe
pct_change	transformc                 C      | |    S rC   minr   r   r   r   r    a      c                 C   r   rC   r   r   r   r   r   r    e  r   applyc                 C   rB   rC   r   r   r   r   r   r    g  rD   c                 C   rB   rC   r   r   r   r   r   r    h  rD   cumsumcummincummaxcumprodanyallr   maxstdprodsemskewkurtc                 C   s.   t d}|t| }|r|dS t| S )Nz'(.*)?'r   )recompilesearchstrgroup)r   xprmr   r   r   idfn  s
   

r   c                 C   s   t | d S )N)r   r   r   r   r   r      r   )paramsidsc                 C   s   | j S )z1
    An NDFrame method returning an NDFrame.
    )param)requestr   r   r   ndframe_method  s   r   zAignore:DataFrame.fillna with 'method' is deprecated:FutureWarningz'ignore:last is deprecated:FutureWarningc                 C   s:   | \}}}|| }ddi|_ ||}|j ddiksJ d S )Nr   r   )attrs)r   cls	init_argsrF   ndframeresultr   r   r   test_finalize_called  s
   

r   datac                 C   ^   d}ddi| _ tjt|d | d}|j ddiksJ W d    d S 1 s(w   Y  d S )Nzfirst is deprecatedr   r   matchr   )r   tmassert_produces_warningFutureWarningfirstr   deprecated_msgr   r   r   r   test_finalize_first  s   

"r   c                 C   r   )Nzlast is deprecatedr   r   r   r   )r   r   r   r   r   r   r   r   r   test_finalize_last  s   	

"r   c                  C   sJ   t d tdddgi} d| jd< | jddd}|jddiks#J d S )Nnumexprr   r   r
   r9   r;   )pytestimportorskippd	DataFramer   r8   dfr   r   r   r   !test_finalize_called_eval_numexpr  s
   

r   annotate)leftrightbothargsc                 C   s&   dt | d j dt | d j dS )N(r   ,r   ))type__name__r   r   r   r   r      s   & )r   c                 C   s  |\}}t |tjtjfri |_t |tjtjfri |_|dkr*t |tr*td |dkr8t |tr8td t |tst |ts|dkr|j	ds|dkrit |t
|ri| jtjj| dt
| d t |t
|s|dkrt |tjr| jtjj| d	d nn|dkrt |tjr| jtjj| d	d nU|dkrt |t
|r| jtjj| d
t
| d t |t
|s|dkrt |tjr| jtjj| d	d n|dkrt |tjr| jtjj| d	d |dv rt |tsddi|_|dv rt |tsddi|_|tjtjtjtjtjtjfv }|rCt |tjrCt |tjrC|j|ddd\}}n|r^t |tjr^t |tjr^|j|ddd\}}|||}|jddiksmJ d S )Nr   z)left is an int and doesn't support .attrsr   z*right is an int and doesn't support .attrsr   rz0 doesn't work when right has attrs and both are r   z= doesn't work when the objects are different Series has attrsz/ doesn't work when left has attrs and both are >   r   r   r   r   >   r   r   F)ru   r   )
isinstancer   r   Seriesr   r>   r   skipr   
startswithr   node
add_markermarkxfailoperatoreqnegtgeltlealign)r   r   r   all_binary_operatorsr   r   is_cmpr   r   r   r   test_binops  s   



""
r   rF   
capitalizecasefoldcatcontainsencodezutf-8endswithextractz(\w)(\d))expandfindfindallr4   lenljustr   lstripr   z\w	normalizeNFCpad	partitionrfindrindexrjust
rpartitionrstripsliceslice_replace)replr   stripswapcase	translateupperwrapzfillisalnumisalphaisdigitisspaceislowerisupperistitle	isnumeric	isdecimalget_dummiesc                 C   s6   t dg}ddi|_| |j}|jddiksJ d S )Na1r   r   )r   r   r   r   rF   sr   r   r   r   test_string_method4  s   6

r  strftimez%Yfloorceil
month_nameday_namec                 C   >   t t jddd}ddi|_| |j}|jddiksJ d S Nr.   r   r/   r   r   )r   r   
date_ranger   dtr  r   r   r   test_datetime_methodp  s   

r)  attr)datetimetimetzyearmonthdayhourminutesecondmicrosecond
nanosecond	dayofweekday_of_week	dayofyearday_of_yearquarteris_month_startis_month_endis_quarter_startis_quarter_endis_year_startis_year_endis_leap_yeardaysinmonthdays_in_monthc                 C   @   t t jddd}ddi|_t|j| }|jddiksJ d S r&  )r   r   r'  r   getattrr(  r*  r  r   r   r   r   test_datetime_property  s   
rG  )ro   secondsmicrosecondsnanoseconds
componentsc                 C   rD  r&  )r   r   timedelta_ranger   rE  r(  rF  r   r   r   test_timedelta_property  s   
rM  total_secondsc                 C   r%  r&  )r   r   rL  r   r(  r  r   r   r   test_timedelta_methods  s   

rO  add_categoriesrZ   
as_orderedas_unorderedc                 C   s
   t | dS )Ncodes)rE  r   r   r   r   r         
 remove_categoriesremove_unused_categoriesrename_categories)r   r   reorder_categoriesset_categoriesc                 C   s<   t jddgdd}ddi|_| |j}|jddiksJ d S )Nr   r   categoryr   r   )r   r   r   r   r  r   r   r   test_categorical_accessor  s   

r[  objc                 C   s   |  dd S )Nc                 S   rB   rC   r   )yr   r   r   r      rD   z<lambda>.<locals>.<lambda>)r   r   r   r   r   r          c                 C   
   |  dS )NrU   aggr   r   r   r   r      rT  c                 C   r_  )NrT   r`  r   r   r   r   r      rT  c                 C   r_  )Nrk   r`  r   r   r   r   r      rT  c                 C   s6   ddi| _ || jddgdd}|j ddiksJ d S )Nr   r   r   F)
group_keysr   groupbyr\  rF   r   r   r   r   test_groupby_finalize  s   
rf  c                 C   s   |  ddgS )NrU   rf   r`  r   r   r   r   r      r^  c                 C   r_  )Nr   r`  r   r   r   r   r      rT  c                 C   r_  )Nvarr`  r   r   r   r   r      rT  c                 C   r_  )Nr   r`  r   r   r   r   r      rT  c                 C   r_  )Nsizer`  r   r   r   r   r      rT  c                 C   r_  )Nohlcr`  r   r   r   r   r      rT  c                 C   s2   ddi| _ || ddg}|j ddiksJ d S )Nr   r   r   rc  re  r   r   r   %test_groupby_finalize_not_implemented  s   
rj  c                  C   s8   t dddgi} t ddg| }|jd u sJ d S )Nr!   r   r
   )r   r   r   __finalize__r!   r   r   r   r   test_finalize_frame_series_name  s   rl  );__doc__r   r   numpynpr   pandasr   pandas._testing_testingr   r   r   not_implemented_mark
MultiIndexfrom_productmi
frame_dataframe_mi_datar   arraymethodcallerNAperiod_ranger'  r   r   r-   filterwarnings	Timedelta
datetime64Period	Timestampnegposinvabsr   floatobject_all_methodsr   fixturer   r   parametrizer   r   r   r   r  r)  rG  rM  rO  r[  rf  rj  rl  r   r   r   r   <module>   s   	 
	
  %)123456789:;<=>?@ABCDEFGHIJKLMNOQW\`hjostuw{|     
                      $  )  .  3  7  8  :  >  ?  A  B  C  D  F  G  H  I  J  K  
L  M  
N  O  P  Q  R  S  U  Z  ^  _  `  a  b  c  d  e  f  g  h  i  j  k  l  m  o  s  t  u  v  w  x  z
  
   
    

            
    
    "    #    "$    %    "&    '    (    )    +
    0
    5
    :
    >    ?    @    A    B    D    H    I    K    L    M    N    O    P    Q    R    T    U    V    W    X    Y    Z    [    \    ]    ^    b	



	



W




	
















 !"#
$
%&'()*
+
,
-
.
/15






(
(