Ë
    ômxiæ  ã                   óœ  — d dl Z d dlmZ d dlmZmZmZ d dlmZ	 d dl
mZ e j                  j                  gZ e j                  d«      Zej#                  e j                  j%                   eej&                  «       ed«      k(  xr  e«       dej&                  › �¬«      «       e j                  j)                  d	«       G d
„ d«      «       Zy)é    N)Úis_platform_arm)Ú	DataFrameÚSeriesÚoption_context)ÚVersionÚnumbaz0.61z&Segfaults on ARM platforms with numba )ÚreasonÚ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/home/htdocs/ttos/venv/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³¼uÀQÈ»{ÑKÓLˆØ"'°XÈ8ÑTˆØ�Z‰Z˜ $ˆZÓ'ˆØ"”˜˜TÓ"ð 
Ø¨-ñ
Ø;Añ
ˆð %”7˜2˜tÓ$Ñ. vÑ.ˆÜ
×Ñ˜f hÕ/ó    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                r0   Útest_cython_vs_numba_getitemz'TestEngine.test_cython_vs_numba_getitem(   s˜   € ð 2‰ˆˆfÜš\´°a³¼uÀQÈ»{ÑKÓLˆØ"'°XÈ8ÑTˆØ�Z‰Z˜ $ˆZÓ'¨Ñ,ˆØ"”˜˜TÓ"ð 
Ø¨-ñ
Ø;Añ
ˆð %”7˜2˜tÓ$Ñ. vÑ.ˆÜ
×Ñ˜v xÕ0r2   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%   r4   )r(   r   r   r   r   r)   r*   r+   Úserr    r-   r.   r/   s                r0   Útest_cython_vs_numba_seriesz&TestEngine.test_cython_vs_numba_series5   sŠ   € ð 2‰ˆˆfÜ”U˜1“X¢Y°UÔ;ˆØ"'°XÈ8ÑTˆØ�[‰[˜q tˆ[Ó,ˆØ"”˜˜TÓ"ð 
Ø¨-ñ
Ø;Añ
ˆð %”7˜2˜tÓ$Ñ. vÑ.ˆÜ
×Ñ˜v xÕ0r2   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         r0   Útest_as_index_false_unsupportedz*TestEngine.test_as_index_false_unsupportedB   sv   € Ø1‰ˆˆfÜš\´°a³¼uÀQÈ»{ÑKÓLˆØ�Z‰Z˜ eˆZÓ,ˆÜ�]‰]Ô.Ð6FÔGñ 	8ØŒG�B˜ÓÑ7 WÐ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!   rA   rE   s         r0   Útest_axis_1_unsupportedz"TestEngine.test_axis_1_unsupportedI   su   € Ø1‰ˆˆfÜš\´°a³¼uÀQÈ»{ÑKÓLˆØ�Z‰Z˜ !ˆZÓ$ˆÜ�]‰]Ô.°hÔ?ñ 	8ØŒG�B˜ÓÑ7 WÐ7°Ò7÷	8÷ 	8ñ 	8úrG   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        r0   Útest_no_engine_doesnt_raisez&TestEngine.test_no_engine_doesnt_raiseP   s€   € äš\´°a³¼uÀQÈ»{ÑKÓLˆØ�Z‰Z˜‹_ˆô Ð/°Ó6ñ 	)Ø—&‘&˜#˜w˜Ó(ˆC÷	)à—6‘6˜3 ˜.Ó)ˆÜ
×Ñ˜c 8Õ,÷	)ð 	)ús   ÁBÂBN)	Ú__name__Ú
__module__Ú__qualname__r1   r5   r<   rF   rJ   rO   r!   r2   r0   r   r      s    „ ò0ò1ò1ò8ò8ó	-r2   r   )rB   Úpandas.compatr   Úpandasr   r   r   Úpandas._testingÚ_testingr%   Úpandas.util.versionr   ÚmarkÚ
single_cpuÚ
pytestmarkÚimportorskipr   ÚappendÚskipifÚ__version__Úfilterwarningsr   r!   r2   r0   ú<module>r`      s¾   ðÛ å )÷ñ õ
 Ý 'à�k‰k×$Ñ$Ð%€
àˆ×Ñ˜GÓ$€Ø 
× Ñ Ø
‡K�K×ÑÙ�×!Ñ!Ó"¡g¨f£oÑ5ÒK¹/Ó:KØ7¸×8IÑ8IÐ7JÐKð ó ôð ‡�×Ñ˜HÓ%÷?-ð ?-ó &ñ?-r2   