
    Mh                         S SK r S SKJrJrJr  S SKJr  \ R                  R                  r
\ R                  " S5        \ R                  R                  S5       " S S5      5       rg)    N)	DataFrameSeriesoption_contextnumbaignorec                   8    \ rS rSrS rS rS rS rS rS r	Sr
g	)

TestEngine   c                     Uu  pg[        / SQ[        S5      [        SS5      S.5      nX#US.n	UR                  SUS9n
[        X5      " SS	U	S
.UD6n[        X5      " S0 UD6n[        R
                  " X5        g )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                Q/var/www/html/env/lib/python3.13/site-packages/pandas/tests/groupby/test_numba.pytest_cython_vs_numba_frame%TestEngine.test_cython_vs_numba_frame   s     2\auQ{KL"'8TZZ$Z'" 
-
;A
 2$.v.
f/    c                     Uu  pg[        / SQ[        S5      [        SS5      S.5      nX#US.n	UR                  SUS9S	   n
[        X5      " SS
U	S.UD6n[        X5      " S0 UD6n[        R
                  " X5        g )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_getitem'TestEngine.test_cython_vs_numba_getitem   s     2\auQ{KL"'8TZZ$Z'," 
-
;A
 2$.v.
v0r1   c                     Uu  pg[        [        S5      / SQSS9nX#US.n	UR                  SUS9n
[        X5      " S
SU	S	.UD6n[        X5      " S
0 UD6n[        R
                  " X5        g )Nr   )r   r   r   foo)indexnamer   r   )levelr   r   r   r   )r   r    r!   r"   r#   r3   )r&   r   r   r   r   r'   r(   r)   serr   r+   r,   r-   s                r.   test_cython_vs_numba_series&TestEngine.test_cython_vs_numba_series,   s|     2U1XYU;"'8T[[qt[," 
-
;A
 2$.v.
v0r1   c                     Uu  p#[        / SQ[        S5      [        SS5      S.5      nUR                  SSS9n[        R                  " [
        S	S
9   [        XR5      " SSS0UD6  S S S 5        g ! , (       d  f       g = f)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_unsupported*TestEngine.test_as_index_false_unsupported9   sh    1\auQ{KLZZeZ,]].6FGB7W77 HGG   A//
A=c                     Uu  p#[        / SQ[        S5      [        SS5      S.5      nUR                  SSS9n[        R                  " [
        SS	9   [        XR5      " SS
S0UD6  S S S 5        g ! , (       d  f       g = f)Nr   r   r   r   r   r   )axiszaxis=1r@   r   r   r   rB   rF   s         r.   test_axis_1_unsupported"TestEngine.test_axis_1_unsupported@   sg    1\auQ{KLZZ!Z$]].h?B7W77 @??rI   c                 4   [        / SQ[        S5      [        SS5      S.5      nUR                  S5      n[        SS5         UR	                  S	S
05      nS S S 5        UR	                  S	S
05      n[
        R                  " WU5        g ! , (       d  f       N9= f)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_raise&TestEngine.test_no_engine_doesnt_raiseG   sy    \auQ{KLZZ_ /6&&#w(C 7663.)
c8, 76s   B		
Br   N)__name__
__module____qualname____firstlineno__r/   r4   r<   rG   rL   rR   __static_attributes__r   r1   r.   r	   r	      s     01188	-r1   r	   )rC   pandasr   r   r   pandas._testing_testingr#   mark
single_cpu
pytestmarkimportorskipfilterwarningsr	   r   r1   r.   <module>ra      s\     
 [[##
   G  H%?- ?- &?-r1   