
    OwgP                     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    `/var/www/horilla/myenv/lib/python3.12/site-packages/pandas/tests/groupby/aggregate/test_other.pypeak_to_peakz5test_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#$ 
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KD
"&'QAaD'#'(aQqT($()qad)	
C  	 F@CAR[[!!#a&*5::<DF	
D  "&'QAaD'#'(aQqT($()qad)	
C \CL$4$4$66C	N
++f

'
'
.C\CL$4$4$66C	N
++f

'
'
.Cs8s3x; (() (()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8D	299((+;;FC4	PB	!		 	 	"Bbhh,,, y!#>E			%%a(//E
;5	IB			%%a(//E
;5	IB			Bb1	2Bjj(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    
=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                  ^   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     /4Bi8T1q5!	$8A8^A	#	$BZZ_S!%%c*Fzz#s#))+H68, 	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	VB#~.lCHHNNZZ_cU#((*F&(+~NC<XHHNNZZ_c3Z(--/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 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   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   	uQxeAhG	HB
,C	)	5 



35%.)%8H0IJ	


 /C	xs	+ 7


3c
#''u67 -C	)	5 0


3ug./0 0
 
7 70 0s$   ,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	TFE?+4&%2IJKL
 
)	5 <	#

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

BGG456 6R RL L
< <6 6s0   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    	299((+44Rh4G	HB 
 
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
    AFF r   c                     | j                   S r   )rQ   r   s    r   r   z*test_series_agg_multikey.<locals>.<lambda>.  s
    agg r   r7   )
r	   r"   r   float64r
   r&   r*   r7   rj   rl   )tsr,   ry   rz   s       r   test_series_agg_multikeyr   *  sh    	
		"BJJ'z,PR/S
B jj*,=>?G[[F{{}H68,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     188==!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    }}QXXZ1--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uubT{H''..q1HUU2YF&(+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    q6Mr   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    	Aq61a&1	2	9	9"((	CB 
 	

O zz% $$U+H  0 %}T*''C8 	2ZZ&**51F	2
fh/0	2 	2s   !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hh|,-0F/GHMAq6
3N"mOnoVH
&&e&
$C#x(VVE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    166!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    166!9>> r   i  c                 @    | j                   d   | j                   d   z
  S )Nr   r  r   s    r   r   z6test_agg_tzaware_non_datetime_result.<locals>.<lambda>  s    166":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
7Cood#--c2	C0	1B	CB W[[,-Fc#A#hS)HHNN68, W[[12FtTl-HHNN68,W[[9:Fr||+R\\q-ABMHHNN2::>>!,bjjnnQ.?@sK!68,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    QVVAY r   )r   r/   c                      | j                   d   S )Nr
  r  r   s    r   r   z.test_agg_timezone_round_trip.<locals>.<lambda>  s    QVVBZ 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)!TB"r||B'?"?!TU	VBjjoc"&&u-2215Gjjoc"&&':;@@CGjjoc"&&(--a0Gb==b==b== LQQRTU;FGy1Y/LAE  
#sae4	5BjjoG	CaBQ$))!,,,,a%**1----%**1---- DC		#	#$6c	B CW]]#67<<TBBBBC 
CaB$))!,,,, DC		#	#$6c	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FQFQF+6	:B	!	!BGryyE 	
KLH HNNZZ]E2F&(+ jjmT2G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
    eAh r   c                     t        |       S r   rR   r   s    r   r   r   (  s
    d1g 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 ZZc
#--i8F:HNN&(+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 ZZ_S!++I6FHNN68,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	nn_D9U
B $C		#	#M	= EC(35599"))DE!!/jsSH
 HM*68,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    	%(3Q#a&35QR8:Ta3q6:TU	VBjju5j9G[[#012F$Qx(!s1v(%(/KQ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    affQi r   re   r   )r   r   r  r&   r*   r   rj   rk   )r+   ry   rz   s      r   test_agg_lambda_with_timezonerM  h  s    	q6\e4\e4	

B ZZ""F,?#@AF	lu	-.QCe$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#     q6Q;Mvvay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	\RVEWX
B j$q'47!345H 
^##BuI.22:>F68?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#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   