U
|
ý°dÏ- ã @ sÆ d Z ddlZddlZddlmZmZ ddlZddlm Z m
|
Z
|
m Z m Z m Z mZmZ ddlmZ ej ddddd d
|
d ejd ej d ¡gddddg
|
¡dd Zdd Zdd Zdd Zdd Zd d Z d d Z ej d d ej fd ej fd ej fdej!fdej"fdej#fdej$fd!d"d# fd$d%d# fg ¡d&d' Z%ej d d ej fd d(d# fd d)d# fdej#fdej$fg¡d*d+ Z&d,d- Z'ej d.d!d$ddg¡ej d/ed0e d1g¡d2d3 Z(ej d4dddddd d d d5d6d7d8d9d:d;d!d$d<d=d>g¡d?d@ Z)ej ddddd d
|
d d dddg
|
¡dAdB Z*ej dCdDdEg¡ej dFdGdHdIdJdKdLdMdNdOg ¡ej d/ej+dPdQdRdSgdTdUej+dPdQdRdSgdVdUej+dWdXdYdZgd[dUej+dWdXdYdZgd\dUej+dDdDdEdEgd]dUg¡d^d_ Z,dS )`z
|
test cython .agg behavior
|
é N)Úis_float_dtypeÚis_integer_dtype)Ú DataFrameÚIndexÚNaTÚSeriesÚ TimedeltaÚ TimestampÚ bdate_rangeÚop_nameÚcountÚsumZstdÚvarZsemÚmeanÚmedianzignore::RuntimeWarning)ZmarksÚprodÚminÚmaxc
|
s8 ddddddddddt jt jg ddgd t j d¡d}t|}t j|jd d
|
d
d f< fd d |jdgdd d¡} fdd|D }td |i}d|j_ |}t
|
||¡ | ddg¡}i }|D ]$\\}}} | d | |i ¡|< qÈt|j jdd}ddg|j_d |_ |d }dkr4t
|
||¡ d S )Nr é g ð?ÚAÚBé é )r r ÚCé é
|
r c s t | S ©N)Úgetattr©Úx)r © úad:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/groupby/aggregate/test_cython.pyÚ<lambda>7 ó z(test_cythonized_aggers.<locals>.<lambda>)Úaxisc s i | ]\}}| |d qS )r r )Ú.0ÚcatÚgroup)Úopr r! Ú
|
<dictcomp>; s z*test_cythonized_aggers.<locals>.<dictcomp>F)Zdropna)r r )ÚnpÚnanÚrandomÚrandnr ÚlocZdropÚgroupbyÚindexÚnameÚtmÚassert_frame_equalÚ
|
setdefaultÚTÚstackÚnamesÚassert_series_equal)
|
r ÚdataÚdfÚgroupedÚexpÚresultZexpdZcat1Zcat2r' r )r( r r! Útest_cythonized_aggers s.
|
|
ý
|
r> c C sd t tj ddd¡tj ddd¡ d¡d} | d¡d ¡ }| d¡d tj¡}t ||¡ d S ) Nr é é2 r Úbool©ÚaÚbrC rD )
|
r r* r, ÚrandintÚastyper/ r Úaggr2 r8 )Úframer= Úexpectedr r r! Útest_cython_agg_booleanO s þÿrJ c C s¾ t tj ddd¡ddgd d} d}tjt|d | d
|
¡d jd d W 5 Q R X t tj ddd¡ddgd d} | d g | d
|
¡jd d }t g | d
|
¡
|
¡ g d}t ||¡ d S )Nr r? r@ ÚfooÚbaré rB úKCannot use numeric_only=True with SeriesGroupBy.mean and non-numeric dtypes©ÚmatchrC rD T©Ú numeric_only)r0 Úcolumns) r r* r, rE ÚpytestÚraisesÚ TypeErrorr/ r Z sort_valuesZdrop_duplicatesr2 r3 )rH Úmsgr= rI r r r! Ú test_cython_agg_nothing_to_agg\ s " " ÿrX c C sf t tj ddd¡ddgd tjdddd d
|
} d }tjt|d | d ¡j
|
j dd W 5 Q R X d S )Nr r? r@ rK rL rM Únowr5 )ÚperiodsÚfreq)rC rD ÚdatesrN rO rD TrQ ) r r* r, rE ÚpdZ
|
date_rangerT rU rV r/ r\ r )rH rW r r r! Ú)test_cython_agg_nothing_to_agg_with_datesl s
|
ýÿr^ c C sf t dddgdddgd} | jddd ¡ | jddd ¡ | jddd ¡ | jddd ¡ d S )
|
Nr r é é r? )r Úyr rS )Úlevelr$ )r r/ r )r: r r r! Ú test_cython_agg_frame_columnsy s
|
rc c
|
C s t ddddddddgddddddddgtj d¡tj d¡d} | d¡d d
|
d ¡}td d d d ddd d gtddgddd d}t ||¡ d S )NrK rL ÚoneÚtwoÚthreeé )r r r ÚDr r c S s | ¡ ¡ S r )Z value_countsÚto_dictr r r r! r" r# z-test_cython_agg_return_dict.<locals>.<lambda>r )re rd rf r ©r1 )r0 r1 )
|
r r* r, r- r/ rG r r r2 r8 )r: ÚtsrI r r r! Útest_cython_agg_return_dict s
|
|
üÿ ýrl c C sX t ddd} tdddddgd | d
|
}| d d ¡}| ¡ }| tj¡}t ||¡ d S ) Nz1/1/2000r@ )rZ r r r rh ÚEr ©r0 c S s | j S r )Úmonthr r r r! r" r# z&test_cython_fail_agg.<locals>.<lambda>)r
|
r r/ r rG r* r2 r8 )Zdrrk r; ZsummedrI r r r! Útest_cython_fail_agg s rp z
|
op, targopÚfirstc C s
|
| j d S ©Nr ©Zilocr r r r! r" « r# r" Úlastc C s
|
| j d S )Néÿÿÿÿrs r r r r! r" ¬ r# c C s^ t tj d¡}tjjdddd t¡}| |¡j| d dd}| |¡ |¡}t
|
||¡ d S )Niè r r@ )ÚsizeT©ZaltrR ) r r* r, r- rE rF Úfloatr/ Ú_cython_agg_generalrG r2 r3 )r( Útargopr: Úlabelsr= rI r r r! Útest__cython_agg_general¡ s
|
r| c C s t | dkrt | ¡S tjS rr )Úlenr* r r+ r r r r! r" ¼ r# c C s t j| ddS )Nr )Zddof)r* r r r r r! r" ½ r# c s t dddg}tddd}|jt |d |¡|d}|j| d dd }|jt |d |¡|d}| fd
|
d ¡}t ||¡ d S ) Né r é r é7 r? ©ÚobservedTrw c s | S r r r ©rz r r! r" Ì r# z/test_cython_agg_empty_buckets.<locals>.<lambda>) r Úranger/ r] Úcutry rG r2 r3 )r( rz r r: ÚgrpsÚgr= rI r r r! Ú test_cython_agg_empty_buckets¸ s r c C s t dddgdgd}tjdddtjd }|jt |d |¡| d
|
jd d d d }tjdddd}t dddddgitj |dd dd}| r||j
|
dk }t ||¡ |jt |d |¡| d
|
jdd d d }t dddddgitj |dd dd}| rü||j
|
dk }t ||¡ d S )Nr~ r r rC )rS r rM r? ©Údtyper r Trw é )r[ é$ )r1 Zorderedrn r r i´ ) r r* ZarangeÚint_r/ r] r
ry Zinterval_rangeZCategoricalIndexrC r2 r3 )r r: r r= Z intervalsrI r r r! Ú$test_cython_agg_empty_buckets_nanopsÐ s6 ÿþ ÿþr r( r9 z2016-10-14 21:00:44.557z17088 days 21:00:44.557c C sZ t ddg|tgd}tddgdd}t d|tgi|d}| d¡ | ¡}t ||¡ d S )Nr r rB rC rj rD rn )r r r r/ Ú aggregater2 r3 )r( r9 r: r0 rI r= r r r! Ú"test_cython_with_timestamp_and_natñ s
|
r rG ZohlcZcumprodZcumsumÚshiftÚanyÚallZquantileZrankZcumminZcummaxc C sr t dddddgdddddgd}d|jjd j_| d
|
g¡ d | i¡}| ¡ d
|
g¡ d | i¡}t ||¡ d S ) Ngffffff@g@gÍÌÌÌÌÌ@gffffff@g @Zsetosa)Ú sepal_lengthÚspeciesFr r r )
|
r Z_mgrZarraysÚflagsZ writeabler/ rG Úcopyr2 Z assert_equal)rG r: r= rI r r r! Ú test_read_only_buffer_source_agg s þÿr c C s t ddgd tjddddddd d
|
d tjg
|
d d d}t| d¡d | }|j|d d¡d}t| d¡d | }| dkr| d¡}t ||¡ d S )Nr r r? r r r_ r` r é rg é ÚInt64r )r r Úfloat64)r r )
|
r r] ÚarrayÚNAr r/ ZassignrF r2 r8 )r r: r= Zdf2rI r r r! Ú test_cython_agg_nullable_int* s
|
"þÿ
|
r Úwith_naTFzop_name, action)r Ú large_int)r Ú always_float)r r¢ )r r¢ )r r¡ )r Úpreserve)r r£ )rq r£ )rt r£ r r r_ r` r r ZInt8g¹?gÉ?g333333Ó?gÙ?ZFloat32ZFloat64Úbooleanc C s$ |rt j| d< tddddg| d}| d¡}|dkr@t ¡ }nh|dkrvt| jrZ| j}q¨t| jrl| j}q¨t ¡ }n2|dkrt| jr| j}q¨t ¡ }n|d kr¨| j}t || }|d
|
j|ksÆt
|
| |¡}|d
|
j|ksât
|
t |d
|
| }|j|kst
|
|d
|
|¡}|j|ks t
|
d S ) Nr_ rC rD )ÚkeyÚcolr¥ Z
|
always_intr¡ r¢ r£ r¦ ) r] r r r/ Z
|
Int64Dtyper r r Z Float64Dtyper ÚAssertionErrorr )r9 r Úactionr r: r; Zexpected_dtyper= r r r! Ú test_cython_agg_EA_known_dtypesM s4
|
|
|
|
|
|
|
|
|
r© )-Ú__doc__Únumpyr* rT Zpandas.core.dtypes.commonr r Zpandasr] r r r r r r r
|
Zpandas._testingZ_testingr2 ÚmarkZ parametrizeÚparamÚfilterwarningsr> rJ rX r^ rc rl rp r r r r r r r r| r r r r r r r© r r r r! Ú<module> sú $ üñþ
|
!
|
|
|
|
÷þ
|
|
|
ûþ
|
|
! ÿ ìþ
|
öþ
|
ôþûþ
|
