U
|
ý°d Ê ã @ s\ d Z ddlZddlZddlmZ ddlZddlZddlZddlm Z ddl
|
m Z ddl Z ddl mZmZmZmZmZmZ ddlmZ ddlmZ dd Zd
|
d Zd d Zdd Z dd Z dd Z dd Z dd Z dd Z!ej" #ddd d d dd gg¡d d Z$d d! Z%d"d# Z&ej" #d$d%d&d'd(gd)i fd*d+gd, e'e(e(e(d-fd.d/gd, e'e(e(e(d-fd%d&d'd(gd0d1d)ifd*d+gd, d0e(e(e(d-fd.d/gd, d0d2d2d2d-fg¡d3d4 Z)ej" #d5d%d6d7gd8d9gd:d;ggd)d)d<fd*d=gd6 gd, d2fd.d6gd6 gd, d2d2d<fg¡d>d? Z*d@dA Z+dBdC Z,dDdE Z-dFdG Z.dHdI Z/dJdK Z0dLdM Z1dNdO Z2dPdQ Z3dRdS Z4ej" #dTdUd dVd dWd dXd dYd dZd g¡d[d\ Z5ej" #d]d^ged_gd^d`fd^daged_gd6ggdgdggd^dagdbfg¡ej" #dcddded)dfd2g¡ej" #dgddded)dfd2g¡ej" #dhdidjdkg¡dldm Z6ej" #d]d^ged_gd^d`fd^daged_gd6ggdgdggd^dagdbfg¡ej" #dndod_dpg¡ej" #dqe7e'e(g¡ej" #dhdidjdkg¡drds Z8dtdu Z9dvdw Z:ej" #dqej;ej<g¡ej" #dxdydzd{d|d}d~g¡dd Z=dd Z>ej" #de ?e@d¡e Add,¡e jBd
d,dde Cdddg¡g¡dd ZDdd ZEdd ZFG dd dZGG dd dZHej" #dddejIfdd_d,gdd6gddgfddd fddd fdd_d,gd_d_gddgfe Jdd|¡e JdejK¡e Jddd ¡d_d,gddgd_d_gfg¡dd ZLd¡d¢ ZMd£d¤ ZNej" #d¥d¦d{gig d{gd§g¡d¨d© ZOdªd« ZPd¬d ZQd®d¯ ZRd°d± ZSej" #d²d³d d´d dµd g¡d¶d· ZTd¸d¹ ZUej" #d²dºd eUg¡d»d¼ ZVd½d¾ ZWej"jXd¿dÀdÁd ZYG dÃdÄ dÄZZdÅdÆ Z[ej" #dÇd{d{dÈd_d&gd^d¦gdÈfdÉd{idÉd^d¦gifdÊd{idÊd_d&gifg¡dËdÌ Z\ej" #dÇd{d|gd{dÈdÍdÎgfd{d{d|gdÈdÏdÐgfdÉd{d|gidÑdÒgfg¡dÓdÔ Z]dÕdÖ Z^d×dØ Z_dÙdÚ Z`dÛdÜ ZadÝdÞ Zbdßdà Zcej" #dádâdãdägdågfdædãdçgdègfg¡dédê Zdej" #dëd}dìdígfd%dîdïgfg¡dðdñ Zeej" #d²d{d|d.g¡dòdó Zfej" #d²d{gd}d|gdad%idôd~d§g¡dõdö Zgej" #d÷d;døgd;dùgd8dúggd8d;gdúdødùggdûfd;døgd;dùgdüdúggd;dügdødùgdúgdûfd^d_gd^d_gdad6gdad,ggd^dagd_d6d,ggdûfejhd^d_gd^d6gdad,gdad,ggd^dagd_d6gd,gdûej"jXdýg¡dþdÿ Zid d Zjdd ZkdS ( zN
|
test .agg behavior / note that .apply is tested generally in test_groupby.py
|
é N)Úpartial)ÚSpecificationError)Úis_integer_dtype)Ú DataFrameÚIndexÚ
|
MultiIndexÚSeriesÚconcatÚ to_datetime)ÚGroupingc C sB t ddddgddddgd} | d ¡d
|
}d d }| |¡ d S ) NÚaÚbÚcé é é é )ÚkeyÚvaluer r c S s t | dkst| ¡ S ©Nr )ÚlenÚAssertionErrorÚsum©Úx© r údd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/groupby/aggregate/test_aggregate.pyÚ
|
dummy_func" s z3test_groupby_agg_no_extra_calls.<locals>.dummy_func)r ÚgroupbyÚagg)ÚdfÚgbr r r r Ú test_groupby_agg_no_extra_calls s r" c C s: | dd dd g¡}| tj¡}| ¡ }t ||¡ d S )Nc S s | j S ©N©Úyearr r r r Ú<lambda>* ó z&test_agg_regression1.<locals>.<lambda>c S s | j S r# ©Úmonthr r r r r& * r' ©r r ÚnpÚmeanÚtmÚassert_frame_equal)ÚtsframeÚgroupedÚresultÚexpectedr r r Útest_agg_regression1) s r3 c C sf | d¡d }d}tjt|d | dd ¡ W 5 Q R X tjt|d | dd ¡ W 5 Q R X d S )NÚAÚCz Must produce aggregated value©Úmatchc S s | ¡ S r# )Zdescriber r r r r& 5 r' z#test_agg_must_agg.<locals>.<lambda>c S s | j d d
S ©Nr ©Úindexr r r r r& 7 r' )r ÚpytestÚraisesÚ Exceptionr )r r0 Úmsgr r r Útest_agg_must_agg0 s r? c C sH dd }| j | j| jg¡ |¡}| ddg¡ ¡ d }t ||¡ d S )Nc S s | ¡ S r# ©r r r r r r& ; r' z(test_agg_ser_multi_key.<locals>.<lambda>r4 ÚBr5 )r5 r r4 rA Ú aggregater r- Úassert_series_equal)r ÚfÚresultsr2 r r r Útest_agg_ser_multi_key: s rF c C s t dddtjddddgdddtjddd d
|
gd tjd d ddddddgddgdd} t dddddddtjddddg d
|
ddddddtjd dddg d d dd tjd dd d dtjd g d!d"dd#tjd$d#dd dtjtjg d%}| ddg¡}|d&d'g ¡ }t || ¡ d S )(Né é r r r é7 éM é! é, é )Úv1Úv2)r é_ )r éc )r rP )r rQ )ÚbigÚdamp)ÚblueÚdry)ÚredrV )rV ÚwetÚby1Úby2©Únamesr9 é é éX rQ rV rT r rR é rW rU rP rS )rN rO rX rY rN rO ) r r+ Únanr Ú from_tuplesr r, r- r. )r2 r Úgr1 r r r Ú$test_groupby_aggregation_mixed_dtypeA sN þø
|
õû ôüÿrc c C s ddddgddt jdgddt jdgddt jdgg} t| t ddddg¡d}|jddd }|jdd
|
}td dddgdd ddgd }t ||¡ d S )NTF)r4 r )r4 r )rA r )rA r )ÚdataÚcolumnsr ©ÚlevelÚaxis)Z numeric_onlyç @r ©r r ) r+ r` r r ra r r r- r. )Úlstr r! r1 r2 r r r Ú+test_groupby_aggregation_multi_level_columnr s
|
üþ rl c C sê | j | tj dd}| jtjks$ttg tjtg tjdd}t |
|
¡ |¡ t | tj
|
¡|¡ tj | tj
|
¡|dd |j |d tj dd}t |jttg dtjdd}t |
|
¡ |¡ t | tj
|
¡|¡ t | tj
|
¡|¡ d S ) NF©Z
|
group_keys©Údtype©ro r: )Zcheck_index_typer4 )Únamero )re ro r: )r r+ r` ro Úfloat64r r r r- rC r r Úapplyr re Úfloatr. )Útsr/ r0 ÚexpZexp_dfr r r Útest_agg_apply_corner s ýrw c C s t ¡ }| dd ¡}|jjd j}t| jt||jjd< | t
|
j ¡}| ¡ }t ||¡ t| jt ||jjd< | t
|
j ¡}| ¡ }t ||¡ d S )Nc S s | j S r# r$ r r r r r& r' z1test_agg_grouping_is_list_tuple.<locals>.<lambda>r )r- ZmakeTimeDataFramer ÚgrouperZ groupingsZgrouping_vectorr r: Úlistr r+ r, r. Útuple)ru r r0 rx r1 r2 r r r Ú test_agg_grouping_is_list_tuple s r{ c C s2 | ddg¡}| tj¡}| ¡ }t ||¡ d S )Nr4 rA r* )Zmframer0 r1 r2 r r r Útest_agg_python_multiindex¯ s r| Ú groupbyfuncc C s | ¡ S r# )Úweekdayr r r r r& ¸ r' r& c C s | j S r# r( r r r r r& ¸ r' c C s¤ | |¡}|d d¡}|d ¡ }t ||¡ | d¡}| ¡ }t ||¡ | ddddd¡}t|d ¡ |d ¡ |d ¡ |d
|
¡ d}t ||¡ d S )
|
Nr4 ÚstdÚvarr, Úsem)r4 rA r5 ÚDrA r5 r ) r r r r- rC rB r r. r r, r )r/ r} r0 r1 r2 r r r Útest_aggregate_str_func· s
|
|
|
|
|
|
üÿr c
|
C st t dddddddgttjddddddgddd}| d¡ ¡ }t ddd gitddgdd
|
dd }t ||¡ d S ) Nr r ZFloat64rn ©r r r r g( µ¦yâ?r ©rq ©r: ro ) r r ÚpdÚNAr r r r- r. )Úany_numeric_ea_dtyper r1 r2 r r r Útest_std_masked_dtypeÔ s þÿ
|
ÿr c C sZ | j dd}|dkr t}d}nt}d| d}tj||d |j|dd W 5 Q R X d S )
|
Nr ©rg )ZidxmaxZidxminz3reduction operation '.*' not allowed for this dtypez
|
Operation z does not support axis=1r6 r ©rh )r Ú TypeErrorÚ
|
ValueErrorr; r< r )r Úreduction_funcr! Úerrorr> r r r Ú%test_agg_str_with_kwarg_axis_1_raisesã s r z(func, expected, dtype, result_dtype_dictr rG rH r] Úint64r gÙlßÌvø @r )ÚiÚjÚkr g @ÚInt64r rr c C s t ddddddggd t ddgd d
|
d gg¡d ||d ¡}|jddd | ¡}t |gd d d
|
d gd |¡}t ||¡ d S )Nr r r r rG é r r r r r ©re ))r r )r r rf )r r Ú from_productÚastyper r r- r. )Úfuncr2 ro Úresult_dtype_dictr r1 r r r Ú(test_multiindex_groupby_mixed_cols_axis1ï s þýÿr z&func, expected_data, result_dtype_dictr r é
|
r_ é é )r r gÍ;f ö?c C s t t d¡ dd¡tdddgddtdd dd gd
|
dd d dd i¡}|jd
|
dd | ¡}t |tdddgddtdd gd
|
dd |¡}t ||¡ d S )Nr_ r r r r Úyr
r r r r )r: re ro r r )rd r: re )
|
r r+ ÚarangeÚreshaper r r r r- r. )r Ú expected_datar r r1 r2 r r r Ú test_groupby_mixed_cols_axis1 s" üûýür¥ c C sà | d¡}dd }| |¡}| jdk ¡ }| jdk ¡ }t|j}tt |g| ¡t ddd}t
|
| d¡|¡ tt |g| ¡t ddd}t
|
| d¡|¡ dd }t | j¡ |¡}t|tsÌtt|d
|
ksÜtd S ) Nr4 c S s | j S r# ©Úsize©Zserr r r r& % r' z-test_aggregate_item_by_item.<locals>.<lambda>ÚfooÚbarZBCD©r: rq c S s | j S r# r¦ r¨ r r r Úaggfun_12 s z-test_aggregate_item_by_item.<locals>.aggfun_1r )r r r4 r r re r r+ Úarrayry r- rC ÚxsZassert_almost_equalr Ú
|
isinstancer ) r r0 Zaggfun_0r1 ZfoosumZbarsumÚKrv r¬ r r r Útest_aggregate_item_by_item" s
|
|
|
r± c C s | ddg¡}dd }tjtdd | |¡ W 5 Q R X |dd | D |¡}| jd d
| jd kf }| ddg¡ |¡}t ||¡ d S )
|
Nr4 rA c S s | j tkrtd| ¡ S )NúTest error message)ro Úobjectr r r¨ r r r r = s
|
z test_wrap_agg_out.<locals>.funcr² r6 c S s g | ]}|d kr|qS )r5 r )Ú.0r r r r Ú
|
<listcomp>D s z%test_wrap_agg_out.<locals>.<listcomp>r5 ) r r; r< r rB Úlocre r- r. )Z three_groupr0 r r1 Z exp_groupedr2 r r r Útest_wrap_agg_out: s r· c C sP dt jfdt jfdt jfg}| d¡d |¡}tdddg}t |j |¡ d S )Nr, ÚmaxÚminr4 r5 )
|
r+ r, r¸ r¹ r r r r- Úassert_index_equalre )r Úfuncsr1 Zexp_colsr r r Ú*test_agg_multiple_functions_maintain_orderJ s r¼ c s¬ t tj dd¡tjdddddddgd d
|
¡ dttj d d ttj d d gi¡} tjdd
|
dd}t
|
ddg¡}t fdddD ¡j }t |||d}t | |¡ d S )Néè r ú1/1/2012ÚS)ÚfreqÚperiodsr4 rA r5 ©r: re Ú3Tç §èH.ÿï?©Úqçâé² q¼?r )r4 Úquantilec s" g | ]} d ¡jj|djqS ©rà rÅ ©Úresampler4 rÈ Úvalues©r´ rÆ ©r r r rµ ` s z9test_agg_multiple_functions_same_name.<locals>.<listcomp>©rÄ rÇ ©re r: )r r+ ÚrandomÚrandnr Ú
|
date_rangerË r r rÈ r ra r ÚTr- r. )r1 Úexpected_indexÚexpected_columnsÚexpected_valuesr2 r rÎ r Ú%test_agg_multiple_functions_same_nameS s& ý
|
ÿÿ ÿrØ c sæ t tj dd¡tjdddddtddd gd
|
d d d ¡ ddt tj
|
ddt tj
|
ddgi¡} tjdd ddd}t j ddddddgd
|
d d gd}t fdddD ¡j}t d ¡j ¡ |g¡}t |||d }t | |¡ d S ) Nr½ r r¾ r¿ Zdti)rÀ rÁ rq r4 rA r5 Úalphar
r rà ÚohlcrÄ rÅ rÇ r )r4 rÚ Úopen)r4 rÚ Úhigh)r4 rÚ Úlow)r4 rÚ Úclose)r4 rÈ r4 rZ c s" g | ]} d ¡jj|djqS rÉ rÊ rÍ rÎ r r rµ s zKtest_agg_multiple_functions_same_name_with_ohlc_present.<locals>.<listcomp>rÏ rÐ )r r+ rÑ rÒ r rÓ r rË r r rÈ r ra r rÔ Zhstackr4 rÚ r- r. )r1 rÕ rÖ Znon_ohlc_expected_valuesr× r2 r rÎ r Ú7test_agg_multiple_functions_same_name_with_ohlc_presenth s: ý
|
ÿú÷ ÿ ÿrß c C s | j ddgd} ddg}ddg}| d¡d |¡}| d¡d |¡}t ||¡ | d¡ |¡}| d¡ |¡}t ||¡ d S ) NrA r5 r )r© r, r )r r r4 r )Zdropr r r- r. )r r» Zex_funcsr1 r2 r r r Ú-test_multiple_functions_tuples_and_non_tuples s rà c
|
C s@ | d¡}| tjtjd¡}| tjtjd¡}t||gddgdd}|jddddjddd }tjtjgtjtjgd}| |¡}t
|
||¡ | tjtjtjgd¡}| tjtjtjgd¡}t
|
||¡ d
|
d }d d }d} t j t | d& tjtjtjdd}| |¡ W 5 Q R X tjg||gd}| |¡ d S )Nr4 ©r5 r r, r r )Úkeysrh r r rf c S s
|
t | ¡S r# ©r+ r, r r r r Ú numpymean s z:test_more_flexible_frame_multi_function.<locals>.numpymeanc S s t j| ddS )Nr )Zddof)r+ r r r r r Únumpystd° s z9test_more_flexible_frame_multi_function.<locals>.numpystdú nested renamer is not supportedr6 ©r© rª )r r r+ r, r r Z swaplevelZ
|
sort_indexrB r- r. r; r< r )
|
r r0 ZexmeanZexstdr2 Údr1 rä rå r> r r r Ú'test_more_flexible_frame_multi_function s&
|
|
ré c C s² | d¡}dddddid}d}tjt|d | |¡ W 5 Q R X ddddd}tjt|d | |¡ W 5 Q R X ddddd}tjt|d | |¡ W 5 Q R X d S ) Nr4 r, r rç r rá ræ r6 )r r; r< r rB )r r0 rè r> r r r Ú test_multi_function_flexible_mix¾ s
|
rê c C s¦ t ddddgddddgd d ddgd} | d¡}tddgdd}|d d d
|
¡}td d g|dd }t ||¡ |d dd
|
¡}td d g|dd }t ||¡ d S )Nr r r r ©r r r r r
r c S s | dk ¡ S r ©Úallr r r r r& Ý r' z1test_groupby_agg_coercing_bools.<locals>.<lambda>FTr« r c S s | ¡ ¡ S r# )Zisnullrí r r r r r& á r' )r r r rB r r- rC )ÚdatZgpr: r1 r2 r r r Ú test_groupby_agg_coercing_boolsÖ s (
|
rï c C sn t dddddgdddddgd} | d¡dg ddi¡}t dddgiddgd jdd
|
d }t ||¡ d S ) Nr4 rA r r ©r4 rA r r r r9 r r )r r r Z rename_axisr- r. )rî r1 r2 r r r Ú"test_groupby_agg_dict_with_getitemæ s ""rñ Úopc C s | ¡ S r# r@ r r r r r& ó r' c C s | ¡ S r# )Úcumsumr r r r r& ô r' c C s
|
| d¡S ©Nr ©Ú transformr r r r r& õ r' c C s
|
| d¡S ©Nró rõ r r r r r& ö r' c C s
|
| d¡S rô ©r r r r r r& ÷ r' c C s
|
| d¡S r÷ rø r r r r r& ø r' c C sd t ddgddgd}| d¡d }| | d¡d j}t|sDt| | d¡j}t|s`td S )Nr FTr r r )r Ú set_indexr ro r r )rò r Úsr1 r r r Útest_bool_agg_dtypeð s rû zkeys, agg_indexr r r
r rZ Ú input_dtypeÚboolÚint32Úfloat32Ú result_dtypeÚmethodrs rB rö c
|
sª t dgdgdgd}|d |¡|d< t| | ¡dg |}| fdd}|dkrbt d d¡n|}t d|d jd gi|d
|
¡} |d krd g| j_t
|
|| ¡ d S ) Nr r Trë r c s | ¡jd S r ©r Úilocr ©r r r r& r' z2test_callable_result_dtype_frame.<locals>.<lambda>rö r r9 rs ) r r Úgetattrr r Ú
|
RangeIndexr re r[ r- r. )
|
râ Ú agg_indexrü r r r rò r1 rÕ r2 r r r Ú test_callable_result_dtype_frame s ÿ
|
r ÚinputTç ð?ro c
|
s t dgdg|gd}t| | ¡d |}| fdd}|dkrNt dd¡n|}t|d jd g|dd ¡} t || ¡ d S )
|
Nr r rë r c s | ¡jd S r r r rn r r r& 5 r' z3test_callable_result_dtype_series.<locals>.<lambda>rö r r« )
|
r r r r r r r r r- rC )
|
râ r r ro r r rò r1 rÕ r2 r rn r Ú!test_callable_result_dtype_series' s r c C sf t ddddgddddgd} | d¡ ddd d
|
d g¡}|jjd }tddd d
|
d g}t ||¡ d S ) Nr r r r rð r4 r r¸ r, rÚ r¹ )r r r re Úlevelsr r- rº )r Úresr1 r2 r r r Ú#test_order_aggregate_multiple_funcs; s
|
r c C sÔ t ddddddgddtjddtjgd | d
|
}| d ¡}| ¡ }t ddddgtjgd dgd dgd gt d gd dddgg¡tddddg| d d| d}t ||¡ |jd dd}| ¡ }|
|
¡ }t ||¡ d S )Nr r r r é rM r r r rn r r rÛ rÜ rÝ rÞ ©ro rq )re r: ro F©Zas_index) r r r r rÚ r r r r- r. Z reset_index)r r r! r1 r2 Zgb2Úresult2Z expected2r r r Útest_ohlc_ea_dtypesG s" $þ
|
&ü r ÚhowÚfirstÚlastr¹ r¸ r, Úmedianc C st t dddgd}| d¡ d|i¡}|j | ¡|_| d¡ d|i¡}|dkr`|j tj¡|_tj||dd d S )
|
Nl &tLW Hd~ r r ©r r¡ r¡ r )r, r T)Z check_exact) r r r r r r+ r r- r. )ro r r r2 r1 r r r Útest_uint64_type_handling] s r c C sT d} t ddddgddddgd}tjt| d | d ¡ d
|
d
|
g¡ W 5 Q R X d S ) NzFunction namesr r r r r rð r6 r4 r¹ )r r; r< r r r )r> r r r r Útest_func_duplicates_raisesk s r r: ÚabcZ2020r )rÁ rÀ )r r )r r )r r c C s` t dddgdddgd| d}| d¡ dtji¡}t ddgddgd d¡}t ||¡ d S )Nr r r )Úgroupr r9 r r )r r r r Znuniquerù r- r. )r: r r1 r2 r r r Ú$test_agg_index_has_complex_internalss s r c C sr t dddddgdddddgdddddgd} | d¡ ¡ }t ddgddgd tddgdd
|
d }t ||¡ d S ) Nr r ZoneZtwoZthreeÚsix)Úkey1Úkey2Úkey3r )r r! r
r9 )r r r¹ r r- r. ©r r1 r2 r r r Útest_agg_split_block s ýÿþr# c
|
C s² t tjdddddddgdd d
|
dgdddd gtjddddd d
|
dgd t¡} | d d d d g¡ ¡ }t t d¡gdgdgdgt d¡gdgd t d g¡td}t
|
||¡ d S )NÚ2000r )rÁ r r r rè r r r Úe)r4 rA r5 r ÚEÚFr r ) r r rÓ r r³ r r¹ Ú Timestampr+ r r- r. r" r r r Ú#test_agg_split_object_part_datetime s0
|
|
|
|
úÿ ÷
|
|
|
ú
|
ö r) c @ sV e Zd Zdd Zdd Zdd Zdd Zej d e
|
j d
|
d d d d
|
d gg¡dd Z dS )ÚTestNamedAggregationSeriesc C s t ddddg}| ddddg¡}|jddd}tdd gddgdd
|
d gt ddg¡d }t ||¡ |jddd }|d d
|
g }t ||¡ d S )Nr r r r r r r¹ r rH r r rÐ )r r ©r r r r r+ r r- r. )Úselfr Úgrr1 r2 r r r Útest_series_named_agg² s ÿ z0TestNamedAggregationSeries.test_series_named_aggc C s\ t ddg ddg¡}tjtdd | ¡ W 5 Q R X | g ¡}tg d}t ||¡ d S )Nr r r ú Must provider6 r ) r r r; r< r r r r- r. ©r, r- r1 r2 r r r Útest_no_args_raises¿ s
|
|
z.TestNamedAggregationSeries.test_no_args_raisesc C s\ t dddg dddg¡}|jddd}tddgddgdt ddg¡d}t ||¡ d S )Nr r r r r r r9 r+ )r, r- r0 r2 r r r Ú*test_series_named_agg_duplicates_no_raisesÉ s $zETestNamedAggregationSeries.test_series_named_agg_duplicates_no_raisesc C sd t dddg dddg¡}|jdd dd d}tddgddgdt ddg¡d }t ||¡ d S )
|
Nr r r r c S s dS r r r r r r r& Ò r' z9TestNamedAggregationSeries.test_mangled.<locals>.<lambda>c S s dS ©Nr r r r r r r& Ò r' r r9 r+ r0 r r r Ú test_mangledÐ s $z'TestNamedAggregationSeries.test_mangledÚinpÚanythingr¹ ©ÚcolumnÚaggfunc)r6 r¹ c C sZ t ddddddddg}dt|j }tjt|d | |j¡j|d W 5 Q R X d S ) Nr r r r rG z func is expected but received r6 ©r ) r ÚtypeÚ__name__r; r< r r rÌ r )r, r5 rú r> r r r Útest_named_agg_nametupleÖ s
|
z3TestNamedAggregationSeries.test_named_agg_nametupleN) r< Ú
|
__module__Ú __qualname__r. r1 r2 r4 r; ÚmarkÚ parametrizer ÚNamedAggr= r r r r r* ± s
|
ýþr* c @ sL e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dS )Ú TestNamedAggregationDataFramec C s
|
t ddddgddddgddd d
|
gd }| d ¡jd dd}t ddgdd
|
gdtddgd dddgd}t ||¡ tjtj dd}| d ¡jddt
|
fdtj fd dd|fd}t dd gddgddgddgdd
|
gdd gdtddgd dd d d ddd gd}t ||¡ d S )!Nr r r r r r rG r rH r\ ©r r4 rA r ©r4 r¸ ©rA r¸ )Úa_maxÚb_maxr
rG rH r éb rÅ ©rA r¹ r4 )Úb_minÚa_minÚa_meanrG rH Úa_98g à?g @g\Âõ(\ï?g×£p=
|
×@rK rL rM rN ) r r r r r- r. Ú functoolsr r+ Ú
|
percentiler¹ r, )r, r r1 r2 Zp98r r r Útest_agg_relabelç s> "ÿý
|
úúö z.TestNamedAggregationDataFrame.test_agg_relabelc C sn t ddddgddddgddd d
|
gd }| d ¡jf d di}t d ddgitddgd dd}t ||¡ d S )Nr r r r r r rG r rH r\ rD r zmy colrE r
r9 ©r r r r r- r. ©r, r r1 r2 r r r Ú test_agg_relabel_non_identifier s "ÿ z=TestNamedAggregationDataFrame.test_agg_relabel_non_identifierc C s t ddddgddddgd}| d¡jddd }t ddgddgd tddgdd
|
d }t ||¡ tjtj d d }tjtj dd }d|_
|
d|_
|
t dddddgdddddgd}| d¡jd|fd|fd}t ddgddgdtddgdd
|
d }t ||¡ d S ) Nr r r r r rð r4 rJ r r
r9 é2 rÅ éF Úquant50Úquant70r r rG )Úcol1Úcol2rY rZ )Z quantile_50Z quantile_70ç ø?ç @g333333û?g@) r r r r r- r. rO r r+ rP r< )r, r r0 r2 rW rX Útestr r r Útest_duplicate_no_raises s$ & "
|
ÿþz6TestNamedAggregationDataFrame.test_duplicate_no_raisesc C s t ddddgddddgdt ddgd d
|
gg¡d }|jdd jd ddd}t ddgddgddgdddgd }t ||¡ d S )Nr r r r r rð r4 rA r r r9 r rE )r4 r¹ )rA r, )ZaaZbbÚccr[ g @)r r r r r r- r. rS r r r Útest_agg_relabel_with_level- s þ ÿ ÿz9TestNamedAggregationDataFrame.test_agg_relabel_with_levelc C s t dddgdddgd}| d¡}d}tjt|d |jdd W 5 Q R X tjt|d | ¡ W 5 Q R X tjt|d |jd
|
d d W 5 Q R X d S ) Nr r r r rð r4 r/ r6 )r© rF )r r r r )r r r; r< r r )r, r r0 r7 r r r Ú test_agg_relabel_other_raises: s
|
z;TestNamedAggregationDataFrame.test_agg_relabel_other_raisesc C sP t ddgddgd}t d¡}tjt|d | d¡jdd W 5 Q R X d S )
|
Nr r r rð z Column(s) ['C'] do not existr6 r4 )r5 r ©r )r ÚreÚescaper; r< ÚKeyErrorr r )r, r r7 r r r Útest_missing_raisesG s
|
z1TestNamedAggregationDataFrame.test_missing_raisesc C s` t ddgddgd}| d¡jt dd¡tjddd d
|
}| d¡jd d d
|
}t ||¡ d S ) Nr r r rð r4 rA r Úcountr7 ©r r )rA r )rA rg ©r r r r rB r- r. rS r r r Útest_agg_namedtupleM s
|
|
ÿz1TestNamedAggregationDataFrame.test_agg_namedtuplec C sv t ddgddgddgd}| d¡jdd d
|
fd d d
|
fd }t ddgddgd tddgddd}t ||¡ d S )Nr r r r r )r4 rA r5 r4 rA c S s dS r r r r r r r& W r' z<TestNamedAggregationDataFrame.test_mangled.<locals>.<lambda>r5 c S s dS r3 r r r r r r& W r' rh r
r9 rR rS r r r r4 U s $&z*TestNamedAggregationDataFrame.test_mangledN) r< r> r? rQ rT r^ r` ra rf rj r4 r r r r rC æ s $ rC zCagg_col1, agg_col2, agg_col3, agg_result1, agg_result2, agg_result3©©r¡ r4 r¸ rl )©r¡ rA r, g @ç @c C s t | S r# )r¸ r r r r r& h r' c C s dS r3 r r r r r r& i r' rm c C s dS r3 r r r r r r& r r' c
|
C s¶ t ddddgddddgddd d
|
gd }t d d dg¡|_tddgd d}| d ¡jdd}t dddgi|d} t || ¡ | d ¡j| ||d}t |||d|d} t || ¡ d S )Nr r r r r r rG r rH r\ rD ©r r rl rm r
rk )rG rG r9 )Zcol_1Zcol_2Zcol_3) r r ra re r r r r- r. )
|
Zagg_col1Zagg_col2Zagg_col3Z agg_result1Z agg_result2Z agg_result3r Úidxr1 r2 r r r Ú"test_agg_relabel_multiindex_column\ s$ !"ÿ
|
ÿ
|
ÿrq c C sj t ddddgddddgddd d
|
gd } t d d dg¡| _tjtdd | d ¡jdd W 5 Q R X d S )Nr r r r r r rG r rH r\ rD ro rl rm z do not existr6 ))ÚYr r¸ r: ) r r ra re r; r< re r r rÎ r r r Ú,test_agg_relabel_multiindex_raises_not_exist s "ÿrs c C s t ddddgddddgddd d
|
gd } t d d dg¡| _| d ¡jddd}tddgd d}t ddgddgd|d}t ||¡ d S )Nr r r r r r rG r rH r\ rD ro rl rm )rl r¹ r r
r9 ) r r ra re r r r r- r. )r r1 rp r2 r r r Ú&test_agg_relabel_multiindex_duplicates s "ÿ
|
ÿrt Úkwargsr rh c C sn t dddgdddgdddgd}| d¡ | ¡}t ddgtddgdddt d d
|
gg¡d }t ||¡ d S ) Nr r r r rë r r r r r¹ r )r r r r r ra r- r. )ru r r1 r2 r r r Ú test_groupby_aggregate_empty_key¬ s "ýrv c C sb t dddgdddgdddgd} | d¡ dg i¡}t tdgg gg g gdd }t ||¡ d S )
|
Nr r r r rë r r )r Úcodesr )r r r r r- r. r" r r r Ú-test_groupby_aggregate_empty_key_empty_return¹ s " rx c C sb t dddgd} | jddgddjdtfd}t dgtg g gg g gddgd d
|
}t ||¡ d S ) Nr r r r Frm )rè rè rZ rÐ )r r r ry r r- r. r" r r r Ú2test_groupby_aggregate_empty_with_multiindex_frameÁ s ÿry c C st t ddddddgdddddd gd
|
} | jd d d }|jtjdddd}t dddgddd gd}t ||¡ d S )Nr r¡ Úzr
|
çé?ri ç @çÍÌÌÌÌÌ @ç è?)r Úvalr Fr r r¹ r7 ©Úmin_val)r r ri ©r r0 r1 r2 r r r Ú9test_grouby_agg_loses_results_with_as_index_false_relabelË s ÿr c C s t ddddddgddddddgdddd d
|
d gd } | jd dgdd}|jtjdddd}t dddgdddgdd dgd}t ||¡ d S )Nr r¡ r r r r
|
r{ ri r| r} r~ )r r r r r Fr r r¹ r7 r )r r r ri r r r r ÚDtest_grouby_agg_loses_results_with_as_index_false_relabel_multiindexÙ s ýÿ ÿr r c C s | ¡ S r# )r, ©rú r r r r& ï r' c C s
|
t | ¡S r# rã r
r r r r& ï r' c C s
|
t | ¡S r# )r+ Znanmeanr
r r r r& ï r' c C s dddgdddgg}t |tjdddgdddggddgd d
|
}| t d dg¡¡ | ¡}d d dddddd dd}t |t d dg¡|jd}t ||¡ d S )Nr r r rG rH r ZSiskoZJanewayrZ r r r
|
g @rj r\ g @ri ))r r )r r )r r r )
|
r r Z from_arraysr r+ r r re r- r. )r rd r r1 Ú expected_dictr2 r r r Útest_multiindex_custom_funcî s ÿþýr c C s t j| ddS ©NgÍÌÌÌÌÌì?rÅ ©r+ rP r
r r r Úmyfunc s r c C s t j| ddS r r r
r r r r& r' c C sz t ddddgddddgddd d
|
gd }| d ¡jd d| fd}t ddgddggddgtddgd dd}t ||¡ d S )NÚcatÚdogç333333"@ç @ç #@ç A@ç @rn çÍÌÌÌÌÌ#@ç Àh@©ÚkindÚheightÚweightr )r r, r )Ú mean_heightÚperc90g"@g0»' 5"@g 4@g5^ºI @r r r
rÐ rR )r Zanimalsr1 r2 r r r Útest_lambda_named_agg s
|
|
|
ýÿ
|
ÿýr c C s¬ t t dgd ¡ dd¡tdtdd} dddg| d < | d ¡ d
|
d ¡}dgdgdggddgddgddggg}t |tddgd d d tdddgd dd}t ||¡ d S )Nr r] r ZXYZr ©rd re r: zgroup 1r Úgroupingc S s | ¡ S r# )Útolistr r r r r& % r' z,test_aggregate_mixed_types.<locals>.<lambda>r³ r ÚXrr ÚZrn r )
|
r r+ r r£ ry r rB r r- r. )r r1 r¤ r2 r r r Útest_aggregate_mixed_types s ÿ&ýr z Not implemented;see GH 31256©Úreasonc C sj dd } t dt dddg¡i}| dddg¡ | ¡}t dtjdtjgddiddgd }t ||¡ d S )
|
Nc S s t | dkrdS tjS d S )Nr r )rí r r r r r r r9 8 s z5test_aggregate_udf_na_extension_type.<locals>.aggfuncr4 r r r r rn r9 )r r r r r r r- r. )r9 r r1 r2 r r r Ú$test_aggregate_udf_na_extension_type/ s
|
$r£ c @ sB e Zd Zdd Zdd Zejjdddd Zd d
|
Z d d Z
|
d S )ÚTestLambdaManglingc C st t ddddgddddgd}| d¡ dd d
|
d d
|
gi¡}t ddgddgd tddgdd d}t ||¡ d S )Nr r r r r rð r4 rA c S s dS r r r r r r r& G r' z/TestLambdaMangling.test_basic.<locals>.<lambda>c S s dS r3 r r r r r r& G r' ))rA ú
|
<lambda_0>)rA ú
|
<lambda_1>r
r9 rR rS r r r Ú
|
test_basicE s þz TestLambdaMangling.test_basicc C sl t ddddg ddddg¡}| dd dd g¡}ddgddgd }t|t ddg¡d
|
}t ||¡ d S ) Nr r r r r c S s dS r r r r r r r& Q r' z?TestLambdaMangling.test_mangle_series_groupby.<locals>.<lambda>c S s dS r3 r r r r r r& Q r' ©r¥ r¦ r9 r+ )r, r- r1 Úexp_datar2 r r r Útest_mangle_series_groupbyO s
|
z-TestLambdaMangling.test_mangle_series_groupbyz GH-26611. kwargs for multi-agg.r¡ c C s ddd}ddd}t ddg ddg¡ ||gd¡}tdgdgd }t ||¡ t ddg ddg¡j||gdd
|
d }td gd gd }t ||¡ d S )Nr c S s | ¡ | | S r# r@ ©r r¡ r r r r r& X r' z5TestLambdaMangling.test_with_kwargs.<locals>.<lambda>r c S s | ¡ || S r# r@ r« r r r r& Y r' r r r r¨ r )r é é )r )r )r r r r r- r. )r, Úf1Úf2r1 r2 r r r Útest_with_kwargsV s
|
|
" &z#TestLambdaMangling.test_with_kwargsc C sÖ t ddddgddddgddd d
|
gd }d d dg}t ddgddgd d
|
gdtddgdd|d}|jddjtjddd dtjdddtjdddd}t ||¡ |jddjdd d fd d d}t ||¡ d S ) Nr r r r r r r rn r r r Úheight_sqr_minÚ
|
height_maxÚ
|
weight_maxç¤p=
|
׳T@ç B@)r± r² r³ r r
r ©Zbyr c S s t | d ¡S r8 ©r+ r¹ r r r r r& z r' z=TestLambdaMangling.test_agg_with_one_lambda.<locals>.<lambda>r7 r¸ r c S s t | d ¡S r8 r· r r r r r& r' ©r r¸ ©r r¸ )r r r r r rB r- r. ©r, r re r2 Zresult1r r r r Útest_agg_with_one_lambdab s<
|
|
|
ýÿ
|
ýù ÿ û
|
ýz+TestLambdaMangling.test_agg_with_one_lambdac C s t ddddgddddgddd d
|
gd }d d dddg}t ddgddgd d
|
gddgddgdtddgdd|d}|jddjddd fdd dd d fd d d fd}t ||¡ |jddjtjdd d d!tjdd"d!tjd d"d!tjdd#d d!tjd d$d d!d}t ||¡ d S )%Nr r r r r r r rn r r r r± r² r³ Ú height_max_2Ú
|
weight_minr´ rµ )r± r² r³ r¼ r½ r r
r r¶ r c S s t | d ¡S r8 r· r r r r r& ¨ r' z=TestLambdaMangling.test_agg_multiple_lambda.<locals>.<lambda>r¸ r¹ c S s
|
t | ¡S r# ©r+ r¸ r r r r r& « r' r c S s
|
t | ¡S r# r· r r r r r& ¬ r' c S s t | d ¡S r8 r· r r r r r& ³ r' r7 r¸ c S s
|
t | ¡S r# r¾ r r r r r& · r' c S s
|
t | ¡S r# r· r r r r r& ¸ r' )r r r r r- r. r rB rº r r r Útest_agg_multiple_lambda sR
|
|
|
ýÿûû÷
|
|
|
û ÿ ù z+TestLambdaMangling.test_agg_multiple_lambdaN) r< r> r? r§ rª r; r@ Úxfailr° r» r¿ r r r r r¤ D s
|
|
'r¤ c C s^ t dddgdddgd} | d¡ ddd i¡}t ddgddgd d¡}t ||¡ d S )
|
Nr¿ ÚWr
|
ri rð r4 rA c S s | | jd ¡S )Néÿÿÿÿ)Úgetr: r r r r r& À r' z+test_groupby_get_by_index.<locals>.<lambda>)r r r rù r- r. )r r r2 r r r Útest_groupby_get_by_index½ s rÄ zgrp_col_dict, exp_data)ÚnrÚcat_ordrÆ rÅ c C s¶ t ddddddddgtd td
|
d }| d d d ¡}|d j ¡ |d< | d¡ | ¡}tjddgddgddd d}t ||d}d|kr¦|d j }|d |¡|d< t
|
||¡ d S )Nr r r r rG r rH r\ ÚaabbccddÚaaaabbbb©rÅ rÆ r Úcategory©r rÆ rÆ r r r F©Ú
|
categoriesZorderedrq ro )rd r: ) r ry r r Ú
|
as_orderedr r r ÚCategoricalIndexro r- r. )Ú grp_col_dictr© Úinput_dfÚ result_dfÚ cat_indexÚ expected_dfro r r r Ú test_groupby_single_agg_cat_colsÅ s* ýÿ ÿ
|
rÕ )r r r )rG r\ r )r r r )rG r rè r )r rè c C s t ddddddddgtd td
|
d }| d d d ¡}|d j ¡ |d< | d¡ | ¡}tjddgddgddd d}g }| ¡ D ]:\}}t
|
|tr®|D ]}| ||g¡ qq| ||g¡ qt t|¡} t || |d}
|
|
|
jD ].} t
|
| tràd| krà|
|
| |d j¡|
|
| < qàt ||
|
¡ d S )Nr r r r rG r rH r\ rÇ rÈ rÉ rÊ rË rÆ r r r FrÌ r )r ry r r rÎ r r r rÏ Úitemsr¯ Úappendr ra rz re ro r- r. ) rÐ r© rÑ rÒ rÓ Zmulti_index_listr Úvr Z multi_indexrÔ Úcolr r r Ú#test_groupby_combined_aggs_cat_colsì s8 ýÿ ÿ
|
|
rÚ c C s\ t ddddgddddgd} | d¡}| dg¡}|j d¡|_| d¡}t ||¡ d S )Nr r r r ró r )r r r re Z droplevelr- r. )r rb r1 r2 r r r Útest_nonagg_agg s
|
|
rÛ c C sh t ddgt ddddddd¡t d dddddd¡gd
|
} | d ¡j ¡ }| d ¡d }t ||¡ d S ) Nr rr iÕ r r r é i`= i½ rð r4 rA )r Údatetimer rA r¸ rù r- rC r" r r r Ú test_aggregate_datetime_objects* s þþÿ rÞ c C s t dddgdddgdddgd} | j d¡| _| dd g¡}|j d
|
d ¡}tjd d gdd}tddg|dd}t
|
||¡ d S )Nr r¡ r r r )Úc0Úc1ÚpÚOrß rà c S s t | dkS r rì r r r r r& A r' z1test_groupby_index_object_dtype.<locals>.<lambda>)r r r )rß rà rZ FTrá r« ) r r: r r rá r r ra r r- rC )r r0 r rÕ r2 r r r Ú test_groupby_index_object_dtype< s " ÿrã c C sT dd } t dgt d¡gd}| dd ¡ | ¡}t dggdgd}t ||¡ d S ) Nc S s | ¡ ¡ rd S t | ¡S r# )Zisnarí r+ r r¨ r r r r N s z)test_timeseries_groupby_agg.<locals>.funcr
|
z2018-01-16 00:00:00+00:00r9 c S s dS r3 r r r r r r& T r' z-test_timeseries_groupby_agg.<locals>.<lambda>r )r r r( r r r- r. )r r r r2 r r r Útest_timeseries_groupby_aggK s
|
rä c C s | dkr d S t ddgdtjgg}| d¡ | ¡}| d¡ d| i¡}| dkrdtj||d dd n t ||¡ t |j|j¡ d S )N)ZcorrwithZnthr r )r§ ZngroupF)Z check_names) r r+ r` r r r- rC r. Zdtypes)r ÚobjZresult_reduced_seriesZresult_reduced_framer r r Ú test_groupby_aggregate_directoryZ s ÿ ÿræ c C sN t dddgdd} t dgdt dg¡d}| dddg¡ ¡ }t ||¡ d S ) Nz1 dayz3 daysÚNaTztimedelta64[ns]rn z2 daysr rp )r r+ r r r, r- rC )rd r2 r1 r r r Ú test_group_mean_timedelta_natp s rè zinput_data, expected_outputz2021-01-01T00:00rç z2021-01-01T02:00z2021-01-01T01:00z2021-01-01T00:00-0100z2021-01-01T02:00-0100z2021-01-01T01:00-0100c C sH t t| }t t|t dg¡d}| dddg¡ ¡ }t ||¡ d S )Nr r9 )r
|
r r+ r r r, r- rC )Z
|
input_dataZexpected_outputrd r2 r1 r r r Ú test_group_mean_datetime64_natz s ré z func, outputy @ 2@y $@ 6@y D@ V@y I@ [@c C sN t t d¡ dd¡ ddg¡}| |jd ¡ | ¡}t |}t ||¡ d S )Nr r r r ù @)
|
r r+ r¢ r£ Údotr r: r r- rC )r Úoutputrd r1 r2 r r r Útest_groupby_complex s rí c C sX t t d¡ dd¡ ddg¡}d}tjt|d | |j d ¡
|
| ¡ W 5 Q R X d S )Nr r r r rê zNo matching signature foundr6 ) r r+ r¢ r£ rë r; r< r r r: r )r rd r> r r r Útest_groupby_complex_raises s rî Úprodc C sX t dddgdddgdddgd }|jd
|
dd }tjtd d | | ¡ W 5 Q R X d S )Nr r r r rG r rH r\ rë r r z"axis other than 0 is not supportedr6 )r r r; r< ÚNotImplementedErrorr )r r r! r r r Útest_multi_axis_1_raises¤ s "rñ ztest, constantr4 rA r5 rj r )Zmarksc C s< t | }| d¡ tj¡}t |}| d¡}t ||¡ d S r )r r r r Úmoderù r- r. )r] ZconstantZdf1r1 r2 r r r Útest_agg_of_mode_list¯ s
|
|
ró c C sÈ t dddgdddgd} | d¡}dd
|
d }dd d }d}tjt|d |j||gdddd W 5 Q R X |j||gddd}t ddgddgddggtdddgddt ddg¡d}t
|
||¡ d S ) Nr r r r r r r r¡ r c S s | ¡ | | S r# r@ ©r r r r r r Úfoo1Í s zAtest__dataframe_groupy_agg_list_like_func_with_args.<locals>.foo1c S s | ¡ | | S r# r@ ©r r r r r r Úfoo2Ð s zAtest__dataframe_groupy_agg_list_like_func_with_args.<locals>.foo2ú/foo1\(\) got an unexpected keyword argument 'b'r6 r rh rb r\ r] r r
)r rõ )r r÷ r )r r )r r ) r r r; r< r r r r ra r- r. )r r! rõ r÷ r> r1 r2 r r r Ú3test__dataframe_groupy_agg_list_like_func_with_argsÈ s
|
|
|
ýrù c C s² t dddg} | | ¡}ddd}ddd}d }tjt|d
|
|j||gddd d W 5 Q R X |j||gdd d }tddgddgddggtdddgddgd}t ||¡ d S )Nr r r r c S s | ¡ | | S r# r@ rô r r r rõ å s z>test__series_groupy_agg_list_like_func_with_args.<locals>.foo1c S s | ¡ | | S r# r@ rö r r r r÷ è s z>test__series_groupy_agg_list_like_func_with_args.<locals>.foo2rø r6 r rh rb r\ r] r rõ r÷ r )r r )r r )
|
r r r; r< r r r r r- r. )rú Zsgbrõ r÷ r> r1 r2 r r r Ú0test__series_groupy_agg_list_like_func_with_argsà s
|
|
|
ÿrú )lÚ__doc__rÝ rO r rc Únumpyr+ r; Z pandas.errorsr Zpandas.core.dtypes.commonr Zpandasr r r r r r r
|
Zpandas._testingZ_testingr- Zpandas.core.groupby.grouperr r" r3 r? rF rc rl rw r{ r| r@ rA r r r Úintrt r r¥ r± r· r¼ rØ rß rà ré rê rï rñ rû r rý r r r r Zuint64r r rÏ ry Zinterval_rangeZ period_rangera r r# r) r* rC r¹ rB r, rq rs rt rv rx ry r r r r r r rÀ r£ r¤ rÄ rÕ rÚ rÛ rÞ rã rä ræ rè ré rí rî rñ Úparamró rù rú r r r r Ú<module> s¸
|
1 ÿ
|
úþ
|
üþ
|
!$
|
úþ
|
(þþ ÿ ÿ(þþ
|
|
üþ
|
5vú
|
|
ú
|
úïþ
|
|
|
ÿ
|
|
|
yýþ
|
ýþ
|
)
|
þþûþ
|
ÿ
|
|
ÿ
|
**0ýüþ
|
|
