
    Pi'                        d dl mZ d dlZd dlZd dlZd dlmZ d dlm	Z	 d dl
mZmZ d dlmZ d dlmZmZ ej                            dg d          ej                            d	 ej        i d
           ej        ddid           ej        ddid           ej        dddd          g          ej                            dddg          d                                     Zd Zej                            dg d          ej                            dddg          d                         Zej                            ed          ej                            dg d          ej                            dddg          d                                      Zej                            d! e ej         eej        "          d#d$ej        fd%ej        fd&d'd(ej        fd)d*ej        fd+ej        fd,ej        fg
           ej         eej        dd-d.g          g d/           ej         ed0                                          g d1                              d2             Zej                            d! e ej         eej        "          d3 eg ej        "          fd4 eg ej        "          fg           ej         eej        dd-d.g          d3 eej        dd-d5g          fd4 eej        dd.d5g          fg           ej         ed0                                          d4 eg d6          fg                              d7             Z ej                            d8 e ej         e            d9 ed:"          fd$ ed:"          fd% ed:"          fd; ee!"          fd< ee!"          fd( ed:"          fd= ed:"          fd* ed:"          fd+ ed:"          fd, ed:"          fg
           ej         eej        dgdd-gg          d9 ed>d.g          fd$ ed>d-g          fd% ed>dg          fd; eddg          fd< eddg          fd( edd?g          fd= ed>d-g          fd* eej        d@g          fd+ eej        dAg          fd, edd?g          fg
                              dB             Z"ej                            d8 e ej         e            d3 e            fd4 e            fg           ej         eej        dgdd-gg          d3 eej        dgdd-gg          fd4 eej        dgdd.gg          fg                              dC             Z#ej                            de          dD             Z$ej                            de          dE             Z%ej                            dFg dG          dH             Z&dS )I    )chainN)WASM)	is_number)	DataFrameSeries)frame_transform_kernelsseries_transform_kernelsfunc)summeanminmaxstdkwdsno_kwds)idaxis   on_axisnumeric_onlyT	func_kwds)r   r   axis_and_func_kwdshowaggapplyc                      t          | |          |fi |} t          | |          di |}t          j        ||           d S )N )getattrtmassert_series_equal)float_framer
   r   r   resultexpecteds         o/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/pandas/tests/apply/test_str.pytest_apply_with_string_funcsr%      s]     'W[#&&t44t44F)w{D))11D11H68,,,,,    c                 p    |                      |          } t          | |                      }||k    sJ d S )N)r   r   )datetime_seriesall_numeric_reductionsr"   r#   s       r$   test_with_string_argsr*   )   sE    ""#9::F?w(>??AAHXr&   op)r   medianr   varc                    t          ddgddgd          } t          ||          |           }| dv rddini }t           t          t          |           |fdd	i||j        
          }t          j        ||           d S )Nr            )ab)r   r-   ddofr   r   )index)r   r   r   npcolumnsr   r    )r+   r   r!   r"   kwargsr#   s         r$   test_apply_np_reducerr9   /   s     1a&1v6677K&W[#&&r**F.00fa[[bFB66!6v66k>Q  H 68,,,,,r&   zNo fp exception support in wasm)reason)absceilcoscumsumexplogsqrtsquare	transformc                 (   d| j         d<   d }|dv rt          }t          j        |d          5   t	          | |          |          } t	          t
          |          |           }d d d            n# 1 swxY w Y   t          j        ||           d S )Ng      )r   r   )r@   rA   F)check_stacklevel)ilocRuntimeWarningr   assert_produces_warningr   r6   assert_frame_equal)r!   r+   r   warnr"   r#   s         r$   test_apply_np_transformerrK   =   s     "KTD	_		#D5	A	A	A 0 0 +c**2.."72r??;//	0 0 0 0 0 0 0 0 0 0 0 0 0 0 0
 &(+++++s   8A22A69A6zseries, func, expected)dtype)r   r   r   r   allT)anyFr   )prodr   r   r-   r,   r/   r0   )
)r      )r   r0   )r   r   rM   rO   T)r   r/   )rP   rQ   )r   r   )r-   r   )r,   r/   za b c))r   abc)r   c)r   r2   rM   rR   c                 ,   t          |t                    rd nt          }t          j        |d          5  |                     |          }d d d            n# 1 swxY w Y   t          |          rt          j        ||d          sJ d S ||k    sJ d S )Nis currently using Series.*matchT)	equal_nan)	
isinstancestrFutureWarningr   rH   r   r   r6   iscloseseriesr
   r#   rJ   r"   s        r$   test_agg_cython_table_seriesr`   T   s    b dC((;44mD		#D0M	N	N	N " "D!!" " " " " " " " " " " " " " " "z&(d;;;;;;;;!!!!!!   AAAcumprodr>   rQ   )r2   abrS   c                     t          |t                    rd nt          }t          j        |d          5  |                     |          }d d d            n# 1 swxY w Y   t          j        ||           d S )NrV   rW   rZ   r[   r\   r   rH   r   r    r^   s        r$   &test_agg_cython_table_transform_seriesrf      s    4 dC((;44mD		#D0M	N	N	N " "D!!" " " " " " " " " " " " " " "68,,,,,ra   zdf, func, expectedr   float64rN   rO   rP   g      ?g      ?g۞?g      ?c                     t          |t                    rd nt          }t          j        |d          5  |                     ||          }d d d            n# 1 swxY w Y   t          j        ||           d S )Nis currently using DataFrame.*rW   r   re   dfr
   r#   r   rJ   r"   s         r$   test_agg_cython_table_framerm      s    N dC((;44mD		#D0P	Q	Q	Q ) )4(() ) ) ) ) ) ) ) ) ) ) ) ) ) ) 68,,,,,s   AA Ac                 (   |dv r|                     d          }t          |t                    rd nt          }t	          j        |d          5  |                     ||          }d d d            n# 1 swxY w Y   t	          j        ||           d S )N)r7   r   rg   ri   rW   rj   )astyperZ   r[   r\   r   rH   r   rI   rk   s         r$   %test_agg_cython_table_transform_framerp      s    & ~??9--dC((;44mD		#D0P	Q	Q	Q ) )4(() ) ) ) ) ) ) ) ) ) ) ) ) ) ) &(+++++s   A22A69A6c                    |dk    r9|                      t          j                            t          d                     |dk    rdgng }t          j        |j        d                   }|dk    rt          nd }d}t          j
        ||          5   |                    |          j        |g|R  }d d d            n# 1 swxY w Y    |j        |dg|R  }t          j        ||           d S )	Nngroupngroup not valid for NDFrameraisesr:   fillna        r   z"SeriesGroupBy.fillna is deprecatedrW   )applymarkerpytestmarkxfail
ValueErrorr6   onesshaper\   r   rH   groupbyrC   r    )	requeststring_seriesr+   argsr}   rJ   msgr#   r"   s	            r$   $test_transform_groupby_kernel_seriesr      sY    
X~~KZ8VWW	
 	
 	
 (NNC55D7=&q)**D(NN==D
.C		#D	4	4	4 D D8=((..8CdCCCD D D D D D D D D D D D D D D$]$R2T222F68,,,,,s   "B??CCc                 >   |dk    r9|                      t          j                            t          d                     |dk    rdgng }t          j        |j        d                   }|                    |          }|dk    rt          nd }d}t          j        ||          5   |j        |g|R  }d d d            n# 1 swxY w Y    |j        |dg|R  }	t          j        |	|           d	|j        vsJ |d
                                         |d	<   t!          |j        j                  dk    sJ t          j        |j        d                   }|                    |          }
 |
j        |g|R  } |j        |dg|R  }t          j        ||           d S )Nrr   rs   rt   rv   rw   r   z%DataFrameGroupBy.fillna is deprecatedrW   EAr   )rx   ry   rz   r{   r|   r6   r}   r~   r   r\   r   rH   rC   rI   r7   copylen_mgrblocks)r   r!   r+   r   r}   gbrJ   op_msgr#   r"   gb2	expected2result2s                r$   #test_transform_groupby_kernel_framer     s   	X~~KZ8VWW	
 	
 	
 (NNC55D7;$Q'((D			T	"	"B(NN==D4F		#D	7	7	7 + +2<*T***+ + + + + + + + + + + + + + + #["2q04000F&(+++ k)))))"3',,..K{&''!++++7;$Q'((D


d
#
#Cb(4(((I#k#B1D111G'9-----s   &CCCmethod)r;   shift
pct_changer>   rankc                     t          dddgi          }|                    |           } t          j        |           |          }t	          j        ||           d S )Nr   r/   )r   rC   operatormethodcallerr   rI   )r   rl   r"   r#   s       r$   test_transform_method_namer   -  s]     
C"a>	"	"B\\&!!F,x$V,,R00H&(+++++r&   )'	itertoolsr   r   numpyr6   ry   pandas.compatr   pandas.core.dtypes.commonr   pandasr   r   pandas._testing_testingr   pandas.tests.apply.commonr   r	   rz   parametrizeparamr%   r*   r9   skipifrK   get_cython_table_paramsrg   nansplitr`   rf   boolrm   rp   r   r   r   r   r&   r$   <module>r      s                    0 / / / / /                     !E!E!EFF
RI&&&fa[Y///nd+<<<a66;OPPP	   011- - 21  GF-   ???@@ 011	- 	- 21 A@	- D!BCC
J
J
J  g 677, , 87  DC
,$ 	E""F$$$ 26"	
 	
 	#"FBFAq!$%%  	
 	
 	#"F7==??##  		
 		
?) ), ,Z
" 
"[, ,Z
" 	E""F$$$FF2RZ888966"BJ7778	
 	
 	#"FBFAq!$%%FFBFAq!#455666261a"3445	
 	
 	#"F7==??##x8J8J8J1K1K&L%M	
 	
  ,- -- ,- 	E""IKKY///0Y///0Y///0T***+T***+i0001i0001Y///0Y///066	2223	
 	
 	#"I{QF+,,Qx(()Qx(()Qx(()d|,,-d|,,-C))*a))*1223}--.661c(++,	
 	
! " "F- -G" "F- 	E""IKK9iikk2Xyy{{4KL	
 	
 	#"I{QF+,,II{QF&;<<=99rvqkAq6%:;;<	
 	
	  , , , 788- - 98-" 677. . 87.@ #S#S#STT, , UT, , ,r&   