U
|
ý°dÕ ã @ sô d dl Zd dlZd dlmZ d dlZd dlmZmZm Z d dl
|
m Z dd Z dd Zdd Zd
|
d Zd d Zej dddg¡dd Zej dddg¡dd Zdd Zej dddg¡edddZej ddd d d d g¡d d! Zd"d# ZdS )$é N)ÚiNaT)Ú DataFrameÚIndexÚSeriesc C s t ddddgddddgddg d} | d ¡ ¡ }d
|
|ks@t| d ¡jd d }d
|
|ks^t| d ¡ ¡ }d
|
|ksxt| d ¡jd d }d
|
|kstd S ) Né é é é é é Zmama)ÚnnÚiiÚssr r F)Z numeric_only)r ÚgroupbyÚmaxÚAssertionErrorÚmin)ZaaÚresult© r úXd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/groupby/test_min_max.pyÚtest_max_min_non_numeric s & r c C sà t dddddgdddddgddd d
|
d gd }| ¡ | sNt|jjdksNt| d ¡}|dg ¡ }tdddgd d}t ddd
|
d gi|d}t
|
||¡ |dg ¡ }tdddgd d}t ddd d gi|d}t
|
||¡ d S )Nr r r
|
ZfooÚbarFÚaÚbÚcÚdÚe)ÚAÚBÚCr r ©Úname©Úindex) r Z_consolidate_inplaceÚlenZ_mgrÚblocksr r r r ÚtmÚassert_frame_equalr )Zusing_array_managerÚdfÚgbr ÚeiÚexpectedr r r Ú$test_max_min_object_multiple_columns s$ ýÿ
|
r, c C sª t jtdddgddjj} ttjdtjgdddg| d}|jdd d
|
d ¡ d }t jtddgd d ddjj}t
|
||¡ | d¡d ¡ }d|j _ t
|
||¡ d S ) Nz
|
2019-05-09z%Y-%m-%d)ÚformatÚ1r r )r r r r F)Zas_indexr r )ÚpdÚ to_datetimer ÚdtÚdater ÚnpÚnanr r r& Úassert_series_equalr# r! )Údatesr( r r+ r r r Útest_min_date_with_nans? s ÿ
|
ÿ
|
r7 c C s´ t dtg} tjddgtjd}| |¡}|jdd}t dditjd}tj||dd |j dd}t dtitjd}tj||dd |j dd}t dtj
|
i}tj||dd d S )Nr ©Údtyper ©Z min_countT©Z check_exactr
|
) r r r3 ÚarrayÚint64r r r& r5 r r4 )ÚserÚkeyr) r r+ r r r Ú test_max_inatQ s
|
r@ c C sn t dtdtd g} | ddddg¡}|jdd}t tjtjtd d}|j tj¡|_t j
|
||dd d S )Nr r r
|
r: )r r r
|
Tr; ) r r r r r3 r4 r# ÚastypeÚint_r& r5 )r> r) r r+ r r r Útest_max_inat_not_all_nae s rC Úfuncr r c C sj ddg}t jdddd}t||d}t| d¡d | }tddgdd }t||dd
|
}t ||¡ d S ) Nr r Ú2020ÚY©ÚperiodsÚfreq©r r r r r )r# r! ) r/ Ú period_ranger Úgetattrr r r r& r5 ©rD ÚgroupsrH r( r Úidxr+ r r r Ú$test_groupby_aggregate_period_columns s rP c C sh ddg}t jdddd}t||d}t| d¡| }tddgdd}td |i|d
|
}t ||¡ d S ) Nr r rE rF rG rJ r r r r" )r/ rK r rL r r r& r' rM r r r Ú#test_groupby_aggregate_period_frame s rQ c C sÜ t ddddgtdtjgd d t¡} | d¡ ¡ }t ddgddgtjtjgd d¡ t¡}t
|
||¡ t ddddgtdt dd t¡} | d¡ ¡ }t ddgddgd d
|
gd d¡ t¡}t
|
||¡ d S ) Nr r Zabcdr )r? Zcol1Zcol2r? r r r r ) r Úlistr3 r4 rA Úobjectr r Ú set_indexr& r' Úrange)r( r r+ r r r Ú#test_aggregate_numeric_object_dtype s8 ÿþÿýüÿ ÿþ ÿþÿrV )rD c C sx t dgddj ¡ }tdg|d}| d¡ d| i¡}tddgitdgddd }|d |j¡|d< t
|
||¡ d S )
|
Nr Úcategoryr8 iÍ ©r r r r r r" ) r ÚcatZ
|
as_orderedr r Zaggr rA r9 r& r' )rD Zdsr( r r+ r r r Ú%test_aggregate_categorical_lost_index s rZ r9 ÚInt64ZInt32ZFloat64ZFloat32Úbooleanc C s | dkrd}n| dkr d}nd}t ddg||d gd}|d | ¡|d < | d
|
¡}| ¡ }|jd d
d
|
¡}t ||¡ | ¡ }|jdd
d
|
¡}t ||¡ |jd d }t d t j
|
gi|j | d } t || ¡ |jd d }
|
t |
|
| ¡ t dddg|t j
|
|d gd} | d | ¡| d < | d
|
¡} | ¡ } t | |¡ | ¡ }t ||¡ | jdd }t || ¡ | jdd }t || ¡ d S )Nr[ l ².¿Dÿ²3 r\ r g @r r )ÚidÚtsr^ r] r
|
r: )r# r9 éd ) r rA r r ÚilocrT r& r' r r/ ÚNAr# )r9 r^ r( r) r r+ Zres_maxZ expected_maxZresult2Z expected2Zres_max2Zdf2Zgb2Zresult3Zres_max3Zresult4Zres_max4r r r Ú test_groupby_min_max_nullable» s<
|
|
rb c C s® t jdgd ddgd} t| t tjdtjd¡d}| d¡}| ¡ }t j ddg| j
|
dd}td t jdt j gd
|
di|d }t ||¡ | ¡ }d |jd < t ||¡ d S )Nr é
|
r )Ú
|
categoriesr8 rX r )r9 r! r ZUInt64r" é )r r )r/ Z Categoricalr r< r3 ZarangeZuint64r r ZCategoricalIndexr9 ra r& r' r r` )rY r( r) ÚresrO r+ r r r Ú(test_min_max_nullable_uint64_empty_groupë s
|
|
rg )Únumpyr3 ZpytestZpandas._libs.tslibsr Zpandasr/ r r r Zpandas._testingZ_testingr& r r, r7 r@ rC ÚmarkZ parametrizerP rQ rV ÚstrrZ rb rg r r r r Ú<module> s(
|
|
|
/
|
