U
|
®ý°d ã @ s® d Z ddlmZmZ ddlZddlZddlZddlmZ ddl Z
|
ddl m Z m Z m Z mZ ddlmZ ddlmZ G dd dZG d d
|
d
|
ZG d d d Zd d Zdd ZdS )z+
|
Tests for scalar Timedelta arithmetic ops
|
é )ÚdatetimeÚ timedeltaN)ÚOutOfBoundsTimedelta)ÚNaTÚ TimedeltaÚ TimestampÚoffsets)Úopsc @ s e Zd ZdZej dedddedde
|
dd¡e
|
dd¡e d¡g¡d d
|
Z ej d ed ed eddde
|
dd¡e
|
dd¡ e ¡ e d¡ g¡dd Zej dejejg¡dd Zdd Zej dejejg¡dd Zej dejejg¡dd Zej dejejg¡d d Zej dejejg¡d d Zd d! Zd"d# Zd$d% Z d&d' Z d(d) Z d*d+ Z d,d- Z d.d/ Z!d0d1 Z"d2d3 Z#d4d5 Z$d6d7 Z%d8d9 Z&ej dejejg¡d:d; Z'ej dejejg¡d<d= Z(d>d? Z)d@dA Z*dBdC Z+dDS )EÚ TestTimedeltaAdditionSubtractionz\
|
Tests for Timedelta methods:
|
|
__add__, __radd__,
|
__sub__, __rsub__
|
Ú ten_secondsé
|
Ús©Úunit©Úsecondsl d( Únsc C sD t d}t d}t d}|| }||ks,t|| }||ks@td S )Nz20130101 09:01:12.123456z20130101 09:01:22.123456ú20130101 09:01:02.123456©r ÚAssertionError)Úselfr ÚbaseÚ expected_addÚ expected_subÚresult© r údd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/scalar/timedelta/test_arithmetic.pyÚtest_td_add_sub_ten_seconds" s z<TestTimedeltaAdditionSubtraction.test_td_add_sub_ten_secondsÚone_day_ten_secsz1 day, 00:00:10z1 days, 00:00:10é ©Údaysr ÚDc C sD t d}t d}t d}|| }||ks,t|| }||ks@td S )Nz20130102 09:01:12.123456z20130103 09:01:22.123456r r )r r r r r r r r r Ú#test_td_add_sub_one_day_ten_seconds8 s zDTestTimedeltaAdditionSubtraction.test_td_add_sub_one_day_ten_secondsÚopc C s¾ t ddd}||tddd}|tjkr6t|ts6t|tdddksJt||td}t|tsft|tdksvt||t d ¡}t|tst|td
|
ks¤t||t }|t ksºtd S ) Nr Údr éà r é z2018-01-12 18:09z2018-01-22 18:09z
|
2018-01-12ú
|
2018-01-22)
|
r r ÚoperatorÚaddÚ
|
isinstancer r ÚnpÚ
|
datetime64r ©r r$ Útdr r r r Ú test_td_add_datetimelike_scalarN s
|
|
z@TestTimedeltaAdditionSubtraction.test_td_add_datetimelike_scalarc C sl t d d¡}d}tjt|d |tddd W 5 Q R X d}tjt|d |tdd W 5 Q R X d S )
|
Nz
|
1700-01-01r z3Cannot cast 259987 from D to 'ns' without overflow.©Úmatchi÷ r" r z>Cannot cast 259987 days 00:00:00 to unit='ns' without overflow©r! )r Úas_unitÚpytestÚraisesr r r )r ÚtsÚmsgr r r Ú test_td_add_timestamp_overflowe s z?TestTimedeltaAdditionSubtraction.test_td_add_timestamp_overflowc C s@ t ddd}||t dd}t|t s*t|t ddks<td S )Nr r% r r3 é ©r r+ r r. r r r Útest_td_add_tdo s z/TestTimedeltaAdditionSubtraction.test_td_add_tdc C s@ t ddd}||tdd}t|t s*t|t ddks<td S )Nr r% r é r3 é ©r r r+ r r. r r r Útest_td_add_pytimedeltaw s z8TestTimedeltaAdditionSubtraction.test_td_add_pytimedeltac C sB t ddd}||t dd¡}t|t s,t|t ddks>td S )Nr r% r éüÿÿÿr" é r3 ©r r, Ú timedelta64r+ r r. r r r Útest_td_add_timedelta64~ s z8TestTimedeltaAdditionSubtraction.test_td_add_timedelta64c C sB t ddd}||t d¡}t|t s*t|t dddks>td S )Nr r% r rB ©r! Úhours©r r ÚHourr+ r r. r r r Útest_td_add_offset
s z3TestTimedeltaAdditionSubtraction.test_td_add_offsetc C s> t ddd}t ddd}|| }t|t s.t||ks:td S ©Nr r% r r r r; ©r r/ Úexpectedr r r r Útest_td_sub_td s
|
z/TestTimedeltaAdditionSubtraction.test_td_sub_tdc C sh t ddd}t ddd}|| ¡ }t|t s2t||ks>t| ¡ | }t|t sXt||ksdtd S rK )r Úto_pytimedeltar+ r rL r r r Útest_td_sub_pytimedelta s z8TestTimedeltaAdditionSubtraction.test_td_sub_pytimedeltac C sh t ddd}t ddd}|| ¡ }t|t s2t||ks>t| ¡ | }t|t sXt||ksdtd S rK )r Úto_timedelta64r+ r rL r r r Útest_td_sub_timedelta64 s z8TestTimedeltaAdditionSubtraction.test_td_sub_timedelta64c C s$ t ddd}|t }|tks td S ©Nr r% r ©r r r ©r r/ r r r r Útest_td_sub_nat¬ s z0TestTimedeltaAdditionSubtraction.test_td_sub_natc C sB t ddd}t d¡}|| }|tks*t|| }|tks>td S ©Nr r% r r ©r r, rD r r )r r/ Útd_natr r r r Útest_td_sub_td64_nat² s
|
z5TestTimedeltaAdditionSubtraction.test_td_sub_td64_natc C s@ t ddd}|t d¡ }t|t s(t|t dddks<td S )Nr r% r r éï ÚhrH rU r r r Útest_td_sub_offset¼ s z3TestTimedeltaAdditionSubtraction.test_td_sub_offsetc
|
C sº t ddd}d}ddt d¡t d¡fD ]}tjt|d || W 5 Q R X tjt|d || W 5 Q R X tjt|d || W 5 Q R X tjt|d || W 5 Q R X q(d S )Nr r% r úunsupported operand typeé ç @r1 )r r, Úint64Úfloat64r5 r6 Ú TypeError)r r/ r8 Úotherr r r Ú test_td_add_sub_numeric_raises s z?TestTimedeltaAdditionSubtraction.test_td_add_sub_numeric_raisesc C s t d}t dg¡}d}tjt|d |t dg¡ W 5 Q R X d ddg¡}tjt|d || W 5 Q R X d}tjt|d || W 5 Q R X d }tjt|d || W 5 Q R X d S )
|
Nú1 dayr z;unsupported operand type\(s\) for \+: 'Timedelta' and 'int'r1 ú|zEunsupported operand type\(s\) for \+: 'numpy.ndarray' and 'Timedelta'z;Concatenation operation is not implemented for NumPy arraysz:unsupported operand type\(s\) for -: 'Timedelta' and 'int'zDunsupported operand type\(s\) for -: 'numpy.ndarray' and 'Timedelta')r r, Úarrayr5 r6 rc Újoin©r r/ rd r8 r r r Útest_td_add_sub_int_ndarrayÏ s$ ùÿ z<TestTimedeltaAdditionSubtraction.test_td_add_sub_int_ndarrayc C s> t ddd}t| }|tks tt d¡| }|tks:td S rW )r r r r, r- rU r r r Útest_td_rsub_natë s
|
z1TestTimedeltaAdditionSubtraction.test_td_rsub_natc C s< t d¡tddd }t|ts$t|tdddks8td S )Nr r r% r iÿÿÿr\ )r rI r r+ r ©r r r r r Útest_td_rsub_offsetó s z4TestTimedeltaAdditionSubtraction.test_td_rsub_offsetc C sH t tdtdg¡}t tdtdg¡}|td }t ||¡ d S )Nú 20130101 9:01ú 20121230 9:02z 20121231 9:01z 20121229 9:02Ú1D©r, rh r r ÚtmÚassert_numpy_array_equal)r ÚarrÚexpÚresr r r Ú,test_td_sub_timedeltalike_object_dtype_arrayø s zMTestTimedeltaAdditionSubtraction.test_td_sub_timedeltalike_object_dtype_arrayc C sl t d}t |tdt dd¡g¡}t |td tdt dd¡td g¡}|td }t ||¡ d S )Nú2021-11-09 09:54:00rq r_ r\ Z0D)r r, rh r rD rs rt )r Únowru rv rw r r r Ú7test_td_sub_mixed_most_timedeltalike_object_dtype_arrayÿ s
|
ýÿ zXTestTimedeltaAdditionSubtraction.test_td_sub_mixed_most_timedeltalike_object_dtype_arrayc C sR t d}t |tdt dd¡g¡}d}tjt|d td| W 5 Q R X d S )Nry rq r_ r\ zAunsupported operand type\(s\) for \-: 'Timedelta' and 'Timestamp'r1 )r r, rh r rD r5 r6 rc )r rz ru r8 r r r Ú8test_td_rsub_mixed_most_timedeltalike_object_dtype_array s
|
zYTestTimedeltaAdditionSubtraction.test_td_rsub_mixed_most_timedeltalike_object_dtype_arrayc C sJ t tdtdg¡}t tdtdg¡}||td}t ||¡ d S )Nro rp z 20130102 9:01z 20121231 9:02rq rr )r r$ ru rv rw r r r Ú,test_td_add_timedeltalike_object_dtype_array s zMTestTimedeltaAdditionSubtraction.test_td_add_timedeltalike_object_dtype_arrayc C sR t d}t |tdg¡}t |td tdg¡}||td}t ||¡ d S )Nry rq Z2D)r r, rh r rs rt )r r$ rz ru rv rw r r r Ú2test_td_add_mixed_timedeltalike_object_dtype_array s
|
zSTestTimedeltaAdditionSubtraction.test_td_add_mixed_timedeltalike_object_dtype_arrayc C s t d}t | ¡ g¡}t t d ¡ g¡}|| }t ||¡ || }t ||¡ || }t ||d ¡ || }t ||d ¡ d S )Nrf ú2 Daysr ©r r, rh rQ rs rt ©r r/ rd rM r r r r Ú test_td_add_sub_td64_ndarray& s z=TestTimedeltaAdditionSubtraction.test_td_add_sub_td64_ndarrayc C sx t d}t dg¡j}t dg¡j}t || |¡ t || |¡ t dg¡j}t | | |¡ t || |¡ d S )Nrf z
|
2000-01-01z
|
2000-01-02z
|
1999-12-31)r ÚpdÚ to_datetimeÚvaluesrs rt )r r/ rd rM r r r Ú test_td_add_sub_dt64_ndarray6 s z=TestTimedeltaAdditionSubtraction.test_td_add_sub_dt64_ndarrayc C s° t d}t |j¡}|| }t|t s*t|d| ks:t|| }t|t sPt|d| ks`t|| }t|t svt|d| kst|| }t|t st|d| ks¬td S )Nrf r_ r )r r, rh Úasm8r+ r )r r/ rd r r r r Útest_td_add_sub_ndarray_0dB s z;TestTimedeltaAdditionSubtraction.test_td_add_sub_ndarray_0dN),Ú__name__Ú
|
__module__Ú __qualname__Ú__doc__r5 ÚmarkÚ parametrizer r r, rD r ZSecondr ZDayr# r) r* r Zraddr0 r9 r< r@ rE rJ rN rP rR rV rZ r] re rk rl rn rx r{ r| r} r~ r r r r r r r r
|
sh
|
|
|
ûþ
|
|
|
ûþ
|
|
|
|
|
|
|
|
|
|
|
r
|
c
|
@ s e Zd ZdZej dee dd¡e d¡g¡ej de
|
j e j g¡dd Zej deje d ¡edg¡ej de
|
j e j g¡d
|
d Zej de
|
j e j g¡d d Zdd Zdd Zdd Zdd Zdd Zdd Zej deje d ¡edg¡dd Zd d Zd d Z d d! Z d"d# Z d$d% Z d&d' Z d(d) Z!d*d+ Z"d,d- Z#d.d/ Z$d0d1 Z%d2d3 Z&d4d5 Z'd6d7 Z(d8d9 Z)d:d; Z*d<d= Z+d>d? Z,d@dA Z-dBdC Z.dDdE Z/dFdG Z0dHdI Z1dJdK Z2dLdM Z3dNdO Z4dPdQ Z5dRdS Z6dTdU Z7dVdW Z8dXdY Z9dZd[ Z:d\d] Z;d^d_ Z<d`da Z=dbdc Z>ddde Z?ej de
|
j e j e
|
j@e jAe jBg¡ej dfe CeDdgeDdhg¡e CeDdieEdjg¡g¡dkdl ZFdmS )nÚ#TestTimedeltaMultiplicationDivisionzà
|
Tests for Timedelta methods:
|
|
__mul__, __rmul__,
|
__div__, __rdiv__,
|
__truediv__, __rtruediv__,
|
__floordiv__, __rfloordiv__,
|
__mod__, __rmod__,
|
__divmod__, __rdivmod__
|
rY r r r$ c C s` t ddd}d dddg¡}d d| d | d
|
d g¡}tjt|d ||| W 5 Q R X d S ) Nr r% r rg znumpy.timedelta64ZNaTTyper z'unsupported operand type\(s\) for \*: 'z' and 'ú'z1ufunc '?multiply'? cannot use operands with typesr1 )r ri r5 r6 rc )r r$ rY r/ Ztypsr8 r r r Útest_td_mul_natf s þÿz3TestTimedeltaMultiplicationDivision.test_td_mul_natÚnanÚNaNc C s& t ddd}|||}|tks"td S rS rT )r r$ r r/ r r r r Útest_td_mul_nanw s
|
z3TestTimedeltaMultiplicationDivision.test_td_mul_nanc C sä t dd}||d}|t ddks&t||d}|t dddksDt||tjtksXt|d |jd |j ksrt|d
|
|jd
|
|j kstd }tjt|d ||t d dd W 5 Q R X tjt|d ||| W 5 Q R X d S )Né ©Úminutesr_ rB g ø?é é ©r r éÿÿÿÿg ð¿r^ r1 r& r )
|
r r r, r r Ú_valuer5 r6 rc r )r r$ r/ r r8 r r r Útest_td_mul_scalar s
|
|
|
z6TestTimedeltaMultiplicationDivision.test_td_mul_scalarc C sT t d}t dg¡}t t d ¡ g¡}|| }t ||¡ || }t ||¡ d S )Nrf r_ r r r r r r Útest_td_mul_numeric_ndarray s z?TestTimedeltaMultiplicationDivision.test_td_mul_numeric_ndarrayc C st t d}t d¡}|jdks tt d}|| }t|t ks@t||ksLt|| }t|t ksdt||ksptd S )Nrf r_ r ú2 days)r r, rh Úndimr Útype)r r/ rd rM rw r r r Ú test_td_mul_numeric_ndarray_0d£ s
|
zBTestTimedeltaMultiplicationDivision.test_td_mul_numeric_ndarray_0dc C sz t d}t t d ¡ g¡}dtj dtj d}tjt|d || W 5 Q R X tjt|d || W 5 Q R X d S )Nrf r z:ufunc '?multiply'? cannot use operands with types dtype\('zm8\[ns\]'\) and dtype\('z m8\[ns\]'\)r1 ) r r, rh rQ rs ZENDIANr5 r6 rc rj r r r Ú test_td_mul_td64_ndarray_invalid± s ÿzDTestTimedeltaMultiplicationDivision.test_td_mul_td64_ndarray_invalidc C sd t ddd}|t d¡ }|dks&t|| dks6t|t dd¡ dksNtt |t ¡s`td S ) Nr r% r r éð é< r\ r )r r rI r r, rD Úisnanr rU r r r Ú test_td_div_timedeltalike_scalarÁ s zDTestTimedeltaMultiplicationDivision.test_td_div_timedeltalike_scalarc C sÌ t d}|t dd¡ }||jd ks*t|t dd¡ }||jd ksLt|t dd¡ }||jksjtt d}|t dd¡ }|dkst|t dd¡ }|dksªt|t dd¡ }||jksÈtd S ) Nz1 days 2 hours 3 nsr r" l "R: r é Ê;r i m )r r, rD r r rU r r r Útest_td_div_td64_non_nanoÍ s z=TestTimedeltaMultiplicationDivision.test_td_div_td64_non_nanoc C s` t ddd}|d }t|t s"t|t ddks4t|d }t|t sJt|t ddks\td S )Nr r% r r_ é r3 r; rU r r r Útest_td_div_numeric_scalarà s z>TestTimedeltaMultiplicationDivision.test_td_div_numeric_scalarc C s8 t ddd}|| }|tks t|| }|tks4td S rS rT )r r r/ r r r r Útest_td_div_nanì s
|
|
z3TestTimedeltaMultiplicationDivision.test_td_div_nanc C sX t d}t t d ¡ g¡}t dg¡}|| }t ||¡ || }t ||d ¡ d S )Nrf r g à?r r r r r r Útest_td_div_td64_ndarrayý s z<TestTimedeltaMultiplicationDivision.test_td_div_td64_ndarrayc C s8 t d}t d¡}|| }t|t s(t||ks4td S )Nrf r )r r, rh r+ r )r r/ rd rw r r r Útest_td_div_ndarray_0d s
|
|
z:TestTimedeltaMultiplicationDivision.test_td_div_ndarray_0dc C sB t ddd}t d¡| }|dks&tt dd¡| dks>td S ) Nr r% r r gq?r¥ r\ g Ð?)r r rI r r, rD rU r r r Ú!test_td_rdiv_timedeltalike_scalar s zETestTimedeltaMultiplicationDivision.test_td_rdiv_timedeltalike_scalarc C s¬ t ddd}t| }t |¡s"td | }t |¡s8tt d¡| }t |¡sTtd}tjt|d t d¡| W 5 Q R X d}tjt|d tj
|
| W 5 Q R X d S )Nr r% r r zGunsupported operand type\(s\) for /: 'numpy.datetime64' and 'Timedelta'r1 ú<unsupported operand type\(s\) for /: 'float' and 'Timedelta') r r r, r¦ r rD r5 r6 rc r- r )r r/ r r8 r r r Útest_td_rdiv_na_scalar s z:TestTimedeltaMultiplicationDivision.test_td_rdiv_na_scalarc C sæ t ddd}tj|gtd}|| }tjdgtjd}t ||¡ t d g¡}|| }t tjg¡}t ||¡ tjtjgtd}d}tj t
|
|d || W 5 Q R X tjtjgtjd}d}tj t
|
|d || W 5 Q R X d S ) Nr r% r ©Údtyper r° r1 z$cannot use operands with types dtype) r r, rh Úobjectrb rs rt r r5 r6 rc )r r/ ru r rM r8 r r r Útest_td_rdiv_ndarray1 s" z8TestTimedeltaMultiplicationDivision.test_td_rdiv_ndarrayc C s, t ddd}t |j¡}|| dks(td S )Nr r% r r )r r, rh r r )r r/ ru r r r Útest_td_rdiv_ndarray_0dH s z;TestTimedeltaMultiplicationDivision.test_td_rdiv_ndarray_0dc C sZ t ddd}t ddd}|| dks(t| | ¡ dks>td| | ¡ dksVtd S )Nr r ©rG r r éþÿÿÿr_ )r r rO rQ ©r r/ Úscalarr r r Ú%test_td_floordiv_timedeltalike_scalarR s
|
zITestTimedeltaMultiplicationDivision.test_td_floordiv_timedeltalike_scalarc C sL t ddd}|tj tks tt |t ¡s0tt |t d¡ ¡sHtd S )Nr r r· r )r r, r r r r¦ rD ©r r/ r r r Ú test_td_floordiv_null_scalar[ s z@TestTimedeltaMultiplicationDivision.test_td_floordiv_null_scalarc C s< t ddd}|t d¡ dks"t|t d¡ dks8td S )Nr r r· r r_ é\ )r r rI r ZMinuter¼ r r r Útest_td_floordiv_offsetsc s z<TestTimedeltaMultiplicationDivision.test_td_floordiv_offsetsc C sL t ddd}d dddg¡}tjt|d |tjd d
|
d W 5 Q R X d S ) Nr r r· rg z.Invalid dtype datetime64\[D\] for __floordiv__z8'dtype' is an invalid keyword argument for this functionz5ufunc '?floor_divide'? cannot use operands with typesr1 ú
|
2016-01-01zdatetime64[us]r² )r ri r5 r6 rc r, r- ©r r/ r8 r r r Ú test_td_floordiv_invalid_scalari s ýÿzCTestTimedeltaMultiplicationDivision.test_td_floordiv_invalid_scalarc C s~ t ddd}t ddd}|d |ks(t|d |ks8t|t d¡ |ksNt|t d¡ |ksdt|t d¡ |ksztd S )Nr r r· r é r_ r` )r r r, rb Úint32Úuint8)r r/ rM r r r Ú test_td_floordiv_numeric_scalarw s zCTestTimedeltaMultiplicationDivision.test_td_floordiv_numeric_scalarc C s¨ t ddd}t ddd}|t | ¡ ¡ dks2td| t | ¡ g¡ }tjdgtjd}t ||¡ d| t | ¡ t d¡g¡ }t dtj g¡}t ||¡ d S )Nr r r· r r² r r )
|
r r, rh rQ r ra rs rt rD r )r r/ rº rw rM r r r Ú$test_td_floordiv_timedeltalike_array s zHTestTimedeltaMultiplicationDivision.test_td_floordiv_timedeltalike_arrayc C s: t ddd}tjdgtjd}|| }|jjdks6td S )Nr r r· r r² Úm)r r ÚSeriesr, ra r³ Úkindr )r r/ Úserrw r r r Ú test_td_floordiv_numeric_series s zCTestTimedeltaMultiplicationDivision.test_td_floordiv_numeric_seriesc C s` t ddd}t ddd}| |¡dks*t| | ¡ ¡dksBtd| | ¡ ¡dks\td S )Nr r· r r r¸ r_ r )r Ú __rfloordiv__r rO rQ r¹ r r r Ú&test_td_rfloordiv_timedeltalike_scalar s
|
zJTestTimedeltaMultiplicationDivision.test_td_rfloordiv_timedeltalike_scalarc C s> t ddd}t | t¡¡s tt | t d¡¡¡s:td S )Nr r· r )r r, r¦ rÍ r r rD r¼ r r r Ú test_td_rfloordiv_null_scalar« s zATestTimedeltaMultiplicationDivision.test_td_rfloordiv_null_scalarc C s t d¡tdd dks td S )Nr é r r_ )r rI r r ©r r r r Útest_td_rfloordiv_offsets² s z=TestTimedeltaMultiplicationDivision.test_td_rfloordiv_offsetsc C sT t ddd}t dd¡}| |¡tks*td}tjt|d || W 5 Q R X d S )Nr r· rÀ ÚuszHunsupported operand type\(s\) for //: 'numpy.datetime64' and 'Timedelta'r1 ) r r, r- rÍ ÚNotImplementedr r5 r6 rc )r r/ Zdt64r8 r r r Ú test_td_rfloordiv_invalid_scalar¶ s ÿzDTestTimedeltaMultiplicationDivision.test_td_rfloordiv_invalid_scalarc C s t ddd}| tj¡tks t| d¡tks2t| d¡tksDt| t d¡¡tks\t| t d¡¡tkstt| t d¡¡tkstd}t j
|
t |d t d¡| W 5 Q R X t j
|
t |d t d¡| W 5 Q R X t j
|
t |d t d¡| W 5 Q R X d S ) Nr r· g @r_ r` r= z9unsupported operand type\(s\) for //: '.*' and 'Timedeltar1 ) r rÍ r, r rÔ r rb rÅ rÄ r5 r6 rc rÁ r r r Ú test_td_rfloordiv_numeric_scalarÄ s zDTestTimedeltaMultiplicationDivision.test_td_rfloordiv_numeric_scalarc C s² t ddd}t ddd}| t | ¡ ¡¡dks4t| t d| ¡ g¡¡}tjdgtjd}t ||¡ t d| ¡ t d¡g¡}| |¡}t dtj
|
g¡}t ||¡ d S )Nr r· r r r² r r ) r rÍ r, rh rQ r ra rs rt rD r )r r/ rº rw rM ru r r r Ú%test_td_rfloordiv_timedeltalike_arrayØ s
|
zITestTimedeltaMultiplicationDivision.test_td_rfloordiv_timedeltalike_arrayc C sH t ddddg¡d }d}tjt|d |tdd d
|
W 5 Q R X d S ) NirPi isPiºtPi vPr¨ ú Invalid dtyper1 r r r )r, rh r5 r6 rc r )r Úintsr8 r r r Útest_td_rfloordiv_intarrayé s z>TestTimedeltaMultiplicationDivision.test_td_rfloordiv_intarrayc C s^ t ddd}tjdgtjd}| |¡}|tks4td}tj t
|
|d || W 5 Q R X d S )Nr r· r r² rØ r1 ) r r rÉ r, ra rÍ rÔ r r5 r6 rc )r r/ rË rw r8 r r r Ú test_td_rfloordiv_numeric_seriesñ s
|
zDTestTimedeltaMultiplicationDivision.test_td_rfloordiv_numeric_seriesc C s| t dd}|t dd }t|t s&t|t ddks8t|tdd }t|t sTt|t dksdt|t }|tksxtd S )Né% ©rG rB r r¥ r r )r r+ r r r rU r r r Útest_mod_timedeltalike s
|
z:TestTimedeltaMultiplicationDivision.test_mod_timedeltalikec C s* t dd}|t dd¡ }|tks&td S )NrÜ rÝ r r rX rU r r r Útest_mod_timedelta64_nat s
|
z<TestTimedeltaMultiplicationDivision.test_mod_timedelta64_natc C s> t dd}|t dd¡ }t|t s(t|t ddks:td S )NrÜ rÝ r_ r\ r rC rU r r r Útest_mod_timedelta64 s
|
z8TestTimedeltaMultiplicationDivision.test_mod_timedelta64c C s< t dd}|t d¡ }t|t s&t|t ddks8td S )NrÜ rÝ rª r_ rH rU r r r Útest_mod_offset s
|
z3TestTimedeltaMultiplicationDivision.test_mod_offsetc C s t dd}|d }t|t s t|t dks0t|d }t|t sFt|t dddksZt|td }t|t stt|t dddkstd S ) NrÜ rÝ r_ r g ¢mBr r: r )r r+ r ÚintrU r r r Útest_mod_numeric' s
|
z4TestTimedeltaMultiplicationDivision.test_mod_numericc C sZ t dd}d}tjt|d |td W 5 Q R X tjt|d |g W 5 Q R X d S )NrÜ rÝ r^ r1 r( )r r5 r6 rc r rÁ r r r Útest_mod_invalid8 s
|
z4TestTimedeltaMultiplicationDivision.test_mod_invalidc C s< t dd}tdd| }t|t s&t|t ddks8td S )Nr r r r r? rU r r r Útest_rmod_pytimedeltaB s
|
z9TestTimedeltaMultiplicationDivision.test_rmod_pytimedeltac C s> t dd}t dd¡| }t|t s(t|t ddks:td S )Nr r rª rÈ r_ rC rU r r r Útest_rmod_timedelta64J s
|
z9TestTimedeltaMultiplicationDivision.test_rmod_timedelta64c C s¬ t dd}d}tjt|d td| W 5 Q R X tjt|d d| W 5 Q R X tjt|d d| W 5 Q R X d}tjt|d t d d
|
g¡| W 5 Q R X d S ) Nr r zunsupported operandr1 r( é ç 0@úInvalid dtype inté é )r r5 r6 rc r r, rh rÁ r r r Útest_rmod_invalidQ s
|
z5TestTimedeltaMultiplicationDivision.test_rmod_invalidc C s t ddd}t|d}|d t dddks.tt|d t s@t|d t dd ksVt|s^tt|tj}|d tkszt|d tkstd S )
|
Nr_ rB rF g $DZåBr r r r rÝ )r Údivmodr r+ r, r r rU r r r Útest_divmod_numericf s
|
z7TestTimedeltaMultiplicationDivision.test_divmod_numericc C sÊ t ddd}t|tdd}|d dks,tt|d t s>t|d t ddksTtt|d}|d t ddksttt|d t st|d t dkstt|t}t |d ¡s¶t|d tksÆtd S ) Nr_ rB rF r r3 r rÝ é6 )r rí r r r+ r r, r¦ rU r r r Ú test_divmodt s
|
|
z/TestTimedeltaMultiplicationDivision.test_divmodc C sX t ddd}t|t d¡}|d dks,tt|d t s>t|d t dd ksTtd S )
|
Nr_ rB rF rA r éòÿÿÿr r¸ rÝ )r rí r rI r r+ rU r r r Útest_divmod_offset s
|
z6TestTimedeltaMultiplicationDivision.test_divmod_offsetc C s< t ddd}d}tjt|d t|td W 5 Q R X d S )Nr_ rB rF zAunsupported operand type\(s\) for //: 'Timedelta' and 'Timestamp'r1 r( )r r5 r6 rc rí r rÁ r r r Útest_divmod_invalid s z7TestTimedeltaMultiplicationDivision.test_divmod_invalidc C sT t tdddtdd}|d dks(tt|d ts:t|d tddksPtd S )Nr_ rB rF r r3 r rÝ )rí r r r r+ rm r r r Útest_rdivmod_pytimedelta s z<TestTimedeltaMultiplicationDivision.test_rdivmod_pytimedeltac C sR t t d¡tdd}|d dks&tt|d ts8t|d tddksNtd S )Nrï rA rÝ r rñ r r¸ )rí r rI r r r+ rm r r r Útest_rdivmod_offset s z7TestTimedeltaMultiplicationDivision.test_rdivmod_offsetc C s´ t dd}d}tjt|d ttd| W 5 Q R X tjt|d td| W 5 Q R X tjt|d td| W 5 Q R X d}tjt|d tt d d
|
g¡| W 5 Q R X d S ) Nr r r^ r1 r( rç rè ré rê rë )r r5 r6 rc rí r r, rh rÁ r r r Útest_rdivmod_invalid¤ s
|
z8TestTimedeltaMultiplicationDivision.test_rdivmod_invalidru ro rp ry rq c C s0 d}t jt|d ||td W 5 Q R X d S )Nz7unsupported operand type|cannot use operands with typesr1 rq )r5 r6 rc r )r r$ ru r8 r r r Ú(test_td_op_timedelta_timedeltalike_array¸ s zLTestTimedeltaMultiplicationDivision.test_td_op_timedelta_timedeltalike_arrayN)Gr r r r r5 r r r r, rD r) Úmulr Zrmulr r rb Úfloatr 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Õ rÖ r× rÚ rÛ rÞ rß rà rá rã rä rå ræ rì rî rð rò ró rô rõ rö ÚtruedivZrdivZrsubrh r r r÷ r r r r r W s ÿ
|
ýþ
|
|
|
|
ÿþþr c @ s^ e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z e
|
j dddg¡dd Z dS )ÚTestTimedeltaComparisonc C s¶ dD ]È}t j |¡j}t j |¡j}|tjk s2t|tjks@t|tjkrNt|tjkr\t|tjksjt|tjkrxt|tjkst|tjkst|tjk r¢t|tjkr°t|tjks¾t|tjkrtqdD ]Þ}t j |¡j}t j |¡j}|tjkst|tjkst|tjk r"t|tjkr2t|tjksBt|tjkrRt|tjk sbt|tjksrt|tjkrt|tjkrt|tjks¢t|tjkrÒtqÒd S )N)r rÓ )Úmsr )r Úmaxr4 Úminr r )r r ZtdmaxZtdminr r r Ú test_compare_pytimedelta_boundsÉ s< z7TestTimedeltaComparison.test_compare_pytimedelta_boundsc C sø t ddd}t |jd¡t |jd¡ }t|}|j|jks@t|j|jksPt||ks\t||krht||k rtt||krt||kst||kst|tdd d¡ }||ks¸t||krÄt||k sÐt||ksÜt||krèt||krôtd S )NiÿÉ;iQ r r" r r r )r r, rD r! r r r r4 )r ZpytdZtd64r/ Ztd2r r r Ú test_compare_pytimedelta_bounds2ñ s$ z8TestTimedeltaComparison.test_compare_pytimedelta_bounds2c C sÌ |}|d}|j }t|ts t||ks,t||kr8t||ksDt||ksPt||k r\t||krht|d| krxt|d| kst|d| kst|d| k s¨t|d| kr¸t|d| krÈtd S )Nr r_ )Údeltar+ r r )r Z tick_classesÚclsÚoffr/ r r r Útest_compare_tick s z)TestTimedeltaComparison.test_compare_tickc C s¬ t d}t d}tj||gtd}||k}tjddgtd}||k ¡ sLttj||g||ggtd}||k}tjddgddggtd}|j|jkst||k ¡ s¨td S )Nr z3 hoursr² FT)r r, rh r´ ÚboolÚallr Úshape)r r/ rd ru rw rM r r r Ú test_comparison_object_array s z4TestTimedeltaComparison.test_comparison_object_arrayc C sD t dt dg}t |¡}|d |k}t ddg¡}t ||¡ d S )Nz0 days 01:00:00r F)r r, rh rs rt )r Zperiodsru r rM r r r Ú test_compare_timedelta_ndarray/ s
|
|
z6TestTimedeltaComparison.test_compare_timedelta_ndarrayc C s t d¡ d¡}t|d }t jdddddgtd}||k}t ||¡ ||k}t ||¡ ||k}t || ¡ ||k}t || ¡ d S )Nrª ztimedelta64[ns]r FTr² )r, ZarangeZastyper rh r rs rt )r ru r/ rM r r r r Útest_compare_td64_ndarray7 s z1TestTimedeltaComparison.test_compare_td64_ndarrayc C s G dd d}t d}|dks"t|dks.t|| ks<t||ddksNt||ddk s`t||ddk rrt||ddkstd S )
|
z
|
Make sure non supported operations on Timedelta returns NonImplemented
|
and yields to other operand (GH#20829).
|
c @ s4 e Zd Zd ddddZdd Zdd Zd d
|
ZdS ) zGTestTimedeltaComparison.test_compare_custom_object.<locals>.CustomClassN)Úreturnc S s
|
|| _ d S ©N©Ú
|
cmp_result)r r r r r Ú__init__Q s zPTestTimedeltaComparison.test_compare_custom_object.<locals>.CustomClass.__init__c S s | j d krtS | j S d S r )r rÔ rÑ r r r Úgeneric_resultT s
|
zVTestTimedeltaComparison.test_compare_custom_object.<locals>.CustomClass.generic_resultc S s | ¡ S r ©r ©r rd r r r Ú__eq__Z s zNTestTimedeltaComparison.test_compare_custom_object.<locals>.CustomClass.__eq__c S s | ¡ S r r r r r r Ú__gt__] s zNTestTimedeltaComparison.test_compare_custom_object.<locals>.CustomClass.__gt__)N)r r r r r r r r r r r Ú CustomClassP s r Ú1sÚstringr Fr TN)r r )r r Útr r r Útest_compare_custom_objectJ s z2TestTimedeltaComparison.test_compare_custom_objectÚvalr r c C s t d}d}tjt|d ||k W 5 Q R X tjt|d ||k W 5 Q R X tjt|d ||k W 5 Q R X tjt|d ||k W 5 Q R X d S )Nr z>not supported between instances of 'Timedelta' and '(int|str)'r1 )r r5 r6 rc )r r r r8 r r r Útest_compare_unknown_typel s z1TestTimedeltaComparison.test_compare_unknown_typeN)r r r rÿ r r r r r
|
r r5 r r r r r r r rû È s ("rû c C sz G dd d} | }t d}| |¡tks.t| |¡tks@t| |¡tksRt| |¡tksdt| |¡tksvtd S )Nc @ s e Zd ZdS )z&test_ops_notimplemented.<locals>.OtherN)r r r r r r r ÚOther| s r rf )r Ú__add__rÔ r Ú__sub__Ú __truediv__Ú__mul__Ú __floordiv__)r rd r/ r r r Útest_ops_notimplemented{ s r" c
|
C s t d} | dfd| ffD ]x\}}d dddg¡}tjt|d || W 5 Q R X d}tjt|d ||k W 5 Q R X ||krt||kstqd S ) Nrf Úarg r^ z3can only concatenate str \(not "Timedelta"\) to strzmust be str, not Timedeltar1 z"not supported between instances of)r ri r5 r6 rc r )r/ ÚleftÚrightr8 r r r Útest_ops_error_str s ýÿ r& )r r r r) Únumpyr, r5 Z pandas.errorsr Zpandasr r r r r Zpandas._testingZ_testingrs Z pandas.corer r
|
r rû r" r& r r r r Ú<module> s* ? u 4
|
