Ë
|
oñúh' ã óþ d dl Zd dlZd dlmZ d dlmZ d dlmZm Z m
|
Z
|
d dl m Z d dlmZ ej j" gZ ej& d« Zej+ ej j- eej. « ed« k( xr e« dej. ¬ « « d
|
Zd Zej j5 d « ej j7 d ddg« ej j7 dddg« ej j7 dddg« d « « « « Z ej j5 d « ej j7 d ddg« ej j7 dddg« d « « « Z d Z ej j7 dddgdddgddg« d « Z d Z d Z!d Z"ej j5 d « d « Z#d! Z$d" Z%d# Z&y)$é N)Úis_platform_arm)ÚNumbaUtilError)Ú DataFrameÚSeriesÚoption_context)ÚVersionÚnumbaz0.61z&Segfaults on ARM platforms with numba )Úreasonc ó® t j d« d } t g d¢g d¢dddg¬« }t j t d ¬
|
« 5 |j d« j | d¬ « d d d « t j t d ¬
|
« 5 |j d« d j | d¬ « d d d « y # 1 sw Y SxY w# 1 sw Y y xY w) Nr c ó | dz S ©Né © ©Úxs ú\H:\Change_password\venv_build\Lib\site-packages\pandas/tests/groupby/transform/test_numba.pyÚincorrect_functionz;test_correct_function_signature.<locals>.incorrect_function s Ø1u ó ©Úar Úbr r ©ç ð?ç @g @g @g @©ÚkeyÚdatar r ©Úcolumnsz The first 2©Úmatch©Úengine©ÚpytestÚ importorskipr Úraisesr ÚgroupbyÚ transform©r r s r Ú test_correct_function_signaturer+ sÈ Ü
|
×ÑÔ òô Ú)Ò3LÑMØô Dô
|
~¨]Ô ;ñ JØ UÓ×%Ñ%Ð&8ÀÐ%ÔI÷Jô
|
~¨]Ô ;ñ RØ UÓFÑ#×-Ñ-Ð.@ÈÐ-ÔQ÷Rð R÷Jð Jú÷Rð Rús Á
|
#B?Â&C Â?Cà Cc ó² t j d« d } t g d¢g d¢dddg¬« }t j t d ¬
|
« 5 |j d« j | dd ¬ « d d d « t j t d ¬
|
« 5 |j d« d j | dd ¬ « d d d « y # 1 sw Y TxY w# 1 sw Y y xY w) Nr c ó | dz S r r ©ÚvaluesÚindexs r r z6test_check_nopython_kwargs.<locals>.incorrect_function. ó ØzÐr r r r r r r znumba does not supportr r )r# r r$ r* s r Útest_check_nopython_kwargsr2 + sÏ Ü
|
×ÑÔ òô Ú)Ò3LÑMØô Dô
|
~Ð-EÔ Fñ OØ UÓ×%Ñ%Ð&8ÀÈAÐ%ÔN÷Oô
|
~Ð-EÔ Fñ WØ UÓFÑ#×-Ñ-Ð.@ÈÐTUÐ-ÔV÷Wð W÷Oð Oú÷Wð Wús Á
|
$CÂ'C ÃC
|
à CÚignoreÚjitTFÚ
|
pandas_objr r Úas_indexc óP t j d« d }| rdd l}|j |« }t g d¢g d¢dddg¬« }|||d } |j d|¬
|
« }
|
|d k( r|
|
d }
|
|
|
j |d| ¬ « } |
|
j d d¬« } t j | | « y )Nr c ó | dz S r r r. s r Úfuncz"test_numba_vs_cython.<locals>.funcD r1 r r r r ©r r r r ©ÚnogilÚparallelÚnopython)r6 r ©r# Ú engine_kwargsc ó | dz S r r r s r ú<lambda>z&test_numba_vs_cython.<locals>.<lambda>V ó
|
¨1¨q©5 r Úcythonr" © r% r& r r4 r r( r) ÚtmÚ assert_equal) r4 r5 r< r= r> r6 r9 r r r@ ÚgroupedÚresultÚexpecteds r Útest_numba_vs_cythonrK < s± ô ×ÑÔ òñ ãàyyä Ú %Ò*CÑDÈqÐRSÈfô Dð $°ÀxÑPMØll1 xlÓ0GØXÒ Ø!*à × Ñ t¨GÀ=Ð Ó QFØ× Ñ ¡ ¸Ð ÓBHäOOFHÕ%r c ól t j d« d }d }| r&dd l}|j |« }|j |« }t g d¢g d¢dddg¬ « }|||d
|
} |j d« }
|
|d k( r|
|
d }
|
|
|
j |d| ¬ « } |
|
j d d¬« } t j | | « |
|
j |d| ¬ « } |
|
j d d¬« } t j | | « |
|
j |d| ¬ « } |
|
j d d¬« } t j | | « y )Nr c ó | dz S r r r. s r Úfunc_1ztest_cache.<locals>.func_1c r1 r c ó | dz S ©Né r r. s r Úfunc_2ztest_cache.<locals>.func_2f r1 r r r r r: r r r; r r? c ó | dz S r r r s r rB z test_cache.<locals>.<lambda>x rC r rD r" c ó | dz S rP r r s r rB z test_cache.<locals>.<lambda>| rC r c ó | dz S r r r s r rB z test_cache.<locals>.<lambda> rC r rE ) r4 r5 r< r= r> rN rR r r r@ rH rI rJ s r Ú
|
test_cacherV [ s6 ô ×ÑÔ òòñ Ûà6Ó"Ø6Ó"ä Ú %Ò*CÑDÈqÐRSÈfô Dð $°ÀxÑPMØll1oGØXÒ Ø!*à × Ñ v¨gÀ]Ð Ó SFØ× Ñ ¡ ¸Ð ÓBHÜOOFHÔ%à × Ñ v¨gÀ]Ð Ó SFØ× Ñ ¡ ¸Ð ÓBHÜOOFHÔ%ð × Ñ v¨gÀ]Ð Ó SFØ× Ñ ¡ ¸Ð ÓBHÜOOFHÕ%r c ó< t j d« d } t g d¢g d¢dddg¬« }|j d« }|j | d¬ « }t d
|
d « 5 |j | d ¬ « }d d d « t j |« y # 1 sw Y xY w) Nr c ó | dz S r r r. s r rN z&test_use_global_config.<locals>.func_1 r1 r r r r: r r r r" zcompute.use_numbaT)r% r& r r( r) r rF Úassert_frame_equal)rN r rH rJ rI s r Útest_use_global_configrZ
s Ü
|
×ÑÔ òô Ú %Ò*CÑDÈqÐRSÈfô Dð ll1oGØ× Ñ °Ð Ó8HÜ Ð+¨TÓ 2ñ 8Ø×"Ñ" 6°$Ð"Ó7÷8ä×Ñ( FÕ+÷8ð 8ús Á BÂBÚagg_funcÚminÚmaxÚsum)ÚBÚCc ó t j d« |\ } }t g d¢g d¢dddg¬« }|j d« } |j | fddi|¤} |j | fdd i|¤}t j ||« |d j | fddi|¤} |d j | fdd i|¤}t j ||« y )
|
Nr r r r: r r r r# rD )r% r& r r( r) rF rY Úassert_series_equal)r[ Únumba_supported_reductionsÚkwargsr rH rI rJ s r Útest_string_cython_vs_numbare sÝ ô ×ÑÔ Ø1ÑHfÜ Ú %Ò*CÑDÈqÐRSÈfô Dð ll1oGà W× Ñ xÑ B°Ð B¸6Ñ BFØ w× Ñ ÑE°(ÐE¸fÑEHÜ×Ñ& (Ô+à !WQZ× !Ñ ! (Ñ E°7Ð E¸fÑ EFØ#wqz×#Ñ# HÑH°XÐHÀÑHHÜ×Ñ6 8Õ,r c ól t j d« d } t g d¢g d¢d« }|j d« d }|j | dd¬ « }t d
|
gd z d¬ « }t j ||« |j | d d¬ « }t dgd z d¬ « }t j ||« y )Nr c ó* | | d j « S ©N)r^ )r/ r0 Úns r Úsum_lastz&test_args_not_cached.<locals>.sum_last® s Øqbc{Ó Ð r )r r r r )r r r r )Úidr rk r r r" r é )Únameé r )r% r& r r( r) r rF rb )rj ÚdfÚ grouped_xrI rJ s r Útest_args_not_cachedrq ª s« ä
|
×ÑÔ ò!ô
|
,ª\Ñ:Ó ;BØ
|
|
4Ó Ñ%IØ × Ñ ¨1°WÐ Ó =FÜseai cÔ*HÜ×Ñ6 8Ô,à × Ñ ¨1°WÐ Ó =FÜseai cÔ*HÜ×Ñ6 8Õ,r c óö t j d« d } t g d¢g d¢dg d¢¬« }|j d« j | d¬ « }t g d
|
¢d gg d¢¬ « }t j ||« y ) Nr c ó |dz
|
S r r r. s r Úfz+test_index_data_correctly_passed.<locals>.fÀ s ØqyÐr )ÚAru r_ )rl rQ é )ÚgroupÚv)éÿÿÿÿéþÿÿÿéýÿÿÿ)r0 rw r" )g Àg Àg Àrx )r r0 ©r% r& r r( r) rF rY )rt ro rI rJ s r Ú test_index_data_correctly_passedr} ¼ sg ä
|
×ÑÔ òô
|
_²9Ñ=Â\Ô RBØ ZZÓ × *Ñ *¨1°WÐ *Ó =FÜÒ+°c°UÂ,ÔOHÜ×Ñ& (Õ+r c ó¬ t j d« dddfd} d}t dg d¢i« }|j d¬ « j | d|¬
|
« }t dg d ¢i« }t j ||« dd}|j d¬ « j | d|¬
|
« }t dg d ¢i« }t j ||« y ) Nr TFc ó z z S rh r )r/ r0 r< r> r= s r Ú func_kwargsz2test_engine_kwargs_not_cached.<locals>.func_kwargsÑ s ø ØxÑ (Ñ*Ð*r ©r> r< r= Úvalue)r r r r )Úlevelr? )r r r )r r r r| )r r@ ro rI rJ r< r> r= s @@@r Ú test_engine_kwargs_not_cachedr É så ú ô ×ÑÔ Ø EØHØHö+ð "*°EÀxÑPMÜ GYÐ'Ó (BØ ZZaZÓ × *Ñ *ØG°=ð +ó Fô '¢?Ð3Ó4HÜ×Ñ& (Ô+à EØ!)°EÀxÑPMØ ZZaZÓ × *Ñ *ØG°=ð +ó Fô '¢?Ð3Ó4HÜ×Ñ& (Õ+r c ó6 t j d« d }t ddddg« j ddg« }|| |d }|j d« j |d|¬
|
« }t ddd dg« j ddg« }t j ||« y ) Nr c ó yr r r. s r Ú
|
numba_funcz+test_multiindex_one_key.<locals>.numba_funcé ó Ør r rn é ©ru r_ r` ru r_ r r? r )r% r& r Ú set_indexr( r) rF rY )r< r= r> r ro r@ rI rJ s r Útest_multiindex_one_keyr å s ä
|
×ÑÔ òô
|
! !¨!Ñ,Ð-Ó .× 8Ñ 8¸#¸s¸Ó DBØ!)°EÀxÑPMØ ZZ_× &Ñ &Ø7°-ð 'ó Fô ¨°Ñ4Ð5Ó6×@Ñ@À#ÀsÀÓLHÜ×Ñ& (Õ+r c ó, t j d« d }t ddddg« j ddg« }|| |d }t j t
|
d
|
¬ « 5 |j ddg« j |d|¬ « d d d « y # 1 sw Y y xY w) Nr c ó yr r r. s r r z;test_multiindex_multi_key_not_supported.<locals>.numba_funcø r r r rn r r ru r_ r zmore than 1 grouping labelsr r? )r% r& r r r' ÚNotImplementedErrorr( r) )r< r= r> r ro r@ s r Ú'test_multiindex_multi_key_not_supportedr õ s Ü
|
×ÑÔ òô
|
! !¨!Ñ,Ð-Ó .× 8Ñ 8¸#¸s¸Ó DBØ!)°EÀxÑPMÜ Ô*Ð2OÔ Pñ
|
Ø
|
|
|
C:Ó ×(Ñ(Ø w°mð )ô
|
÷
|
÷
|
ñ
|
ús Á&B
|
Â
|
Bc ó¶ t j d« | \ }}t g d¢g d¢t j j d« j d« t j j d« j d« d« }|j ddg« } |j |fd di|¤} |j |fd d
|
i|¤}t j ||« y ) Nr ©ÚfooÚbarr r r r r r ©Úoner ÚtwoÚthreer r r r rn é ©ru r_ r` ÚDru r_ r# rD © r% r& r ÚnpÚrandomÚ default_rngÚstandard_normalr( r) rF rY )rc Ú reductionrd ro ÚgbÚres_aggÚ expected_aggs r Ú test_multilabel_numba_vs_cythonr¥ sÁ Ü
|
×ÑÔ Ø2ÑIvÜ âIÚMÜ×&Ñ& qÓ)×9Ñ9¸!Ó<Ü×&Ñ& qÓ)×9Ñ9¸!Ó<ñ
|
ó
|
Bð
|
S#JÓ BØbll9Ñ?¨WÐ?¸Ñ?GØ 2<< ÑE°(ÐE¸fÑELÜ×Ñ' <Õ0r c ó¨ t j d« t g d¢g d¢t j j d« j d« t j j d« j d« d« } | j ddg« }|j d d¬
|
« }|j d d ¬
|
« }t j ||« y ) Nr r r rn r r ru r_ c ól | | j « z
|
| j « | j « z
|
z S rh ©r\ r] r. s r rB z5test_multilabel_udf_numba_vs_cython.<locals>.<lambda> s' v¨¯
|
©
|
« Ñ 4¸¿¹» ÈÏ
|
É
|
Ë Ñ9TÑ U r r" c ól | | j « z
|
| j « | j « z
|
z S rh r¨ r s r rB z5test_multilabel_udf_numba_vs_cython.<locals>.<lambda>$ s& 1quuw; 1§5¡5£7¨Q¯U©U«WÑ#4Ñ5 r rD r )ro r¢ rI rJ s r Ú#test_multilabel_udf_numba_vs_cythonrª s¸ Ü
|
×ÑÔ Ü âIÚMÜ×&Ñ& qÓ)×9Ñ9¸!Ó<Ü×&Ñ& qÓ)×9Ñ9¸!Ó<ñ
|
ó
|
Bð
|
S#JÓ BØ \\ÙUØð ó Fð ||Ù5¸hð ó Hô ×Ñ& (Õ+r )'Únumpyr r% Ú pandas.compatr Ú pandas.errorsr Úpandasr r r Úpandas._testingÚ_testingrF Úpandas.util.versionr ÚmarkÚ
|
single_cpuÚ
|
pytestmarkr& r ÚappendÚskipifÚ __version__r+ r2 ÚfilterwarningsÚ parametrizerK rV rZ re rq r} r r r r¥ rª r r r ú<module>rº s ðÛ Û å )Ý (÷ñ õ
|
Ý 'àkk×$Ñ$Ð %
|
à×ÑGÓ$Ø
|
× Ñ Ø
|
KK×ÑÙ×!Ñ!Ó"¡g¨f£oÑ5ÒK¹/Ó:KØ7¸×8IÑ8IÐ7JÐKð ó ôòRò"Wð" ×ÑHÓ%à×Ñ u Ó.Ø×Ñ ¨°+Ð'>Ó?Ø×Ñ d¨E ]Ó3ñ&ó 4ó @ó /ó &ð
|
&ð4 ×ÑHÓ%à×Ñ u Ó.Ø×Ñ ¨°+Ð'>Ó?ñ#&ó @ó /ó &ð#&òL ,ð" ×ÑØ% ¨u°e¨nÀ5Ñ(IÐJóñ-óð-ò"-ò$
|
,ò,ð8 ×ÑHÓ%ñ ,ó &ð ,ò
|
ò 1ó",r
|
