
    Owg                         d dl Z d dlmZmZmZ d dlmZ e j                  j                  Z
 e j                  d       e j                  j                  d       G d d             Zy)    N)	DataFrameSeriesoption_contextnumbaignorec                   0    e Zd Zd Zd Zd Zd Zd Zd Zy)
TestEnginec                    |\  }}t        g dt        d      t        dd      d      }|||d}	|j                  d|      }
 t        |
|      dd	|	d
|} t        |
|      di |}t	        j
                  ||       y )N      r   r            abcnogilparallelnopythonr   sortr   engineengine_kwargs )r   rangegroupbygetattrtmassert_frame_equalselfr   r   r   r   numba_supported_reductionsfunckwargsdfr   gbresultexpecteds                V/var/www/horilla/myenv/lib/python3.12/site-packages/pandas/tests/groupby/test_numba.pytest_cython_vs_numba_framez%TestEngine.test_cython_vs_numba_frame   s     2f\auQ{KL"'X8TZZ$Z'"T" 
-
;A
 %72t$.v.
fh/    c                    |\  }}t        g dt        d      t        dd      d      }|||d}	|j                  d|      d	   }
 t        |
|      dd
|	d|} t        |
|      di |}t	        j
                  ||       y )Nr   r   r   r   r   r   r   r   r   r   r   r   )r   r   r    r!   r"   assert_series_equalr$   s                r-   test_cython_vs_numba_getitemz'TestEngine.test_cython_vs_numba_getitem   s     2f\auQ{KL"'X8TZZ$Z',"T" 
-
;A
 %72t$.v.
vx0r/   c                     |\  }}t        t        d      g dd      }|||d}	|j                  d|      }
 t        |
|      d
d|	d	|} t        |
|      d
i |}t	        j
                  ||       y )Nr   )r   r   r   foo)indexnamer   r   )levelr   r   r   r   )r   r   r    r!   r"   r1   )r%   r   r   r   r   r&   r'   r(   serr   r*   r+   r,   s                r-   test_cython_vs_numba_seriesz&TestEngine.test_cython_vs_numba_series,   s     2fU1XYU;"'X8T[[qt[,"T" 
-
;A
 %72t$.v.
vx0r/   c                    |\  }}t        g dt        d      t        dd      d      }|j                  dd      }t        j                  t
        d	
      5   t        ||      dddi| d d d        y # 1 sw Y   y xY w)Nr   r   r   r   r   r   F)as_indexzas_index=Falsematchr   r   r   r   r   r    pytestraisesNotImplementedErrorr!   r%   r&   r'   r(   r)   r*   s         r-   test_as_index_false_unsupportedz*TestEngine.test_as_index_false_unsupported9   sv    1f\auQ{KLZZeZ,]].6FG 	8GB7W77	8 	8 	8   A66A?c                    |\  }}t        g dt        d      t        dd      d      }|j                  dd      }t        j                  t
        d	      5   t        ||      dd
di| d d d        y # 1 sw Y   y xY w)Nr   r   r   r   r   r   )axiszaxis=1r<   r   r   r   r>   rB   s         r-   test_axis_1_unsupportedz"TestEngine.test_axis_1_unsupported@   su    1f\auQ{KLZZ!Z$]].h? 	8GB7W77	8 	8 	8rD   c                 (   t        g dt        d      t        dd      d      }|j                  d      }t        dd      5  |j	                  d	d
i      }d d d        |j	                  d	d
i      }t        j                  |       y # 1 sw Y   3xY w)Nr   r   r   r   r   r   zcompute.use_numbaTr   first)r   r   r    r   aggr"   r#   )r%   r)   r*   resr,   s        r-   test_no_engine_doesnt_raisez&TestEngine.test_no_engine_doesnt_raiseG   s    \auQ{KLZZ_ /6 	)&&#w(C	)663.)
c8,	) 	)s   BBN)	__name__
__module____qualname__r.   r2   r9   rC   rG   rL   r   r/   r-   r	   r	      s     01188	-r/   r	   )r?   pandasr   r   r   pandas._testing_testingr"   mark
single_cpu
pytestmarkimportorskipfilterwarningsr	   r   r/   r-   <module>rX      s^     
 [[##
   G  H%?- ?- &?-r/   