U
|
ý°d \ ã 8 @ s` d dl m Z d dlZd dlZd dlZd dlmZmZmZm Z d dl
|
m Z dd Z dd Zej dd gd d
|
gg¡ejjd d d dD e d¡e d¡e d¡e d¡e d¡gejdddejdddejdddejdddejdddge d¡e d ¡ e d¡e d ¡ e d¡e d ¡ e d¡e d ¡ e d¡e d ¡ ge d¡ d¡e d¡ d¡e d¡ d¡e d¡ d¡e d¡ d¡gg dd dej ddddd d d d d gfdddd d d d d!gfdddd d d d d gfdddd!d!d"d!d gfd#ddd d d d d gfd#ddd"d"d d"d!gfd#ddd$d$d d$d gfd#ddd%d%d"d%d gfd&ddd$d$d d$d gfd&ddd%d%d d%d!gfd&ddd d d d d gfd&ddd d d"d d gfd'ddd d d d$d gfd'ddd"d d d%d!gfd'ddd$d d d d gfd'ddd%d!d"d d gfd(ddd d d$d d gfd(ddd)d)d d)d*gfd(ddd$d$d d$d gfd(ddd d d)d d*gfg¡d+d, Zej dd gd d
|
gg¡ej d ej ej ejd ejejejgg¡ej d-ddd.d/d/ejd0ejd1d1gfddd2d3d3d/d d/d4d4gfddd5d/d/d4d$d4d1d1gfddd.d1d1ejd0ejd/d/gfddd2d4d4d/d d/d3d3gfddd5d1d1d4d$d4d/d/gfd#dd.d d ejd$ejd d gfd#dd2d$d$d d d d6d6gfd#dd5d d d6d$d6d d gfd#dd.d d ejd$ejd d gfd#dd2d6d6d d d d$d$gfd#dd5d d d6d$d6d d gfd&dd.d d ejd$ejd d gfd&dd2d d d d d d7d7gfd&dd5d d d7d$d7d d gfd&dd.d d ejd$ejd d gfd&dd2d7d7d d d d d gfd&dd5d d d7d$d7d d gfd'dd.d d ejd$ejd d gfd'dd2d$d d d d d6d7gfd'dd5d d d6d$d7d d gfd'dd.d d ejd$ejd d gfd'dd2d6d7d d d d$d gfd'dd5d d d6d$d7d d gfd(dd.d d ejd ejd$d$gfd(dd2d d d d$d d d gfd(dd5d d d d d d$d$gfd(dd.d$d$ejd ejd d gfd(dd2d d d d$d d d gfd(dd5d$d$d d d d d gfg ¡d8d9 Zej dd gd d
|
gg¡ejjd d:d d;D e d¡e d¡eje d¡e d¡e d¡ejejgejdddejdddejejdddejdddejdddejejge d¡e d ¡ e d¡e d ¡ eje d¡e d ¡ e d¡e d ¡ e d¡e d ¡ ejejge d¡ d¡e d¡ d¡eje d¡ d¡e d¡ d¡e d¡ d¡ejejgg d<d dej d=ddd.dd d ejd d d ejejgfddd.dd d ejd d d!ejejgfddd.dd d ejd d d ejejgfddd.dd!d!ejd"d!d ejejgfd#dd.dd d ejd d d ejejgfd#dd.dd"d"ejd d"d!ejejgfd#dd.dd$d$ejd d$d ejejgfd#dd.dd%d%ejd"d%d ejejgfd&dd.dd$d$ejd d$d ejejgfd&dd.dd%d%ejd d%d!ejejgfd&dd.dd d ejd d d ejejgfd&dd.dd d ejd"d d ejejgfd'dd.dd d ejd d$d ejejgfd'dd.dd"d ejd d%d!ejejgfd'dd.dd$d ejd d d ejejgfd'dd.dd%d!ejd"d d ejejgfd(dd.dd d ejd$d d ejejgfd(dd.dd)d)ejd d)d*ejejgfd(dd.dd$d$ejd d$d ejejgfd(dd.dd d ejd)d d*ejejgfddd5dd d d7d d d d7d7gfddd5dd>d>d?d@d>dAd?d?gfddd5dd d d7d d d d7d7gfddd5ddAdAd?dBdAd>d?d?gfd#dd5dd d d6d d d d6d6gfd#dd5ddBdBdCd@dBdAdCdCgfd#dd5dd$d$d6d d$d d6d6gfd#dd5ddDdDdCdBdDd>dCdCgfd&dd5dd$d$dEd d$d dEdEgfd&dd5ddDdDd d@dDdAd d gfd&dd5dd d dEd d d dEdEgfd&dd5dd@d@d dBd@d>d d gfd'dd5dd d d6d d$d d7dEgfd'dd5ddBd>dCd@dDdAd?d gfd'dd5dd$d d6d d d d7dEgfd'dd5ddDdAdCdBd@d>d?d gfd(dd5dd d d d$d d d d gfd(dd5dd>d>d dCd>dAd d gfd(dd5dd$d$d d d$d d d gfd(dd5ddCdCd d>dCdAd d gfg(¡dFdG Zej dHdd$d$d$d$d$gfdd%d%d%d%d%gfg¡dIdJ Zej dKdLdMdNdOdPdQg¡ej dRddg¡dSdT Zej dUdd#d&d'd(g¡ej dVddg¡ej dWd.d2d5g¡ej dXddg¡ej d dYdYdZdYd[gdYejdZejd[gg¡d\d] Zej dWdd^d_g¡ej dUdd#d&d'd(g¡ej dVddg¡ej dXddg¡ej d dYdYdZdYd[gdYejdZejd[gd_ejd`ejd0gg¡dadb Zdcdd Z ej ded_d`gd_d_gd d gfd_d_d`d`gd_d`d_d`gdAd dAd gfd_d_d`d`gd_d`d_ejgdAd d ejgfd_d_d`gd_d`ejgdAd ejgfg¡dfdg Z dhdi Z ej djddg¡dkdl Z dmdn Z dodp Z!dqdr Z"dsdt Z#dudv Z$dS )wé )ÚdatetimeN)Ú DataFrameÚNaTÚSeriesÚconcatc C s~ t jg dd} t| }| ¡ }d}| | ¡}tjt|d | ¡ W 5 Q R X | | ¡}tjt|d | ¡ W 5 Q R X d S )NF©Zorderedz0Cannot perform rank with non-ordered Categorical©Úmatch) ÚpdÚ Categoricalr Zto_frameÚgroupbyÚpytestÚraisesÚ TypeErrorÚrank)ÚcatZserÚdfÚmsgÚgbÚgb2© r úUd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/groupby/test_rank.pyÚ)test_rank_unordered_categorical_typeerror s
|
|
r c C s t dd¡} t dd¡}tjjdddd}tjjdddd}ttj d¡| |¡| |¡d}| dd g¡j
|
¡ }d
|
d | dd g¡D }t |dd }| |j ¡}t ||¡ | dd g¡j j
|
d d}dd | dd g¡D }t |dd }| |j ¡}t ||¡ d S )Né
|
éd é r iô )Úsize)ÚvalueÚkey1Úkey2r r c S s g | ]\}}|j ¡ qS r ©r r ©Ú.0ÚkeyZpiecer r r Ú
|
<listcomp>1 s z#test_rank_apply.<locals>.<listcomp>©ÚaxisT©Úpctc S s g | ]\}}|j jd dqS )Tr' r r! r r r r$ 8 s )ÚtmZ rands_arrayÚnpÚrandomÚrandintr ZrandnZtaker r r r ZreindexÚindexÚassert_series_equal)Zlev1Zlev2Zlab1Zlab2r ÚresultÚexpectedr r r Útest_rank_apply! s,
|
ýÿ ÿ r1 ÚgrpsZquxZquuxÚvalsc C s$ g | ] }t jd d dd dg|dqS ©é é é ©Údtype)r* Úarray©r" r9 r r r r$ C s ÿr$ ) Úi8Úi4Úi2Úi1Úu8Zu4Úu2Úu1Úf8Úf4Úf2z
|
2018-01-02z
|
2018-01-08z
|
2018-01-06z
|
US/Pacific©ÚtzÚDc C s t | d S ©Nr ©Útype©Úxr r r Ú<lambda>e ó rN )Úidsz ties_method,ascending,pct,expZaverageTFç @g @g @gÙ?ç ð?gé?gÉ?Úminç @g333333ã?ÚmaxÚfirstÚdenseçUUUUUUÕ?çUUUUUUå?c C s t | t|¡}|}t|t| }t|t jr@t j||jd}t||d}| d¡j
|
|||d} t|t| dgd}
|
t | |
|
¡ d S )Nr8 ©r# Úvalr# )ÚmethodÚ ascendingr( r[ ©Úcolumns© r* ÚrepeatÚlenÚlistÚ
|
isinstanceZndarrayr: r9 r r r r) Úassert_frame_equal) r2 r3 Ú ties_methodr] r( Úexpr# Ú orig_valsr r/ Úexp_dfr r r Útest_rank_args@ s A rj z#ties_method,ascending,na_option,expZkeepç ø?é g @Útopç @g @Úbottomg @g @c
|
C sf t | t|¡}|t| }t||d}| d¡j|||d}t|t| dgd} t || ¡ d S )NrZ r# )r\ r] Ú na_optionr[ r^ )r* ra rb r r r r) re )
|
r2 r3 rf r] rp rg r# r r/ ri r r r Útest_infs_n_nans s )
|
ÿrq c C s0 g | ](}t jd d t jdd dt jt jg|dqS r4 )r* r: Únanr; r r r r$ Å s ÿ)rC rD rE c C s t | d S rI rJ rL r r r rN ó rO z'ties_method,ascending,na_option,pct,expg Ð?g ì?g ä?ç à?g À?g è?g Ø?g @c C s t | t|¡}|}t|t| }t|t jr@t j||jd}t||d} | d¡j
|
||||d}
|
t|t| dgd} t |
|
| ¡ d S )Nr8 rZ r# ©r\ r] rp r( r[ r^ r` ) r2 r3 rf r] rp r( rg r# rh r r/ ri r r r Útest_rank_args_missing s r
|
ÿru zpct,expc C s\ t ddddddddddg
|
dgd d}| d¡j| d}t |d d gd
|
}t ||¡ d S ) NÚaÚbé r rZ r# r' r5 r[ r^ ©r r r r) re )r( rg r r/ ri r r r Útest_rank_resets_each_groupà s "ÿrz r9 Úint64Úint32Zuint64Zuint32Úfloat64Úfloat32Úupperc C s¨ |r(| d ¡ | dd
} | dd¡} tdgd dgd d}|d | ¡|d< |d j| ksft| d ¡ ¡ }td
|
d
|
d
|
d
|
gdgd }|r| d ¡}t ||¡ d S ) Nr rx ZUiZUIrv é rZ r[ r# g @r^ ZFloat64)
|
r Úreplacer Úastyper9 ÚAssertionErrorr r r) re )r9 r r r/ ri r r r Útest_rank_avg_even_valsÏ s
|
r rf r] rp r( ÚbarÚfooZbazc C s² t dgd |d}|d ¡ }| d¡}|j| |||d}| ¡ rht dgd dtjdtjd gd} n t dgd ddddd gd} | d¡}
|
|
|
j| |||d} t || ¡ d S )
|
Nr é rZ r[ r# rt r r5 rx ) r Zisnar r Úanyr* rr r) re ) rf r] rp r( r3 r Úmaskr ÚresZdf2r Úaltr r r Útest_rank_object_dtypeã s
|
$
|
ÿr Úbadrx r5 c C sN t dgd |d}d}tjt|d | d¡j| |||d W 5 Q R X d S )Nr r rZ z3na_option must be one of 'keep', 'top', or 'bottom'r r# rt )r r r Ú
|
ValueErrorr r )rf r] rp r( r3 r r r r r Útest_rank_naoption_raisesÿ s
|
ÿr c C s d} t dddgdtjdgd}| | ¡jjdd}td tjdgd
|
d }t ||¡ | | ¡jdd}t d
|
d tjdgi}t ||¡ d S ) NÚAr rx rR rQ ©r ÚBTr' rs r ©Úname)
|
r r* rr r r r r r) r. re )Úcolumnr r/ r0 r r r Útest_rank_empty_group s r z"input_key,input_value,output_valuec C s> t | |d}| d¡jddd}t d|i}t ||¡ d S )Nr r rW T©r\ r( r ry )Z input_keyZ input_valueZ output_valuer r/ r0 r r r Útest_rank_zero_div# s r c C s t dddgt tj¡jt tj¡jt tj¡jgttdddtgd} | d¡ ¡ }t dddgtj
|
dtj
|
gd}t ||¡ d S ) Nrx r5 iÑ )ÚgrpÚint_colÚ datetimeliker rR rQ )r r ) r r* Ziinfor{ rS rU r r r r ÚNaNr) re ©r r/ r0 r r r Útest_rank_min_int5 s ýùÿ ÿr Úuse_nanc C s | r
|
t jnd}tddgddgd|gd|ggddgd}| dg¡d jdd d
|
}| rntd dt jt jgdd }ntd dddgdd }t ||¡ d S )Nrl éÿÿÿÿrx r5 Úgroupr[ r^ rW Tr rs r rX rY )r* rr r r r r r) r. )r Z
|
fill_valuer r/ r0 r r r Ú.test_rank_pct_equal_values_on_group_transitionK s" üù þr¢ c C s t tddgddgdtddgdd gd
|
d dd } | jd dd}|jdd }t | d jdd | d jdd gdddgd}t ||¡ d S )Nrl r rx r5 )Úcol1Úcol2r r7 é r6 )Úcol3Zcol4)rv rw r% r ©Úlevelr& rv rw )r& Úkeys)r r r r r) re )r r r/ r0 r r r Útest_rank_multiindexd s þû þúrª c C s¤ t ddddgddddgd d
|
d d gd ddddgd} | jddd}|jdd}t| jd jdd| jd jddgdd}t ||¡ |jdd}| |¡r td S ©Nrx rl r r¥ r5 r r7 r6 rk rn g @g @)r rx r5 rv rw ©r- r r§ r% ) r r r r Úlocr) re Úequalsr )r r r r0 r r r r Ú test_groupby_axis0_rank_axis1| s "
|
þ , r¯ c C s~ t ddddgddddgd d
|
d d gd ddddgd} | jddd}|jdd}| ddg tj¡}|d |d< t ||¡ d S r« )r r Zcummaxr r* r} r) re )r r Zcmaxr0 r r r Ú test_groupby_axis0_cummax_axis1 s "
|
þ r° c C s t dddtjgddtjdddgd d } | | jd
|
g¡jjd d d }t dddtjgtjdddgd d d}t
|
||¡ d S )NrR rQ rT )r r Z20170101z
|
US/EasternrF r r¬ r T)r] r( r )r- r ) r r* rr r
|
Ú Timestampr r- r r r r) r. r r r r Útest_non_unique_index s þ ýr² c C s t jdddtjddgdd} t jdddtjd d
|
gdd}td dd dd dg| |d }| d ¡}| ¡ }| t¡ d ¡ ¡ }t
|
||¡ d S )Nrv rw ÚcTr rx r5 rl r r r )r£ r¤ r¦ r£ ) r
|
r r* rr r r r r Úobjectr) re )r Zcat2r r r r0 r r r Útest_rank_categorical¯ s
|
rµ )%r Únumpyr* r Zpandasr
|
r r r r Zpandas._testingZ_testingr) r r1 ÚmarkZ parametrizer± Z to_periodrj Úinfrr rq ru rz r r r r r r r¢ rª r¯ r° r² rµ r r r r Ú<module> sä þû ûûûêü"Ü&ìþ "ÿ âþ# þø ø ø øáü.Ð2ûûûû""û"""û"ûûûûûøûûøû û û û û û û û Èþ > "ÿ
|
ÿ ÿ ýþ
|
$ üþ
|
|
|
