U
|
ý°dU3 ã @ s¦ d dl mZ d dlmZ d dlZd dlZd dlmZm Z m
|
Z
|
m Z m Z m Z mZmZ d dlmZ G dd dZdd Zd d
|
Zd d Zd d Zdd Zdd Zdd ZdS )é )Úproduct)Úascii_lowercaseN)Ú DataFrameÚIndexÚ
|
MultiIndexÚPeriodÚSeriesÚ TimedeltaÚ TimestampÚ
|
date_rangec @ s e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z dd Z dd Zdd Zdd Zd d Zd d Zd!d" Zej d#d$d% ed&d'D d(d% ed&d'D d)d% ed&d'D d*d% ed&d'D d+d% ed&d'D g¡d,d- Zd.d/ Zd0d1 Zd2d3 Zd4S )5Ú TestCountingc C sh t dgdgdgdgdggdgd}| d¡}|j}tdddddg}t || ¡ ¡ t || ¡ ¡ d S ) NÚaÚbÚA©Úcolumnsr é é é ©r Úgroupbyr r ÚtmÚassert_series_equalÚcumcount©ÚselfÚdfÚgÚsgÚexpected© r úYd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/groupby/test_counting.pyÚ test_cumcount s "
|
zTestCounting.test_cumcountc C sN t jdd}ttdjdd}tdd}t || ¡ ¡ t || ¡ ¡ d S ©Nr )Úlevel©ÚdtypeÚint64)r r r Úobjectr r r ©r ÚgeÚseÚer r r! Útest_cumcount_empty s
|
|
z TestCounting.test_cumcount_emptyc C sz t dgdgdgdgdggdgdgd d}| d¡}|j}tddddd gdgd d
|
}t || ¡ ¡ t || ¡ ¡ d S © Nr r r r é ©r Úindexr r r ©r1 r r r r r! Útest_cumcount_dupe_index) s ÿ
|
z%TestCounting.test_cumcount_dupe_indexc C s t ddgddgddgddgddgg¡}tdgdgdgdgdggdg|d}| d¡}|j}tdddddg|d }t || ¡ ¡ t || ¡ ¡ d S )
|
Nr r r r r r r0 r r2 ) r Ú from_tuplesr r r r r r r ©r Úmir r r r r r r! Útest_cumcount_mi5 s ($
|
z TestCounting.test_cumcount_mic C s t dgdgdgdgdggdgdgd d}| dddddg¡}|j}tddddd gdgd d
|
}t || ¡ ¡ t || ¡ ¡ d S r. r r r r r! Ú test_cumcount_groupby_not_col@ s ÿ z*TestCounting.test_cumcount_groupby_not_colc C sV t dtdi}| d¡}|j}tdddddg}t || ¡ ¡ t || ¡ ¡ d S )Nr Úaaabar r ©r Úlistr r r r r Úngroupr r r r! Ú test_ngroupL s
|
zTestCounting.test_ngroupc C sT t dtdi}| d¡}|j}ttddd}t || ¡ ¡ t || ¡ ¡ d S )Nr Zabcder/ r' r% ) r r; r r r Úranger r r< r r r r! Útest_ngroup_distinctV s
|
z!TestCounting.test_ngroup_distinctc C sT t ddgd i}| d¡}|j}tdgd }t || ¡ ¡ t || ¡ ¡ d S )Nr r r/ )r r r r r r r< r r r r! Útest_ngroup_one_group` s
|
z"TestCounting.test_ngroup_one_groupc C sN t jdd}ttdjdd}tdd}t || ¡ ¡ t || ¡ ¡ d S r# )r r r r( r r r< r) r r r! Útest_ngroup_emptyj s
|
|
z TestCounting.test_ngroup_emptyc C s@ t dtdi}ttd}t | |¡ ¡ | |¡ ¡ ¡ d S )Nr r9 )r r; r r r r r< )r r Úsr r r! Ú test_ngroup_series_matches_framet s z-TestCounting.test_ngroup_series_matches_framec C sj t dtdidgd d}| d¡}|j}tdddddgdgd d}t || ¡ ¡ t || ¡ ¡ d S ©Nr r9 r r/ r2 r r: r r r r! Útest_ngroup_dupe_indexz s
|
z#TestCounting.test_ngroup_dupe_indexc C s t ddgddgddgddgddgg¡}tdtdi|d}| d¡}|j}tdddddg|d}t || ¡ ¡ t || ¡ ¡ d S )Nr r r r r9 r2 )
|
r r4 r r; r r r r r r< r5 r r r! Útest_ngroup_mi s (
|
zTestCounting.test_ngroup_mic C st t dtdidgd d}| dddddg¡}|j}tdddddgdgd d}t || ¡ ¡ t || ¡ ¡ d S rD r: r r r r! Útest_ngroup_groupby_not_col s z(TestCounting.test_ngroup_groupby_not_colc C s t dddddgdgd}| dg¡}tdddddg}tdddddg}t ||jd | ¡ t ||jdd¡ t ||jd d¡ d S )
|
Nr r r r r r T)Ú ascendingF)r r r r r Zngroupsr< )r r r rH Z
|
descendingr r r! Útest_ngroup_descending s z#TestCounting.test_ngroup_descendingc C s t ddgddgddgddgddggddgd}| ddg¡}| ¡ }| ¡ }tdd d
|
dd g}tdddd dg}t ||¡ t ||¡ d S ) Nr ÚxÚyr r ÚXr r r r r )r r r< r r r r )r r r Zg_ngroupZ
|
g_cumcountZexpected_ngroupZexpected_cumcountr r r! Ú test_ngroup_matches_cumcount£ s þ z)TestCounting.test_ngroup_matches_cumcountc s t tdddD ]xtdi}| dg¡}tt fddD }fddtD }t | ¡ t
|
|¡ t | ¡ t
|
|¡ qd S )Nr é )Úrepeatr c s g | ]} |¡qS r r2 )Ú.0Úval)Úorderr r! Ú
|
<listcomp>¹ s z:TestCounting.test_ngroup_cumcount_pair.<locals>.<listcomp>c s" g | ]\}} d |
|¡qS ©N)Úcount)rP ÚirQ )Úpr r! rS º s ) r r> r r ÚsortedÚsetÚ enumerater r r< r r )r r r ZngroupdZ
|
cumcountedr )rR rW r! Útest_ngroup_cumcount_pair² s z&TestCounting.test_ngroup_cumcount_pairc C sÄ t j d¡ tdt j tdd¡i}|jd|d}d|d< d|d< t|D ]>\}\}}||j|j df< t|j D ]\}}||j|df< qrqLt
|
t |d j | ¡ ¡ t
|
t |d j | ¡ ¡ d S ) Nr r Zabcdeféd )ÚsortéÿÿÿÿZgroup_idZ group_index)ÚnpÚrandomÚseedr Úchoicer; r rZ Úlocr1 r r r Úvaluesr< r ) r r] r r rV Ú_ÚgroupÚjÚindr r r! Ú"test_ngroup_respects_groupby_order¿ s z/TestCounting.test_ngroup_respects_groupby_orderÚ datetimelikec C s g | ]}t d |ddqS )ú2016-05-Ú02dz 20:09:25+00:00©r
|
©rP rV r r r! rS Ñ s zTestCounting.<listcomp>r rN c C s g | ]}t d |ddqS )rk rl ú 20:09:25rm rn r r r! rS Ò s c C s" g | ]}t d |ddddqS )rk rl ro ÚUTC)Útzrm rn r r r! rS Ó s c C s g | ]}t |d dqS )Úh)Úunit)r ©rP rJ r r r! rS Ô s c C s g | ]}t d d|dqS )Z2Wiá )ÚfreqÚyearÚmonth)r rt r r r! rS Õ s c C sR t dddg|d}| d¡ ¡ }t dddgiddgd}d|j_t ||¡ d S ) Nr r )rJ rK rJ rK r r r2 )r r rU r1 Únamer Úassert_frame_equal)r rj r Úresr r r r! Ú test_count_with_datetimelikeÎ s
|
z)TestCounting.test_count_with_datetimelikec C sx t tjtjgddgddgd}| ddg¡j ¡ }tg ddggg g gddgd}tg |tjd d
|
}t j
|
||d d d S ) Nr r r r )r ÚBÚCr r| )ÚlevelsÚcodesÚnamesr} )r1 r& rx F)Zcheck_index_type) r r_ Únanr r} rU r r r' r r )r r Úresultr6 r r r r! Ú(test_count_with_only_nans_in_first_groupâ s
|
z5TestCounting.test_count_with_only_nans_in_first_groupc C sl t dddddgddtjddgd}| dg¡ ¡ }t tdd d
|
d gdd d ddddgid}t ||¡ d S )Nr r/ rN r r )r r| r| g g @g @g @©rx r )r1 Údata)r r_ ÚNaNr rU r r ry )r r rz r r r r! Ú4test_count_groupby_column_with_nan_in_groupby_columnê s $ ÿzATestCounting.test_count_groupby_column_with_nan_in_groupby_columnc C s t dddd}tt d¡|t d¡gd}| dd ¡}| ¡ }tt d¡t d¡|gd}| dd ¡}| ¡ }t ||¡ d S ) Nz1/1/2012Z5miné
|
)Ústartru Úperiodsr2 c S s | d d dkS )Nr r r r ©rJ r r r! Ú<lambda>ø ó z@TestCounting.test_groupby_count_dateparseerror.<locals>.<lambda>c S s | d d dkS )Nr r r r r r r! r ü r )r r r_ Úaranger rU r r )r ÚdrZserZgroupedr r r r r! Ú!test_groupby_count_dateparseerroró s z.TestCounting.test_groupby_count_dateparseerrorN) Ú__name__Ú
|
__module__Ú __qualname__r" r- r3 r7 r8 r= r? r@ rA rC rE rF rG rI rM r[ ri ÚpytestÚmarkZ parametrizer> r{ r r r r r r r! r s<
|
|
|
|
|
|
|
|
|
ûþ
|
|
|
r c C s\ t tdt d¡ d¡d} tddgtddgdd d
|
d }| d¡j ¡ }t
|
||¡ d S ) NÚababrN ztimedelta64[ns])r Údeltr r r r r r ©r1 rx ) r r; r_ r Úastyper r r r rU r r )r r r r r r! Ú#test_groupby_timedelta_cython_count s ÿ r c C sJ d} t d| d dd}ttj tt| ¡tj dd| ¡tj | ¡ d¡tj d d| ¡tj || ¡tj | ¡ d¡tj | ¡ d¡tj || ¡tj |d
|
¡ tj tt| ¡d }|j
|
d d dg¡D ]"}tj |j tj | | d ¡|f< qÄ|d d¡|d< d d d d gfD ]<}| |¡ ¡ }| |¡ tj¡j |d
|
d}t ||¡ qd S )Ni z
|
2015-08-30r ÚT)r ru r r/ r iöÿÿÿr ) Ú1stÚ2ndZ3rdÚ4thZ5thZ6thZ7thZ8thÚ9thr r r r Úcategory)Zaxis)r r r_ r` rb r; r ÚrandintZrandnÚroundr Zdropr rc r r rU Úapplyr ry )Únr r ÚcolÚkeyÚleftÚrightr r r! Ú
|
test_count s* ÷ÿ r© c C s¼ t dddgdtjdgdtjtjggdddgd } | d¡ ¡ }| jdd
|
d ¡ }t ddgd d ggddgddgd }d|j_t || ¡ ¡ t ||¡ | d¡d ¡ }t
|
||d ¡ d S )Nr r ZfooÚbarr r r| r} r F)Zas_indexr r0 ) r r_ r r rU r1 rx r ry Z reset_indexr )r Zcount_asZ count_not_asr Zcount_Br r r! Útest_count_non_nulls( s þ" r« c C sØ t dgd dgd dgd dgd d} | d¡j ¡ }tddgtddgdddd}t ||¡ t dtj tj gdgd dgd dgd d} | d¡j ¡ }td dgtddgdddd}t ||¡ d S )
|
Nr r r r )r Úcr¬ r r r )
|
r r r rU r r r r r_ r ©r r r r r r! Útest_count_object< s . 2 r® c C sª t t j ddd¡t j ddd¡f¡} t| ddddgd }t j||dk< | ddg¡ ¡ }d
|
D ]F}|d |¡|d< |d |¡|d< | ddg¡ ¡ }t
|
||¡ q^d S ) Nr r/ )r\ r r r r r¬ Údr )Úfloat32r( ) r_ Zhstackr` r¡ r r r rU r r ry )Úvalsr r Útr r r r! Útest_count_cross_typeH s ÿr³ c C s t t ddddgtj¡t ddddgtj¡t dddd gtj¡td
|
d } | d ¡ ¡ }t ddgddgddgd t tdd dd}t
|
||¡ d S )Nr r r r\ r é rN r/ é r )r r r¬ Úgrpr¶ )r r r¬ Úabr r2 ) r r_ ÚarrayZint8Zuint32Úint16r; r rU r r ry r r r r! Útest_lower_int_prec_countY s üÿ ÿrº c s G dd dt G fddd t fddtdD tdd } | d
|
¡ ¡ }td d d gittd d
|
dd}t ||¡ d S )Nc @ s e Zd ZdS )zAtest_count_uses_size_on_exception.<locals>.RaisingObjectExceptionN)r r r r r r r! ÚRaisingObjectExceptionj s r» c s0 e Zd Zddd fdd ZfddZ ZS ) z8test_count_uses_size_on_exception.<locals>.RaisingObjectúI will raise inside CythonN)Úreturnc s t ¡ || _d S rT )ÚsuperÚ__init__Úmsg)r rÀ ©Ú __class__r r! r¿ n s
|
zAtest_count_uses_size_on_exception.<locals>.RaisingObject.__init__c s | j d S rT )rÀ )r Úother©r» r r! Ú__eq__r s z?test_count_uses_size_on_exception.<locals>.RaisingObject.__eq__)r¼ )r r r r¿ rÅ Ú __classcell__r rÄ rÁ r! Ú RaisingObjectm s rÇ c s g | ]
|
} qS r r )rP re )rÇ r r! rS v s z5test_count_uses_size_on_exception.<locals>.<listcomp>rN r )r r¶ r¶ r r r· r r2 ) Ú Exceptionr r> r; r rU r r ry r r )rÇ r» r! Ú!test_count_uses_size_on_exceptioni s $ rÉ )Ú itertoolsr Ústringr Únumpyr_ r Zpandasr r r r r r r
|
r Zpandas._testingZ_testingr r r r© r« r® r³ rº rÉ r r r r! Ú<module> s (
|
o
|
