
    m*i                        S SK r S SKJr  S SKJrJrJr  S SKJr	  S SK
Jr  \ R                  R                  /r\ R                  " S5      r\R#                  \ R                  R%                  \" \R&                  5      \" S5      :H  =(       a    \" 5       S\R&                   3S95        \ R                  R)                  S	5       " S
 S5      5       rg)    N)is_platform_arm)	DataFrameSeriesoption_context)Versionnumbaz0.61z&Segfaults on ARM platforms with numba )reason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                _/var/www/html/land-doc-ocr/venv/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                r1   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0r4   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&   r6   )r)   r   r   r   r   r*   r+   r,   serr!   r.   r/   r0   s                r1   test_cython_vs_numba_series&TestEngine.test_cython_vs_numba_series5   s|     2U1XYU;"'8T[[qt[," 
-
;A
 2$.v.
v0r4   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         r1   test_as_index_false_unsupported*TestEngine.test_as_index_false_unsupportedB   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=1rC   r    r   r"   rE   rI   s         r1   test_axis_1_unsupported"TestEngine.test_axis_1_unsupportedI   sg    1\auQ{KLZZ!Z$]].h?B7W77 @??rL   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.   resr0   s        r1   test_no_engine_doesnt_raise&TestEngine.test_no_engine_doesnt_raiseP   sy    \auQ{KLZZ_ /6&&#w(C 7663.)
c8, 76s   B		
Br"   N)__name__
__module____qualname____firstlineno__r2   r7   r?   rJ   rO   rU   __static_attributes__r"   r4   r1   r   r      s     01188	-r4   r   )rF   pandas.compatr   pandasr   r   r   pandas._testing_testingr&   pandas.util.versionr   mark
single_cpu
pytestmarkimportorskipr   appendskipif__version__filterwarningsr   r"   r4   r1   <module>ri      s     ) 
  'kk$$%
G$ 
  
KK!!"gfo5K/:K78I8I7JK   H%?- ?- &?-r4   