
    Mh*                     |   S SK Jr  S SKrS SKrS SKrS SKJr  S SKJ	r	J
r
Jr  S SKJr  \R                  R!                  SSS/5      S 5       rS	 rS
 rS r\R                  R!                  SSS/5      S 5       r\R                  R!                  S/ SQ5      \R                  R!                  SSSS00SSS/00/5      S 5       5       r\R                  R!                  SSSS/0SS/SS/S./5      S 5       rS r\R                  R!                  SSS/5      S  5       r\R                  R!                  S/ SQ5      \R                  R!                  SSS0SS/0/5      S! 5       5       rS" r\R                  R!                  S#S$S%SS&S'/44S(/5      S) 5       rS* rS+ r\R                  R!                  S,\R>                  " \	" S&S'/S'S&//5      S-\ //5      5      S. 5       r!\R                  R!                  S/\" \R>                  " \
" S0RE                  5       5      S\ 4S1\ 4S2\ 4S3\ 4S4\ 4S-\ 4/5      5      5      S5 5       r#S6 r$S7 r%\R                  R!                  SS8 S9 S: /5      S; 5       r&S< r'\R                  RQ                  S=5      \R                  R!                  S>S?S/S@4\RR                  SS.SA4/5      SB 5       5       r*\R                  R!                  SSS/SS?//5      SC 5       r+\R                  R!                  SSS/S?S//5      SD 5       r,\R                  R!                  SESF SG SH SI /5      SJ 5       r-g)K    )chainN)SpecificationError)	DataFrameSeries
date_rangeresult_typefoo   c           	          [        [        R                  " [        R                  " SSS9S5      R	                  SS5      S-   / SQS9nS	n[
        R                  " [        US
9   UR                  S SU S9  S S S 5        g ! , (       d  f       g = f)N   int64dtype   r
   ABCcolumnszUinvalid value for result_type, must be one of {None, 'reduce', 'broadcast', 'expand'}matchc                 
    / SQ$ )Nr
      r    xs    U/var/www/html/env/lib/python3.13/site-packages/pandas/tests/apply/test_invalid_arg.py<lambda>(test_result_type_error.<locals>.<lambda>&   s    9    axisr   	r   nptilearangereshapepytestraises
ValueErrorapply)r   dfmsgs      r    test_result_type_errorr1      su     

		!7+Q/772>B
B	2  
z	-
$1+F 
.	-	-s   #A??
Bc                      [        / SQ/ SQ/ SQ// SQS9n Sn[        R                  " [        US9   U R	                  S S	5        S S S 5        g ! , (       d  f       g = f)
Nr   )      r   )      	   )ar8   c)indexz)No axis named 2 for object type DataFramer   c                     U $ Nr   r   s    r    r!   /test_apply_invalid_axis_value.<locals>.<lambda>-   s    1r#   r   r   r+   r,   r-   r.   r/   r0   s     r    test_apply_invalid_axis_valuer@   )   sB    	Iy)4O	LB
5C	z	-
a  
.	-	-s   A
Ac                      [        SS/SS/S.5      n Sn[        R                  " [        US9   U R	                  5         S S S 5        g ! , (       d  f       g = f)Nr   r
   r   r   r   zMust provider   )r   r+   r,   	TypeErroraggr?   s     r    test_agg_raisesrE   0   sA    	!Qq!f-	.B
C	y	,
 
-	,	,s   A
Ac                      [        / SQ5      n Sn[        R                  " [        US9   U R	                  S SS9  S S S 5        g ! , (       d  f       g = f)Nr   z)na_action must either be 'ignore' or Noner   c                     U $ r<   r   r   s    r    r!   8test_map_with_invalid_na_action_raises.<locals>.<lambda>>   s    r#   ____	na_actionr   r+   r,   r-   map)sr0   s     r    &test_map_with_invalid_na_action_raisesrO   9   s;    yA
5C	z	-	kV, 
.	-	-s   A
Ainput_na_actionrI   Tc                     [        / SQ5      nSU  S3n[        R                  " [        US9   UR	                  SS0U S9  S S S 5        g ! , (       d  f       g = f)Nr   z+na_action must either be 'ignore' or None, z was passedr   r
   r   rJ   rL   )rP   rN   r0   s      r    2test_map_arg_is_dict_with_invalid_na_action_raisesrR   A   sK     	yA77H
TC	z	-	q!f0 
.	-	-s   A		
Amethod)r.   rD   	transformfuncr   r   sumc                     U " SS/05      nSn[         R                  " [        US9   [        X15      " U5        S S S 5        g ! , (       d  f       g = f)Nr   r
   nested renamer is not supportedr   )r+   r,   r   getattr)frame_or_seriesrS   rU   objr   s        r    test_nested_renamerr\   J   sA     3*
%C-E	)	7T" 
8	7	7s   A
Arenamerminmaxmean)r	   barc                     [        [        S5      SSS9nSn[        R                  " [        US9   UR                  U 5        S S S 5        g ! , (       d  f       g = f)Nr   r   series)r   namerX   r   )r   ranger+   r,   r   rD   )r]   rN   r0   s      r    test_series_nested_renamerrf   T   sA    
 	uQxwX6A
+C	)	5	g 
6	5	5s   A
Ac            	      .   [        [        R                  R                  S5      R	                  S5      / SQ[        SSS9S9n Sn[        R                  " [        US	9   U R                  R                  S
SS/05        S S S 5        g ! , (       d  f       g = f)Nr   )
   r   r   z1/1/2000rh   )periods)r   r:   rX   r   r	   rV   r`   )r   r'   randomdefault_rngstandard_normalr   r+   r,   r   r   rD   )tsdfr0   s     r    test_apply_dict_deprrn   _   sq    
		a 009R0D
 ,C	)	5

EE6?+, 
6	5	5s    B
BrD   rT   c                     [        [        S5      SS.5      nSn[        R                  " [        US9   [        X5      " SS0SS0S.5        S S S 5        g ! , (       d  f       g = f)	Nr4   rB   rX   r   r	   r^   ra   r_   )r   re   r+   r,   r   rY   )rS   r/   r0   s      r    test_dict_nested_renaming_deprrp   j   sQ    	q*	+B -C	)	55%.u~FG 
6	5	5s   A
A#c                     [        SS/05      n[        R                  " S5      n[        R                  " [
        US9   [        X 5      " U5        S S S 5        g ! , (       d  f       g = f)Nr   r
   zColumn(s) ['B'] do not existr   )r   reescaper+   r,   KeyErrorrY   )rS   rU   r[   r   s       r    test_missing_columnru   t   sJ     S1#J
CII45E	xu	-T" 
.	-	-s   A
A'c                      [        SS/05      n Sn[        R                  " [        US9   U R	                  [
        [        [
        S.5        S S S 5        g ! , (       d  f       g = f)Nr8   1z#Column\(s\) \[1, 'b'\] do not existr   )r8   r
   b)r   r+   r,   rt   rT   intstrr?   s     r    'test_transform_mixed_column_name_dtypesr{   ~   sD    	C#<	 B
0C	xs	+
33S12 
,	+	+s   "A
A#z	how, args)
pct_changer   	nsmallestr8   rx   )tailr
   c                     [        SS/SS/S.5      nSU  S3n[        R                  " [        US9   UR	                  U SUS	9  S S S 5        g ! , (       d  f       g = f)
Nr
   r   r   r3   )r8   rx   z
Operation z does not support axis=1r   )r%   argsr>   )howr   r/   r0   s       r    test_apply_str_axis_1_raisesr      sV    
 
!Qq!f-	.Bse3
4C	z	-
14( 
.	-	-s   A
Ac                      Sn [         R                  " [        U S9   [        S/5      R	                  SSS9  S S S 5        g ! , (       d  f       g = f)Nz&No axis named 1 for object type Seriesr   r
   rV   r%   )r+   r,   r-   r   rT   )r0   s    r    test_transform_axis_1_raisesr      s:    
2C	z	-se!, 
.	-	-s   A  
A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 [        R
                  U R                  S'   S nS	n[        R                  " [        US
9   U R                  USS9  S S S 5        g ! , (       d  f       g = f)N)r	   r	   r	   r	   ra   ra   ra   ra   r	   r	   r	   )oner   r   twor   r   r   r   r   r   r   )dullr   shinyr   r   r   r   r   r   r   r   r      )r   r   r   DEF)r3   r   c                 T    U S   R                  S5      (       a  U S   S:X  a  SU S'   U $ )Nr   shinr   r	   r5   r   )
startswith)rows    r    rT   .test_apply_modify_traceback.<locals>.transform   s0    s8v&&3s8u+<CH
r#   z,'float' object has no attribute 'startswith'r   r
   r   )r   r'   rj   rk   rl   nanlocr+   r,   AttributeErrorr.   )datarT   r0   s      r    test_apply_modify_tracebackr      s     &&q)99"=&&q)99"=&&q)99"=U+	
-D^ vvDHHV
 9C	~S	1

91
% 
2	1	1s   C22
D zdf, func, expectedcumprodc                 ^   U(       a  SS K nX%R                  R                  4nSn[        U[        5      (       a  S O[
        n[        R                  " X&S9   [        R                  " USS9   U R                  XS9  S S S 5        S S S 5        g ! , (       d  f       N= f! , (       d  f       g = f)Nr   z>can't multiply sequence by non-int of type 'str'|has no kernelr   zusing DataFrame.cumprodr   )pyarrowlibArrowNotImplementedError
isinstancerz   FutureWarningr+   r,   tmassert_produces_warningrD   )r/   rU   expectedr%   using_infer_stringpar0   warns           r    "test_agg_cython_table_raises_framer      s}     ff==>
JCdC((4mD	x	+''4MNFF4F# O 
,	+NN 
,	+s$   B,B<B
B	B
B,zseries, func, expectedza b cprodstdvarmedianc                    SnUS:X  d&  U[         R                  L d  U[         R                  L a  SnU(       a  SS KnX%R                  R
                  4nUS-   n[        U[        5      (       a  S O[        n[        R                  " X$S9   [        R                  " USS9   U R                  U5        S S S 5        S S S 5        g ! , (       d  f       N= f! , (       d  f       g = f)Nz?[Cc]ould not convert|can't multiply sequence by non-int of typer   z)Cannot convert \['a' 'b' 'c'\] to numericr   z|does not support|has no kernelr   zis currently using Series.*)r'   	nanmedianr   r   r   r   r   rz   r   r+   r,   r   r   rD   )rc   rU   r   r   r0   r   r   s          r    #test_agg_cython_table_raises_seriesr      s    $ MCx42<</42993D:ff==>
1
1CdC((4mD	x	+''4QRJJt S 
,	+RR 
,	+s$   	CC1C
C	C
C!c                      [        SS /05      n [        R                  " S5      n[        R                  " [
        US9   U R                  SS 05        S S S 5        g ! , (       d  f       g = f)Nr8   zint() argument must be a stringr   c                 2    [        U R                  S   5      $ Nr   ry   ilocr   s    r    r!   'test_agg_none_to_type.<locals>.<lambda>  s    s166!9~r#   )r   rr   rs   r+   r,   rC   rD   r?   s     r    test_agg_none_to_typer     sM    	C$=	!B
))5
6C	y	,
-./ 
-	,	,s   A
A*c                      [        SS /05      n Sn[        R                  " [        US9   U R	                  SS 05        S S S 5        g ! , (       d  f       g = f)Nr8   zargument must be ar   c                 2    [        U R                  S   5      $ r   r   r   s    r    r!   -test_transform_none_to_type.<locals>.<lambda>  s    S^r#   )r   r+   r,   rC   rT   r?   s     r    test_transform_none_to_typer     sC    	C$=	!B
C	y	,
c345 
-	,	,s   A
Ac                 R    [         R                  " SS/5      R                  SS5      $ )Nr
   r   r   )r'   arrayr*   r   s    r    r!   r!     s    "((Aq6"**2q1r#   c                 
    SS/$ Nr
   r   r   r   s    r    r!   r!      s    1a&r#   c                     [        SS/5      $ r   )r   r   s    r    r!   r!   !  s    &!Q.r#   c           	         [        [        R                  " [        R                  " SSS9S5      R	                  SS5      S-   / SQS9nS	n[
        R                  " [        US
9   UR                  U SSS9  S S S 5        g ! , (       d  f       g = f)Nr   r   r   r   r   r
   r   r   z2too many dims to broadcast|cannot broadcast resultr   	broadcastr$   r&   )rU   r/   r0   s      r    test_apply_broadcast_errorr     so     

		!7+Q/772>B
B ?C	z	-
A;7 
.	-	-s   #A>>
Bc                     Sn[         R                  " [        US9   [        R                  " SS9   UR                  SS/U S9  S S S 5        S S S 5        g ! , (       d  f       N= f! , (       d  f       g = f)Nz3cannot combine transform and aggregation operationsr   ignoreallr_   sqrtr   r+   r,   r-   r'   errstaterD   )r%   float_framer0   s      r    test_transform_and_agg_err_aggr   0  sR    
?C	z	-[[X&OOUFO$O7 ' 
.	-&& 
.	-s"   A&AA&
A#	A&&
A4zignore::FutureWarningz	func, msgr   z(cannot combine transform and aggregationz2cannot perform both aggregation and transformationc                     [         R                  " [        US9   [        R                  " SS9   U R                  U5        S S S 5        S S S 5        g ! , (       d  f       N= f! , (       d  f       g = f)Nr   r   r   r   string_seriesrU   r0   s      r    !test_transform_and_agg_err_seriesr   8  sI     
z	-[[X&d# ' 
.	-&& 
.	-s"   A#AA#
A 	A##
A1c                     Sn[         R                  " [        US9   UR                  X S9  S S S 5        g ! , (       d  f       g = f)NFunction did not transformr   r   r+   r,   r-   rT   )r%   r   rU   r0   s       r    test_transform_wont_agg_framer   J  s4     'C	z	-d. 
.	-	-s	   5
Ac                     Sn[         R                  " [        US9   U R                  U5        S S S 5        g ! , (       d  f       g = f)Nr   r   r   r   s      r    test_transform_wont_agg_seriesr   S  s1     'C	z	-% 
.	-	-s	   7
A
op_wrapperc                     U $ r<   r   r   s    r    r!   r!   ^  s    Qr#   c                     U /$ r<   r   r   s    r    r!   r!   ^  s    1#r#   c                 
    SU 0$ Nr   r   r   s    r    r!   r!   ^  s    #qr#   c                     SU /0$ r   r   r   s    r    r!   r!   ^  s    cTUSVZr#   c                     U" U 5      n[        S/ SQ05      n[        R                  " XA5      nSn[        R                  " [
        US9   UR                  U5        S S S 5        g ! , (       d  f       g = f)Nr   r   r   r   )r   r   get_objr+   r,   r-   rT   )all_reductionsrZ   r   opr[   r0   s         r    test_transform_reducer_raisesr   ]  sV    
 
N	#B
S)$
%C
**S
*C
&C	z	-b 
.	-	-s   	A$$
A2).	itertoolsr   rr   numpyr'   r+   pandas.errorsr   pandasr   r   r   pandas._testing_testingr   markparametrizer1   r@   rE   rO   rR   r\   rf   rn   rp   ru   r{   r   r   r   get_cython_table_paramsrC   r   splitr   r   r   r   r   filterwarningsr   r   r   r   r   r   r#   r    <module>r      s    	   , 
  
3G 4G!- *VTN;1 <1 #@A3e"5cE7^7L!MN# O B# eU^ueneV_MN	- E;#78H 9H #@A3,ug!?@# A B#3 ${Qc
O&DkR))-9&x C:Sz*+y).D-E$$ 	
""7==?###	"	"9%I&
	
 ! (06 
1 	8	88 34
%DEGGE*@	
	$	 5$ 5%.5&/!BC/ D/ 5%.65/!BC& D& ;/ACWX		r#   