Ë
|
oñúh64 ã ó¼ d dl Zd dlZd dlmZ d dlmZ d dlmZm Z m
|
Z
|
m Z m Z d dl mZ d dlmZ ej$ j& gZ ej* d« Zej/ ej$ j1 eej2 « ed« k( xr e« dej2 ¬ « « d
|
Zd Zej$ j9 d « ej$ j; d ddg« ej$ j; dddg« ej$ j; dddg« d « « « « Z ej$ j9 d « ej$ j; d ddg« ej$ j; dddg« d « « « Z d Z ej$ j; ddddgiddidddgdd id e
|
d d¬ « ig« d « Z!ej$ j; d!dd" idf ejD dd# d$ giddgej$ jG d%¬ « ¬&« g« d' « Z$ej$ j; ddddgiddiddd(g« d) « Z%ej$ j& ej$ j; d* e g d+¢« dddgif e g d+¢« ddif eg d+¢g d,¢d d d-g¬.« dddgif eg d+¢g d,¢d d d-g¬.« ddif eg d+¢g d,¢d d d-g¬.« dddgdd if eg d+¢g d,¢d d d-g¬.« d/ e
|
d d¬ « ifg« d0 « « Z&d1 Z'd2 Z(d3 Z)ej$ j9 d « d4 « Z*d5 Z+d6 Z,d7 Z-y)8é N)Úis_platform_arm)ÚNumbaUtilError)Ú DataFrameÚIndexÚNamedAggÚ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 ó t | « dz S ©Ng@©Úsum©Úxs ú\H:\Change_password\venv_build\Lib\site-packages\pandas/tests/groupby/aggregate/test_numba.pyÚincorrect_functionz;test_correct_function_signature.<locals>.incorrect_function s Ü1v|Ðó ©Úar Úbr r ©ç ð?ç @g @g @g @©ÚkeyÚdatar r ©Úcolumnsz The first 2©Úmatch©Úengine©ÚpytestÚ importorskipr Úraisesr ÚgroupbyÚagg©r r s r Ú test_correct_function_signaturer- sÈ Ü
|
×ÑÔ ò ô Ú)Ò3LÑMØô Dô
|
~¨]Ô ;ñ DØ UÓ× Ñ Ð 2¸7Ð ÔC÷Dô
|
~¨]Ô ;ñ LØ UÓFÑ#×'Ñ'Ð(:À7Ð'ÔK÷Lð L÷Dð Dú÷Lð Lú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 ó t | « dz S r r ©ÚvaluesÚindexs r r z6test_check_nopython_kwargs.<locals>.incorrect_function0 s Ü6{SÑ Ð r r r r r r r znumba does not supportr" é )r% r r& r, s r Útest_check_nopython_kwargsr4 - sÎ Ü
|
×ÑÔ ò!ô Ú)Ò3LÑMØô Dô
|
~Ð-EÔ Fñ IØ UÓ× Ñ Ð 2¸7ÀaÐ ÔH÷Iô
|
~Ð-EÔ Fñ QØ UÓFÑ#×'Ñ'Ð(:À7ÈaÐ'ÔP÷Qð Q÷Ið Iú÷Qð Qú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 ó2 t j | « dz S r ©ÚnpÚmeanr0 s r Ú
|
func_numbaz(test_numba_vs_cython.<locals>.func_numbaF ó Üwwv Ñ$Ð$r r r r ©r r3 r3 r ©ÚnogilÚparallelÚnopython)r8 r ©r% Ú engine_kwargsc ó2 t j | « dz S r r; r s r ú<lambda>z&test_numba_vs_cython.<locals>.<lambda>X ó ¤R§W¡W¨Q£Z°#Ñ%5 r Úcythonr$ © r' r( r r6 r r* r+ ÚtmÚ assert_equal) r6 r7 rB rC rD r8 r> r r rF ÚgroupedÚresultÚexpecteds r Útest_numba_vs_cythonrQ > s ô ×ÑÔ ò%ñ ãàYYzÓ*
|
ä Ú %Ò*CÑDÈqÐRSÈfô Dð $°ÀxÑPMØll1 xlÓ0GØXÒ Ø!*à [[¨GÀ=[Ó QFØ{{Ñ5¸h{ÓGHä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 ó2 t j | « dz
|
S ©Ng333333 @r; r0 s r Úfunc_1ztest_cache.<locals>.func_1e r? r c ó2 t j | « dz S r r; r0 s r Úfunc_2ztest_cache.<locals>.func_2h r? r r r r r@ r3 r rA r rE c ó2 t j | « dz
|
S rT r; r s r rH z test_cache.<locals>.<lambda>z rI r rJ r$ c ó2 t j | « dz S r r; r s r rH z test_cache.<locals>.<lambda> rI r c ó2 t j | « dz
|
S rT r; r s r rH z test_cache.<locals>.<lambda> rI r rK ) r6 r7 rB rC rD rU rW r r rF rN rO rP s r Ú
|
test_cacher[ ] s) ô ×ÑÔ ò%ò%ñ Ûà6Ó"Ø6Ó"ä Ú %Ò*CÑDÈqÐRSÈfô Dð $°ÀxÑPMØll1oGØXÒ Ø!*à [[¨¸}[Ó MFØ{{Ñ5¸h{ÓGHÜOOFHÔ%ð [[¨¸}[Ó MFØ{{Ñ5¸h{ÓGHÜOOFHÔ%ð [[¨¸}[Ó MFØ{{Ñ5¸h{ÓGHÜ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 ó2 t j | « dz
|
S rT r; r0 s r rU z&test_use_global_config.<locals>.func_1 r? r r r r@ r r3 r r$ zcompute.use_numbaT)r' r( r r* r+ r rL Úassert_frame_equal)rU r rN rP rO s r Útest_use_global_configr_ s Ü
|
×ÑÔ ò%ô Ú %Ò*CÑDÈqÐRSÈfô Dð ll1oGØ{{6¨'{Ó2HÜ Ð+¨TÓ 2ñ 2ØV¨DÓ1÷2ä×Ñ( FÕ+÷2ð 2ús Á BÂBÚ
|
agg_kwargsÚfuncÚminÚmaxr )r3 é Úbminr3 )ÚcolumnÚaggfuncc ó t j d« t g d¢g d¢g d¢dg d¢¬« }|j d« } |j d
|
i | ¤ddi¤} |j d
|
i | ¤dd i¤}t j ||« y ) Nr r r ©r3 rd é é é ©r r3 rd r r r% rJ © ©r' r( r r* r+ rL r^ )r` r rN rO rP s r Ú$test_multifunc_numba_vs_cython_framerp s~ ô ×ÑÔ Ü â(Ú(Ú ñ
|
ò
|
ô Dð ll1oGØ W[[Ñ 6:Ñ 6¨gÒ 6FØw{{Ñ9ZÑ9°Ò9HÜ×Ñ& (Õ+r zagg_kwargs,expected_funcc ó" | j « S ©Nr r0 s r rH rH ´ s ¨¯
|
©
|
« r c ó" | j « S rr r r0 s r rH rH ¹ ó ¨&¯*©*«, r c ó" | j « S rr ©rb r0 s r rH rH º rt r z2This doesn't work yet! Fails in nopython pipeline!)Úmarksc ó t j d« t g d¢g d¢g d¢dg d¢¬« }|j d« } |j d i | ¤ddi¤}|j |d ¬
|
« }t j ||d ¬ « y )Nr r r ri rm r r r% rJ r$ F)Ú check_dtypern ro )r` Ú expected_funcr rN rO rP s r Ú test_multifunc_numba_udf_framer{ ± s| ô( ×ÑÔ Ü â(Ú(Ú ñ
|
ò
|
ô Dð ll1oGØ W[[Ñ 6:Ñ 6¨gÒ 6FØ{{=°{Ó:Hô ×Ñ& (¸Ö>r )Úmin_valÚmax_valc óH t j d« g d¢}t g d¢« }|j |« }d| d< |j di | ¤}d| d< |j di | ¤}t |t « rt j ||« y t j ||« y )Nr r r r% rJ rn )
|
r' r( r r* r+ Ú
|
isinstancer rL r^ Úassert_series_equal)r` Úlabelsr rN rO rP s r Ú%test_multifunc_numba_vs_cython_seriesr Ö s ô
|
×ÑÔ Ú &FÜ Ò+Ó ,DØll6Ó"GØ"JxÑØ W[[Ñ &:Ñ &FØ#JxÑØw{{Ñ(ZÑ(HÜ(IÔ&Ü
|
× Ñ f hÕ/ä
|
× Ñ v xÕ0r zdata,agg_kwargsr ri rd r Úmin_colc ó0 t j d« g d¢}| j |« } |j di |¤dddid¤} |j di |¤ddi¤}t |t
|
« rt j ||« y t j ||« y )Nr r rC TrE r% rn ) r' r( r* r+ r r rL r^ r )r r` r rN rO rP s r Ú&test_multifunc_numba_kwarg_propagationr
é s ôB ×ÑÔ Ú &FØll6Ó"GØ W[[Ñ X:Ð X¨gÀjÐRVÐEWÓ XFØw{{Ñ8ZÑ8°Ò8HÜ(IÔ&Ü
|
× Ñ f hÕ/ä
|
× Ñ v xÕ0r c ó¤ t j d« d } t g d¢g d¢d« }|j d« d }|j | dd¬ « }t d
|
gd z dt d dgd¬ « ¬« }t j ||« |j | d d¬ « }t dgd z dt d dgd¬ « ¬« }t j ||« y )Nr c ó* | | d j « S rr r )r1 r2 Úns r Úsum_lastz&test_args_not_cached.<locals>.sum_last s Øqbc{Ó Ð r )r r r3 r3 )r3 r3 r3 r3 )Úidr r r r3 r$ r rd r ©Úname)r r2 r ) r' r( r r* r+ r r rL r )r ÚdfÚ grouped_xrO rP s r Útest_args_not_cachedr s¿ ä
|
×ÑÔ ò!ô
|
,ª\Ñ:Ó ;BØ
|
|
4Ó Ñ%IØ ]]8 Q¨w]Ó 7FÜseai c´¸¸1°vÀDÔ1IÔJHÜ×Ñ6 8Ô,à ]]8 Q¨w]Ó 7FÜseai c´¸¸1°vÀDÔ1IÔJHÜ×Ñ6 8Õ,r c ó t j d« d } t g d¢g d¢dg d¢¬« }|j d« j | d¬ « }t d
|
d gd gt d dgd¬« ¬« }t j ||« y )Nr c ó, t j |« S rr r; r0 s r Úfz+test_index_data_correctly_passed.<locals>.f+ s Üwwu~Ð r )ÚAr ÚB)rk rl é )ÚgroupÚv)éÿÿÿÿéþÿÿÿéýÿÿÿ)r2 r r$ g ø¿g Àr r r r )r! r2 )r' r( r r* Ú aggregater rL r^ )r r rO rP s r Ú test_index_data_correctly_passedr ' sz ä
|
×ÑÔ ò ô
|
_²9Ñ=Â\Ô RBØ ZZÓ × *Ñ *¨1°WÐ *Ó =FÜØ t se¬5°#°s°À'Ô+JôHô ×Ñ& (Õ+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 rr rn )r1 r2 rB rD rC s r Ú func_kwargsz2test_engine_kwargs_not_cached.<locals>.func_kwargs> s ø ØxÑ (Ñ*Ð*r ©rD rB rC Úvalue)r r r r )ÚlevelrE )r r r )r r r )r' r( r r* r rL r^ )r rF r rO rP rB rD rC s @@@r Ú test_engine_kwargs_not_cachedr£ 6 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 ó, t j d« d }t ddddg« j ddg« }|| |d }|j d« j |d|¬
|
« }t d gt dgd¬ « d g¬« }t j ||« y )Nr c ó y©Nr3 rn r0 s r Ú
|
numba_funcz+test_multiindex_one_key.<locals>.numba_funcV ó Ør r3 rd rj ©r r ÚCr r r rE r r rª )r2 r! ) r' r( r Ú set_indexr* r+ r rL r^ )rB rC rD r§ r rF rO rP s r Útest_multiindex_one_keyr¬ R s ä
|
×ÑÔ òô
|
! !¨!Ñ,Ð-Ó .× 8Ñ 8¸#¸s¸Ó DBØ!)°EÀxÑPMØ ZZ_× Ñ Ø7°-ð !ó Fô #¤e¨Q¨C°cÔ&:ÀSÀEÔJHÜ×Ñ& (Õ+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¦ rn r0 s r r§ z;test_multiindex_multi_key_not_supported.<locals>.numba_funce r¨ r r3 rd rj r© r r r zmore than 1 grouping labelsr" rE )r' r( r r« r) ÚNotImplementedErrorr* r+ )rB rC rD r§ r rF s r Ú'test_multiindex_multi_key_not_supportedr° b s Ü
|
×ÑÔ òô
|
! !¨!Ñ,Ð-Ó .× 8Ñ 8¸#¸s¸Ó DBØ!)°EÀxÑPMÜ Ô*Ð2OÔ Pñ
|
Ø
|
|
|
C:Ó ×"Ñ"Ø w°mð #ô
|
÷
|
÷
|
ñ
|
ús Á&B
|
Â
|
Bc ó2 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 ||« t ||« d d di|¤} t ||« d d d
|
i|¤}t j ||« y ) Nr ©ÚfooÚbarr³ r´ r³ r´ r³ r³ ©Úoner¶ ÚtwoÚthreer· r· r¶ r¸ rd é ©r r rª ÚDr r r% rJ rn ) r' r( r r< ÚrandomÚ default_rngÚstandard_normalr* r+ rL r^ Úgetattr) Únumba_supported_reductionsÚ reductionÚkwargsr ÚgbÚres_aggÚ expected_aggÚ
|
direct_resÚdirect_expecteds r Ú test_multilabel_numba_vs_cythonrÈ p s Ü
|
×ÑÔ Ø2ÑIvÜ âIÚMÜ×&Ñ& qÓ)×9Ñ9¸!Ó<Ü×&Ñ& qÓ)×9Ñ9¸!Ó<ñ
|
ó
|
Bð
|
S#JÓ BØbffYÑ9 wÐ9°&Ñ9GØ266)Ñ?¨HÐ?¸Ñ?LÜ×Ñ' <Ô0à'YÓ'ÑA¨wÐA¸&ÑAJØ,gb )Ó,ÑG°HÐGÀÑGOÜ×Ñ* oÕ6r 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µ rd r¹ rº r r c ó" | j « S rr rv r0 s r rH z5test_multilabel_udf_numba_vs_cython.<locals>.<lambda> rt r r$ c ó" | j « S rr rv r s r rH z5test_multilabel_udf_numba_vs_cython.<locals>.<lambda> s §¡£ r rJ ) r' r( r r< r¼ r½ r¾ r* r+ rL r^ )r rà rO rP s r Ú#test_multilabel_udf_numba_vs_cythonrÌ
s© Ü
|
×ÑÔ Ü âIÚMÜ×&Ñ& qÓ)×9Ñ9¸!Ó<Ü×&Ñ& qÓ)×9Ñ9¸!Ó<ñ
|
ó
|
Bð
|
S#JÓ BØ VVÑ6¸wVÓ GFØvvÑ'°vÓ9HÜ×Ñ& (Õ+r ).Únumpyr< r' Ú pandas.compatr Ú pandas.errorsr Úpandasr r r r r Úpandas._testingÚ_testingrL Úpandas.util.versionr
|
ÚmarkÚ
|
single_cpuÚ
|
pytestmarkr( r ÚappendÚskipifÚ __version__r- r4 ÚfilterwarningsÚ parametrizerQ r[ r_ rp ÚparamÚxfailr{ r r
r r r£ r¬ r° rÈ rÌ rn r r ú<module>rÞ sÿ ðÛ Û å )Ý (÷õ õ Ý 'àkk×$Ñ$Ð %
|
à×ÑGÓ$Ø
|
× Ñ Ø
|
KK×ÑÙ×!Ñ!Ó"¡g¨f£oÑ5ÒK¹/Ó:KØ7¸×8IÑ8IÐ7JÐKð ó ôòLò"Qð" ×ÑHÓ%à×Ñ u Ó.Ø×Ñ ¨°+Ð'>Ó?Ø×Ñ d¨E ]Ó3ñ&ó 4ó @ó /ó &ð
|
&ð4 ×ÑHÓ%à×Ñ u Ó.Ø×Ñ ¨°+Ð'>Ó?ñ$&ó @ó /ó &ð$&òN ,ð ×ÑØà % Ð Ø Ø eU^¨Ñ.Ð/Ø ¨EÔ2Ð3ð óñ ,óð ,ð ×ÑØ à
|
Ñ4Ð 5°uÐ=à àÙ6Ù6ðð ð ENØ++×#Ñ#ØKð $ó ô
|
ðóñ&?ó'ð&?ð$ ×ÑØØ uenÐ ¨¸EÈeÑ0TÐUóñ 1ó ð 1ð ×ÑØ×ÑØá Ò)Ó *¨V°e¸U°^Ð,DÐEÙ Ò)Ó *¨V°U¨OÐ<á Ú-²/ÑBÈQÐPQÈFô ð eU^Ð $ð
|
ñ Ú-²/ÑBÈQÐPQÈFô ð UOð
|
ñ Ú-²/ÑBÈQÐPQÈFô ð % ¨EÑ2Ð 3ð
|
ñ Ú-²/ÑBÈQÐPQÈFô ð ¨°5Ô9Ð :ð
|
ð+ó ñ> 1ó? ó ð@ 1ò-ò$ ,ò ,ð8 ×ÑHÓ%ñ ,ó &ð ,ò
|
ò 7ó* ,r
|
