U
|
®ý°d\ ã @ sj d dl Zd dlZd dlmZ d dlmZmZ d dlm Z m
|
Z
|
m Z m Z m Z mZ d dlmZ G dd dZdS )é N)Ú is_numpy_dev)Úis_complex_dtypeÚis_extension_array_dtype)ÚNAÚPeriodÚSeriesÚ TimedeltaÚ TimestampÚ
|
date_rangec @ sd e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z dd Z dS )ÚTestSeriesDescribec C s` t dddddgdd}| ¡ }t dd| ¡ dddddgdd d
|
d d d dddgd}t ||¡ d S )Nr é é é é Zint_data©Únameé ÚcountÚmeanÚstdÚminú25%ú50%ú75%Úmax©r Úindex)r Údescriber ÚtmÚassert_series_equal©ÚselfÚserÚresultÚexpected© r% ú`d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/series/methods/test_describe.pyÚtest_describe_ints s ýz%TestSeriesDescribe.test_describe_intsc C sL t dddddgdd}| ¡ }t ddddgddd d
|
d gd }t ||¡ d S ) NTFZ bool_datar r r r r ÚuniqueÚtopÚfreqr ©r r r r r r% r% r& Útest_describe_bools! s
|
|
ÿz&TestSeriesDescribe.test_describe_boolsc C sL t dddddgdd}| ¡ }t dddd gdd
|
d d d gd}t ||¡ d S )NÚaÚbÚcÚdZstr_datar r r r r r( r) r* r r+ r r% r% r& Útest_describe_strs) s
|
|
ÿz%TestSeriesDescribe.test_describe_strsc C s t tdtdtdtdtdgdd}| ¡ }t d|d | ¡ |d
|
|d |d |d |d gdddddddddgd}t ||¡ d S )Nz1 daysz2 daysz3 daysz4 daysz5 daysZtimedelta_datar r r r r r r r r r r r r r r r )r r r r r r r r% r% r& Útest_describe_timedelta641 s ûø
|
.ýz,TestSeriesDescribe.test_describe_timedelta64c C s^ t tddtddtddgdd}| ¡ }t dd|d dgdd d
|
d d gd }t ||¡ d S )Nz2020-01ÚMz2019-12Z period_datar r r r r r( r) r* r )r r r r r r r% r% r& Útest_describe_periodD s þ
|
ýz'TestSeriesDescribe.test_describe_periodc C s t d d gtd}| ¡ }t ddtjtjgtddddgd}t ||¡ |d d
¡ }t ||¡ t |jd ¡svt t |jd ¡st d S )
|
N©Údtyper r r( r) r* ©r6 r r r )
|
r Úobjectr ÚnpÚnanr r ÚisnanZilocÚAssertionError©r! Úsr# r$ r% r% r& Útest_describe_empty_objectQ s
|
ý z-TestSeriesDescribe.test_describe_empty_objectc
|
C s¢ |}t |}tddd}tddd}tt|||d|d}| ¡ }tdtddd |¡| |¡|d |d |d | |¡g|dd d
|
d d d dgd}t ||¡ d S )Néâ r r ©Útzr r r r r r r r r r r )Ústrr r r
|
r Z tz_localizer r ) r! Ztz_naive_fixturerB r ÚstartÚendr> r# r$ r% r% r& Útest_describe_with_tzb s& ù õ z(TestSeriesDescribe.test_describe_with_tzc C s¨ d }}t ddd}t ddd}tt|||d|d}| ¡ }tdt d|dt d|dt d |dt d|dt d
|
|dt d |dg|d d dddddgd}t ||¡ d S )NZCETr@ r r rA r z2018-01-03 00:00:00z2018-01-01 00:00:00z2018-01-02 00:00:00z2018-01-04 00:00:00z2018-01-05 00:00:00r r r r r r r r )r r r
|
r r r )r! r rB rD rE r> r# r$ r% r% r& Ú test_describe_with_tz_numericy s$
|
|
|
|
|
|
ù õ z0TestSeriesDescribe.test_describe_with_tz_numericc C sh t tddd}| ¡ }t dtdtdtdtdtdtdgd d
|
d d d ddgd}t ||¡ d S )NZ2012r )Zperiodsz
|
2012-01-02z
|
2012-01-01z2012-01-01T12:00:00z2012-01-02T12:00:00z
|
2012-01-03r r r r r r r )r )r r
|
r r r r r= r% r% r& Ú*test_datetime_is_numeric_includes_datetime s ù ö z=TestSeriesDescribe.test_datetime_is_numeric_includes_datetimec
|
C sª t |rd}nt|rdnd }tddg|d}|dkr`tr`tjtdd | ¡ W 5 Q R X d S | ¡ }tdd | ¡ dd
|
d d d gd dddddddg|d}t
|
||¡ d S )NÚFloat64Z
|
complex128r r r5 z$^a must be an array of real numbers$)Úmatchg @g à?g Ð?g è?g ð?r r r r r r r r )r r6 ) r r r r ÚpytestZraisesÚ TypeErrorr r r r )r! Zany_numeric_dtyper6 r" r# r$ r% r% r& Útest_numeric_result_dtype¡ s4 ÿø
|
ôz,TestSeriesDescribe.test_numeric_result_dtypec C sj t dgdd}t d ¡ | ¡ }W 5 Q R X t ddtdddddgddddd d
|
d d d gd}t ||¡ d S )Ng rI r5 r r r r r r r r r r r7 )r r Zassert_produces_warningr r r r r% r% r& Ú test_describe_one_element_eaÀ s ýz/TestSeriesDescribe.test_describe_one_element_eaN)Ú__name__Ú
|
__module__Ú __qualname__r' r, r1 r2 r4 r? rF rG rH rM rN r% r% r% r& r s
|
r )Únumpyr9 rK Z pandas.compatr Zpandas.core.dtypes.commonr r Zpandasr r r r r r
|
Zpandas._testingZ_testingr r r% r% r% r& Ú<module> s
|
