
    MhP                        S r SSKrSSKJr  SSKrSSKrSSKJ	r	  SSK
rSSK
JrJrJrJrJrJrJr  SSKJr  SSKJr  S rS rS	 rS
 rS rS rS rS rS r S r!S r"S r#S r$S r%S r&S r'\RP                  RS                  SSS/5      S 5       r*S r+S r,\RP                  RS                  S\-\" SSS S!.05      4\.\" S/ SQ/ S QS!.05      4S" \" SSS S!.05      4S# \" S/ SQ/ S QS!.05      4/5      S$ 5       r/\RP                  RS                  S\-\" SS /S%S&/SS'94\.\" / SQ/ S Q/S%S&/SS'94S( \" SS /S%S&/SS'94S) \" / SQ/ S Q/S%S&/SS'94/5      S* 5       r0S+ r1S, r2S- r3\RP                  RS                  S.\4\5\6\7\8\9\:\;/5      S/ 5       r<g)0z
test all other .agg behavior
    N)partial)SpecificationError)	DataFrameIndex
MultiIndexPeriodIndexSeries
date_rangeperiod_range)pprint_thingc                     [        [        R                  R                  S5      R	                  S5      [        R                  R                  S5      R	                  S5      / SQ/ SQS.5      n U R                  S5      nS n[        R                  " [        SS	9   UR                  U/5        S S S 5        [        R                  " [        SS	9   UR                  U5        S S S 5        g ! , (       d  f       NB= f! , (       d  f       g = f)
N      )ar   br   r   )onetwor   r   r   )data1data2key1key2r   c                 D    U R                  5       U R                  5       -
  $ N)maxmin)arrs    [/var/www/html/env/lib/python3.13/site-packages/pandas/tests/groupby/aggregate/test_other.pypeak_to_peak5test_agg_partial_failure_raises.<locals>.peak_to_peak)   s    wwy3779$$    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     
YY**1-==a@YY**1-==a@-7		

B jj G% 
y(B	C\N# 
D 
y(B	CL! 
D	C 
D	C 
D	Cs   C)C:)
C7:
Dc            	      P   / SQ/ SQ/ SQ/n [        U  Vs/ s H  oS   PM	     snU  Vs/ s H  oS   PM	     snU  Vs/ s H  oS   PM	     snS.5      nU  Vs/ s HI  nUS   US   (       a1  [        R                  R                  US   S5      R	                  5       OS US   /PMK     n n[        U  Vs/ s H  oS   PM	     snU  Vs/ s H  oS   PM	     snU  Vs/ s H  oS   PM	     snS.5      nUS	   US	   R                  5       -  US
'   UR                  S5      R                  S5      nUS	   US	   R                  5       -  US
'   UR                  S5      R                  S5      n[        U5      [        U5      :X  d   eg s  snf s  snf s  snf s  snf s  snf s  snf s  snf )N)   
2012-01-01      ?)r   z
2012-01-02g       @)   Ng      @r   r0   r   )keydatevaluez%Y-%m-%dr6   weightsr5   sum)	r   dtdatetimestrptimer5   r8   r'   	aggregatelen)dataxdf1rowdf2gb1gb2s          r   test_agg_datetimes_mixedrE   3   s   "$:NKD
"&'$QaD$'#'(4aqT4($()DqdD)	
C  C	 F@CAR[[!!#a&*5::<DF	

  	  "&'$QaD$'#'(4aqT4($()DqdD)	
C \CL$4$4$66C	N
++f

'
'
.C\CL$4$4$66C	N
++f

'
'
.Cs8s3x; (() (()s(   FF

FAF3FF
F#c                     [        SSSS9n [        [        R                  R	                  S5      R                  S5      U S9nUR                  SS	9R                  5       n[        UR                  [        5      (       d   e[        S
SSS9n[        [        R                  R	                  S5      R                  [        U5      5      US9n[        [        R                  R	                  S5      R                  [        U5      5      US9n[        R                  " XES.5      nUR                  UR                  R                  5      n[        U5        g )Nz2012-1-1Mr3   )freqperiodsr   )r3   r   indexr   )levelz1999-01r   )startrI   rH   )s1s2)r   r   r#   r$   r%   r&   r'   r8   
isinstancerK   r   r	   r=   	from_dictmonthlist)prngr,   rsrK   rN   rO   r-   s          r   test_agg_period_indexrV   X   s    
a8D	299((+;;FC4	PB	!		 	 	"Bbhh,,,, y!#>E			%%a(//E
;5	IB			%%a(//E
;5	IB			B1	2Bjj(GMr    c            	         [        / SQ[        SSSS9S.5      n S U R                  / SQS4'   U R                  / S	Q   R                  S
5      nU R	                  S
5      n[
        R                  " UR                  5       U5        [
        R                  " UR                  S5      U5        [
        R                  " UR                  SS05      U5        [
        R                  " UR                  R                  5       US   5        [
        R                  " UR                  R                  S5      US   5        U R                  / SQ   R                  S
5      nU R	                  S
5      n[
        R                  " UR                  5       U5        [
        R                  " UR                  S5      U5        [
        R                  " UR                  SS05      U5        [
        R                  " UR                  R                  5       US   5        [
        R                  " UR                  R                  S5      US   5        [        / SQ[        [        S5      S
S9SS9n[
        R                  " UR                  R                  [        5      U5        [
        R                  " UR                  R!                  5       U5        [        / SQ[        [        S5      S
S9SS9n[
        R                  " UR                  R#                  5       U5        g )N)ArX   BrY   CrZ   Dr[   z1/1/2011   h)rI   rH   )classtime)r   r0   r   r   r_   )r   r3         r^   first)r   r3   r`      last)r   r   r   r   ABCDnamerK   rg   )r   r0   r0   r   )r   r
   loc	set_indexr'   tmassert_frame_equalrb   r+   assert_series_equalr_   rd   r	   r   rS   r=   sizecount)r,   expr-   s      r   *test_agg_dict_parameter_cast_result_dtypesrq   g   s8    
=z13?	

B $(BFF<  &&

(
(
1Cjj!G'--/3/'++g.4'++vw&78#>7<<--/V=7<<++G4c&kB &&

(
(
1Cjj!G',,.#.'++f-s3'++vv&67=7<<,,.F<7<<++F3S[A U4<g%FV
TC7<<++C0#67<<,,.4
U4<g%FV
TC7<<--/5r    c                  d   [        S5       V s/ s H  n [        R                  " SU S-   S5      PM      nn [        S5      n[	        X!S.5      nUR                  S5      S   R                  [        5      nUR                  S5      S   R                  5       n[        R                  " XE5        g s  sn f )N   i  r0   aaabbbbbbccd)XYru   rv   )ranger9   r:   rS   r   r'   r+   r=   ro   rk   rm   )r?   uvr,   resultexpecteds         r   test_agg_cast_results_dtypesr|      s     /4Bi8iT1q5!	$iA8^A	#	$BZZ_S!%%c*Fzz#s#))+H6, 	9s   %B-c                     [        / SQ/ SQ/ SQS.5      n [        S/ SQ0/ SQS9nSUR                  l        U R                  S5      S/   R	                  5       n[
        R                  " X!5        [        / SQ/ SQS	./ SQS9nSUR                  l        U R                  S5      SS
/   R	                  5       n[
        R                  " X!5        g )N)r0   r   r3   r`   r   )r0   r   r   r`   r   r   r   cr   )r0   g      @r`   r   )r0   r   r`   r   rJ   r   )r   r   r   )r   rK   rg   r'   meanrk   rl   )r,   r{   rz   s      r   test_aggregate_float64_no_int64r      s    	_U	VB#~.lCHHNNZZ_cU#((*F&+~NC<XHHNNZZ_c3Z(--/F&+r    c                  H   [        / SQ/ SQ[        R                  R                  S5      R	                  S5      S-   [        R
                  " S5      S.5      n U R                  SS/5      nUS	   R                  5       nUS	   R                  5       nUS
   R                  5       nUS
   R                  5       nUS
   R                  SS/5      n[        R                  " XT/SS9nSS/Ul        [        R                  " XgSS9  UR                  SS/5      n[        R                  " X2XT/SS9n[        R                   " S	S
/SS//5      Ul        [        R                  " XgSS9  US
S	/   R                  SS/5      n[        R                  " XTX2/SS9n[        R                   " S
S	/SS//5      Ul        [        R                  " XgSS9  UR                  SSS.5      n[        R                  " XR/SS9n[        R                  " XgSS9  UR                  SS/SS/S.5      n[        R                  " X#XE/SS9n[        R                   " S	S
/SS//5      Ul        Sn["        R$                  " [&        US9   US
S	/   R                  SSS.5        S S S 5        g ! , (       d  f       g = f)Nfoobarr   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r\   r2   rX   rY   rZ   r[   rX   rY   rZ   r[   r8   r   r0   )axisT)
check_likerZ   r[   z&Column\(s\) \['r', 'r2'\] do not existr!   )rr2)r   r#   r$   r%   r&   aranger'   r   r8   r+   pdconcatcolumnsrk   rl   r   from_productr(   r)   KeyError)	r,   r-   c_meanc_sumd_meand_sumrz   r{   msgs	            r   test_aggregate_api_consistencyr      sa    
II&&q)99!<sB1		

B jj#s$GS\ FCLES\ FCLES\ufo.Fyy%q1HvH&t<[[%)Fyy%7a@H!..c
UFO/LMH&t<c3Z $$eV_5Fyy%7a@H!..c
UFO/LMH&t<[[vE23Fyy%q1H&t<[[fe_EFFyy&7a@H!..c
VUO/LMH
3C	xs	+c
e6 :; 
,	+	+s   0J
J!c                     [        / SQ[        S5      [        S5      S.5      n Sn[        R                  " [        US9   U R                  S5      R                  SSS	/0S
SS/0S.5        S S S 5        Sn[        R                  " [        US9   U R                  S5      SS/   R                  SS	05        S S S 5        Sn[        R                  " [        US9   U R                  S5      R                  R                  SS05        S S S 5        g ! , (       d  f       N= f! , (       d  f       Np= f! , (       d  f       g = f)N)r0   r0   r0   r   r   r   rX   rY   rZ   nested renamer is not supportedr!   rX   r   r8   r   r   ro   r   )rY   rZ   z!Column\(s\) \['ma'\] do not existrY   rZ   ma)	r   rw   r(   r)   r   r'   r+   r   rY   )r,   r   s     r   "test_agg_dict_renaming_deprecationr      s    	uQxeAhG	HB
,C	)	5


35%.)%8H0IJ	
 
6
 /C	xs	+


3c
#''u6 
, -C	)	5


3ug./ 
6	5 
6	5 
,	+ 
6	5s#   ,D(D -D1
D 
D.1
D?c                     [        / SQ/ SQ[        R                  R                  S5      R	                  S5      S-   [        R
                  " S5      S.5      n U R                  SS/5      nS	n[        R                  " [        US
9   US   R                  SSS/05        S S S 5        [        R                  " [        US
9   US   R                  SSS.5        S S S 5        g ! , (       d  f       NH= f! , (       d  f       g = f)Nr   r   r   r\   r2   r   rX   rY   r   r!   r[   rZ   r8   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%() 
6 
)	5	#

U+, 
6	5 
6	5 
6	5s   C! C2!
C/2
D c                     [        / SQ/ SQ[        R                  R                  S5      R	                  S5      S-   [        R
                  " S5      S.5      n U R                  SS/5      nS	n[        R                  " [        US
9   UR                  SSS/0SSS/0S.5        S S S 5        [        R                  " [        US
9   UR                  SSS/0SSS/0S.5        S S S 5        [        R                  " [        US
9   US   R                  [        R                  [        R                  S.5        S S S 5        [        R                  " [        US
9   US   R                  [        R                  [        R                  S.5        S S S 5        g ! , (       d  f       N= f! , (       d  f       N= f! , (       d  f       N= f! , (       d  f       g = f)Nr   r   r   r\   r2   r   rX   rY   r   r!   rZ   r   r8   r[   )r1r   rar   rbr   )result1result2)r[   r   )r   r#   r$   r%   r&   r   r'   r(   r)   r   r<   r+   r8   r   r   s      r   test_agg_nested_dictsr     sZ   	II&&q)99!<sB1		

B 	

C:A
,C	)	5	C&%1#9OPQ 
6 
)	5	TFE?+4&%2IJK 
6
 
)	5	#

rvv"'':; 
6 
)	5	#

BGG45 
6	5 
6	5 
6	5
 
6	5 
6	5s0   FF%4F64G
F"%
F36
G
Gc                     [        [        R                  R                  S5      R	                  SSS95      n S n[
        R                  " [        SS9   U R                  S5      R                  U5        S S S 5        g ! , (       d  f       g = f)	Nr   
   )   r   )rn   c                 `    [        S5        [        U R                  5       5        [        S5      e)Nz(----------------------------------------test)r   	to_stringr*   )r,   s    r   raiseException=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     sb    	299((+44Rh4G	HB 
 
y	/


1.) 
0	/	/s   !A==
Bc                     [        [        R                  " S[        R                  S9[	        SSS9S9n U R                  S S /5      nUR                  S5      nUR                  5       n[        R                  " X#5        g )	Nr   dtypez
2020-01-01)rI   rJ   c                     U R                   $ r   )yearr?   s    r   <lambda>*test_series_agg_multikey.<locals>.<lambda>.  s    AFFr    c                     U R                   $ r   )rR   r   s    r   r   r   .  s    aggr    r8   )
r	   r#   r   float64r
   r'   r+   r8   rk   rm   )tsr-   rz   r{   s       r   test_series_agg_multikeyr   *  sf    	
		"BJJ'z,PR/S
B jj*,=>?G[[F{{}H6,r    c            
         [        / SQ/ SQ/ SQ[        R                  R                  S5      R	                  S5      [        R                  R                  S5      R	                  S5      [        R                  R                  S5      R	                  S5      S.5      n S nU R                  SS	/5      R                  U5      nU R                  SS	/5      R                  S
 5      n[        R                  " X#5        g )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      )rX   rY   rZ   r[   EFc                 N    [        U R                  R                  5      S:  d   eg)Nr   r   )r=   valuesbaser   s    r   bad.test_series_agg_multi_pure_python.<locals>.bade  s     188==!A%%%r    rX   rY   c                     g)Nr    r   s    r   r   3test_series_agg_multi_pure_python.<locals>.<lambda>j  s    er    )	r   r#   r$   r%   r&   r'   r+   rk   rl   )r>   r   rz   r{   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                     S n [        / SQ/ SQ[        R                  " SSS5      [        R                  " SSS5      [        R                  " SSS5      [        R                  " SSS5      /S.5      nUR                  S	5      nUR	                  U /5      nUR
                  R                  S
   Ul        UR	                  U 5      n[        R                  " XC5        g )Nc                 H    [         R                  " U R                  5       SS9$ )Nr0   )q)r#   
percentiledropna)r   s    r   P1 test_agg_consistency.<locals>.P1q  s    }}QXXZ1--r    )r0   r   r3   r`   )r            i  r   r   r   )col1col2r5   r5   r   )	r   r9   r5   r'   r+   r   levelsrk   rl   )r   r,   r   r{   rz   s        r   test_agg_consistencyr   n  s    . 
 $a$a$a$a$			

B 	

6AuubT{H''..q1HUU2YF&+r    c                  :   [        SS/SS/S.5      R                  [        R                  5      n  " S S5      n[        [        R                  S S	 [        [        5      U" 5       /nU R                  S
5      R                  S5      nU H}  nU[        L d  U[        R                  L a  [        OS nSn[        R                  " XVS9   U R                  S
5      R                  U5      nS S S 5        [        R                  " WU5        M     g ! , (       d  f       N*= f)Nr0   r   r3   r`   )r   r   c                       \ rS rSrS rSrg)$test_agg_callables.<locals>.fn_classi  c                     [        U5      $ r   r8   )selfr?   s     r   __call__-test_agg_callables.<locals>.fn_class.__call__  s    q6Mr    r   N)__name__
__module____qualname____firstlineno__r   __static_attributes__r   r    r   fn_classr     s    	r    r   c                     [        U 5      $ r   r   r   s    r   r   $test_agg_callables.<locals>.<lambda>  s    #a&r    c                 "    U R                  5       $ r   r   r   s    r   r   r     s
    !%%'r    r   r8   zusing DataFrameGroupBy.sumr!   )r   astyper#   int64r8   r   r'   r+   FutureWarningrk   assert_produces_warningrl   )r,   r   equiv_callablesr{   ecallwarnr   rz   s           r   test_agg_callablesr     s    	Aq61a&1	2	9	9"((	CB 
 	

O zz% $$U+H  %}T*''8ZZ&**51F 9
fh/ ! 98s   !D
D	c            	          [        S[        R                  " / SQ5      /S[        R                  " / SQ5      /S[        R                  " / SQ5      //SS/S9n U R                  S5      n[        R                  " / S	Q5      /[        R                  " / SQ5      //n[	        SS/SS
9nS/n[        X#US9nUR                  SS9n[        R                  " Xe5        UR                  SSS9n[        R                  " Xu5        g )Nr0   )r   r      )(   2   <   r   )r   r   r   category	arraydata)r   )r   F   Z   rf   rK   r   Fnumeric_onlyr8   )	r   r#   arrayr'   r   r8   rk   rl   r+   )r,   gbexpected_dataexpected_indexexpected_columnr{   altrz   s           r   test_agg_over_numpy_arraysr    s    	&'&'&'	

 [)
B 
J	Bhh|,-0F/GHMAq6
3N"mOoVH
&&e&
$C#(VVEV.F&+r    	as_periodTFc                 f   [        SSSS9nU (       a   UR                  S 5      R                  S5      n[        / SQUS.5      nUR	                  S5      nUS	   R                  S
 5      n[        US S S2   S	S9nSUR                  l        [        R                  " XE5        US	   R                  S 5      n[        SS/S	S9nSUR                  l        [        R                  " XE5        US	   R                  S 5      n[        [        R                  " SS9[        R                  " SS9/S	S9nSUR                  l        U (       aW  [        [        R                  R                  S5      [        R                  R                  S5      /S	S9nSUR                  l        [        R                  " XE5        g )Nr1   r`   UTC)rI   tzr[   )r   r   r0   r0   r   r   r   r   c                      U R                   S   $ Nr   ilocr   s    r   r   6test_agg_tzaware_non_datetime_result.<locals>.<lambda>  s    166!9r    r   rf   c                 4    U R                   S   R                  $ r  )r  r   r   s    r   r   r    s    166!9>>r    i  c                 @    U R                   S   U R                   S   -
  $ )Nr   r  r   s    r   r   r    s    166":q	#9r    r0   )days)r
   tz_localize	to_periodr   r'   r+   r	   rK   rg   rk   rm   r   	TimedeltaoffsetsDay)r	  dtir,   r  rz   r{   s         r   $test_agg_tzaware_non_datetime_resultr    sV    \1
7Cood#--c2	C0	1B	CB W[[,-Fc#A#hS)HHNN6, W[[12FtTl-HHNN6,W[[9:Fr||+R\\q-ABMHHNN2::>>!,bjjnnQ.?@sK!6,r    c                     [         R                  " SSS9n [        S[        S5       Vs/ s H  o[        R
                  " US9-   PM     snS.5      nUR                  S5      S	   R                  S
5      R                  S   nUR                  S5      S	   R                  S 5      R                  S   nUR                  S5      S	   R                  5       R                  S   nX0:X  d   eX@:X  d   eXP:X  d   e[        SS5       Vs/ s H  n[         R                  " SUS S3SS9PM     nn[        SS	/S-  US.5      nUR                  S5      nUS   R                  S   n XR                  S5      S   R                  S   :X  d   eXR                  S5      S   R                  S   :X  d   eXR                  5       S   R                  S   :X  d   eSn	[        R                  " [        U	S9   XR!                  S 5      R                  S   :X  d   e S S S 5        US   R                  S   n XR#                  5       S   R                  S   :X  d   eSn	[        R                  " [        U	S9   XR!                  S 5      R                  S   :X  d   e S S S 5        g s  snf s  snf ! , (       d  f       N= f! , (       d  f       g = f)Nz2016-01-01 12:00:00z
US/Pacificr  r0   r   )minutesr  r   r   r   r   c                 .    [         R                  " U 5      $ r   )r#   r   r   s    r   r   .test_agg_timezone_round_trip.<locals>.<lambda>  s    r    r   z	2016-01-0dz	 12:00:00r   rX   rY   rX   rY   z7DataFrameGroupBy.apply operated on the grouping columnsr!   c                      U R                   S   $ r  r  r   s    r   r   r"    s    QVVAYr    )r   r0   c                      U R                   S   $ )Nr  r  r   s    r   r   r"    s    QVVBZr    )r   	Timestampr   rw   r9   	timedeltar'   r+   r  r   nthheadrb   rk   r   DeprecationWarningapplyrd   )
r   nnr,   r   r   result3idatesr-   r   s
             r   test_agg_timezone_round_tripr1    s   	+	=B	%PR)!T)Br||B'?"?)!TU	VBjjoc"&&u-2215Gjjoc"&&':;@@CGjjoc"&&(--a0G====== LQQRTU;KVay1Y/LA; 
  
#sae4	5BjjoG	CaBQ$))!,,,,a%**1----%**1---- DC		#	#$6c	B]]#67<<TBBBB 
C 
CaB$))!,,,, DC		#	#$6c	B]]#78==dCCCC 
C	B? "U 
C	B 
C	Bs#    J
$J!-%J&-%J7&
J47
Kc                     [        SS/SS/SS//[        S9n U S-   n [        / S	Q[        R                  S9n[        S/ S
Q0U[        S9nSUR
                  l        U R                  S5      R                  SS9n[        R                  " X25        U R                  S5      R                  SS9nU/    n[        R                  " XE5        g )Nr0   r   r3   r`   r   ra   r   l    )l            l           l           )l           l           l           )rK   r   r   Fr   T)r   objectr   r#   uint64rK   rg   r'   r8   rk   rl   )r,   rK   r{   rz   r   	expected2s         r   test_sum_uint64_overflowr6    s     
QFQFQF+6	:B	!	!BGryyE 	
KLH HNNZZ]E2F&+ jjmT2GI'-r    zstructure, expectedrZ   )r0   r0   r0   )r3   r`   r`   ))r0   r0   )r3   r`   c                     [        U 5      $ r   tupler   s    r   r   r   $  s    eAhr    c                     [        U 5      $ r   rS   r   s    r   r   r   (  s    d1gr    c                     [        / SQ/ SQ/ SQS.5      nUR                  SS/5      R                  U 5      nSS/UR                  l        [
        R                  " X15        g )Nr0   r0   r0   r3   r3   r3   r0   r0   r0   r`   r`   r`   r0   r0   r0   r3   r`   r`   r   rX   rY   )r   r'   r<   rK   namesrk   rl   	structurer{   r,   rz   s       r   test_agg_structs_dataframerC    sW      
 '9@RS
B ZZc
#--i8F:HNN&+r    r0   r3   rh   c                     [        U 5      $ r   r8  r   s    r   r   r   <  s    58r    c                     [        U 5      $ r   r;  r   s    r   r   r   =  s    47r    c                     [        / SQ/ SQ/ SQS.5      nUR                  S5      S   R                  U 5      nSUR                  l        [
        R                  " X15        g )Nr=  r>  r?  r   rX   rZ   )r   r'   r<   rK   rg   rk   rm   rA  s       r   test_agg_structs_seriesrG  7  sS     
 '9@RS
B ZZ_S!++I6FHNN6,r    c           	         / SQn[        [        R                  " / SQUS9/ SQS.5      nSn[        R                  " [
        US9   UR                  SU S	9R                  R                  [        R                  5      nS S S 5        [        / S
Q[        R                  " / SQUSS9SS9nU (       a  XUS:g     n[        R                  " WU5        g ! , (       d  f       NW= f)Nr~   )r   r   r   )
categories)r0   r   r3   r$  zusing SeriesGroupBy.sumr!   rX   )observed)r3   r3   r   )rI  rg   rY   rh   r   )r   r   Categoricalrk   r   r   r'   rY   r+   r#   nansumr	   CategoricalIndexrm   )rJ  rI  r,   r   rz   r{   s         r   test_agg_category_nansumrN  K  s     J	nn_D9U
B $C		#	#M	=C(35599"))D 
>!!/jsSH
 M*68, 
>	=s   8C
Cc            	         [        [        S5       V s/ s H  n [        U 5      PM     sn [        S5       V s/ s H  n [        U 5      PM     sn S.5      nUR                  SSSS9nUR	                  SS 05      n[        [        S5       V s/ s H  n [        U 5      PM     sn [        S5       V s/ s H  n [        U 5      /PM     sn S.5      n[
        R                  " X45        g s  sn f s  sn f s  sn f s  sn f )Nr3   r$  rX   F)as_indexsortrY   c                     [        U 5      $ r   r;  r   s    r   r   )test_agg_list_like_func.<locals>.<lambda>a  s    ar    )r   rw   strr'   r+   rk   rl   )r?   r,   r-   rz   r{   s        r   test_agg_list_like_funcrU  ]  s    	%(3(Q#a&(35QR8:T8a3q68:TU	VBjju5j9G[[#012F$Qx(x!s1vx(%(/K(QQ(/KLH &+ 4:T )/Ks   C&C+
C00C5
c            	      8   [        SS/[        R                  " SSS9[        R                  " SSS9/S.5      n U R                  S5      R	                  SS	 05      n[        [        R                  " SSS9/[        S/SS
9S/S9n[        R                  " X5        g )Nr0   z
2018-01-01r  r  z
2018-01-02)tagr5   rW  r5   c                 $    U R                  S5      $ )Nr0   )r*  )es    r   r   /test_agg_lambda_with_timezone.<locals>.<lambda>s  s    affQir    rf   r   )r   r   r'  r'   r+   r   rk   rl   )r,   rz   r{   s      r   test_agg_lambda_with_timezoner[  h  s    	q6\e4\e4	

B ZZ""F,?#@AF	lu	-.QCe$H
 &+r    err_clsc                   ^  SSK JnJnJn  U" 5       S S n[	        / SQ/ SQU" U5      S.5      n[        U" US   US   /5      5      nU 4S jnUS	   R                  US
   5      R                  U5      n[        R                  " XSS9  g )Nr   )DecimalArray	make_data
to_decimalr   )r   r   r   r0   r0   )r   r0   r   r0   r0   )id1id2decimalsr3   c                 D   > [        U 5      S:X  a  TeU R                  S   $ r  )r=   r  )r?   r\  s    r   
weird_func1test_groupby_agg_err_catching.<locals>.weird_func  s#     q6Q;Mvvayr    rc  ra  F)check_names)
$pandas.tests.extension.decimal.arrayr^  r_  r`  r   r	   r'   r+   rk   rm   )	r\  r^  r_  r`  r>   r,   r{   re  rz   s	   `        r   test_groupby_agg_err_catchingri  |  s    $  ;r?D	\RVEWX
B j$q'47!345H 
^##BuI.22:>F6?r    )=__doc__r:   r9   	functoolsr   numpyr#   r(   pandas.errorsr   pandasr   r   r   r   r   r	   r
   r   pandas._testing_testingrk   pandas.io.formats.printingr   r.   rE   rV   rq   r|   r   r   r   r   r   r   r   r   r   r   r  markparametrizer  r1  r6  r9  rS   rC  rG  rN  rU  r[  NotImplementedErrorRuntimeErrorr   
IndexErrorOSError
ValueErrorArithmeticErrorAttributeErrorri  r   r    r   <module>r{     s4       ,     3"." J#6L	-,",<^0&-*6:	*-6,r,804,6 tUm4- 5->#DL.2 		3I FGHI	y#	9EFGHsY	BCD	

 sY	BCD	
,, 		9-aV#FG	vy),QFEF	VY	$:1a&sST	FIy#9!QcRS	---$,,( 	@@r    