U
|
ý°d ã @ s° d dl Z d dlmZ d dlm mZ d dlmZm Z m
|
Z
|
d dl m Z e d¡dd Ze d¡dd Ze d¡e j d ¡e j d
|
d d g¡e j d ddg¡e j dd d g¡dd Ze d¡e j d ¡e j d
|
d d g¡e j d ddg¡dd Ze d¡dd Ze d¡e j dddgdddgddg¡d d Ze d¡d d Ze d¡d d! Ze d¡d"d# Ze d¡e j d ¡d$d% Ze d¡d&d' Z dS )(é N)ÚNumbaUtilError)Ú DataFrameÚSeriesÚoption_contextÚnumbac C s dd } t dddddgddddd gd
|
d d gd }tjtdd | d ¡j| dd W 5 Q R X tjtdd | d ¡d j| dd W 5 Q R X d S )Nc S s | d S ©Né © ©Úxr r ú`d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/groupby/transform/test_numba.pyÚincorrect_function s z;test_correct_function_signature.<locals>.incorrect_functionÚaÚbç ð?ç @ç @ç @ç @©ÚkeyÚdatar r ©Úcolumnsz The first 2©Úmatchr ©Úengine©r ÚpytestÚraisesr ÚgroupbyÚ transform©r r r r r Ú test_correct_function_signature s þ r$ c C s dd } t dddddgddddd gd
|
d d gd }tjtdd | d ¡j| ddd W 5 Q R X tjtdd | d ¡d j| ddd W 5 Q R X d S )Nc S s | d S r r ©ÚvaluesÚindexr r r r s z6test_check_nopython_kwargs.<locals>.incorrect_functionr r r r r r r r r r r znumba does not supportr r r )r r r r# r r r Útest_check_nopython_kwargs s þ r( ÚignoreÚjitTFÚ
|
pandas_objr r Úas_indexc C s¤ dd }| r dd l }| |¡}tdddddgdddd d
|
gd dd gd }|||d} |jd|d}
|
|dkrr|
|
d }
|
|
|
j|d| d} |
|
jdd dd} t | | ¡ d S )Nc S s | d S r r r% r r r Úfunc5 s z"test_numba_vs_cython.<locals>.funcr r r r r r r r ©r r r r ©ÚnogilÚparallelÚnopython)r, r r ©r Ú engine_kwargsc S s | d S r r r
|
r r r Ú<lambda>G ó z&test_numba_vs_cython.<locals>.<lambda>Úcythonr ©r r* r r! r" ÚtmZ assert_equal) r* r+ r0 r1 r2 r, r- r r r4 ÚgroupedÚresultÚexpectedr r r Útest_numba_vs_cython. s
|
ÿ r= c C s dd }dd }| r0dd l }| |¡}| |¡}tdddddgdd d
|
d d gd ddgd}|||d} | d¡}
|
|dkr|
|
d }
|
|
|
j|d| d} |
|
jdd dd} t | | ¡ |
|
j|d| d} |
|
jdd dd} t | | ¡ |
|
j|d| d} |
|
jdd dd} t | | ¡ d S )Nc S s | d S r r r% r r r Úfunc_1S s ztest_cache.<locals>.func_1c S s | d S ©Né r r% r r r Úfunc_2V s ztest_cache.<locals>.func_2r r r r r r r r r. r r r/ r r r3 c S s | d S r r r
|
r r r r5 h r6 z test_cache.<locals>.<lambda>r7 r c S s | d S r? r r
|
r r r r5 l r6 c S s | d S r r r
|
r r r r5 q r6 r8 ) r* r+ r0 r1 r2 r> rA r r r4 r: r; r< r r r Ú
|
test_cacheL s.
|
|
ÿ
|
rB c C s~ dd } t dddddgddddd gd
|
d d gd }| d ¡}|j| dd}tdd |j| d d}W 5 Q R X t ||¡ d S )Nc S s | d S r r r% r r r r> w s z&test_use_global_config.<locals>.func_1r r r r r r r r. r r r r r zcompute.use_numbaT)r r! r" r r9 Úassert_frame_equal)r> r r: r< r; r r r Útest_use_global_configu s ÿ
|
rD Úagg_funcÚminÚmaxÚsum)ÚBÚCc C s t dddddgdddddgdd d
|
gd }| d ¡}tjtd d |j| dd W 5 Q R X tjtd d |d
|
j| dd W 5 Q R X d S )Nr r r r r r r r. r r r zNumba engine canr r r )r r! r r ÚNotImplementedErrorr" )rE r r: r r r Ú test_multifunc_notimplimented s ÿ
|
rL c C s dd } t ddddgddddgd}| d¡d }|j| ddd }td
|
gd dd }t ||¡ |j| d dd }tdgd dd }t ||¡ d S )Nc S s | | d
¡ S ©N)rH )r& r' Únr r r Úsum_last s z&test_args_not_cached.<locals>.sum_lastr r )Úidr rP r r r r é )Únameé r )r r! r" r r9 Zassert_series_equal)rO ÚdfZ grouped_xr; r< r r r Útest_args_not_cached s rU c C sl dd } t dddgdddgdd d
|
d gd }| d ¡j| dd}t dddgdgd d
|
d gd}t ||¡ d S )Nc S s |d S r r r% r r r Úf¨ s z+test_index_data_correctly_passed.<locals>.fÚArI rQ r@ é )ÚgroupÚvéÿÿÿÿéþÿÿÿéýÿÿÿ)r' rY r r g Àg Àg ÀrZ )r r' ©r r! r" r9 rC )rV rT r; r< r r r Ú test_index_data_correctly_passed¥ s
|
$ r_ c sº d dd fdd} d}t ddddgi}|jddj| d |d
|
}t dd d d gi}t ||¡ d d}|jddj| d |d
|
}t dd d d gi}t ||¡ d S ) NTFc s S rM r r% ©r0 r2 r1 r r Ú func_kwargs¹ s z2test_engine_kwargs_not_cached.<locals>.func_kwargs©r2 r0 r1 Úvaluer )Úlevelr r3 r r r^ )ra r4 rT r; r< r r` r Ú test_engine_kwargs_not_cached± s, ÿ ÿre c C sr dd }t ddddg ddg¡}|| |d }| d¡j|d
|
|d }t ddd dg ddg¡}t ||¡ d S ) Nc S s dS r r r% r r r Ú
|
numba_funcÐ s z+test_multiindex_one_key.<locals>.numba_funcr rS é ©rW rI rJ rW rI rb r r3 r )r Ú set_indexr! r" r9 rC )r0 r1 r2 rf rT r4 r; r< r r r Útest_multiindex_one_keyÍ s
|
ÿ rj c C sh dd }t ddddg ddg¡}|| |d }tjtd
|
d | ddg¡j|d |d W 5 Q R X d S )Nc S s dS r r r% r r r rf Þ s z;test_multiindex_multi_key_not_supported.<locals>.numba_funcr rS rg rh rW rI rb zMore than 1 grouping labelsr r r3 )r ri r r rK r! r" )r0 r1 r2 rf rT r4 r r r Ú'test_multiindex_multi_key_not_supportedÜ s ÿrk ) r Z pandas.errorsr Z pandas.util._test_decoratorsÚutilZ_test_decoratorsÚtdZpandasr r r Zpandas._testingZ_testingr9 Z
|
skip_if_nor$ r( ÚmarkÚfilterwarningsZ parametrizer= rB rD rL rU r_ re rj rk r r r r Ú<module> sL
|
|
|
|
$
|
ÿ
|
|
|
|
|
