Ë
    ¤eh½P  ã                   ó6  — d Z ddlZddlmZ ddlZddlZddlm	Z	 ddl
Zddl
mZmZmZmZmZmZmZ ddlmZ ddlmZ d„ Zd„ Zd	„ Zd
„ Zd„ Zd„ Zd„ Zd„ Zd„ Z d„ Z!d„ Z"d„ Z#d„ Z$d„ Z%d„ Z&d„ Z'ejP                  jS                  dddg«      d„ «       Z*d„ Z+d„ Z,ejP                  jS                  de- eddd d!œi«      fe. edg d¢g d ¢d!œi«      fd"„  eddd d!œi«      fd#„  edg d¢g d ¢d!œi«      fg«      d$„ «       Z/ejP                  jS                  de- edd gd%d&gd¬'«      fe. eg d¢g d ¢gd%d&gd¬'«      fd(„  edd gd%d&gd¬'«      fd)„  eg d¢g d ¢gd%d&gd¬'«      fg«      d*„ «       Z0d+„ Z1d,„ Z2d-„ Z3ejP                  jS                  d.e4e5e6e7e8e9e:e;g«      d/„ «       Z<y)0z
test all other .agg behavior
é    N)Úpartial)ÚSpecificationError)Ú	DataFrameÚIndexÚ
MultiIndexÚPeriodIndexÚSeriesÚ
date_rangeÚperiod_range)Úpprint_thingc                  ó
  — t        t        j                  j                  d«      j	                  d«      t        j                  j                  d«      j	                  d«      g d¢g d¢dœ«      } | j                  d«      }d„ }t        j                  t        d¬	«      5  |j                  |g«       d d d «       t        j                  t        d¬	«      5  |j                  |«       d d d «       y # 1 sw Y   Œ?xY w# 1 sw Y   y xY w)
Né   é   )Úar   Úbr   r   )ÚoneÚtwor   r   r   )Údata1Údata2Úkey1Úkey2r   c                 óD   — | j                  «       | j                  «       z
  S ©N)ÚmaxÚmin)Úarrs    úg/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/pandas/tests/groupby/aggregate/test_other.pyÚpeak_to_peakz5test_agg_partial_failure_raises.<locals>.peak_to_peak)   s   € Ø�w‰w‹y˜3Ÿ7™7›9Ñ$Ð$ó    zunsupported operand type©Úmatch)
r   ÚnpÚrandomÚdefault_rngÚstandard_normalÚgroupbyÚpytestÚraisesÚ	TypeErrorÚagg)ÚdfÚgroupedr   s      r   Útest_agg_partial_failure_raisesr-      sÓ   € ô 
ä—Y‘Y×*Ñ*¨1Ó-×=Ñ=¸aÓ@Ü—Y‘Y×*Ñ*¨1Ó-×=Ñ=¸aÓ@Ú-Ú7ñ		
ó
€Bð �j‰j˜Ó €Gò%ô 
�‰”yÐ(BÔ	Cñ $Ø�‰�\�NÔ#÷$ô 
�‰”yÐ(BÔ	Cñ "Ø�‰�LÔ!÷"ð "÷$ð $ú÷"ð "ús   ÂC-ÃC9Ã-C6Ã9Dc            	      ó&  — g d¢g d¢g d¢g} t        | D �cg c]  }|d   ‘Œ	 c}| D �cg c]  }|d   ‘Œ	 c}| D �cg c]  }|d   ‘Œ	 c}dœ«      }| D �cg c]C  }|d   |d   r1t        j                  j                  |d   d«      j	                  «       nd |d   g‘ŒE } }t        | D �cg c]  }|d   ‘Œ	 c}| D �cg c]  }|d   ‘Œ	 c}| D �cg c]  }|d   ‘Œ	 c}dœ«      }|d	   |d	   j                  «       z  |d
<   |j                  d«      j                  d«      }|d	   |d	   j                  «       z  |d
<   |j                  d«      j                  d«      }t        |«      t        |«      k(  sJ ‚y c c}w c c}w c c}w c c}w c c}w c c}w c c}w )N)é   ú
2012-01-01ç      ð?)r   z
2012-01-02g       @)é   Ng      @r   r/   r   )ÚkeyÚdateÚvaluez%Y-%m-%dr5   Úweightsr4   Úsum)	r   ÚdtÚdatetimeÚstrptimer4   r7   r&   Ú	aggregateÚlen)ÚdataÚxÚdf1ÚrowÚdf2Úgb1Úgb2s          r   Útest_agg_datetimes_mixedrD   3   s˜  € Ú"Ò$:ºNÐK€Dä
à"&Ö'˜Q�A�a“DÒ'Ø#'Ö(˜a�Q�q“TÒ(Ø$(Ö)˜q�a˜“dÒ)ñ	
ó€Cð öð ð	 �‰FØ@CÀAÂŒR�[‰[×!Ñ! # a¡&¨*Ó5×:Ñ:Ô<ÈDØ�‰Fò	
ð€Dð ô à"&Ö'˜Q�A�a“DÒ'Ø#'Ö(˜a�Q�q“TÒ(Ø$(Ö)˜q�a˜“dÒ)ñ	
ó€Cð ˜‘\ C¨¡L×$4Ñ$4Ó$6Ñ6€Cˆ	�NØ
�+‰+�fÓ
×
'Ñ
'¨Ó
.€Cà˜‘\ C¨¡L×$4Ñ$4Ó$6Ñ6€Cˆ	�NØ
�+‰+�fÓ
×
'Ñ
'¨Ó
.€Cäˆs‹8”s˜3“xÒÐÑùò; (ùÚ(ùÚ)ùòùò (ùÚ(ùÚ)s(   •E0§E5
¹E:ÁAE?Â%FÂ7F	
Ã	Fc                  ó¾  — t        ddd¬«      } t        t        j                  j	                  d«      j                  d«      | ¬«      }|j                  d¬	«      j                  «       }t        |j                  t        «      sJ ‚t        d
dd¬«      }t        t        j                  j	                  d«      j                  t        |«      «      |¬«      }t        t        j                  j	                  d«      j                  t        |«      «      |¬«      }t        j                  ||dœ«      }|j                  |j                  j                  «      }t        |«       y )Nz2012-1-1ÚMr2   )ÚfreqÚperiodsr   )r2   r   ©Úindexr   )Úlevelz1999-01r   )ÚstartrH   rG   )Ús1Ús2)r   r   r"   r#   r$   r%   r&   r7   Ú
isinstancerJ   r   r	   r<   Ú	from_dictÚmonthÚlist)Úprngr+   ÚrsrJ   rM   rN   r,   s          r   Útest_agg_period_indexrU   X   sù   € Ü˜
¨°aÔ8€DÜ	”2—9‘9×(Ñ(¨Ó+×;Ñ;¸FÓCÈ4Ô	P€BØ	�‰˜!ˆÓ	×	 Ñ	 Ó	"€BÜ�b—h‘h¤Ô,Ð,Ð,ô ˜y°!¸#Ô>€EÜ	”—	‘	×%Ñ% aÓ(×/Ñ/´°E³
Ó;À5Ô	I€BÜ	”—	‘	×%Ñ% aÓ(×/Ñ/´°E³
Ó;À5Ô	I€BÜ	×	Ñ	 B¨bÑ1Ó	2€BØ�j‰j˜Ÿ™Ÿ™Ó(€GÜˆ…Mr   c                  ó   — t        g d¢t        ddd¬«      dœ«      } d | j                  g d¢df<   | j                  g d	¢   j                  d
«      }| j	                  d
«      }t        j                  |j                  «       |«       t        j                  |j                  d«      |«       t        j                  |j                  ddi«      |«       t        j                  |j                  j                  «       |d   «       t        j                  |j                  j                  d«      |d   «       | j                  g d¢   j                  d
«      }| j	                  d
«      }t        j                  |j                  «       |«       t        j                  |j                  d«      |«       t        j                  |j                  ddi«      |«       t        j                  |j                  j                  «       |d   «       t        j                  |j                  j                  d«      |d   «       t        g d¢t        t        d«      d
¬«      d¬«      }t        j                  |j                  j                  t        «      |«       t        j                  |j                  j!                  «       |«       t        g d¢t        t        d«      d
¬«      d¬«      }t        j                  |j                  j#                  «       |«       y )N)ÚArW   ÚBrX   ÚCrY   ÚDrZ   z1/1/2011é   Úh)rH   rG   )ÚclassÚtime)r   r/   r   r   r^   )r   r2   é   é   r]   Úfirst)r   r2   r_   é   Úlast)r   r   r   r   ÚABCD©Úname©rJ   rf   )r   r/   r/   r   )r   r
   ÚlocÚ	set_indexr&   ÚtmÚassert_frame_equalra   r*   Úassert_series_equalr^   rc   r	   r   rR   r<   ÚsizeÚcount)r+   Úexpr,   s      r   Ú*test_agg_dict_parameter_cast_result_dtypesrp   g   s8  € ô 
â=Ü˜z°1¸3Ô?ñ	
ó
€Bð $(€B‡F�FŠ<˜ÐÑ ð �&‰&’Ñ
×
(Ñ
(¨Ó
1€CØ�j‰j˜Ó!€GÜ×Ñ˜'Ÿ-™-›/¨3Ô/Ü×Ñ˜'Ÿ+™+ gÓ.°Ô4Ü×Ñ˜'Ÿ+™+ v¨wÐ&7Ó8¸#Ô>Ü×Ñ˜7Ÿ<™<×-Ñ-Ó/°°V±Ô=Ü×Ñ˜7Ÿ<™<×+Ñ+¨GÓ4°c¸&±kÔBð �&‰&’Ñ
×
(Ñ
(¨Ó
1€CØ�j‰j˜Ó!€GÜ×Ñ˜'Ÿ,™,›.¨#Ô.Ü×Ñ˜'Ÿ+™+ fÓ-¨sÔ3Ü×Ñ˜'Ÿ+™+ v¨vÐ&6Ó7¸Ô=Ü×Ñ˜7Ÿ<™<×,Ñ,Ó.°°F±Ô<Ü×Ñ˜7Ÿ<™<×+Ñ+¨FÓ3°S¸±[ÔAô ’¤U¬4°«<¸gÔ%FÈVÔ
T€CÜ×Ñ˜7Ÿ<™<×+Ñ+¬CÓ0°#Ô6Ü×Ñ˜7Ÿ<™<×,Ñ,Ó.°Ô4ä
’¤U¬4°«<¸gÔ%FÈVÔ
T€CÜ×Ñ˜7Ÿ<™<×-Ñ-Ó/°Õ5r   c                  ó^  — t        d«      D � cg c]  } t        j                  d| dz   d«      ‘Œ }} t        d«      }t	        ||dœ«      }|j                  d«      d   j                  t        «      }|j                  d«      d   j                  «       }t        j                  ||«       y c c} w )Né   iß  r/   Úaaabbbbbbccd)ÚXÚYrt   ru   )Úranger8   r9   rR   r   r&   r*   r<   rn   rj   rl   )r>   ÚuÚvr+   ÚresultÚexpecteds         r   Útest_agg_cast_results_dtypesr{   �   s’   € ô /4°B«iÖ8¨Œ�‰�T˜1˜q™5 !Õ	$Ð8€AÐ8Üˆ^Ó€AÜ	˜ Ñ#Ó	$€Bà�Z‰Z˜‹_˜SÑ!×%Ñ%¤cÓ*€FØ�z‰z˜#‹˜sÑ#×)Ñ)Ó+€HÜ×Ñ˜6 8Õ,ùò 	9s   Ž!B*c                  óª  — t        g d¢g d¢g d¢dœ«      } t        dg d¢ig d¢¬«      }d|j                  _        | j                  d«      dg   j	                  «       }t        j                  ||«       t        g d¢g d¢d	œg d¢¬«      }d|j                  _        | j                  d«      dd
g   j	                  «       }t        j                  ||«       y )N)r/   r   r2   r_   r   )r/   r   r   r_   r   ©r   r   Úcr   )r/   g      @r_   r   )r/   r   r_   r   rI   r   )r   r~   r~   )r   rJ   rf   r&   Úmeanrj   rk   )r+   rz   ry   s      r   Útest_aggregate_float64_no_int64r€   ™   s©   € ä	šªÂ_ÑUÓ	V€Bä˜#š~Ð.²lÔC€HØ€H‡N�NÔà�Z‰Z˜‹_˜c˜UÑ#×(Ñ(Ó*€FÜ×Ñ˜& (Ô+äš~²NÑCÊ<ÔX€HØ€H‡N�NÔà�Z‰Z˜‹_˜c 3˜ZÑ(×-Ñ-Ó/€FÜ×Ñ˜& (Õ+r   c                  óv  — t        g d¢g d¢t        j                  j                  d«      j	                  d«      dz   t        j
                  d«      dœ«      } | j                  ddg«      }|d	   j                  «       }|d	   j                  «       }|d
   j                  «       }|d
   j                  «       }|d
   j                  ddg«      }t        j                  ||gd¬«      }ddg|_        t        j                  ||d¬«       |j                  ddg«      }t        j                  ||||gd¬«      }t        j                   d	d
gddgg«      |_        t        j                  ||d¬«       |d
d	g   j                  ddg«      }t        j                  ||||gd¬«      }t        j                   d
d	gddgg«      |_        t        j                  ||d¬«       |j                  dddœ«      }t        j                  ||gd¬«      }t        j                  ||d¬«       |j                  ddgddgdœ«      }t        j                  ||||gd¬«      }t        j                   d	d
gddgg«      |_        d}t#        j$                  t&        |¬«      5  |d
d	g   j                  dddœ«       d d d «       y # 1 sw Y   y xY w)N©ÚfooÚbarrƒ   r„   rƒ   r„   rƒ   rƒ   ©r   r   r   r   r   r   r   r   r   r[   r1   ©rW   rX   rY   rZ   rW   rX   rY   rZ   r7   r   r/   )ÚaxisT)Ú
check_like©rY   rZ   z&Column\(s\) \['r', 'r2'\] do not existr    )ÚrÚr2)r   r"   r#   r$   r%   Úaranger&   r   r7   r*   ÚpdÚconcatÚcolumnsrj   rk   r   Úfrom_productr'   r(   ÚKeyError)	r+   r,   Úc_meanÚc_sumÚd_meanÚd_sumry   rz   Úmsgs	            r   Útest_aggregate_api_consistencyr—   ª   s‚  € ô 
âIÚIÜ—‘×&Ñ& qÓ)×9Ñ9¸!Ó<¸sÑBÜ—‘˜1“ñ		
ó
€Bð �j‰j˜#˜s˜Ó$€GØ�S‰\×ÑÓ €FØ�C‰L×ÑÓ€EØ�S‰\×ÑÓ €FØ�C‰L×ÑÓ€Eà�S‰\×Ñ˜u f˜oÓ.€FÜ�y‰y˜% ˜¨qÔ1€HØ˜v�€HÔÜ×Ñ˜& (°tÕ<à�[‰[˜% ˜Ó)€FÜ�y‰y˜% ¨°Ð7¸aÔ@€HÜ!×.Ñ.°°c°
¸UÀF¸OÐ/LÓM€HÔÜ×Ñ˜& (°tÕ<à�c˜3�ZÑ ×$Ñ$ e¨V _Ó5€FÜ�y‰y˜% ¨°Ð7¸aÔ@€HÜ!×.Ñ.°°c°
¸UÀF¸OÐ/LÓM€HÔÜ×Ñ˜& (°tÕ<à�[‰[˜v¨EÑ2Ó3€FÜ�y‰y˜% ˜¨qÔ1€HÜ×Ñ˜& (°tÕ<à�[‰[ ¨˜°f¸e°_ÑEÓF€FÜ�y‰y˜& %¨°Ð7¸aÔ@€HÜ!×.Ñ.°°c°
¸VÀU¸OÐ/LÓM€HÔà
3€CÜ	�‰”x sÔ	+ñ <Ø��c�
Ñ×Ñ e°6Ñ :Ô;÷<÷ <ñ <ús   ÊJ/Ê/J8c                  ór  — t        g d¢t        d«      t        d«      dœ«      } d}t        j                  t        |¬«      5  | j                  d«      j                  ddd	gid
ddgidœ«       d d d «       d}t        j                  t        |¬«      5  | j                  d«      ddg   j                  dd	i«       d d d «       d}t        j                  t        |¬«      5  | j                  d«      j                  j                  ddi«       d d d «       y # 1 sw Y   Œ©xY w# 1 sw Y   ŒhxY w# 1 sw Y   y xY w)N)r/   r/   r/   r   r   r   ©rW   rX   rY   únested renamer is not supportedr    rW   rƒ   r7   r   r„   rn   r   )rX   rY   z!Column\(s\) \['ma'\] do not existrX   rY   Úma)	r   rv   r'   r(   r   r&   r*   r‘   rX   )r+   r–   s     r   Ú"test_agg_dict_renaming_deprecationrœ   Ù   s  € ä	š¬u°Q«x¼eÀA»hÑGÓ	H€Bà
,€CÜ	�‰Ô)°Ô	5ñ 
Ø
�
‰
�3‹×ÑØ˜5 %˜.Ð)°¸À%Ð8HÐ0IÑJô	
÷
ð
 /€CÜ	�‰”x sÔ	+ñ 7Ø
�
‰
�3‹˜˜c˜
Ñ#×'Ñ'¨¨u¨Ô6÷7ð -€CÜ	�‰Ô)°Ô	5ñ 0Ø
�
‰
�3‹×Ñ×Ñ˜u gÐ.Ô/÷0ð 0÷
ð 
ú÷7ð 7ú÷0ð 0ús$   Á,DÂ(D!Ã-D-ÄDÄ!D*Ä-D6c                  óø  — t        g d¢g d¢t        j                  j                  d«      j	                  d«      dz   t        j
                  d«      dœ«      } | j                  ddg«      }d	}t        j                  t        |¬
«      5  |d   j                  dddgi«       d d d «       t        j                  t        |¬
«      5  |d   j                  dddœ«       d d d «       y # 1 sw Y   ŒExY w# 1 sw Y   y xY w)Nr‚   r…   r   r[   r1   r†   rW   rX   rš   r    rZ   rY   r7   Ústdr‰   )r   r"   r#   r$   r%   rŒ   r&   r'   r(   r   r*   ©r+   Úgr–   s      r   Útest_agg_compatr¡   ì   sÚ   € ä	âIÚIÜ—‘×&Ñ& qÓ)×9Ñ9¸!Ó<¸sÑBÜ—‘˜1“ñ		
ó
€Bð 	�
‰
�C˜�:Ó€Aà
,€CÜ	�‰Ô)°Ô	5ñ *Ø	ˆ#‰�
‰
�C˜% ˜Ð(Ô)÷*ô 
�‰Ô)°Ô	5ñ -Ø	ˆ#‰�
‰
˜ UÑ+Ô,÷-ð -÷*ð *ú÷-ð -ús   ÂC$ÃC0Ã$C-Ã0C9c                  ó–  — t        g d¢g d¢t        j                  j                  d«      j	                  d«      dz   t        j
                  d«      dœ«      } | j                  ddg«      }d	}t        j                  t        |¬
«      5  |j                  dddgidddgidœ«       d d d «       t        j                  t        |¬
«      5  |j                  dddgidddgidœ«       d d d «       t        j                  t        |¬
«      5  |d   j                  t        j                  t        j                  dœ«       d d d «       t        j                  t        |¬
«      5  |d   j                  t        j                  t        j                  dœ«       d d d «       y # 1 sw Y   ŒøxY w# 1 sw Y   ŒÄxY w# 1 sw Y   ŒyxY w# 1 sw Y   y xY w)Nr‚   r…   r   r[   r1   r†   rW   rX   rš   r    rY   r   r7   rZ   )Úr1r‹   Úrarž   Úrbr‰   )Úresult1Úresult2)rZ   r§   )r   r"   r#   r$   r%   rŒ   r&   r'   r(   r   r;   r*   r7   r   rŸ   s      r   Útest_agg_nested_dictsr¨     s€  € ä	âIÚIÜ—‘×&Ñ& qÓ)×9Ñ9¸!Ó<¸sÑBÜ—‘˜1“ñ		
ó
€Bð 	�
‰
�C˜�:Ó€Aà
,€CÜ	�‰Ô)°Ô	5ñ RØ	�‰˜C &¨% Ð1¸#ÀÈ¸Ð9OÑPÔQ÷Rô 
�‰Ô)°Ô	5ñ LØ	�‰�T˜F E˜?Ð+°4¸&À%¸Ð2IÑJÔK÷Lô
 
�‰Ô)°Ô	5ñ <Ø	ˆ#‰�
‰
œrŸv™v´"·'±'Ñ:Ô;÷<ô 
�‰Ô)°Ô	5ñ 6Ø	ˆ#‰�
‰
œŸ™¬B¯G©GÑ4Ô5÷6ð 6÷Rð Rú÷Lð Lú÷
<ð <ú÷6ð 6ús0   ÂFÃF'Ä4F3Å4F?ÆF$Æ'F0Æ3F<Æ?Gc                  ó  — t        t        j                  j                  d«      j	                  dd¬«      «      } d„ }t        j                  t        d¬«      5  | j                  d«      j                  |«       d d d «       y # 1 sw Y   y xY w)	Nr   é
   )é   rª   )rm   c                 ó`   — t        d«       t        | j                  «       «       t        d«      ‚)Nz(----------------------------------------Útest)r   Ú	to_stringr)   )r+   s    r   ÚraiseExceptionz=test_agg_item_by_item_raise_typeerror.<locals>.raiseException!  s$   € ÜÐ?Ô@Ü�R—\‘\“^Ô$Ü˜ÓÐr   r­   r    r   )
r   r"   r#   r$   Úintegersr'   r(   r)   r&   r*   )r+   r¯   s     r   Ú%test_agg_item_by_item_raise_typeerrorr±     sk   € Ü	”2—9‘9×(Ñ(¨Ó+×4Ñ4°R¸hÐ4ÓGÓ	H€Bò ô
 
�‰”y¨Ô	/ñ *Ø
�
‰
�1‹×Ñ˜.Ô)÷*÷ *ñ *ús   Á!BÂBc                  ó  — t        t        j                  dt        j                  ¬«      t	        dd¬«      ¬«      } | j                  d„ d„ g«      }|j                  d«      }|j                  «       }t        j                  ||«       y )	Nrª   ©Údtypez
2020-01-01)rH   rI   c                 ó   — | j                   S r   )Úyear©r>   s    r   ú<lambda>z*test_series_agg_multikey.<locals>.<lambda>.  s
   €  A§F¡F€ r   c                 ó   — | j                   S r   )rQ   r·   s    r   r¸   z*test_series_agg_multikey.<locals>.<lambda>.  s
   € °a·g±g€ r   r7   )
r	   r"   rŒ   Úfloat64r
   r&   r*   r7   rj   rl   )Útsr,   ry   rz   s       r   Útest_series_agg_multikeyr¼   *  sh   € Ü	Ü
�	‰	�"œBŸJ™JÔ'¬z¸,ÐPRÔ/Sô
€Bð �j‰jÑ*Ñ,=Ð>Ó?€Gà�[‰[˜Ó€FØ�{‰{‹}€HÜ×Ñ˜6 8Õ,r   c            
      óø  — t        g d¢g d¢g d¢t        j                  j                  d«      j	                  d«      t        j                  j                  d«      j	                  d«      t        j                  j                  d«      j	                  d«      dœ«      } d„ }| j                  dd	g«      j                  |«      }| j                  dd	g«      j                  d
„ «      }t        j                  ||«       y )N)rƒ   rƒ   rƒ   rƒ   r„   r„   r„   r„   rƒ   rƒ   rƒ   )r   r   r   r   r   r   r   r   r   r   r   )Údullr¾   Úshinyr¾   r¾   r¿   r¿   r¾   r¿   r¿   r¿   r   é   )rW   rX   rY   rZ   ÚEÚFc                 óL   — t        | j                  j                  «      dkD  sJ ‚y)Nr   rƒ   )r<   ÚvaluesÚbaser·   s    r   Úbadz.test_series_agg_multi_pure_python.<locals>.bade  s    € Ü�1—8‘8—=‘=Ó! AÒ%Ð%Ð%Ør   rW   rX   c                  ó   — y)Nrƒ   © r·   s    r   r¸   z3test_series_agg_multi_pure_python.<locals>.<lambda>j  s   � r   )	r   r"   r#   r$   r%   r&   r*   rj   rk   )r=   rÆ   ry   rz   s       r   Ú!test_series_agg_multi_pure_pythonrÉ   5  sÌ   € Üòòòô —‘×&Ñ& qÓ)×9Ñ9¸"Ó=Ü—‘×&Ñ& qÓ)×9Ñ9¸"Ó=Ü—‘×&Ñ& qÓ)×9Ñ9¸"Ó=ñU+	
ó-€Dò^ð �\‰\˜3 ˜*Ó%×)Ñ)¨#Ó.€FØ�|‰|˜S #˜JÓ'×+Ñ+©OÓ<€HÜ×Ñ˜& (Õ+r   c                  ó°  — d„ } t        g d¢g d¢t        j                  ddd«      t        j                  ddd«      t        j                  ddd«      t        j                  ddd«      gdœ«      }|j                  d	«      }|j	                  | g«      }|j
                  j                  d
   |_        |j	                  | «      }t        j                  ||«       y )Nc                 óL   — t        j                  | j                  «       d¬«      S )Nr/   )Úq)r"   Ú
percentileÚdropna)r   s    r   ÚP1z test_agg_consistency.<locals>.P1q  s   € Ü�}‰}˜QŸX™X›Z¨1Ô-Ð-r   )r/   r   r2   r_   )rª   é   é   é   iÝ  r   rª   rÀ   )Úcol1Úcol2r4   r4   r   )	r   r8   r4   r&   r*   r�   Úlevelsrj   rk   )rÏ   r+   r    rz   ry   s        r   Útest_agg_consistencyrÖ   n  s¸   € ò.ô 
â Ú$ä—‘˜˜a Ó$Ü—‘˜˜a Ó$Ü—‘˜˜a Ó$Ü—‘˜˜a Ó$ð	ñ		
ó
€Bð 	�
‰
�6Ó€Aà�u‰u�b�T‹{€HØ×'Ñ'×.Ñ.¨qÑ1€HÔà�U‰U�2‹Y€FÜ×Ñ˜& (Õ+r   c                  ó,  — t        ddgddgdœ«      j                  t        j                  «      }  G d„ d«      }t        t        j                  d„ d	„ t        t        «       |«       g}| j                  d
«      j                  d«      }|D ]|  }|t        u s|t        j                  u rt        nd }d}t        j                  ||¬«      5  | j                  d
«      j                  |«      }d d d «       t        j                  |«       Œ~ y # 1 sw Y   Œ"xY w)Nr/   r   r2   r_   )rƒ   r„   c                   ó   — e Zd Zd„ Zy)ú$test_agg_callables.<locals>.fn_classc                 ó   — t        |«      S r   ©r7   )Úselfr>   s     r   Ú__call__z-test_agg_callables.<locals>.fn_class.__call__�  s   € Ü�q“6ˆMr   N)Ú__name__Ú
__module__Ú__qualname__rÝ   rÈ   r   r   Úfn_classrÙ   Ž  s   „ ó	r   rá   c                 ó   — t        | «      S r   rÛ   r·   s    r   r¸   z$test_agg_callables.<locals>.<lambda>•  s
   € ”#�a“&€ r   c                 ó"   — | j                  «       S r   rÛ   r·   s    r   r¸   z$test_agg_callables.<locals>.<lambda>–  s   € �!—%‘%“'€ r   rƒ   r7   zusing DataFrameGroupBy.sumr    )r   Úastyper"   Úint64r7   r   r&   r*   ÚFutureWarningrj   Úassert_produces_warningrk   )r+   rá   Úequiv_callablesrz   ÚecallÚwarnr–   ry   s           r   Útest_agg_callablesrë   Š  së   € ä	˜A˜q˜6¨1¨a¨&Ñ1Ó	2×	9Ñ	9¼"¿(¹(Ó	C€B÷ñ ô
 	Ü
�‰ÙÙÜ”‹Ù‹
ð€Oð �z‰z˜%Ó ×$Ñ$ UÓ+€HØ ò 0ˆØ %¬¡°¼¿¹±�}ÀTˆØ*ˆÜ×'Ñ'¨°CÔ8ñ 	2Ø—Z‘Z Ó&×*Ñ*¨5Ó1ˆF÷	2ä
×Ñ˜f hÕ/ñ0÷	2ð 	2ús   Ã!D
Ä
D	c            	      ó  — t        dt        j                  g d¢«      gdt        j                  g d¢«      gdt        j                  g d¢«      ggddg¬«      } | j                  d«      }t        j                  g d	¢«      gt        j                  g d¢«      gg}t	        ddgd¬
«      }dg}t        |||¬«      }|j                  d¬«      }t        j                  ||«       |j                  dd¬«      }t        j                  ||«       y )Nr/   )rª   r«   é   )é(   é2   é<   r   )r«   rí   rî   ÚcategoryÚ	arraydata)r�   )rï   éF   éZ   re   ©rJ   r�   F©Únumeric_onlyr7   )	r   r"   Úarrayr&   r   r7   rj   rk   r*   )r+   ÚgbÚexpected_dataÚexpected_indexÚexpected_columnrz   Úaltry   s           r   Útest_agg_over_numpy_arraysrþ   ¤  sç   € ä	à”—‘šÓ&Ð'Ø”—‘šÓ&Ð'Ø”—‘šÓ&Ð'ð	
ð
 ˜[Ð)ô
€Bð 
�‰�JÓ	€Bä—h‘hš|Ó,Ð-´·±ºÓ0FÐ/GÐH€MÜ˜A˜q˜6¨
Ô3€NØ"�m€OÜ˜¨nÀoÔV€Hà
�&‰&˜eˆ&Ó
$€CÜ×Ñ˜#˜xÔ(à�V‰V�E¨ˆVÓ.€FÜ×Ñ˜& (Õ+r   Ú	as_periodTFc                 óx  — t        ddd¬«      }| r |j                  d «      j                  d«      }t        g d¢|dœ«      }|j	                  d«      }|d	   j                  d
„ «      }t        |d d d…   d	¬«      }d|j                  _        t        j                  ||«       |d	   j                  d„ «      }t        ddgd	¬«      }d|j                  _        t        j                  ||«       |d	   j                  d„ «      }t        t        j                  d¬«      t        j                  d¬«      gd	¬«      }d|j                  _        | rZt        t        j                  j                  d«      t        j                  j                  d«      gd	¬«      }d|j                  _        t        j                  ||«       y )Nr0   r_   ÚUTC)rH   ÚtzrZ   )r   r   r/   r/   ©r   r   r   r   c                 ó    — | j                   d   S ©Nr   ©Úilocr·   s    r   r¸   z6test_agg_tzaware_non_datetime_result.<locals>.<lambda>Ë  s   €  1§6¡6¨!¡9€ r   r   re   c                 ó4   — | j                   d   j                  S r  )r  r¶   r·   s    r   r¸   z6test_agg_tzaware_non_datetime_result.<locals>.<lambda>Ñ  s   €  1§6¡6¨!¡9§>¡>€ r   iÜ  c                 ó@   — | j                   d   | j                   d   z
  S )Néÿÿÿÿr   r  r·   s    r   r¸   z6test_agg_tzaware_non_datetime_result.<locals>.<lambda>Ö  s   €  1§6¡6¨"¡:°·±°q±	Ñ#9€ r   r/   )Údays)r
   Útz_localizeÚ	to_periodr   r&   r*   r	   rJ   rf   rj   rl   r�   Ú	TimedeltaÚoffsetsÚDay)rÿ   Údtir+   rù   ry   rz   s         r   Ú$test_agg_tzaware_non_datetime_resultr  ¿  s\  € ô �\¨1°Ô
7€CÙØ�o‰o˜dÓ#×-Ñ-¨cÓ2ˆä	š¨CÑ0Ó	1€BØ	�‰�C‹€Bð �‰W�[‰[Ñ,Ó-€FÜ�c™#˜A˜#‘h SÔ)€HØ€H‡N�NÔÜ×Ñ˜6 8Ô,ð �‰W�[‰[Ñ1Ó2€FÜ�t˜T�l¨Ô-€HØ€H‡N�NÔÜ×Ñ˜6 8Ô,à�‰W�[‰[Ñ9Ó:€FÜ”r—|‘|¨Ô+¬R¯\©\¸qÔ-AÐBÈÔM€HØ€H‡N�NÔÙÜœ2Ÿ:™:Ÿ>™>¨!Ó,¬b¯j©j¯n©n¸QÓ.?Ð@ÀsÔKˆØ!ˆ�‰ÔÜ×Ñ˜6 8Õ,r   c            
      ó†  — t        j                  dd¬«      } t        dt        d«      D �cg c]  }| t	        j
                  |¬«      z   ‘Œ c}dœ«      }|j                  d«      d	   j                  d
«      j                  d   }|j                  d«      d	   j                  d„ «      j                  d   }|j                  d«      d	   j                  «       j                  d   }|| k(  sJ ‚|| k(  sJ ‚|| k(  sJ ‚t        dd«      D �cg c]  }t        j                  d|d›d�d¬«      ‘Œ  }}t        dd	gdz  |dœ«      }|j                  d«      }|d   j                  d   } | |j                  d«      d   j                  d   k(  sJ ‚| |j                  d«      d   j                  d   k(  sJ ‚| |j                  «       d   j                  d   k(  sJ ‚d}	t        j                  t        |	¬«      5  | |j!                  d„ «      j                  d   k(  sJ ‚	 d d d «       |d   j                  d   } | |j#                  «       d   j                  d   k(  sJ ‚d}	t        j                  t        |	¬«      5  | |j!                  d„ «      j                  d   k(  sJ ‚	 d d d «       y c c}w c c}w # 1 sw Y   Œ–xY w# 1 sw Y   y xY w)Nz2016-01-01 12:00:00z
US/Pacific©r  r/   rª   )Úminutesr  r   r   r   r   c                 ó,   — t        j                  | «      S r   )r"   r   r·   s    r   r¸   z.test_agg_timezone_round_trip.<locals>.<lambda>å  s   € ´·±¸³€ r   r   z	2016-01-0Údz	 12:00:00r   ©rW   rX   rW   rX   z7DataFrameGroupBy.apply operated on the grouping columnsr    c                 ó    — | j                   d   S r  r  r·   s    r   r¸   z.test_agg_timezone_round_trip.<locals>.<lambda>ú  s   € ¨Q¯V©V°A©Y€ r   )r   r/   c                 ó    — | j                   d   S )Nr
  r  r·   s    r   r¸   z.test_agg_timezone_round_trip.<locals>.<lambda>  s   € ¨Q¯V©V°B©Z€ r   )r�   Ú	Timestampr   rv   r8   Ú	timedeltar&   r*   r  r   ÚnthÚheadra   rj   rç   ÚDeprecationWarningÚapplyrc   )
r»   Únnr+   r¦   r§   Úresult3ÚiÚdatesr,   r–   s
             r   Útest_agg_timezone_round_tripr%  ß  s°  € ä	�‰Ð+°Ô	=€BÜ	˜Ì%ÐPRË)Ö!TÀB "¤r§|¡|¸BÔ'?Ó"?Ò!TÑUÓ	V€Bà�j‰j˜‹o˜cÑ"×&Ñ& uÓ-×2Ñ2°1Ñ5€GØ�j‰j˜‹o˜cÑ"×&Ñ&Ñ':Ó;×@Ñ@ÀÑC€GØ�j‰j˜‹o˜cÑ"×&Ñ&Ó(×-Ñ-¨aÑ0€Gà�bŠ=Ðˆ=Ø�bŠ=Ðˆ=Ø�bŠ=Ðˆ=ô LQÐQRÐTUË;öØFGŒ�‰�y  1  YÐ/°LÖAð€Eð ô 
˜#˜s˜ a™¨eÑ4Ó	5€BØ�j‰j˜‹o€Gà	ˆC‰�‰�a‰€BØ�—‘˜Q“ Ñ$×)Ñ)¨!Ñ,Ò,Ð,Ð,Ø�—‘˜a“ Ñ%×*Ñ*¨1Ñ-Ò-Ð-Ð-Ø�—‘“ Ñ%×*Ñ*¨1Ñ-Ò-Ð-Ð-ð D€CÜ	×	#Ñ	#Ô$6¸cÔ	Bñ CØ�W—]‘]Ñ#6Ó7×<Ñ<¸TÑBÒBÐBÑB÷Cð 
ˆC‰�‰�a‰€BØ�—‘“ Ñ$×)Ñ)¨!Ñ,Ò,Ð,Ð,ð D€CÜ	×	#Ñ	#Ô$6¸cÔ	Bñ DØ�W—]‘]Ñ#7Ó8×=Ñ=¸dÑCÒCÐCÑC÷Dð Dùò? "Uùò÷Cð Cú÷Dð Dús#   « J!
Ä#J&Ç0%J+É2%J7Ê+J4Ê7K c                  ó°  — t        ddgddgddggt        ¬«      } | dz   } t        g d	¢t        j                  ¬«      }t        dg d
¢i|t        ¬«      }d|j
                  _        | j                  d«      j                  d¬«      }t        j                  ||«       | j                  d«      j                  d¬«      }|g    }t        j                  ||«       y )Nr/   r   r2   r_   r   r`   r³   l   ÿÿÿÿ )l            l           l           )l           l           l           )rJ   r´   r   Frö   T)r   Úobjectr   r"   Úuint64rJ   rf   r&   r7   rj   rk   )r+   rJ   rz   ry   r§   Ú	expected2s         r   Útest_sum_uint64_overflowr*    sË   € ô 
�Q˜�F˜Q ˜F Q¨ FÐ+´6Ô	:€BØ	Ð!Ñ	!€BäÚGÌrÏyÉyô€Eô Ø	
ÒKÐLØÜô€Hð €H‡N�NÔØ�Z‰Z˜‹]×Ñ¨EÐÓ2€FÜ×Ñ˜& (Ô+ð �j‰j˜‹m×Ñ¨TÐÓ2€GØ˜‘€IÜ×Ñ˜' 9Õ-r   zstructure, expectedrY   )r/   r/   r/   )r2   r_   r_   ))r/   r/   )r2   r_   c                 ó   — t        | «      S r   ©Útupler·   s    r   r¸   r¸   $  s
   € ”e˜A“h€ r   c                 ó   — t        | «      S r   ©rR   r·   s    r   r¸   r¸   (  s
   € ”d˜1“g€ r   c                 óÄ   — t        g d¢g d¢g d¢dœ«      }|j                  ddg«      j                  | «      }ddg|j                  _        t        j                  ||«       y )N©r/   r/   r/   r2   r2   r2   ©r/   r/   r/   r_   r_   r_   ©r/   r/   r/   r2   r_   r_   r™   rW   rX   )r   r&   r;   rJ   Únamesrj   rk   ©Ú	structurerz   r+   ry   s       r   Útest_agg_structs_dataframer7    sY   € ô  
Ú Ò'9Ò@RÑSó
€Bð �Z‰Z˜˜c˜
Ó#×-Ñ-¨iÓ8€FØ ˜:€H‡N�NÔÜ×Ñ˜& (Õ+r   r/   r2   rg   c                 ó   — t        | «      S r   r,  r·   s    r   r¸   r¸   <  s
   € ”5˜“8€ r   c                 ó   — t        | «      S r   r/  r·   s    r   r¸   r¸   =  s
   € ”4˜“7€ r   c                 óÂ   — t        g d¢g d¢g d¢dœ«      }|j                  d«      d   j                  | «      }d|j                  _        t        j                  ||«       y )Nr1  r2  r3  r™   rW   rY   )r   r&   r;   rJ   rf   rj   rl   r5  s       r   Útest_agg_structs_seriesr;  7  sU   € ô 
Ú Ò'9Ò@RÑSó
€Bð �Z‰Z˜‹_˜SÑ!×+Ñ+¨IÓ6€FØ€H‡N�NÔÜ×Ñ˜6 8Õ,r   c                 óÂ  — g d¢}t        t        j                  g d¢|¬«      g d¢dœ«      }d}t        j                  t
        |¬«      5  |j                  d| ¬	«      j                  j                  t        j                  «      }d d d «       t        g d
¢t        j                  g d¢|d¬«      d¬«      }| r||dk7     }t        j                  |«       y # 1 sw Y   ŒRxY w)Nr}   )r   r   r   )Ú
categories)r/   r   r2   r  zusing SeriesGroupBy.sumr    rW   )Úobserved)r2   r2   r   )r=  rf   rX   rg   r   )r   r�   ÚCategoricalrj   rç   ræ   r&   rX   r*   r"   Únansumr	   ÚCategoricalIndexrl   )r>  r=  r+   r–   ry   rz   s         r   Útest_agg_category_nansumrB  K  sÀ   € Ú €JÜ	Ü�n‰nš_¸ÔDÊ9ÑUó
€Bð $€CÜ	×	#Ñ	#¤M¸Ô	=ñ EØ—‘˜C¨(�Ó3×5Ñ5×9Ñ9¼"¿)¹)ÓDˆ÷EäÚÜ×!Ñ!¢/¸jÈsÔSØô€Hñ
 Ø˜H¨™MÑ*ˆÜ×Ñ˜6 8Õ,÷Eð Eús   Á	;CÃCc            	      óâ  — t        t        d«      D � cg c]  } t        | «      ‘Œ c} t        d«      D � cg c]  } t        | «      ‘Œ c} dœ«      }|j                  ddd¬«      }|j	                  dd„ i«      }t        t        d«      D � cg c]  } t        | «      ‘Œ c} t        d«      D � cg c]  } t        | «      g‘Œ c} dœ«      }t        j                  ||«       y c c} w c c} w c c} w c c} w )Nr2   r  rW   F)Úas_indexÚsortrX   c                 ó   — t        | «      S r   r/  r·   s    r   r¸   z)test_agg_list_like_func.<locals>.<lambda>a  s
   € ¬¨a«€ r   )r   rv   Ústrr&   r*   rj   rk   )r>   r+   r,   ry   rz   s        r   Útest_agg_list_like_funcrH  ]  s¼   € ä	¬%°«(Ö3 Qœ#˜a�&Ò3Ì5ÐQRË8Ö:TÀa¼3¸q½6Ò:TÑUÓ	V€BØ�j‰j˜ u°5ˆjÓ9€GØ�[‰[˜#Ñ0Ð1Ó2€FÜÜ$ Q›xÖ(˜!Œs�1�vÒ(Ä%ÈÃ(Ö/K¸Q´°Q³²Ò/KÑLó€Hô ×Ñ˜& (Õ+ùò 4ùÒ:Tùò )ùÒ/Ks   “C´C"
Â	C'Â*C,
c                  óP  — t        ddgt        j                  dd¬«      t        j                  dd¬«      gdœ«      } | j                  d«      j	                  dd	„ i«      }t        t        j                  dd¬«      gt        dgd¬
«      dg¬«      }t        j                  ||«       y )Nr/   z
2018-01-01r  r  z
2018-01-02)Útagr4   rJ  r4   c                 ó$   — | j                  d«      S )Nr/   )r  )Úes    r   r¸   z/test_agg_lambda_with_timezone.<locals>.<lambda>s  s   € °a·f±f¸Q³i€ r   re   rõ   )r   r�   r  r&   r*   r   rj   rk   )r+   ry   rz   s      r   Útest_agg_lambda_with_timezonerM  h  s˜   € ä	à�q�6ä—‘˜\¨eÔ4Ü—‘˜\¨eÔ4ðñ	
ó
€Bð �Z‰Z˜Ó×"Ñ" FÑ,?Ð#@ÓA€FÜÜ	�‰�l uÔ	-Ð.Ü�Q�C˜eÔ$Ø�ô€Hô
 ×Ñ˜& (Õ+r   Úerr_clsc                 ó  ‡ — ddl m}m}m}  |«       d d }t	        g d¢g d¢ ||«      dœ«      }t         ||d   |d   g«      «      }ˆ fd„}|d	   j                  |d
   «      j                  |«      }t        j                  ||d¬«       y )Nr   )ÚDecimalArrayÚ	make_dataÚ
to_decimalr   )r   r   r   r/   r/   )r   r/   r   r/   r/   )Úid1Úid2Údecimalsr2   c                 óB   •— t        | «      dk(  r‰‚| j                  d   S r  )r<   r  )r>   rN  s    €r   Ú
weird_funcz1test_groupby_agg_err_catching.<locals>.weird_func›  s#   ø€ ô ˆq‹6�QŠ;ØˆMØ�v‰v�a‰yÐr   rU  rS  F)Úcheck_names)
Ú$pandas.tests.extension.decimal.arrayrP  rQ  rR  r   r	   r&   r*   rj   rl   )	rN  rP  rQ  rR  r=   r+   rz   rW  ry   s	   `        r   Útest_groupby_agg_err_catchingrZ  |  s�   ø€ ÷$ñ ñ ‹;�r˜ˆ?€DÜ	ÚªÁ\ÐRVÓEWÑXó
€Bô ‘j $ q¡'¨4°©7Ð!3Ó4Ó5€Hôð �
‰^×#Ñ# B u¡IÓ.×2Ñ2°:Ó>€FÜ×Ñ˜6 8¸Ö?r   )=Ú__doc__r9   r8   Ú	functoolsr   Únumpyr"   r'   Úpandas.errorsr   Úpandasr�   r   r   r   r   r	   r
   r   Úpandas._testingÚ_testingrj   Úpandas.io.formats.printingr   r-   rD   rU   rp   r{   r€   r—   rœ   r¡   r¨   r±   r¼   rÉ   rÖ   rë   rþ   ÚmarkÚparametrizer  r%  r*  r-  rR   r7  r;  rB  rH  rM  ÚNotImplementedErrorÚRuntimeErrorr‘   Ú
IndexErrorÚOSErrorÚ
ValueErrorÚArithmeticErrorÚAttributeErrorrZ  rÈ   r   r   ú<module>rl     s4  ðñó Ý ã Û å ,ã ÷÷ ñ õ å 3ò"ò." òJò#6òL	-ò,ò",<ò^0ò&-ò*6ò:	*ò-ò6,òr,ò80ò4,ð6 ‡�×Ñ˜ t¨U mÓ4ñ-ó 5ð-ò>#DòL.ð2 ‡�×ÑØà	‘	˜3¨¸IÑ FÐGÓHÐIØ	‰y˜#ª	º9ÑEÐFÓGÐHáÙ�s Y¸	ÑBÐCÓDð	
ñ
 Ù�s¢Yº	ÑBÐCÓDð	
ðóñ,óð,ð ‡�×ÑØà	‘˜	 9Ð-°a¸°VÀ#ÔFÐGØ	‰v’y¢)Ð,°Q¸°FÀÔEÐFÙ	™V Y°	Ð$:À1ÀaÀ&ÈsÔSÐTÙ	™F¢IªyÐ#9À!ÀQÀÈcÔRÐSð	óñ-óð-ò-ò$,ò,ð( ‡�×ÑØàØØØØØØØð	óñ@óñ@r   