U
|
ý°dù ã @ sZ d dl Zd dlZd dlZd dlmZ d dlmZ d dl m
|
Z
|
d dl m Z G dd dZ dS )é N)Ú Timedelta)Únanops)ÚTimedeltaArrayc
|
@ sÌ e Zd Zej ddddddg¡ej ddd g¡d
|
d Zej ddd g¡d d Zdd Zdd Z dd Z
|
dd Z ej de de d¡e jddde jdddg¡d d Zd d Zd d! Zd"d# Zd$S )%ÚTestReductionsÚnameÚstdÚminÚmaxÚmedianÚmeanÚskipnaTFc C sP t g ¡}|j}t|||d}|t jks.tt|||d}|t jksLtd S )N©r )ÚpdÚTimedeltaIndexÚarrayÚgetattrÚNaTÚAssertionError)Úselfr r ÚtdiÚarrÚresult© r úed:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/arrays/timedeltas/test_reductions.pyÚtest_reductions_empty s
|
z$TestReductions.test_reductions_emptyc C sh t g ¡}|j}|j|d}t|ts*t|tdks:t|j|d}t|tsTt|tdksdtd S )Nr r )r r r ÚsumÚ
|
isinstancer r )r r r r r r r r Útest_sum_empty s
|
z TestReductions.test_sum_emptyc C s t ddddddg¡}| ¡ }td}||ks2t| ¡ }td}||ksNt|jdd}|tjksht|jdd}|tjkstd S )NÚ3Hr Ú2HÚ5HÚ4HFr )r Z_from_sequencer r r r r r )r r r Úexpectedr r r Ú test_min_max% s zTestReductions.test_min_maxc C s0 t ddddddg¡}|j}|jdd}tdd }t|ts@t||ksLt|jdd}t|tsft||ksrt|jd
|
d}|t jkst|jd
|
d}|t jks¦t|jd d }|t jksÀt|jd d }|t jksÚt|jd d }t|tsôt||kst|jd d }t|ts t||ks,td S )Nr r r r r! Tr é ©ÚhoursFé )Z min_counté )r r r r r r r r ©r r r r r" r r r Útest_sum6 s.
|
zTestReductions.test_sumc C sp t dddddg¡}|j}t |¡}tdd}t|ts<t||ksHtt |¡}t|ts`t||ksltd S )Nr r r r! r$ r% )r r r Únpr r r r r) r r r Ú
|
test_npsumW s
|
|
|
zTestReductions.test_npsumc C sÈ t d¡ t j¡ d¡ d¡ dd¡}d|d< t|}|jdd }|tj ksPt
|
|jd
|
dd }t t d d tj g¡j }t ||¡ |jddd }t t dd t dd t dd tj g¡j }t ||¡ d S )Né zm8[s]zm8[ns]é é ZNat)éÿÿÿÿr0 Fr r )Úaxisr é )Úsecondsr( é r' )r+ ZarangeZastypeÚint64ÚviewÚreshaper r r r r r r Z_valuesÚtmÚ assert_timedelta_array_equal)r r Útdar r" r r r Útest_sum_2d_skipna_falsee s" & üÿz'TestReductions.test_sum_2d_skipna_falseÚaddr z
|
2021-01-01ÚUTC)Útzz
|
Asia/Tokyoc C s6 t ddddddg¡| }|j}|jdd}tdd}t|tsDt||ksPt|jdd}t|tsjt||ksvtt|d d d kr¶tj t
|
|¡dd}t|t
|
j sªt||ks¶t|jd
|
d}|t j ksÐt|jd
|
d}|t j ksêtt|d d d kr2tj t
|
|¡d
|
d}t|t
|
j s"tt
|
|¡s2td S ) NÚ0Hr! r r Tr r/ r% r> F)r r r r r r r r r Znanstdr+ ZasarrayZ timedelta64r Zisnat)r r< r r r r" r r r Útest_std~ s*
|
|
zTestReductions.test_stdc C sª t ddddddg¡}|j}|jdd}tdd }t|ts@t||ksLt|jdd}t|tsft||ksrt|jd
|
d}|t jkst|jd
|
d}|t jks¦td S ) Nr? r r Ú5H06mr Tr r/ r% F)r r r r
|
r r r r r) r r r Ú test_median¤ s
|
zTestReductions.test_medianc C s t ddddddg¡}|j}t| ¡ j ¡ }| ¡ }||ksBt|jdd}|t jks\t| ¡ jdd}||ksxt|jdd }||kstd S )
|
Nr? r r rA r Fr r ©r1 ) r r Ú_datar ZdropnaZ_ndarrayr r r )r r r r" r r r r Ú test_mean· s zTestReductions.test_meanc C s t jddd}|j dd¡}|jdd}|d }t ||¡ |jdd}|d d
df td d
|
}t ||¡ |jd d}| ¡ }||kstd S ) Nz14 daysé )Zperiodsé r/ r rC r( r2 r% ) r Ztimedelta_rangerD r7 r r8 r9 r r )r r r: r r" r r r Ú test_mean_2dÉ s zTestReductions.test_mean_2dN)Ú__name__Ú
|
__module__Ú __qualname__ÚpytestÚmarkZ parametrizer r r# r* r, r; r r Ú Timestampr@ rB rE rH r r r r r s*
|
|
! üþ
|
r )Únumpyr+ rL Zpandasr r Zpandas._testingZ_testingr8 Z pandas.corer Zpandas.core.arraysr r r r r r Ú<module> s
|
