Ë
|
oñúhÙ( ã ó d dl Zd dlZd dlmZ d dlmZmZm Z m
|
Z
|
m Z d dl m Z d dlmZ d dlmZ G d d« Z G d d e« Z G d
|
d e« Zej, j/ d d dg« d « Zd Zej, j/ dg d¢« d « Zd Zd Z d Z d Z d Z d Z ej, j/ dejB gejB gejB ggejB gejB ggfejB gejB gejB ggejB gejD ggfejB gejB gejB ggejB gejD ggfejB gejB gejB ggejB gejB ggfg« d « Z#y) é N)Úgroupby)Ú group_cumprodÚ group_cumsumÚ
|
group_meanÚ group_sumÚ group_var)Úensure_platform_int)Úisnac ó* e Zd Zd Zd Zd Zd Zd Zy)ÚGroupVarTestMixinc óú t j j d« }t j t j d« z j | j « }t j dd¬« }d|j d« j | j « z }t j t j d« d« j d « }t j |« j d
|
d ¬ « j d d ¬« dz d d
t j f }|dz }| j ||||« t j ||| j « sJ t# j$ ||« y )Né ©é é r Úint64©Údtypeé
|
)é r )é Úintp)r r ÚF©Úorderr )ÚaxisÚddofr )ÚnpÚrandomÚ default_rngÚnanÚonesÚastyper ÚzerosÚtileÚarangeÚsqueezeÚreshapeÚstdÚnewaxisÚalgoÚallcloseÚrtolÚtmÚassert_numpy_array_equal©ÚselfÚprngÚoutÚcountsÚvaluesÚlabelsÚ expected_outÚexpected_countss úWH:\Change_password\venv_build\Lib\site-packages\pandas/tests/groupby/test_libgroupby.pyÚtest_group_var_generic_1dz+GroupVarTestMixin.test_group_var_generic_1d s Üyy×$Ñ$ QÓ'ävv Ñ'×/Ñ/°·
|
±
|
Ó;Ü! 7Ô+Ødkk 'Ó*×1Ñ1°$·*±*Ó=Ñ=Ü 1 tÓ,×3Ñ3°FÓ;ô JJvÓ × &Ñ & v°SÐ &Ó 9× =Ñ =À1È1Ð =Ó MÐQRÑ RÚ
|
RZZ-ñ ð ! 1*à #vv vÔ.Ü{{3 ¨d¯i©iÔ8Ð8Ð8Ü
|
×#Ñ# F¨OÕ<ó c ón t j j d« }t j t j d« z j | j « }t j dd¬« }d|j d« j | j « z }t j dd ¬« }t j |j d¬
|
« dz gg« }|dz }| j ||||« t j ||| j « sJ t j ||« y ) Nr ©r r r r r r r r r )r )r r r r! r" r# r r$ Úarrayr) r+ r, r- r. r/ r0 s r9 Ú%test_group_var_generic_1d_flat_labelsz7GroupVarTestMixin.test_group_var_generic_1d_flat_labels% sæ Üyy×$Ñ$ QÓ'ävv Ñ'×/Ñ/°·
|
±
|
Ó;Ü! 7Ô+Ødkk &Ó)×0Ñ0°·±Ó<Ñ<Ü! 6Ô*äxx &§*¡*°! *Ó"4¸Ñ"9Ð!:Ð ;Ó< Ø 1*à #vv vÔ.ä{{3 ¨d¯i©iÔ8Ð8Ð8Ü
|
×#Ñ# F¨OÕ<r; c ó´ t j j d« }t j t j d« z j | j « }t j dd¬« }d|j d« j | j « z }t j t j d« d« j d « }t j |j ddd« d
|
d ¬ « dz }|dz }| j ||||« t j ||| j « sJ t j ||« y ) Nr ©r r r r r r ©r r ©r r r r ©r r )r r r r! r" r# r r$ r% r& r) r( r+ r, r- r. r/ r0 s r9 Ú$test_group_var_generic_2d_all_finitez6GroupVarTestMixin.test_group_var_generic_2d_all_finite5 sù Üyy×$Ñ$ QÓ'ävv Ñ'×/Ñ/°·
|
±
|
Ó;Ü! 7Ô+Ødkk 'Ó*×1Ñ1°$·*±*Ó=Ñ=Ü 1 tÓ,×3Ñ3°FÓ;ävvfnn¨Q°°1Ó 5¸AÀAÔFÈ!ÑK Ø 1*à #vv vÔ.Ü{{3 ¨d¯i©iÔ8Ð8Ð8Ü
|
×#Ñ# F¨OÕ<r; c ó t j j d« }t j t j d« z j | j « }t j dd¬« }d|j d« j | j « z }t j |d d
df< t j t j d« d « j d
|
« }t j |d d
d f j ddd ¬ « j dd¬« dz t j t j d« z g« j j | j « }|dz }| j ||||« t j ||d¬« t j" ||« y )Nr rA r r r r rB r rC r r r r rD gíµ ÷Æ >©r- )r r r r! r" r# r r$ r% r& Úvstackr( r) ÚTr+ r. Úassert_almost_equalr/ r0 s r9 Ú"test_group_var_generic_2d_some_nanz4GroupVarTestMixin.test_group_var_generic_2d_some_nanD sP Üyy×$Ñ$ QÓ'ävv Ñ'×/Ñ/°·
|
±
|
Ó;Ü! 7Ô+Ødkk 'Ó*×1Ñ1°$·*±*Ó=Ñ=Üvvq!t Ü 1 tÓ,×3Ñ3°FÓ;äyyàq!t ×$Ñ$ Q¨°Ð$Ó5×9Ñ9¸qÀqÐ9ÓIÈQÑNÜ Ñ#ð ó
|
÷
|
!FF4::Ó ð ð ! 1*à #vv vÔ.Ü
|
× Ñ s L°wÕ?Ü
|
×#Ñ# F¨OÕ<r; c ó t j t j gg| j ¬« }t j dgd¬« }dt j d| j ¬« z }t j
|
dd¬« }| j ||||« |d dk( sJ |d dk\ sJ t j |d d « y )
|
Nr r r gUUUÕª¦ê?)r r r r ©r r g ) r r> r! r r" r$ r+ r. rJ )r1 r3 r4 r5 r6 s r9 Útest_group_var_constantz)GroupVarTestMixin.test_group_var_constantY s¡ ô hhz¨¯©Ô4Ü1# WÔ-Ø"¤R§W¡W¨V¸4¿:¹:Ô%FÑFÜ! 6Ô*à #vv vÔ.àayA~Ð ~Ø4yA~Ð ~Ü
|
× Ñ s 4y¨#Õ.r; N)Ú__name__Ú
|
__module__Ú __qualname__r: r? rE rK rN © r; r9 r r s ò=ò"=ò =ò =ó* /r; r c óB e Zd ZdZ ee« Zej Z dZ
|
d Z y)ÚTestGroupVarFloat64Tgñhãµøä>c óÜ t j j d« }t j t j gg| j
|
¬« }t j dgd¬« }|j d« dz j | j
|
« }d|_ t j dd¬« }| j ||||« |d dk( sJ t j |d d
|
d ¬ « y ) Nr r r r é@B l J)£)rV r r rM gUUUUUUµ?gü©ñÒMb@?rG ) r r r r> r! r r# Úshaper$ r+ r. rJ )r1 r2 r3 r4 r5 r6 s r9 Útest_group_var_large_inputsz/TestGroupVarFloat64.test_group_var_large_inputso s· Üyy×$Ñ$ QÓ'ähhz¨¯©Ô4Ü1# WÔ-Ø++eÓ$ vÑ-×5Ñ5°d·j±jÓAØ! Ü% vÔ.à #vv vÔ.àayEÒ!Ð!Ð!Ü
|
× Ñ s 4y¨(¸Ö@r; N) rO rP rQ Ú__test__Ú staticmethodr r+ r Úfloat64r r- rX rR r; r9 rT rT h s% ØHá Ó "DØ JJEØ Dó Ar; rT c ó< e Zd ZdZ ee« Zej Z dZ
|
y)ÚTestGroupVarFloat32Tg{®Gáz?N) rO rP rQ rY rZ r r+ r Úfloat32r r- rR r; r9 r] r] ~ s ØHá Ó "DØ JJEØ Dr; r] r r^ r[ c
|
óÞ t j t j j d« j d« | ¬« }t j g d¢« }t j
|
d| « }t j
|
t |« t j ¬« }t t j t j d« t j t j d|f « « « }t j } ||||d d
d f |« d }t j ||d d « ||d d
|
« ||d
|
d « g« }t j ||« t j" |t j g d ¢t j ¬« « t j$ |d d ||||d d
d f |« t j$ |d< t j ||« y ) Nr é r )é é r` )r é r r c óÌ t | « j « r$t j t j d« S | d | j « | j « | d gS )Nrc r éÿÿÿÿ)r
|
Úallr Úrepeatr! ÚmaxÚmin)Úgroups r9 Ú_ohlcz test_group_ohlc.<locals>._ohlc sH Ü ;??Ô Ü99RVV QÓ'Ð 'Øa%))+ u§y¡y£{°E¸"±IÐ>Ð>r; ra rb )ra ra é )r r> r r Ústandard_normalr$ Úlenr r rg r& ÚdiffÚr_Ú
|
libgroupbyÚ
|
group_ohlcr. rJ r/ r! ) r ÚobjÚbinsr3 r4 r6 Úfuncrk Úexpecteds r9 Útest_group_ohlcrw sa ä
|
((299×(Ñ(¨Ó+×;Ñ;¸BÓ?ÀuÔ
|
MCä 88KÓ DÜ
|
((65Ó
|
!CÜ XXc#h¤b§h¡hÔ /FÜ ¤§¡¬2¯9©9°Q«<¼¿¹ÄÇÁÀqÈ$ÀwÁ Ó9PÓ!QÓ RFä × Ñ DÙfc!T'l FÔ+ò?ô
|
xxs 2 Aw ©¨s°1°R¨yÓ)9¹5ÀÀRÀSÀ»?ÐKÓLHä×Ñ3 Ô)Ü× Ñ ¬¯©²Ä"Ç(Á(Ô(KÔLäffCGÙfc!T'l FÔ+Ü&&HQKÜ×Ñ3 Õ)r; c ó$ d}t j dgdgdgdgg|¬« }t j |« }t j g d¢t j ¬« }d} | |||||« t j
|
||« |dd
d f d¬
|
« y) a
|
Check a group transform that executes a cumulative function.
|
|
Parameters
|
----------
|
pd_op : callable
|
The pandas cumulative function.
|
np_op : callable
|
The analogous one in NumPy.
|
dtype : type
|
The specified dtype of the data.
|
Fr r r rc r )r r r r Nr )Ú check_dtype)r r> Ú
|
zeros_liker r. r/ )Úpd_opÚnp_opr Úis_datetimelikeÚdataÚanswerr6 Úngroupss r9 Ú(_check_cython_group_transform_cumulativer ¢ s ð Oä 88aS1# s Q CÐ(°Ô 6DÜ ]]4Ó Fä XXl¬"¯'©'Ô 2FØGÙ &$ ¨ Ô9ä× Ñ ¡ d£ ¨V²A°q°D©\ÀuÖMr; Únp_dtype)r Úuint64r^ r[ c ó t j | « j }t t j }}t |||« y ©N)r r Útyper Úcumsumr )r r r{ r| s r9 Ú"test_cython_group_transform_cumsumr » s1 ô HHXÓ × #Ñ #EÜ ¤§¡5EÜ,¨U°E¸5ÕAr; c ój t j } t t j }}t ||| « y r
)r r[ r Úcumprodr )r r{ r| s r9 Ú#test_cython_group_transform_cumprodr à s$ ä JJEÜ ¤"§*¡*5EÜ,¨U°E¸5ÕAr; c
|
ó d} t j g d¢t j ¬« }d}t j dgdgdgt j gdggd¬« }t j |« }|j t j « t ||||| « t j ddd t j d
|
gd¬« }t j |d d
d f |« t j |« }|j t j « t ||||| « t j ddd t j d gd¬« }t j |d d
d f |« d } t j t j dd« gdz d¬« d d
d f }t j |d¬« }t ||j d« ||| « t j t j dd« t j dd« t j dd« t j dd« t j dd« g« }t j |d d
d f j d« |« y )NF)r r r r r r r r r rc r[ ra é r r TÚnsr úm8[ns]r ) r r> r r! rz Úfillr r. r/ r Ú timedelta64Úview)r} r6 r r~ Úactualrv s r9 Ú!test_cython_group_transform_algosr Ê sï àOô XXo¬R¯W©WÔ 5FØGä 88aS1# s¤R§V¡V H¨q¨cÐ2¸)Ô DDÜ ]]4Ó FØ
|
KKÔÜ&$ ¨° ÔAÜxxAq¤"§&¡&¨"Ð-°YÔ?HÜ× Ñ ¢q¨! t¡ ¨hÔ7ä ]]4Ó FØ
|
KKÔÜv w°Ô@ÜxxAq¤"§&¡&¨"Ð-°YÔ?HÜ× Ñ ¢q¨! t¡ ¨hÔ7ð OÜ 88R^^ A tÓ,Ð-°Ñ1¸Ô BÂ1ÀdÀ7Ñ KDÜ ]]4 wÔ /FÜ 7Ó+¨V°W¸oÔNÜxxä NN1dÓ #Ü NN1dÓ #Ü NN1dÓ #Ü NN1dÓ #Ü NN1dÓ #ð
|
óHô × Ñ ¢q¨! t¡ × 1Ñ 1°(Ó ;¸XÕFr; c ó8 t j dd¬« } t j dgd¬« }t j t j dd« t j d d« t j d
|
« gd ¬« d d
d f j d« j d« }t j t |« t j ¬« }t | |||d ¬ « t j | d d
df t j dgd¬« « y )Nr= r[ ©rW r r r r r r rc ÚNaTr T©r} r ) r r$ r> r r r# rn r r r. r/ ©r r4 r~ r6 s r9 Ú#test_cython_group_mean_datetimeliker ð sÓ Ü XXF¨)Ô 4FÜ XXqc Ô )Fä
|
Ü ^^AtÓ $¤b§n¡n°Q¸Ó&=¼r¿~¹~ÈeÓ?TÐ UØô
|
ò T'ñ ÷
|
g ß Ó ð ô XXc$i¤r§w¡wÔ /Fävvt V¸TÕBä× Ñ ¢q¨! t¡ ¬b¯h©h¸°sÀ)Ô.LÕMr; c óh t j dd¬« } t j dd¬« }t j dd¬« d d
d f }t j dt j ¬« }t j t
|
d¬« 5 t | |||d d
|
¬ « d d d « y # 1 sw Y y xY w) Nr= r[ r r r r Ú min_count)ÚmatchTr )r} r )r r$ r ÚpytestÚraisesÚAssertionErrorr r s r9 Ú&test_cython_group_mean_wrong_min_countr¡ s Ü XXF¨)Ô 4FÜ XXawÔ 'FÜ 88AYÔ 'ª¨4¨Ñ 0DÜ XXarwwÔ 'Fä ~¨[Ô 9ñ TÜ66 4¨ÀÐQRÕS÷T÷ Tñ Tús  B(Â(B1c óf t j dd¬« } t j dgd¬« }t j t j d« t j d« gd¬« d d
d f j d« j d« }t j t |« t j ¬« }t | |||d ¬
|
« t j | d d
df t j t j t j |d |d « d « d¬« « y ) Nr= r[ r r r r r r Fr r r ) r r$ r> r r r# rn r r r. r/ ÚdivideÚaddr s r9 Ú:test_cython_group_mean_not_datetimelike_but_has_NaT_valuesr¥ sã Ü XXF¨)Ô 4FÜ XXqc Ô )Fä
|
Ü ^^EÓ "¤B§N¡N°5Ó$9Ð :Øô
|
ò T'ñ ÷
|
g ß Ó ð ô XXc$i¤r§w¡wÔ /Fävvt V¸UÕCä× Ñ Øq!t bhhryy¬¯©°°Q±¸¸a¹Ó)AÀ1Ó EÈYÔWõr; c óv t j t j t j gt j t j ggd¬« } t j ddgd¬« }t j t j dgddgddgddgdd gd
|
t j ggd¬« }t j g d ¢t j ¬« }t | |||d ¬ « t j t j dgdt j ggd¬« }t j | |« y )Nr[ r r r g ð?g @g @g @g @r )r r r r r r Fr r )r r> r! Úinfr r r. r/ )r r4 r~ r6 rv s r9 Ú.test_cython_group_mean_Inf_at_begining_and_endr¨ sæ ä XX¤§¡Ð'¬"¯&©&´"·&±&Ð)9Ð:À)Ô LFÜ XXq!f GÔ ,FÜ 88Ü
|
&&# c
|
S¨# J°°c°
|
¸SÀ#¸JÈÌBÏFÉFÈ ÐTØô Dô XXÒ(´·±Ô 8Fävvt V¸UÕCäxx"&& ! q¬"¯&©& kÐ2¸)ÔDHä× Ñ ØØõr; z values, outc ó t j t j gt j ggd¬« }t j ddgd¬« }t j | d¬« }t j g d¢t j ¬« }t ||||d d¬« t j |d¬« }t j ||« y )Nr[ r r r )r r r Fr )r r> r! r r r. r/ )r5 r3 r r4 r~ r6 rv s r9 Ú-test_cython_group_sum_Inf_at_begining_and_endrª 4 s ô XXx¤"§&¡& Ð*°)Ô <FÜ XXq!f GÔ ,FÜ 88F )Ô ,DÜ XXi¤r§w¡wÔ /Fä ffd F¨DÀ%ÕHäxx 9Ô-Hä× Ñ ØØõr; )$Únumpyr r Ú pandas._libsr rq Úpandas._libs.groupbyr r r r r Úpandas.core.dtypes.commonr Úpandasr
|
Úpandas._testingÚ_testingr. r rT r] ÚmarkÚ parametrizerw r r r r r r¡ r¥ r¨ r§ r! rª rR r; r9 ú<module>r´ sÅ ðÛ Û å .÷õ õ :å Ý ÷R/ñ R/ôjAÐ+ô Aô,Ð+ô ð ×Ñ 9¨iÐ"8Ó9ñ*ó :ð*ò6Nð2 ×ÑÒ%NÓOñBó PðBòBò#GòLNò$Tòò(ð( ×ÑØà 66(RVVHrvvhÐ '¨2¯6©6¨(°R·V±V°HÐ)=Ð>Ø 66(RVVH §¡wiÐ (¨B¯F©F¨8°b·f±f°XÐ*>Ð?Ø 66(bffWI §¡xÐ (¨B¯F©F¨8°b·f±f°XÐ*>Ð?Ø 66(bffWI §¡ yÐ )¨R¯V©V¨H¸¿¹°w°iÐ+@ÐAð óñóñr;
|
