U
|
ý°d# ã @ s d dl Zd dlZd dlmZmZmZmZ d dlm Z
|
ej dddddgg¡dd Z ej dd d d d gd dddgdd ddgd dddgejddddgej jd g¡d
|
d Zej dddddgg¡ej d d dg¡dd Zdd Zdd Zdd Zej dd dg¡dd ZdS )é N)Ú DataFrameÚIndexÚ PeriodIndexÚSeriesÚbyÚAÚBc C s: | j |d}| ¡ }|D ] \}}|| t|kstqd S )N©r )ÚgroupbyÚsizeÚlenÚAssertionError)Údfr ÚgroupedÚresultÚkeyÚgroup© r úUd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/groupby/test_size.pyÚ test_size s r é )Zmarksc s fddt ¡D }|r.dd | ¡ D }t|dd}|rF| ¡ }t |j¡rttdd D st|j t
|
j ¡|_| j |||d}| ¡ }t ||¡ d S ) Nc s$ i | ] t fd dD qS )c 3 s | ]}| kV qd S ©Nr )Ú.0Úvalue©r r r Ú <genexpr>! s z.test_size_axis_1.<locals>.<dictcomp>.<genexpr>)Úsum)r r r r Ú
|
<dictcomp>! s z$test_size_axis_1.<locals>.<dictcomp>c S s i | ]\}}|d k r||qS r r )r r r r r r r # s Úint64)Údtypec s s | ]}|d kV qd S r r )r Úxr r r r ' s z#test_size_axis_1.<locals>.<genexpr>)r ZaxisÚsortÚdropna)ÚdictÚfromkeysÚitemsr Z
|
sort_indexÚtmZis_integer_dtypeÚindexÚanyÚastypeÚnpÚint_r
|
r Úassert_series_equal) r Zaxis_1r r! r" ÚcountsÚexpectedr r r r r Útest_size_axis_1 s r/ r! TFc C s\ t tj dd¡tdd}|j|| d ¡ }|j|| dd dd ¡}tj ||d d
|
d S ) Né )iè é ÚABC©Úcolumns)r r! ÚCc S s
|
| j d S )Nr )Úshape)Úar r r Ú<lambda>4 ó z test_size_sort.<locals>.<lambda>F)Z check_names)
|
r r* ÚrandomÚchoiceÚlistr
|
r Úapplyr& r, )r! r r ÚleftÚrightr r r Útest_size_sort/ s r@ c C s< t ddgd} tdtg ddd}t | d¡ ¡ |¡ d S )Nr r r3 r ©Úname©r r' )r r r r& r, r
|
r )r Úoutr r r Útest_size_series_dataframe8 s rE c C sB t dd d gi} | d¡ ¡ }tdtg ddd}t ||¡ d S )Nr r rA rC )r r
|
r r r r& r, )r r r. r r r Útest_size_groupby_all_null? s rF c C s> t dgtdgdddd} | jdd}| ¡ }t || ¡ d S )Nr Ú2000r ÚD)rB Úfreq)r' )Úlevel)r r r
|
r r& r, )ZserÚgrpr r r r Útest_size_period_indexH s rL Úas_indexc C s´ t ddgddggddgd}|d d¡|d< |jddg| d ¡ }t dddgdddgdddgdddggddd gd}|d d¡|d< | r¤| ddg¡d d ¡}t ||¡ d S )
|
Nr é r r r3 Úcategory)rM r r )r r) r
|
r Z set_indexÚrenamer& Z assert_equal)rM r r r. r r r Útest_size_on_categoricalP s " ÿrQ )Únumpyr* ZpytestZpandasr r r r Zpandas._testingZ_testingr& ÚmarkZ parametrizer ÚparamZxfailr/ r@ rE rF rL rQ r r r r Ú<module> s.
|
|
|
|
|
ûþ
|
|
|
