U
|
ý°dm> ã @ sÌ d dl Zd dlZd dlZd dlmZmZ d dlmZ ej
|
ddddddg¡ej
|
d d
|
d d d dgdd d d d
|
gfd
|
d d d gd d d d
|
gfd
|
d d d dgd d d d
|
gfdddddgdddddgfdej dej dgdej dej dgfej dej dej gej dej dej gfej ddddej ddddddd
fej dddd d¡ej ddddddd
d¡fej gd ej gd fg ¡ej
|
dd ddd d
|
g¡d d Zd d Zd!d" Zd#d$ Zd%d& Zej
|
d'd(d)g¡ej
|
d*d gd d
|
gg¡ej
|
ddd+gg¡d,d- Zd.d/ Zd0d1 Zd2d3 Zej
|
d4dej dej ged ddgd5dgfdej ddged ddgd5d6gfd7d8d8ej ged d7d8gd d9gfd gd:gd gd;gfg g ejg d<d=ejg d<d=fg¡d>d? Zej
|
d@ejd
|
d dgd dAd=ejdBdCdgd dDd=g¡ej
|
ddd5ddgg¡dEdF Zej
|
ddd5ddgg¡ej
|
dGdBdCg¡dHdI Z dJdK Z dLdM Z ej
|
dNdOdPdQg¡dRdS Z ej
|
dddg¡dTdU Z ej
|
dVdWdXg¡dYdZ Z!d[d\ Z"d]d^ Z#d_d` Z$dadb Z%dS )cé N)Ú DataFrameÚIndexÚ interpolationZlinearÚlowerÚhigherÚnearestZmidpointz a_vals,b_valsé é é é é ç ð?ç @ç @ç @ç @z1/1/18ÚD)ÚfreqÚperiodséÿÿÿÿÚsÚqç Ð?ç à?ç è?c C s | dkr@|dkr@t |tr@|ddddgkr@|j tjjdd¡ t t |¡t |¡g¡}t |¡j
|
|| d }t |¡j
|
|| d }t d
|
gt | d gt | |d }t ||gd gt d
|
d gddd} |jjdkrò| jjd jdkrò| |j¡} | d¡j
|
|| d }
|
t |
|
| ¡ d S )Nr r r r
|
r r zBUnclear numpy expectation for nearest result with equidistant data)Úreason)r ÚaÚb©ÚkeyÚvalr r ©Úname)ÚcolumnsÚindexÚMr )Ú
|
isinstanceÚlistÚnodeZ
|
add_markerÚpytestÚmarkZxfailÚpdÚconcatÚSeriesÚquantiler Úlenr ÚdtypeÚkindZdtypesÚvaluesÚastypeÚgroupbyÚtmÚassert_frame_equal) r Za_valsZb_valsr ÚrequestZall_valsZ
|
a_expectedZ
|
b_expectedÚdfÚexpectedÚresult© r; úYd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/groupby/test_quantile.pyÚ test_quantile s4 ÿþýüÿÿ& ÿ r= c C s t ddddddgi} tjdddddgtjd}| |¡ dg¡}tj ddgdgg¡}t ddd gi|d
|
}t
|
||¡ t ddddgdd d d gd} tj ddgddgg¡}tjddddgtjd}| |¡ ddg¡}t ddddgddddgd|d
|
}t
|
||¡ d S )NÚAr r r r
|
r ©r0 r ç @©r$ r é é ©r> ÚBr g @g @g @g @g @g @© r ÚnpÚarrayÚint64r4 r. r+ Ú
|
MultiIndexÚ from_productr5 r6 )r8 r r: r$ r9 r; r; r< Útest_quantile_arrayI s ÿrL c C s¬ t j d¡jdddt jd} t| tdd}| d¡ dd g¡}td
|
d d d ddddddg
|
ddddddddddg
|
dt j
|
j dddd d gdd ggdd gd d }t ||¡ d S ) Nr r )é
|
r
|
)Úsizer0 ÚABC©r# r> g333333Ó?çffffffæ?gÍÌÌÌÌÌì?gÍÌÌÌÌÌ @g@g333333 @çù?g333333@gffffff@g@ç g333333ó?gffffff@gÍÌÌÌÌÌü?r gffffffþ?gÍÌÌÌÌÌ@)rE ÚCr r r
|
r ©ÚnamesrA )rG ÚrandomZ RandomStateÚrandintrI r r' r4 r. r+ rJ rK r5 r6 )Zarrr8 r: r9 r; r; r< Útest_quantile_array2^ s þ ÿû rY c C sô t dddgdddgd} tjdddgtjd}| j|d d
|
d d d g¡}t d dddddgddddddgdtj ddgd d d gg¡d}t
|
||¡ | j|d d
|
d d g¡}t dd ddgddddgdtj ddgd d gg¡d}t
|
||¡ d S )Nr r r r
|
r r rD r? F)Úsortr r r r ç ø?ç @r g @rA rF )r8 r r: r9 r; r; r< Útest_quantile_array_no_sorto s þ þr] c C s t dddgdddgdddgdddgd } | d
|
d g¡ d d g¡}tjjddddgd
|
d d gd}t d d ddgddddgd|d}t ||¡ d S )Nr r r r
|
r r r r )r> rE ÚcÚdr^ r_ r r )r r r )r r r )r r r )r r r rU r g
|
@g @r rD rA )r r4 r. r+ rJ Ú from_tuplesr5 r6 )r8 r: r$ r9 r; r; r< Ú#test_quantile_array_multiple_levels s $ÿ
|
þ ÿra Ú
|
frame_size)r r
|
)éd rM r4 g333333ã?c s | \}t t fddt|D ¡td}t t|d¡gt g }fddtt|dD gt tttt|d g }tj || d g d} fddtt|dD } fddtD } t ||| d }
|
|
|
¡ ¡} t | |
|
¡ d S )
|
Nc s g | ]} |d g qS )r r; )Ú.0Ú_)Úncolr; r< Ú
|
<listcomp> s zJtest_groupby_quantile_with_arraylike_q_and_int_columns.<locals>.<listcomp>rP r c s g | ]} D ]}|q qS r; r; ©rd Úxre ©r r; r< rg s )ÚlevelsÚcodesrV c s, g | ]$}D ]}t |gt q qS r; )Úfloatr/ rh ©r4 rf r r; r< rg s c s g | ]}| kr|qS r; r; )rd ri )r4 r; r< rg ¢ s ©r$ r# )r rG rH ÚrangeÚarangeÚminr/ r' r+ rJ r4 r. r5 r6 ) rb r4 r Znrowr8 Z
|
idx_levelsZ idx_codesZexpected_indexZexpected_valuesZexpected_columnsr9 r: r; rn r< Ú6test_groupby_quantile_with_arraylike_q_and_int_columns s, ( $ÿ ÿ ÿ ÿrs c C sN t ddgddgddggddgd} tjtdd | d¡ ¡ W 5 Q R X d S )
|
NZfoor r r^ r r rP z+cannot be performed against 'object' dtypes©Úmatch)r r) ÚraisesÚ TypeErrorr4 r. ©r8 r; r; r< Útest_quantile_raises« s "ry c C s t ddddddgtdd} | ddddddg¡}tjtdd | d¡ W 5 Q R X tjtdd | d ¡ W 5 Q R X d S )
|
Nr r rB ©r r zGot '50.0' insteadrt é2 zGot '-1.0' insteadr )r rp r4 r) rv Ú
|
ValueErrorr. )r8 Úgr; r; r< Ú$test_quantile_out_of_bounds_q_raises² s r~ c C sH t dt jdg¡} t| tdd}| d¡}tdD ] }| ¡ q6d S )Nr r
|
r r rc )rG rH Únanr rp r4 r. )Údatar8 Úgrpre r; r; r< Ú/test_quantile_missing_group_values_no_segfaults½ s
|
|
r z$key, val, expected_key, expected_valrS r@ r r r[ é* g E@Úfloat64r? c C s^ t | |d}t |t|dddgd}| d¡}| d¡}t ||¡ | ¡ }t ||¡ d S )Nr r r! r ro r )r r r4 r. r5 r6 )r r Z expected_keyZ expected_valr8 r9 r r: r; r; r< Ú2test_quantile_missing_group_values_correct_resultsÈ s
|
ÿ
|
|
r
r2 ZInt64TFÚbooleanc C s t dgd dgd | d}| d¡d |¡}t|trdtjjddg|fdd gd}dd d
|
g}ntddgdd }d g}tj |d |dd d}t
|
||¡ d S )Nri r
|
Úyrz r r rU rS r r r! r ÚFloat64©r$ r" r0 ) r r4 r. r& r' r+ rJ rK r r- r5 Úassert_series_equal)r2 r r8 r: ÚidxZtrue_quantilesr9 r; r; r< Ú$test_groupby_quantile_nullable_arrayå s
|
|
r Ú numeric_onlyc C s t dgdgdgd}|rR| d¡j| |d}| d¡dg | ¡}t ||¡ n.tjtdd | d¡j| |d W 5 Q R X d S )
|
Nr r ri )r r r^ r )r r z7'quantile' cannot be performed against 'object' dtypes!rt )r r4 r. r5 r6 r) rv rw )r r r8 r: r9 r; r; r< Ú-test_groupby_quantile_raises_on_invalid_dtypeý s ÿr c C sÊ t ddgdtjgd| d}| d¡d d¡}tdg| dd }| d
|
krN| }nd }tjdg||dd }t ||¡ | d¡d dd g¡}tjdgd tj
|
j |dd gfdd gdd|d}t ||¡ d S )Nr çÉ?©ri r r? ri r r r )r0 r" )ÚFloat32r ©r0 r$ r" r r rU r ) r rG r r4 r. r r+ r- r5 r rJ rK )Zany_float_dtyper8 r: Z exp_indexÚexpected_dtyper9 r; r; r< Ú test_groupby_quantile_NA_float s ür c C s t ddgddgd| d}| d¡d d¡}tjd gd
|
tdgd| d dd }t ||¡ | d¡ d¡}t dd id
|
tdgd| d d }t ||¡ d S )Nr r r r r? ri r r r\ r )r" r0 r ©r0 r$ ) r r4 r. r+ r- r r5 r r6 )Zany_int_ea_dtyper8 r: r9 r; r; r< Ú test_groupby_quantile_NA_int$ s ü ÿr zinterpolation, val1, val2)r r r )r r r
|
)r r r c C s t ddddgdddtjgd|d}| d¡jddg| d }t d
|
||tjtjgi|tjjtjddddg|dddddggdd gd d }t ||¡ d S ) Nr r r
|
rz r? r r rQ ©r r r rU r ©
|
r r+ ÚNAr4 r. rJ Z from_arraysr- r5 r6 )r Zval1Zval2Úany_numeric_ea_dtyper8 r: r9 r; r; r< Ú)test_groupby_quantile_all_na_group_masked7 s ÿ þýr c C sª t ddddgdddtjgd|d}| d¡jddg| d }|d
|
krJ|}nd }t d d dtjtjgi|tjjtjddddg|dddddggdd gdd}t ||¡ d S )Nr r r
|
rz r? r r r r r r r r r@ rU r r )r r r8 r: r r9 r; r; r< Ú0test_groupby_quantile_all_na_group_masked_interpM s& ÿ
|
þûý r r0 r r c C sj t ddgtjgd d| d}| d¡d d¡}tjtjg| tdg| ddd }d|j _
|
t ||¡ d S )
|
Nr r r r? ri r r r r ) r r+ r r4 r. r- rG r r r$ r" r5 r )r0 r8 r: r9 r; r; r< Ú"test_groupby_quantile_allNA_columnj s ÿr c C sr t tjt d¡ddddddgd} | d¡ d¡}t d t d
|
¡t d ¡gitddgdd d }t
|
||¡ d S )Nr r )Úunitr r )ÚvalueÚgroupr g®Gáz®ï?r z0 days 00:00:00.990000z0 days 00:00:02.990000r! rA ) r r+ Z to_timedeltarG rq r4 r. Z Timedeltar r5 r6 ©r8 r: r9 r; r; r< Ú test_groupby_timedelta_quantilev s ÿþÿù r¢ c C s t t d¡ dd¡tdtjtdddd} | jdd d
|
jd d gd }t ddddgddddgddddggtdtj j
|
ddd d gdd gd d}t ||¡ d S ) Né r
|
r ZXYZZABABÚcolr! ro r )Zaxisçé?r rj rR gÙ?gÍÌÌÌÌÌ@gffffffö?gffffff@g@gffffff@g@g333333#@gÍÌÌÌÌÌ @g333333%@gÍÌÌÌÌÌ"@)r> r¥ )r> r )rE r¥ )rE r rU ) r rG rq Zreshaper' r+ r- r4 r. rJ r` r5 r6 r¡ r; r; r< Ú test_columns_groupby_quantile s$ ý
|
|
|
ý
|
ÿù r¦ c C sÄ t tjddddd d¡ttddttdd d
|
} | d ¡ d d g¡}t ddddddddddddgtjj t
|
d¡d ft
|
d¡d ft
|
d¡d ft
|
d¡d fgddd}t ||¡ d S ) Nz2020-04-19 00:00:00Z1Trc ÚUTC)Ústartr r ÚtzZ1Hr ée éÉ )Ú timestampÚcategoryr r¬ r r¥ g)@g333333\@)r r gH@gfffffb@g333333Q@ge@gÍÌÌÌÌ W@gfffffh@z2020-04-19 00:00:00+00:00z2020-04-19 01:00:00+00:00)r¬ NrU rA ) r r+ Ú
|
date_rangeÚfloorr' rp r4 r. rJ r` Ú Timestampr5 r6 r¡ r; r; r< Ú test_timestamp_groupby_quantile s< ÿþ ûÿ
|
ü üùùr± c s¾ t jddd} t | ¡}| ¡ | d¡ d< | d¡ d< | | d d < t j jd
|
< t t
|
d ¡d ¡} |¡}| d ¡} fddt d D }t|j}|j tj¡|_t ||¡ d S )Nz
|
2016-01-01iè )r z
|
US/Pacificr r r r r
|
r r éÈ r c s$ i | ] }| j |d d
d¡qS )Nr r )Úilocr. )rd Úirx r; r< Ú
|
<dictcomp>Ò s z7test_groupby_quantile_dt64tz_period.<locals>.<dictcomp>)r+ r® r- Zto_frameZ tz_localizeZ to_periodZNaTr³ rG Ztilerq r4 r. rp r ÚTr$ r3 Úint_r5 r6 )ZdtiZserZbyÚgbr: Úexpr9 r; rx r< Ú#test_groupby_quantile_dt64tz_period s
|
|
|
|
rº )&ÚnumpyrG r) Zpandasr+ r r Zpandas._testingZ_testingr5 r* Z parametrizer r® Zas_unitr= rL rY r] ra rs ry r~ r rp rH r
r r r r r r r r¢ r¦ r± rº r; r; r; r< Ú<module> s ÿ"&þ þìþ " ûþ
|
|
þþ ÿ
|
|
|
#
|
