gL i ddlZddlmZddlmZmZmZddlmZ ddl m Z ejjgZejdZej#ejj%e ej&e dk(xredej&ejj)d Gd d Zy) N)is_platform_arm) DataFrameSeriesoption_context)Versionnumbaz0.61z&Segfaults on ARM platforms with numba )reasonignorec0eZdZdZdZdZdZdZdZy) TestEnginec|\}}tgdtdtddd}|||d} |jd|} t| |d d | d |} t| |d i|} t j | | y) Nrrabcnogilparallelnopythonrsortrengine engine_kwargs)rrangegroupbygetattrtmassert_frame_equal selfrrrrnumba_supported_reductionsfunckwargsdfr gbresultexpecteds e/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/pandas/tests/groupby/test_numba.pytest_cython_vs_numba_framez%TestEngine.test_cython_vs_numba_frames2 f \auQ{K L"'X8T ZZ$Z '"T" - ;A %72t$.v. fh/c|\}}tgdtdtddd}|||d} |jd|d } t| |d d | d |} t| |d i|} t j | | y) Nrrrrrrrrrrrr!)rr"r#r$r%assert_series_equalr's r0test_cython_vs_numba_getitemz'TestEngine.test_cython_vs_numba_getitem(s2 f \auQ{K L"'X8T ZZ$Z ' ,"T" - ;A %72t$.v. vx0r2c|\}}ttdgdd}|||d} |jd|} t| |d d| d |} t| |d i|} t j | | y) Nr)rrrfoo)indexnamerr)levelrrrr!)rr"r#r$r%r4) r(rrrrr)r*r+serr r-r.r/s r0test_cython_vs_numba_seriesz&TestEngine.test_cython_vs_numba_series5s2 fU1XYU;"'X8T [[qt[ ,"T" - ;A %72t$.v. vx0r2c|\}}tgdtdtddd}|jdd}tjt d 5t ||d d d i|dddy#1swYyxYw)NrrrrrrF)as_indexzas_index=Falsematchrrr!rr"r#pytestraisesNotImplementedErrorr$r(r)r*r+r,r-s r0test_as_index_false_unsupportedz*TestEngine.test_as_index_false_unsupportedBsv1 f \auQ{K L ZZeZ , ]].6F G 8 GB  7W 7 7 8 8 8 A66A?c|\}}tgdtdtddd}|jdd}tjt d 5t ||d d d i|dddy#1swYyxYw) Nrrrrrr)axiszaxis=1r?rrr!rArEs r0test_axis_1_unsupportedz"TestEngine.test_axis_1_unsupportedIsu1 f \auQ{K L ZZ!Z $ ]].h ? 8 GB  7W 7 7 8 8 8rGc(tgdtdtddd}|jd}tdd5|j d d i}ddd|j d d i}t j |y#1swY3xYw) Nrrrrrrzcompute.use_numbaTrfirst)rr"r#raggr%r&)r(r,r-resr/s r0test_no_engine_doesnt_raisez&TestEngine.test_no_engine_doesnt_raisePs \auQ{K L ZZ_/ 6 )&&#w(C )663.) c8, ) )s BBN) __name__ __module__ __qualname__r1r5r<rFrJrOr!r2r0r r s  0 1 188 -r2r )rB pandas.compatrpandasrrrpandas._testing_testingr%pandas.util.versionrmark single_cpu pytestmark importorskiprappendskipif __version__filterwarningsr r!r2r0r`s ) 'kk$$ % G$  KK!!"gfo5K/:K78I8I7JKH%?-?-&?-r2