o h @sXddlZddlmZmZmZddlmZejj Z e dej dGdddZ dS)N) DataFrameSeriesoption_contextnumbaignorec@s<eZdZddZddZddZddZd d Zd d Zd S) TestEnginec Cs~|\}}tgdtdtddd}|||d} |jd|d} t| |d d | d |} t| |d i|} t| | dS) Nr r abcnogilparallelnopythonrsortrengine engine_kwargs)rrangegroupbygetattrtmassert_frame_equal selfrrrrnumba_supported_reductionsfunckwargsdfrgbresultexpectedrrQ/var/www/html/env/lib/python3.10/site-packages/pandas/tests/groupby/test_numba.pytest_cython_vs_numba_frames  z%TestEngine.test_cython_vs_numba_framec Cs|\}}tgdtdtddd}|||d} |jd|dd } t| |d d | d |} t| |d i|} t| | dS) Nrr r r rrrrrrrr)rrrrrassert_series_equalr!rrr*test_cython_vs_numba_getitems  z'TestEngine.test_cython_vs_numba_getitemc Csv|\}}ttdgddd}|||d} |jd|d} t| |d d| d |} t| |d i|} t| | dS) Nr )r r r foo)indexnamerr)levelrrrr)rrrrrr,) r"rrrrr#r$r%serrr'r(r)rrr*test_cython_vs_numba_series,s  z&TestEngine.test_cython_vs_numba_seriescCs|\}}tgdtdtddd}|jddd}tjtd d t||d d d i|WddS1s9wYdS)Nrr r r rrF)as_indexzas_index=FalsematchrrrrrrpytestraisesNotImplementedErrorrr"r#r$r%r&r'rrr*test_as_index_false_unsupported9 "z*TestEngine.test_as_index_false_unsupportedcCs|\}}tgdtdtddd}|jddd}tjtdd t||d d d i|WddS1s9wYdS) Nrr r r rr)axiszaxis=1r5rrrr7r;rrr*test_axis_1_unsupported@r=z"TestEngine.test_axis_1_unsupportedcCs~tgdtdtddd}|d}tdd|d d i}Wdn1s+wY|d d i}t||dS) Nrr r r rrzcompute.use_numbaTrfirst)rrrraggrr )r"r&r'resr)rrr*test_no_engine_doesnt_raiseGs  z&TestEngine.test_no_engine_doesnt_raiseN) __name__ __module__ __qualname__r+r-r3r<r?rCrrrr*rs    r)r8pandasrrrpandas._testing_testingrmark single_cpu pytestmark importorskipfilterwarningsrrrrr*s