a Pf @sXddlZddlmZmZmZddlmZejj Z e dej dGdddZ dS)N) DataFrameSeriesoption_contextnumbaignorec@s<eZdZddZddZddZddZd d Zd d Zd S) TestEnginec Cs~|\}}tgdtdtddd}|||d} |jd|d} t| |fd | d |} t| |fi|} t| | dS) Nr r abcnogilparallelnopythonrsortrengine engine_kwargs)rrangegroupbygetattrtmassert_frame_equal selfrrrrnumba_supported_reductionsfunckwargsdfrgbresultexpectedr)X/var/www/ai-form-bot/venv/lib/python3.9/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| |fd | d |} t| |fi|} t| | dS) Nrr r r rrrrrrr)rrrrrassert_series_equalr r)r)r*test_cython_vs_numba_getitems  z'TestEngine.test_cython_vs_numba_getitemc Csv|\}}ttdgddd}|||d} |jd|d} t| |fd| d |} t| |fi|} t| | dS) Nr )r r r Zfoo)indexnamerr)levelrrr)rrrrrr,) r!rrrrr"r#r$Zserrr&r'r(r)r)r*test_cython_vs_numba_series,s  z&TestEngine.test_cython_vs_numba_seriescCs~|\}}tgdtdtddd}|jddd}tjtd d (t||fd d i|Wdn1sp0YdS) Nrr r r rrF)Zas_indexzas_index=FalsematchrrrrrpytestZraisesNotImplementedErrorrr!r"r#r$r%r&r)r)r*test_as_index_false_unsupported9s z*TestEngine.test_as_index_false_unsupportedcCs~|\}}tgdtdtddd}|jddd}tjtdd (t||fd d i|Wdn1sp0YdS) Nrr r r rr)Zaxiszaxis=1r2rrr4r7r)r)r*test_axis_1_unsupported@s z"TestEngine.test_axis_1_unsupportedcCs~tgdtdtddd}|d}tdd|d d i}Wdn1sV0Y|d d i}t||dS) Nrr r r rrzcompute.use_numbaTrfirst)rrrrZaggrr)r!r%r&resr(r)r)r*test_no_engine_doesnt_raiseGs   ,z&TestEngine.test_no_engine_doesnt_raiseN) __name__ __module__ __qualname__r+r-r1r8r9r<r)r)r)r*rs    r)r5ZpandasrrrZpandas._testingZ_testingrmarkZ single_cpuZ pytestmarkZ importorskipfilterwarningsrr)r)r)r*s