U
|
¬ý°d,) ã @ sz d dl mZ d dlZd dlZd dlZd dlmZ d dlm Z
|
d dl m Z m Z G dd dZG dd dZG d d
|
d
|
ZdS ) é )Ú timedeltaN)Ú Timedelta)Ú DatetimeArrayÚTimedeltaArrayc @ sî e Zd Zejdddgddd Zejdd Zd d
|
Zd d Zej
|
d e j ¡dd Z dd Zdd Zdd Zej
|
de dd¡e dd¡g¡dd Zdd Zd d Zd d Zd!d" Zd#d$ Zd%d& Zd'd( Zd)d* Zd+d, Z d-d. Z d/S )0Ú TestNonNanoÚsÚmsÚus)Úparamsc C s |j S ©N)Úparam)ÚselfÚrequest© r úZd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/arrays/test_timedeltas.pyÚunit s zTestNonNano.unitc C s. t jdt jd d| d¡}tj||jdS )Né ©Údtypeúm8[ú])ÚnpÚarangeÚint64Úviewr Ú _simple_newr )r r Úarrr r r Útda s zTestNonNano.tdac C sT t jdt jd d| d¡}tj||jd}|j|jks>t|d j|ksPtd S )Nr r r r r ) r r r r r r r ÚAssertionErrorr )r r r r r r r Ú test_non_nano s zTestNonNano.test_non_nanoc C sV t jtdd | d¡ W 5 Q R X t |¡}t jtdd | d¡ W 5 Q R X d S )NzSupported units©ÚmatchÚD)ÚpytestÚraisesÚ
|
ValueErrorÚas_unitÚpdÚIndex)r r Útdir r r Útest_as_unit_raises s
|
|
z TestNonNano.test_as_unit_raisesÚfieldc C s@ |j d¡}tj||jd}t||}t||}t ||¡ d S ©Núm8[ns]r )Ú_ndarrayÚastyper r r ÚgetattrÚtmÚassert_numpy_array_equal)r r r+ Úas_nanoÚtda_nanoÚresultÚexpectedr r r Ú test_fields) s
|
|
|
zTestNonNano.test_fieldsc C s< |j d¡}tj||jd}| ¡ }| ¡ }t ||¡ d S r, )r. r/ r r r Úto_pytimedeltar1 r2 )r r r3 r4 r5 r6 r r r Útest_to_pytimedelta2 s
|
z TestNonNano.test_to_pytimedeltac C s< |j d¡}tj||jd}| ¡ }| ¡ }t ||¡ d S r, )r. r/ r r r Ú total_secondsr1 r2 )r r r r3 r4 r5 r6 r r r Útest_total_seconds: s
|
z TestNonNano.test_total_secondsc C s4 t d ¡ }t t dg¡ ¡ d }||ks0td S )Nz2 minr )r r: r' Úarrayr )r r6 r5 r r r Ú"test_timedelta_array_total_secondsB s z.TestNonNano.test_timedelta_array_total_secondsÚnatÚNaTÚnsc C sp || }t |tst|j|jks&t| ¡ ¡ s6t|| }t |tsLt|j|jks\t| ¡ ¡ sltd S r )Ú
|
isinstancer r Ú_cresoÚisnaÚall)r r> r r5 r r r Ú test_add_nat_datetimelike_scalarI s z,TestNonNano.test_add_nat_datetimelike_scalarc C st |t j }t|tst|j|jks(t| ¡ ¡ s8tt j| }t|tsPt|j|jks`t| ¡ ¡ sptd S r )r' r? rA r r rB rC rD )r r r5 r r r Útest_add_pdnatW s
|
|
zTestNonNano.test_add_pdnatc C s¬ t jd|d d¡}| d¡| }|| }t ||¡ || }t ||¡ |td7 }|j|j }tj ||j
|
d d¡ |j ¡}|| }t ||¡ || }t ||¡ d S )Nz
|
2016-01-01)Útzr@ é r ÚUTC)r' Ú Timestampr& r1 Ú assert_extension_array_equalr r. Zasm8r r r Z tz_localizeZ
|
tz_convertrG )r r Ztz_naive_fixtureÚtsr6 ÚresZ
|
exp_valuesr5 r r r Ú test_add_datetimelike_scalard s$ ÿþÿ z(TestNonNano.test_add_datetimelike_scalarc C sB d}|| }t j|j| |jd}t ||¡ |j|jks>td S ©Né r ©r r r. r r1 rK rB r ©r r Úotherr5 r6 r r r Útest_mul_scalar| s
|
zTestNonNano.test_mul_scalarc C sL t t|¡}|| }tj|j| |jd}t ||¡ |j |j ksHt
|
d S ©Nr © r r Úlenr r r. r r1 rK rB r rR r r r Útest_mul_listlike s
|
z TestNonNano.test_mul_listlikec C sR t t|¡}|| t¡ }tj|j| |jd}t
|
||¡ |j |j ksNt d S rU ) r r rW r/ Úobjectr r r. r r1 rK rB r rR r r r Útest_mul_listlike_object s
|
z$TestNonNano.test_mul_listlike_objectc C sB d}|| }t j|j| |jd}t ||¡ |j|jks>td S rO rQ rR r r r Útest_div_numeric_scalar s
|
z#TestNonNano.test_div_numeric_scalarc C s4 t dd}|| }|jt dd¡ }t ||¡ d S )NrH ©Úsecondsr )r r. r Z timedelta64r1 r2 rR r r r Útest_div_td_scalar s
|
z TestNonNano.test_div_td_scalarc C sL t t|¡}|| }tj|j| |jd}t ||¡ |j |j ksHt
|
d S rU rV rR r r r Útest_div_numeric_array s
|
z"TestNonNano.test_div_numeric_arrayc C s2 |j |j d }|| }|j | }t ||¡ d S )Néÿÿÿÿ)r. r1 r2 rR r r r Útest_div_td_array¥ s
|
z TestNonNano.test_div_td_arrayc C sn | d¡}|d }|| }t ||¡ || }t ||¡ |d }|| }t ||¡ || }t ||¡ d S )Nr- rP r )r/ r1 rK )r r r4 r6 rM r r r Útest_add_timedeltaarraylike« s
|
z'TestNonNano.test_add_timedeltaarraylikeN) Ú__name__Ú
|
__module__Ú __qualname__r# Zfixturer r r r* ÚmarkÚ parametrizer Z
|
_field_opsr7 r9 r; r= r Ú
|
datetime64rE rF rN rT rX rZ r[ r^ r_ ra rb r r r r r s4
|
|
|
ÿ
|
r c @ sô e Zd Zej deejej ddg¡dd Z
|
dd Z ej de d d
|
e d d
|
¡ e d d
|
¡ g¡d d Zej d d e d ¡de d¡e d¡dejdddd d d e d¡d d d d¡e d¡ d¡g ¡ej ddd g¡d d Zd S ) ÚTestTimedeltaArrayr Zuint32Zuint64c C st t tdtdg¡}t |¡tjkrNtjtdd | |¡ W 5 Q R X d S | |¡}|j
|
d¡}t ||¡ d S )NÚ1HZ2HzDo obj.astype\('int64'\)r Úi8)r Z_from_sequencer r r r r# r$ Ú TypeErrorr/ r. r r1 r2 )r r r r5 r6 r r r Útest_astype_int½ s
|
z"TestTimedeltaArray.test_astype_intc C s2 t tjdddd}td|d< |jd ks.td S )Nrj rP ÚH©ZperiodsÚfreqr )r r' Útimedelta_ranger rp r )r Úar r r Útest_setitem_clears_freqÊ s z+TestTimedeltaArray.test_setitem_clears_freqÚobjrH r\ c C s@ t jdddd}t||jd}||d< |d tddks<td S ) Nú2 Daysé rn ro ©rp r rH r\ )r' rq r rp r r )r rt r) r r r r Útest_setitem_objectsÏ s
|
z'TestTimedeltaArray.test_setitem_objectsrS g ð?r? z
|
2021-01-01Úinvalidé
|
rk r é é é Ê;zdatetime64[ns]r" ÚindexTFc C sj t jdddd d d }t|dd}|r4t |¡}d d
|
d g¡}tjt|d | |¡ W 5 Q R X d S ) Nrz rk r r{ r| r} r" rw ú|z0searchsorted requires compatible dtype or scalarz<value should be a 'Timedelta', 'NaT', or array of those. Gotr )
|
r r r r' r( Újoinr# r$ rl Z searchsorted)r rS r~ Údatar Úmsgr r r Ú test_searchsorted_invalid_typesß s
|
þÿz2TestTimedeltaArray.test_searchsorted_invalid_typesN)rc rd re r# rf rg Úintr Úint32r rm rs r Zto_timedelta64r8 rx rh r' rJ r r Z to_periodr r r r r ri ¼ s4
|
ýþ
|
÷þri c @ s, e Zd Zdd Zdd Zdd Zdd Zd S )
|
Ú TestUnaryOpsc C sf t jdddgdd}t|}t jdddgdd}t|}t|}t ||¡ t |¡}t ||¡ d S )Nìýÿÿÿ qa r? ì @âB1r- r ì qa )r r< r Úabsr1 Ú assert_timedelta_array_equal©r Úvalsr Zevalsr6 r5 Úresult2r r r Útest_absÿ s
|
zTestUnaryOps.test_absc C sh t jdddgdd}t|}|
|
}t ||¡ t ||¡r>tt |¡}t ||¡ t ||¡rdtd S )Nr r? r r- r )r r< r r1 r Z shares_memoryr Zpositive)r r r r5 r r r r Útest_pos s
|
zTestUnaryOps.test_posc C sd t jdddgdd}t|}t jdddgdd}t|}| }t ||¡ t |¡}t ||¡ d S )Nr r? r r- r r lýÿÿÿ @âB1)r r< r r1 r Únegativer r r r Útest_neg s
|
zTestUnaryOps.test_negc C s^ t jdddd}t||jd}t|j |j d}| }t ||¡ t |¡}t ||¡ d S )Nru rv rn ro rw ) r' rq r rp Ú_datar1 r r r )r r) r r6 r5 r r r r Ú test_neg_freq% s
|
zTestUnaryOps.test_neg_freqN)rc rd re r r r r r r r r r þ s r )Údatetimer Únumpyr r# Zpandasr' r Zpandas._testingZ_testingr1 Zpandas.core.arraysr r r ri r r r r r Ú<module> s .B
|
