U
|
ý°dÀ ã @ s^ d dl Z d dlm mZ d dlmZmZ d dlm Z
|
e d¡e j d¡G dd dZdS )é N)Ú DataFrameÚSeriesÚnumbaÚignorec @ s4 e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd S ) Ú
|
TestEnginec C s |\}}t ddddgtdtddd}|||d} |jd|d }
|
t|
|
|f d
|
| d |} t|
|
|f |} |d k} tj| | | d d S )Né é é é é ©ÚaÚbÚc©ÚnogilÚparallelÚnopythonr ©Úsortr ©ÚengineÚ engine_kwargs©ÚsumÚminÚmax©Ú check_dtype)r ÚrangeÚgroupbyÚgetattrÚtmZassert_frame_equal©Úselfr r r r Únumba_supported_reductionsÚfuncÚkwargsÚdfr ÚgbÚresultÚexpectedr © r, úVd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/groupby/test_numba.pyÚtest_cython_vs_numba_frame s "
|
ÿÿz%TestEngine.test_cython_vs_numba_framec C s |\}}t ddddgtdtddd}|||d} |jd|d d
|
}
|
t|
|
|f d | d |} t|
|
|f |} |d k} tj| | | d d S )Nr r r r
|
r r r r r r r r r r )r r r r! r" Úassert_series_equalr# r, r, r- Ú test_cython_vs_numba_getitem s "
|
ÿÿz'TestEngine.test_cython_vs_numba_getitemc C s |\}}t tddddgdd}|||d} |jd|d}
|
t|
|
|f d | d
|
|} t|
|
|f |} |d k} tj| | | d d S ) Nr r
|
r Zfoo)ÚindexÚnamer r )Úlevelr r r r r )r r r r! r" r/ )r$ r r r r r% r& r' Zserr r) r* r+ r r, r, r- Útest_cython_vs_numba_series. s
|
ÿÿz&TestEngine.test_cython_vs_numba_seriesc C sn |\}}t ddddgtdtddd}|jddd }tjtd
|
d t||f d d i| W 5 Q R X d S )Nr r r r
|
r r r F)Zas_indexzas_index=False©Úmatchr r ©r r r ÚpytestZraisesÚNotImplementedErrorr! ©r$ r% r& r' r( r) r, r, r- Ú test_as_index_false_unsupported= s
|
"z*TestEngine.test_as_index_false_unsupportedc C sn |\}}t ddddgtdtddd}|jddd}tjtd d
|
t||f d d i| W 5 Q R X d S ) Nr r r r
|
r r r )Zaxiszaxis=1r5 r r r7 r: r, r, r- Útest_axis_1_unsupportedD s
|
"z"TestEngine.test_axis_1_unsupportedN)Ú__name__Ú
|
__module__Ú __qualname__r. r0 r4 r; r< r, r, r, r- r s
|
r )r8 Z pandas.util._test_decoratorsÚutilZ_test_decoratorsÚtdZpandasr r Zpandas._testingZ_testingr" Z
|
skip_if_noÚmarkÚfilterwarningsr r, r, r, r- Ú<module> s
|
|
