U
|
ý°d5: ã @ sR d dl Zd dlZd dlZd dlmZmZmZmZm Z d dl
|
m Z G dd dZ dS )é N)Ú CategoricalÚ DataFrameÚSeriesÚ 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 Zej dd d d d gd d gg¡d d Zd!d" Zd#d$ Zd%d& Zd'S )(ÚTestDataFrameDescribec
|
C s¶ t dddddgdddddgdd d
|
d d gd }| ¡ }t ddd
|
|j ¡ dd d
|
d d giddddddddgd}t ||¡ |jdgd}t ddd dd gidd d d gd}t ||¡ d S )!NÚaÚbÚcÚdÚeTFé
|
é é é( é2 )Z string_dataÚ bool_dataÚint_datar é ÚcountÚmeanÚstdÚminú25%ú50%ú75%Úmax©ÚindexÚbool©Úincluder é é ÚuniqueÚtopÚfreq)r Údescriber r ÚtmÚassert_frame_equal©ÚselfÚdfÚresultÚexpected© r/ ú_d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/methods/test_describe.pyÚ!test_describe_bool_in_mixed_frame s$ ýÿ
|
þ
|
ÿz7TestDataFrameDescribe.test_describe_bool_in_mixed_framec C sp t dd d gitd}| ¡ }t dddtjtjgitddddgd}t ||¡ |jd d
¡ }t ||¡ d S ) NÚA©Údtyper r r$ r% r& )r4 r )r Úobjectr' ÚnpÚnanr( r) Úilocr* r/ r/ r0 Útest_describe_empty_object* s
|
ý z0TestDataFrameDescribe.test_describe_empty_objectc
|
C s( t ddddgddddgd}| ¡ }t ddddgddddgdddd d
|
gd }t ||¡ t dddddgd d dddgd}| ¡ }t ddd|j ¡ d d dddgiddddddddgd }t ||¡ t ddddgddddgd}| ¡ }t ddddgddddgdddd d
|
gd }t ||¡ d S ) NFT)Z bool_data_1Z bool_data_2é r" r# r r$ r% r& r r é )r r r r r r r r r r r r r r
|
)r Zstr_data)r r' r( r) r r r* r/ r/ r0 Útest_describe_bool_frame8 s>
|
|
þÿ
|
þ þÿ þ ÿ
|
þz.TestDataFrameDescribe.test_describe_bool_framec C s( t dtj ddd¡i}dd tdddD }t||}|jdgdd }tj|j tdd
|
dd |d |d < |}|
|
¡ }t |j dkst tddddgdddgdd}t|}|
|
¡ }tddddgddddgd}t ||¡ ttddddg}t |ddddgd}|
|
¡ }t |d j|d j¡ d S ) NÚvaluer i' éd c S s g | ]}| d |d qS )z - ió r/ )Ú.0Úir/ r/ r0 Ú
|
<listcomp>` s zCTestDataFrameDescribe.test_describe_categorical.<locals>.<listcomp>iô T)ZbyZ ascendingi) F)ÚrightÚlabelsZ value_groupr; r r r
|
)Ú
|
categoriesÚorderedr: r" r# r r$ r% r& r )ÚcatÚsrF rG )r r6 ÚrandomÚrandintÚranger Z sort_valuesÚpdZcutr= r' ÚlenÚcolumnsÚAssertionErrorr r( Zassert_series_equalZassert_numpy_array_equalÚvalues) r+ r, rC Z
|
cat_labelsrF r- rG r. Zdf3r/ r/ r0 Útest_describe_categorical^ s4
|
|
ÿ
|
|
ÿ z/TestDataFrameDescribe.test_describe_categoricalc C sv t dtg i}| ¡ }t dddtjtjgiddddgdd}t ||¡ t |jd ¡s^t t |jd
|
¡srt d S ) NZ empty_colr r r$ r% r& r5 ©r r4 )r" r )r# r )
|
r r r' r6 r7 r( r) Úisnanr8 rN r* r/ r/ r0 Ú&test_describe_empty_categorical_column} s
|
ý z<TestDataFrameDescribe.test_describe_empty_categorical_columnc
|
C sà t jdddgddd}tddd d
|
d gddd d
|
d gd d d ddgd|d}| ¡ }t jddgdddgddd}tdd |j ¡ ddd d
|
d gdd |j ¡ ddd d
|
d gddddddddd g|d }t ||¡ t |j
|
j |j
|
j ¡ d S ) NÚint1Úint2ÚobjTÚXXX)rE Únamer r r r r r2 r ÚXr; )rT rU rV )rM )rD rE rX r )rT rU r r r r r r r r )r rM ) rK ZCategoricalIndexr r' rT r rU r( r) Zassert_categorical_equalrM rO ©r+ rM r, r- Z exp_columnsr. r/ r/ r0 Ú!test_describe_categorical_columns s0 ýúüþú z7TestDataFrameDescribe.test_describe_categorical_columnsc
|
C s t jdddgdddd}tdd d
|
d d gdd d
|
d d gd dd ddgd}||_| ¡ }t jddgdddd}tdd
|
|jd d
df ¡ dd d
|
d d gdd
|
|jd d
df ¡ dd d
|
d d gdddddddddgd }||_t ||¡ |jj dksòt
|
|jj |jj kst
|
d S ) Nz
|
2011-01-01z
|
2011-02-01z
|
2011-03-01ZMSz
|
US/EasternrW )r& ÚtzrX r r r r r r2 r rY r; )r r; r" r )r r; r r r r r r r r r ) rK Z DatetimeIndexr rM r' r8 r r( r) r& rN r\ rZ r/ r/ r0 Ú test_describe_datetime_columns¬ s< ü ýÿ ÿ$$þû z4TestDataFrameDescribe.test_describe_datetime_columnsc C s
|
t jdddd}t jdddd}t||d}tdt d¡|jd d
d f ¡ t d¡t d
|
¡t d¡t d ¡t d ¡gdt d ¡|jd d
df ¡ t d¡t d¡t d ¡t d¡t d¡gdddddddddgd}| ¡ }t ||¡ d}t ||kst
|
d S ) Nz1 daysÚDr )r& Úperiodsz1 hoursÚH)Út1Út2z3 daysr z2 daysz4 daysz5 daysz3 hoursr; z2 hoursz4 hoursz5 hoursr r r r r r r r r a t1 t2
|
count 5 5
|
mean 3 days 00:00:00 0 days 03:00:00
|
std 1 days 13:56:50.394919273 0 days 01:34:52.099788303
|
min 1 days 00:00:00 0 days 01:00:00
|
25% 2 days 00:00:00 0 days 02:00:00
|
50% 3 days 00:00:00 0 days 03:00:00
|
75% 4 days 00:00:00 0 days 04:00:00
|
max 5 days 00:00:00 0 days 05:00:00) rK Ztimedelta_ranger Z Timedeltar8 r r' r( r) ÚreprrN )r+ ra rb r, r. r- Zexp_reprr/ r/ r0 Ú test_describe_timedelta_valuesÌ s< ø øõé ÿ z4TestDataFrameDescribe.test_describe_timedelta_valuesc
|
C sÎ |}t td}tddd}tddd}t t|||d}t||d}tddddddd d
|
gdtddd |¡| |¡|d |d |d | |¡tjgdd d d dddddgd}|jdd} t
|
| |¡ d S )Nr éâ r; ©r\ ©Ús1Ús2r" r r# r: gßávhXLù?r r r r r r r r r Úallr ) r rJ r r r Z tz_localizer6 r7 r' r( r) )
|
r+ Ztz_naive_fixturer\ rh ÚstartÚendri r, r. r- r/ r/ r0 Útest_describe_tz_valuesü s, øþ ò z-TestDataFrameDescribe.test_describe_tz_valuesc
|
C s t tddddddgd}| ¡ }t dtdtdtd tdtd
|
td tjgdddd dd ddgdddddddddgd}t ||¡ d S )NZ2012r# )r_ r; r" ©r r z
|
2012-01-02z
|
2012-01-01z2012-01-01T12:00:00z2012-01-02T12:00:00z
|
2012-01-03ç ø?ç @r r r r r r r r r )r r r' r r6 r7 r( r) r* r/ r/ r0 Ú*test_datetime_is_numeric_includes_datetime s" ø
|
õ òz@TestDataFrameDescribe.test_datetime_is_numeric_includes_datetimec C sª d}t td}tddd}tddd}t t|||d}t||d}| ¡ }| ¡ }ddd d
|
d d d dg} tj||gdddgdj| dd}
|
|jdd} t
|
| |
|
¡ d S )NZCETr re r; rf rg r r r r r r r r rh ri )ÚaxisÚkeysF)Úcopyrj r ) r rJ r r r r' rK ÚconcatZreindexr( r) ) r+ r\ rh rk rl ri r, Zs1_Zs2_Úidxr. r- r/ r/ r0 Útest_describe_tz_values2- s. ø
|
ÿ z.TestDataFrameDescribe.test_describe_tz_values2c C s t ddgi}t ddd¡}|j|d}t dddtjdfdd |D did d
|
d d d dddddddddddgd}t ||¡ d S )NÚxr; r é )Z percentilesç ð?c s s | ]
|
}d V qdS )rz Nr/ )r? Ú_r/ r/ r0 Ú <genexpr>O s zNTestDataFrameDescribe.test_describe_percentiles_integer_idx.<locals>.<genexpr>r r r r z0%z10%z20%z30%z40%r z60%z70%z80%z90%z100%r r )rz )r r6 Zlinspacer' ÚNaNr( r) )r+ r, Zpctr- r. r/ r/ r0 Ú%test_describe_percentiles_integer_idxH s0 ðþz;TestDataFrameDescribe.test_describe_percentiles_integer_idxc C sX t dddiidddiig}t dddddidgiddd d
|
gd }| ¡ }t ||¡ d S ) NÚtestr Ú1Ú2r" r; r r$ r% r& r )r r' r( r) )r+ r, r. r- r/ r/ r0 Ú8test_describe_does_not_raise_error_for_dictlike_elementse s
|
ÿzNTestDataFrameDescribe.test_describe_does_not_raise_error_for_dictlike_elementsÚexcluderx ÚyÚzc C sF t dgdgdgd}d}tjt|d |jd|d W 5 Q R X d S )
|
zU
|
When include is 'all', then setting exclude != None is not allowed.
|
r; r" r# )rx r r
z*exclude must be None when include is 'all')Úmatchrj ©r! r N)r ÚpytestZraisesÚ
|
ValueErrorr' )r+ r r, Úmsgr/ r/ r0 Ú2test_describe_when_include_all_exclude_not_allowedn s zHTestDataFrameDescribe.test_describe_when_include_all_exclude_not_allowedc C sr t dddgdddgdddggdddgdd}| ¡ }|jd d
df ¡ }tj|||g|jdd }t ||¡ d S )
|
Nr; r" r# Úbarr Úfloat64)rM r4 r )rs rr )r r' r8 rK ru rM r( r) )r+ r, r- Zserr. r/ r/ r0 Ú$test_describe_with_duplicate_columnsx s ýz:TestDataFrameDescribe.test_describe_with_duplicate_columnsc
|
C sz t dtjtjgtjd|d}| ¡ }t ddtjgdgd dgtjgd ddd d
|
d d d ddgdd}t ||¡ d S )Nr; rn r3 rz r g é r r r r r r r r ZFloat64rQ )r rK ZNAr' r( r) )r+ Zany_numeric_ea_dtyper, r- r. r/ r/ r0 Útest_ea_with_na s (ýz%TestDataFrameDescribe.test_ea_with_nac
|
C sÐ t d¡}ttdddgt | ¡ ¡dtdddgt | ¡ ¡dtdddgt | ¡ ¡dd}|j t | ¡ ¡t | ¡ ¡d}tdddddd dd
|
dgid d d dddddgt |
|
¡ ¡d}t ||¡ d S )NZpyarrowr; r" r# r3 )r r r
|
r r ro rp r r r r r r r r rQ ) r Z importorskipr r rK Z
|
ArrowDtypeZint8Úint16Úint32r' r r( r) )r+ Úpar, r- r. r/ r/ r0 Ú test_describe_exclude_pa_dtype s"
|
ýÿ ÿ ýz4TestDataFrameDescribe.test_describe_exclude_pa_dtypeN)Ú__name__Ú
|
__module__Ú __qualname__r1 r9 r< rP rS r[ r] rd rm rq rw r~ r r ÚmarkZ parametrizer r r r r/ r/ r/ r0 r s$ & 0
|
r )Únumpyr6 r ZpandasrK r r r r r Zpandas._testingZ_testingr( r r/ r/ r/ r0 Ú<module> s
|
|
