
    Wh                     f   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
  ej        ddg          d             Z ej        ddg          d             Z ej        ddg          d	             Z ej        ddg          d
             Zej        d             Zej        d             Zej        d             Zej        d             Zej        d             Zej        d             Zej        d             Z ej                    d             Z ej                    d             Z ej         ee	                    d             Z ej         ee
                    d             Z ej         ee	           ee
          z             d             Z ej        ddg          d             Z ej        dg          d             Z ej        dg          d             Z ej        di fdddifddd ifdddifddd ifdi fd i fd!i fgg d"#          d$             Z dS )%    N	DataFrame)reduction_kernelstransformation_kernelsTF)paramsc                     | j         S Nparamrequests    c/var/www/html/movieo_spanner_bot/venv/lib/python3.11/site-packages/pandas/tests/groupby/conftest.pysortr      
    =    c                     | j         S r	   r
   r   s    r   as_indexr      r   r   c                     | j         S r	   r
   r   s    r   dropnar      r   r   c                     | j         S r	   r
   r   s    r   observedr      r   r   c                     | S r	    ) multiindex_dataframe_random_datas    r   mframer       s    ++r   c                      t          g dg dt          j                            d          t          j                            d          d          S )Nfoobarr   r   r   r   r   r   oner!   twothreer"   r"   r!   r#      ABCDr   nprandomrandnr   r   r   dfr.   %   sR    IIIMMM####		
 	
  r   c                  (    t          j                    S r	   )tmmakeTimeSeriesr   r   r   tsr2   1   s    r   c                  (    t          j                    S r	   )r0   getTimeSeriesDatar   r   r   tsdr5   6   s    !!!r   c                      t          |           S r	   r   )r5   s    r   tsframer7   ;   s    S>>r   c            
          t          g dg dt          j                            d          t          j        t          j                            d          d          d          S )Nr   r    r$   float32)dtyper%   )r   r+   r,   r-   arrayr   r   r   df_mixed_floatsr<   @   sa    IIIMMM##")//!,,I>>>		
 	
  r   c            
          t          g dg dg dt          j                            d          t          j                            d          t          j                            d          d          S )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&   r'   r(   r)   EFr*   r   r   r   three_grouprC   L   s           $$$$$$U+	
 +	
- - -r   c            	          g dg dg dg dg dg dg dg dg} t          | g d	
          }|                    d          S )N)r   aa0_at_0)   bb0_at_1)   rE   a1_at_2)   rH   b1_at_3)   cc0_at_4)   rE   a2_at_5)   rE   a3_at_6)   rE   a4_at_7)IndexGroupValue)columnsrW   )r   	set_index)datar.   s     r   slice_test_dfr]   ~   s     		D 
4!<!<!<	=	=	=B<<   r   c                 0    |                      dd          S )NrX   F)r   )groupby)r]   s    r   slice_test_groupedr`      s      5 999r   c                     | j         S )zT
    yields the string names of all groupby reduction functions, one at a time.
    r
   r   s    r   reduction_funcrb      s    
 =r   c                     | j         S )z@yields the string names of all groupby transformation functions.r
   r   s    r   transformation_funcrd           =r   c                     | j         S )z5yields both aggregation and transformation functions.r
   r   s    r   groupby_funcrg      re   r   c                     | j         S )z'parallel keyword argument for numba.jitr
   r   s    r   parallelri      re   r   c                     | j         S )z$nogil keyword argument for numba.jitr
   r   s    r   nogilrk      re   r   c                     | j         S )z'nopython keyword argument for numba.jitr
   r   s    r   nopythonrm      re   r   meanvarddofrG   stdsumminmax)rn   var_1var_0std_1std_0rr   rs   rt   )r   idsc                     | j         S )z(reductions supported with engine='numba'r
   r   s    r   numba_supported_reductionsr{      s     =r   )!numpyr+   pytestpandasr   pandas._testing_testingr0   pandas.core.groupby.baser   r   fixturer   r   r   r   r   r.   r2   r5   r7   r<   rC   r]   r`   sortedrb   rd   rg   ri   rk   rm   r{   r   r   r   <module>r      s                           e}%%%  &% e}%%%  &% e}%%%  &% e}%%%  &% , , ,       " " "       . . .b ! ! ! : : : vv/00111  21 vv455666  76
 vv/0066:P3Q3QQRRR  SR
 e}%%%  &% w   
 v  
 									 	JII      r   