U
|
¬ý°dµ: ã @ s d dl m Z mZ d dlZd dlZd dlmZmZ d dlZ d dlm
|
Z
|
m Z m Z m Z mZmZmZmZmZmZ d dlmZ d dlmZ d dlmZmZm Z dd Z d d
|
Z G d d d Z G d d dZ G dd dZ!G dd dZ"G dd dZ#G dd dZ$G dd dZ%dd Z&dS )é )ÚdatetimeÚ timedeltaN)ÚOutOfBoundsDatetimeÚPerformanceWarning)
|
Ú DataFrameÚ DatetimeIndexÚIndexÚNaTÚSeriesÚ TimedeltaÚTimedeltaIndexÚ TimestampÚoffsetsÚtimedelta_range)Ú PandasArray)Úassert_invalid_addsub_typeÚassert_invalid_comparisonÚget_upcast_boxc C s t | ¡}||kstdS )zT
|
Helper to check the dtype for a Series, Index, or single-column DataFrame.
|
N)ÚtmZ get_dtypeÚAssertionError)ÚobjZexpected_dtypeÚdtype© r ú_d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/arithmetic/test_timedelta64.pyÚ assert_dtype& s
|
r c C s8 | t kr|d }n"| tjtjfkr,|d }n|d }|S )Nr é é )r r Úto_arrayÚpdÚarray)ÚboxÚnamesÚexnamer r r Úget_expected_name/ s
|
|
r# c @ s^ e Zd Zdd Zej deddeddedd ¡ e
|
d¡g¡dd Z ej d d
|
d e d e d d ¡e d ¡ ¡ e d ¡ ¡ e d ¡ e d¡g¡dd Zej deede d¡e d¡ ej¡e d¡ e¡ej ddd dje ej ddd¡eej dddej ddd e¡ej ddddjej dddd e¡g
|
¡dd Z dd Z d S ) Ú#TestTimedelta64ArrayLikeComparisonsc C s| |}|t tjfkr|ntj}tddd}t | ¡ d ¡}t ||¡}||k}t ddddg¡}t ||¡}t ||¡ d S )NÚ2Hé ©Úperiodsr TF)
|
r r r ÚnpÚndarrayr Úto_numpyr Ú box_expectedÚ assert_equal)ÚselfÚbox_with_arrayr ÚxboxÚtdiÚotherÚresÚexpectedr r r Ú test_compare_timedelta64_zerodimD s zDTestTimedelta64ArrayLikeComparisons.test_compare_timedelta64_zerodimÚ td_scalarr ©Údaysé c C sp |}|t tjfkr|ntj}ttddtddg}t ||¡}||k}tddg}t ||¡}t ||¡ d S )Nr r7 r FT)
|
r r r r) r* r
|
r r r, r- )r. r/ r6 r r0 ÚserÚactualr4 r r r Ú!test_compare_timedeltalike_scalarR s zETestTimedelta64ArrayLikeComparisons.test_compare_timedeltalike_scalarÚinvalidl x
|
Ii Úaú
|
2021-01-01ÚUTCr& c C s, |}t ddd}t ||¡}t||| d S ©Nú1 daysé
|
r' )r r r, r )r. r/ r= r Úrngr r r r Ú test_td64_comparisons_invalidg s zATestTimedelta64ArrayLikeComparisons.test_td64_comparisons_invalidr2 rC z
|
1970-01-01©r( Útzr' z
|
1971-01-01ÚD©Úfreqr( c C s* t dddj}t ||¡}t||| d S rA )r Ú_datar r, r )r. r2 r/ rD r r r Ú"test_td64arr_cmp_arraylike_invalid} s zFTestTimedelta64ArrayLikeComparisons.test_td64arr_cmp_arraylike_invalidc C sø t dddj}t ddd|d tdg¡}||k}t d d d d
|
d g¡}t ||¡ ||k}t || ¡ 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 ) NrB é r' r r r é r? FTzDInvalid comparison between|Cannot compare type|not supported between©Úmatch)
|
r rK r) r r r Úassert_numpy_array_equalÚpytestÚraisesÚ TypeError)r. rD r2 Úresultr4 Úmsgr r r Ú test_td64arr_cmp_mixed_invalid s zBTestTimedelta64ArrayLikeComparisons.test_td64arr_cmp_mixed_invalidN) Ú__name__Ú
|
__module__Ú __qualname__r5 rR ÚmarkÚ parametrizer r Zto_timedelta64r ÚHourr< r ÚnowÚ to_datetime64Ú to_pydatetimeÚdater) r rE ÚlistÚrangeÚarangeÚastypeÚfloat32Úobjectr Ú
|
date_rangeZ period_rangerL rW r r r r r$ ? sN üþ
|
|
øþ
|
|
öþ
|
r$ c @ s® e Zd Zej ddeg¡dd Zej dedde e ddge
|
e
|
d d
|
¡e
|
d d
|
¡e
|
d ¡e
|
d ¡e
|
d d
|
¡e
|
d d ¡ e
|
dd
|
¡e
|
dd ¡ g¡g¡dd Z dd ZdS )Ú TestTimedelta64ArrayComparisonsr Nc C s, t tdttdg}t tttdg}|| }}|tkrP| t¡| t¡ }}||k}t dddg¡}t ||¡ ||k}t dddg¡}t ||¡ t dddg¡}t |tk|¡ t t|k|¡ t dddg¡}t |tk|¡ t t|k|¡ t dddg¡}t |tk |¡ t t|k|¡ d S )NrB ú3 daysFT) r r r rg re r) r r rQ )r. r ÚleftÚrightÚlhsÚrhsrU r4 r r r Ú test_comp_nat s(
|
z-TestTimedelta64ArrayComparisons.test_comp_natÚidx2z2 dayz1 day 00:00:02z5 days 00:00:03r rH Znatr ÚsrM rN c C s t dtdtddg}||k }t ddddddg¡}t ||¡ ||k}t ddddddg¡}t ||¡ ||k}t ddddddg¡}t ||¡ ||k}t ddddddg¡}t ||¡ ||k}t ddddddg¡}t ||¡ ||k}t ddddddg¡}t ||¡ d S )Nú1 dayz1 day 00:00:01z5 day 00:00:03TF)r r r) r r rQ )r. rp Zidx1rU r4 r r r Útest_comparisons_natÊ s6 úÿ z4TestTimedelta64ArrayComparisons.test_comparisons_natc C sb t ddd}||d k }t dddgdgd ¡}t ||¡ |t|k}||k}t ||¡ d S )NrB rC r' rN TFé )r r) r r rQ rb )r. rD rU r4 Úexpr r r Útest_comparisons_coverage s z9TestTimedelta64ArrayComparisons.test_comparisons_coverage)rX rY rZ rR r[ r\ rg ro r r r) r Ú timedelta64rs rv r r r r ri ª s(
|
ÿ
|
|
úÿüþ
|
%ri c @ s~ e Zd Zdd Zdd Zdd Zdd Zd d
|
Zej
|
d d d g¡dd Z dd Z dd Z dd Zdd Zdd Zdd Zd S ) Ú!TestTimedelta64ArithmeticUnsortedc C s t dddddgddd}|d t |d¡fD ]D}t|t s>tt ddd d
|
d gddd}t ||¡ |jdks,tq,|d t |d¡fD ]D}t|t stt d dd ddgddd}t ||¡ |jdkstq| t |¡fD ]D}t|t sîtt dddddgddd}t ||¡ |jdksÜtqÜt dddd dgddd}t
|
|t |¡fD ]J}t|t s`tt dd dd dgd dd}t ||¡ |jd ksLtqLd S )Nr% Z4HZ6HZ8HZ10HÚx©rJ Únamer Z12HZ16HZ20HZ1HZ3HZ5HÚHz-2Hz-4Hz-6Hz-8Hz-10Hz-1HZ0H) r r) ÚmultiplyÚ
|
isinstancer r Úassert_index_equalrJ ÚdivideÚnegativeÚabsÚabsolute)r. ÚidxrU ru r r r Útest_ufunc_coercions s4 ÿ z6TestTimedelta64ArithmeticUnsorted.test_ufunc_coercionsc C s¼ t dtdgdd}tjdddd}td}td}d }tjt|d
|
|| W 5 Q R X tjt|d
|
|| W 5 Q R X d }tjt|d
|
|| W 5 Q R X d }tjt|d
|
|| W 5 Q R X || }t d ddgdd}t
|
||¡ || }t d ddgdd}t
|
||¡ || }t d tdgdd}t j
|
||dd || }t d tdgdd}t j
|
||dd || }t dddgddd}t j
|
||dd || }t dtdgdd}t
|
||¡ d S )NrB ú2 daysÚfoo©r{ Ú20130101rN Úbar©r( r{ z0cannot subtract a datelike from a TimedeltaArrayrO z#unsupported operand type\(s\) for -z((bad|unsupported) operand type for unaryú0 daysú-1 daysú-2 daysF)Z check_namesÚ20121231Ú20130102rH rz Z20121230) r r r rh r r rR rS rT r r r )r. r1 ÚdtiÚtdÚdtrV rU r4 r r r Útest_subtraction_ops4 sJ ÿz6TestTimedelta64ArithmeticUnsorted.test_subtraction_opsc C sø t jddd}t ||¡}td}| ¡ }t jddd d¡}t ||¡}td d¡}td d¡}| ¡ }td} dd }
|
|| } td } |
|
| | || } td } |
|
| | || } td } |
|
| | d
|
} tj t
|
| d || W 5 Q R X d } tj t
|
| d || W 5 Q R X d } tj t
|
| d || W 5 Q R X d
|
} tj t
|
| d || W 5 Q R X tj t
|
| d || W 5 Q R X tj t
|
| d || W 5 Q R X d } tj t
|
| d || W 5 Q R X tj t
|
| d || W 5 Q R X || } t d ddg} t | |¡} t | | ¡ || } t d ddg} t | |¡} t | | ¡ || } t d ddg} t | |¡} t | | ¡ || } t d ddg} t | |¡} t | | ¡ | | } td } |
|
| | || } t dddgdd} t | |¡} t | | ¡ d S )Nr rN r' z
|
US/EasternZCETrB c S s | |ks t t| tst d S ©N)r r~ r ©rU r4 r r r Ú_checkp s zNTestTimedelta64ArithmeticUnsorted.test_subtraction_ops_with_tz.<locals>._checkr z<Cannot subtract tz-naive and tz-aware datetime-like objects.rO z6can't subtract offset-naive and offset-aware datetimesz%Cannot subtract tz-naive and tz-awarer r r r r ©rG )r rh r r, r r` Z tz_localizer rR rS rT r r- r )r. r/ r Útsr Zdti_tzZts_tzZts_tz2Zdt_tzr r rU r4 rV r r r Ú test_subtraction_ops_with_tzc s
|
|
|
|
z>TestTimedelta64ArithmeticUnsorted.test_subtraction_ops_with_tzc C s t dtdgdd}tjdddd}|| }t d td gdd}t ||¡ || }t dtd
|
gdd}t ||¡ || }td tdg}t ||¡ d S ) NrB r r r r rN r r r ú4 daysr )r r r rh r r r )r. r1 r rU r4 r r r Útest_dti_tdi_numeric_ops· s z:TestTimedelta64ArithmeticUnsorted.test_dti_tdi_numeric_opsc C sÜ t dtdgdd}tjdddd}td}td}|| }td td
|
gdd}t ||¡ || }td td
|
gdd}t ||¡ || }t dtd gdd}t ||¡ || }t dtd gdd}t ||¡ d }t j
|
t |d ||dd
W 5 Q R X t j
|
t |d |dd
| W 5 Q R X d}t j
|
t |d |t dddgtjd W 5 Q R X || }td tdg}t ||¡ || }td tdg}t ||¡ || }td }||ksºt|| }td }||ksØtd S )NrB r r r r rN r r r Z20130103rj z$cannot add indices of unequal lengthrO r r ú3Addition/subtraction of integers and integer-arraysr ©r Z20130105)r r r rh r r r r r rR rS Ú
|
ValueErrorrT r r) Úint64r )r. r1 r r r rU r4 rV r r r Útest_addition_opsÈ sH " z3TestTimedelta64ArithmeticUnsorted.test_addition_opsrJ rH ÚBc C sæ t jdd|d}|td }|td }| d¡}t ||¡ |dkrt jj d¡}|j |ksbt
|
|j |kspt
|
|j |ks¾t
|
n>|j t jj d¡kst
|
|j d ks¦t
|
|j t jj d¡ks¾t
|
|td }|td }t ||¡ d S )Nz1/1/2000é2 ©r( rJ r éÿÿÿÿÚinferrH ) r rh r Ú
|
_with_freqr r Ztseriesr ÚDayrJ r Z BusinessDay)r. rJ ÚindexZshiftedÚbackr4 rU r r r Útest_timedelta s
|
z0TestTimedelta64ArithmeticUnsorted.test_timedeltac C s¤ t dd¡}t|}|t d¡ }t|t d¡ }|t d¡ }t|t d¡ }|j|jksdt |
|
d ¡}t ||¡ |j|jkst |
|
d ¡}t ||¡ d S )NZ2013Z2014r i áõ) r rh r
|
r r] r r) rw rJ r r§ r r )r. rD rq Zresult1Zresult2Zresult3Zresult4r r r Ú test_timedelta_tick_arithmetic s
|
|
z@TestTimedelta64ArithmeticUnsorted.test_timedelta_tick_arithmeticc C s| t dtdg}|j}tjdddd}|| }|| }t ||¡ || }|| }t ||¡ || }|| }t ||¡ d S )NrB r z
|
1999-12-31rN rH r¤ )r r r r rh r r )r. r1 Ztdar rU r4 r r r Útest_tda_add_sub_index( s z8TestTimedelta64ArithmeticUnsorted.test_tda_add_sub_indexc C sx |}t jdd|d}| d ¡}|| }t ||¡}t ||¡}t t¡ || t¡ }W 5 Q R X t || t¡¡ d S )Nú
|
2016-01-01rN rF )
|
r rh r§ r r, Úassert_produces_warningr re rg r- ) r. r/ Útz_naive_fixturer r r1 r r2 rU r r r Ú test_tda_add_dt64_object_array; s
|
z@TestTimedelta64ArithmeticUnsorted.test_tda_add_dt64_object_arrayc C s` t dd}t dddd}t ||¡}t ||¡}|}||7 }t ||¡ |tk r\t ||¡ d S ©NrB ú10 daysz1 days 02:00:00z10 days 02:00:00rH ©rJ ©r r r, r- r ©r. Ú two_hoursr/ rD r4 Zorig_rngr r r Útest_tdi_iadd_timedeltalikeM s
|
z=TestTimedelta64ArithmeticUnsorted.test_tdi_iadd_timedeltalikec C s\ t dd}t dd}t ||¡}t ||¡}|}||8 }t ||¡ |tk rXt ||¡ d S ©NrB r³ z0 days 22:00:00z9 days 22:00:00rµ r¶ r r r Útest_tdi_isub_timedeltalike\ s
|
|
z=TestTimedelta64ArithmeticUnsorted.test_tdi_isub_timedeltalikec C sd t ddddd}|d|j }t ddddd}t ||¡ |jdksHt|d|j }t d dddd}t ||¡ |jdkst|d }t ddd
|
dd}t ||¡ |jd
|
ks²t|d }t d dd dd}t ||¡ |jd ksät| }t d dd dd}t ||¡ |jd kstt d dd dd}t|}tdd dd dgdd}t ||¡ |jd ks`td S )Nr rM Z2Dry )r( rJ r{ r r r r Z4DrB rH z-2Dr r )r rJ r r r r r )r. rD rU ru r r r Útest_tdi_ops_attributesm s: ÿ z9TestTimedelta64ArithmeticUnsorted.test_tdi_ops_attributesN)rX rY rZ r
r r r r¡ rR r[ r\ r« r¬ r r± r¸ rº r» r r r r rx s /T8
|
rx c @ s. e Zd Zej dddg¡dd Zdd ZdS ) ÚTestAddSubNaTMaskingÚstr_tsz
|
1950-01-01z
|
1980-01-01c C s t ddg¡}t ||¡}t|}|| ¡ | ¡ d¡| ¡ d¡g}|D ]8}|| }|tkrr|j d t
|
kst qJ|d t
|
ksJt qJd S )Nz24658 days 11:15:00r zdatetime64[ns]z datetime64[D])r r r ) r Ú to_timedeltar r, r r` r_ re r Úilocr r ) r. r/ r½ ZtdinatZtdobjr Z ts_variantsÚvariantr3 r r r Ú$test_tdarr_add_timestamp_nat_masking s üz9TestAddSubNaTMasking.test_tdarr_add_timestamp_nat_maskingc C sø t jtdd t dd¡td W 5 Q R X t jtdd tdt dd¡ W 5 Q R X tjd }d}t jt|d t dgd¡td W 5 Q R X t jt|d tdt dgd¡ W 5 Q R X t jt|d t |g¡t d W 5 Q R X t jt|d t d |g¡t d W 5 Q R X t jt|d& t |d d
|
g¡t d |d g¡ W 5 Q R X t
|
tg}t tg¡t d }t ||¡ t
|
d tg}t d tg¡t d }t ||¡ t
|
ttdg}t td d
|
g¡t d td g¡ }t ||¡ d S )NZ10155196800000000000rO iT rH Ú2000r zOverflow in int64 additionrB z5 daysú1 hoursz 7 secondsz4 hoursr z5 hours) rR rS r r r¾ r r Ú_valueÚ OverflowErrorr r r r )r. Z_NaTrV ru rU r r r Útest_tdi_add_overflow« s<
|
ÿÿ
|
ÿz*TestAddSubNaTMasking.test_tdi_add_overflowN)rX rY rZ rR r[ r\ rÁ rÆ r r r r r¼ s
|
r¼ c @ s e Zd Zdd Zdd Zdd Zdd Zej d e
|
e e j g¡d
|
d Zd d Zdd Zdd Zej dddddg¡ej dddg¡dd Zej dd d d e d ¡g¡d d! Zejj d"e d d d#g¡ed d d#ged d d#ged d d#gggd$d% d&d'd( Zd)d* Zd+d, Zd-d. Zd/d0 Z d1d2 Z ej d3e d4¡e g¡d5d6 Z d7d8 Z!d9d: Z"d;d< Z#d=d> Z$d?d@ Z%ej dAe jeeg¡dBdC Z&dDdE Z'dS )FÚ TestTimedeltaArraylikeAddSubOpsc C s< t tdg¡ d¡}|t }ttgdd}t ||¡ d S )Nú00:00:01zm8[s]r )r r¾ r
|
re r r Úassert_series_equal)r. r: rU r4 r r r Útest_sub_nat_retain_unit× s z8TestTimedeltaArraylikeAddSubOps.test_sub_nat_retain_unitc C sØ t tdg¡}t tdg¡}d}tjt|d t ttg¡ W 5 Q R X t ttgdd¡}tdg t j¡}tdg t j¡}tjt|d ttg t j¡ W 5 Q R X ttj g t j¡}t d¡}t d¡} t d¡}
|
|| } | | ksøt
|
| | } | |kst
|
|| } t | |¡ || } t | |¡ || } t | |¡ || } t | |¡ || } t | |¡ | | } t | |¡ ||
|
} t | |¡ |
|
| } t | |¡ ||
|
} t | |¡ |
|
| } t | |¡ d}tjt|d |t j W 5 Q R X tjt|d t j| W 5 Q R X tjt|d |t j W 5 Q R X tjt|d t j | W 5 Q R X |t } t | |¡ |t } t | |¡ || } t | |¡ || } t | |¡ || } t | |¡ || } t | |¡ || } t | |¡ || } t | |¡ || } t | |¡ || } t | |¡ ||
|
} t | |¡ ||
|
} t | |¡ d }tjt|d |t j W 5 Q R X tjt|d |t j W 5 Q R X |t } t | |¡ |t } t | |¡ d S )
|
NrÈ z00:00:02z?dtype datetime64\[ns\] cannot be converted to timedelta64\[ns\]rO úm8[ns]r r úunsupported operand typez8cannot subtract a datelike from|unsupported operand type)r r¾ r
|
rR rS rT r r ÚapplyrÄ r r rÉ r) ÚnanZassert_frame_equal) r. Ús1Ús2rV ZsnZdf1Zdf2ÚdfnZscalar1Zscalar2Z timedelta_NaTr; r r r Ú&test_timedelta_ops_with_missing_valuesà s
|
|
|
|
|
zFTestTimedeltaArraylikeAddSubOps.test_timedelta_ops_with_missing_valuesc C s t jdddd}t jdddd}t|t| }td|j d¡ d¡}t ||¡ |jdksbtt d |i}td
|
d t
|
dD }|jdkst|d |d ¡ }|jdks´t|d | }|jd ksÎt|d ¡ }t |tsèt|d |d ¡ } | jdks t| |d }td tdtdg}
|
t|
|
d d} t || ¡ |d tddd }tdd t
|
dD d d} t || ¡ |jdksttddddd} |d | } | jdks¼t| | } t |d | ¡ tdd}|d | } | | } t | |d ¡ | jd ksttddd}|d | } | | } t |d | ¡ | jd ksRt|d t tddd¡ }|d t tddd¡7 < |d |kstd S )Nz2012-1-1rN rH r¤ z2012-1-2g ÀS$¥ÓBr útimedelta64[ns]ÚAc S s g | ]}t |d qS )r7 ©r ©Ú.0Úir r r Ú
|
<listcomp>T s zNTestTimedeltaArraylikeAddSubOps.test_operators_timedelta64.<locals>.<listcomp>zM8[ns]Z20111230Z20120101Z20120103r iÑ r c S s g | ]}t d | dqS )i± r7 rÕ rÖ r r r rÙ m s rË r& r7 rM ©ÚminutesÚsecondsr )r rh r
|
r© re r rÉ r r r rc ÚshiftÚmaxr~ r r r r) rw )r. Zv1Zv2ÚrsZxpZdfr rU ZmaxaZresultbÚvaluesr4 ÚdZresultaÚvaluer r r Útest_operators_timedelta64J sV
|
z:TestTimedeltaArraylikeAddSubOps.test_operators_timedelta64c C s t ttdg}t ttgdd}t tgdd}t |t |¡ t t | |¡ t || |¡ t | | |¡ t |t |¡ t t| |¡ t || |¡ t || |¡ t |t |¡ t t| |¡ t || |¡ t || |¡ t |t |¡ t t| |¡ t || |¡ t || |¡ t |d |¡ t d| |¡ t |d |¡ t d| |¡ t |d t ttdg¡ t d| t ttdg¡ t |tj |¡ t tj| |¡ t |d t ttd g¡ t |d
|
t ttd g¡ t |tj |¡ d S ) NÚ1srÓ r ç ð?r ç ø?z1.5sr z0.5sç @)r
|
r r r rÉ r) rÎ )r. Ztimedelta_seriesZnat_series_dtype_timedeltaZsingle_nat_dtype_timedeltar r r Útest_timedelta64_ops_nat s ÿ ÿ ÿ ÿþþ ÿ ÿ ÿ ÿþþ ÿ ÿ z8TestTimedeltaArraylikeAddSubOps.test_timedelta64_ops_natÚclsc C s |}t d|d}|tkr"| ¡ }n$|tjkrB|d k r8d S | ¡ }n|}tddd}tjdd|d}t
|
||¡} t
|
||¡}t || |¡ t | | |¡ tjddd |d
|
}
|
t
|
|
|
|¡}
|
t || |
|
¡ t || |
|
¡ d } t j t| d | | W 5 Q R X d S ) Nz
|
2012-01-01r rr rN r' z
|
2012-01-02rF z
|
2011-12-31z-1D)r( rJ rG zcannot subtract a datelikerO )r r r` r) Ú
|
datetime64r_ r r rh r r, r- rR rS rT ) r. ré r/ r° rG Z dt_scalarr r1 r4 ÚtdarrZ expected2rV r r r Ú(test_td64arr_add_sub_datetimelike_scalará s,
|
|
|
zHTestTimedeltaArraylikeAddSubOps.test_td64arr_add_sub_datetimelike_scalarc C s` t d¡}tddd}tdddg}t ||¡}t ||¡}t || |¡ t || |¡ d S )Nr rr rN r' )r) rê r r r r, r- )r. r/ r2 r1 r4 Útdserr r r Ú test_td64arr_add_datetime64_nat s
|
z?TestTimedeltaArraylikeAddSubOps.test_td64arr_add_datetime64_natc C s t jddd}tdgd }|j}t|| }t ||¡}t ||¡}d}tjt |d || W 5 Q R X || }t
|
||¡ d S )Nr® rN r' ú-1 Dayz cannot subtract a datelike fromrO ) r rh r rà r r r, rR rS rT r- )r. r/ r r1 Údtarrr4 rV rU r r r Útest_td64arr_sub_dt64_array s z;TestTimedeltaArraylikeAddSubOps.test_td64arr_sub_dt64_arrayc C sr t jddd}tdgd }|j}t|| }t ||¡}t ||¡}|| }t ||¡ || }t ||¡ d S )Nr® rN r' rï )r rh r rà r r r, r- )r. r/ r r1 rð r4 rU r r r Útest_td64arr_add_dt64_array# s z;TestTimedeltaArraylikeAddSubOps.test_td64arr_add_dt64_arrayÚpi_freqrH ÚWÚQr| Útdi_freqNc
|
C s t ddg|d}td| }| |¡}|d }t ||¡}t ||¡}d} tjt| d || W 5 Q R X tjt| d || W 5 Q R X d S )Nrà z2 hoursr´ z2018-03-07 17:16:40r z(cannot subtract|unsupported operand typerO )r r Z to_periodr r, rR rS rT )
|
r. r/ Zbox_with_array2rö ró r1 r ÚpiZperrV r r r Útest_td64arr_sub_periodlike4 s
|
z;TestTimedeltaArraylikeAddSubOps.test_td64arr_sub_periodliker2 r> r ræ r c C s, t dddgdd}t ||¡}t|| d S ©Nú59 Daysr rË r )r
|
r r, r )r. r/ r2 rí rë r r r Ú*test_td64arr_addsub_numeric_scalar_invalidI s zJTestTimedeltaArraylikeAddSubOps.test_td64arr_addsub_numeric_scalar_invalidÚvecrN c C s
|
t | jS r ©ÚtyperX ©ry r r r Ú<lambda>c ó z(TestTimedeltaArraylikeAddSubOps.<lambda>©Úidsc C s6 t dddgdd}t ||¡}| |¡}t|| d S rù )r
|
r r, re r )r. r/ rü Úany_real_numpy_dtyperí rë Úvectorr r r Ú'test_td64arr_addsub_numeric_arr_invalid[ s
|
zGTestTimedeltaArraylikeAddSubOps.test_td64arr_addsub_numeric_arr_invalidc C sr t dddd}t ||¡}d}t||| tjt|d ||7 }W 5 Q R X tjt|d ||8 }W 5 Q R X d S )Nú1 days 09:00:00r| rC rI ú Addition/subtraction of integersrO )r r r, r rR rS rT )r. r/ ÚonerD rë rV r r r Útest_td64arr_add_sub_intn s z8TestTimedeltaArraylikeAddSubOps.test_td64arr_add_sub_intc C sX |}|t jkrtjn|}tdddd}t ||¡}t dddg|¡}d}t||| d S )Nr r| rN rI r& r r )r r r) r* r r r, r )r. r/ r r0 rD rë r2 rV r r r Ú"test_td64arr_add_sub_integer_array| s zBTestTimedeltaArraylikeAddSubOps.test_td64arr_add_sub_integer_arrayc C sX |}|t jkrtjn|}tdddg}t ||¡}t dddg|¡}d}t||| d S )Nú1 Dayr z3 Hoursé r¥ é r )r r r) r* r r r, r )r. r/ r r0 r1 rë r2 rV r r r Ú)test_td64arr_addsub_integer_array_no_freq s zITestTimedeltaArraylikeAddSubOps.test_td64arr_addsub_integer_array_no_freqc C s¢ |}t jddd}|| d¡ }|j}d| }t ||¡}t ||¡}|| }t ||¡ || }t ||¡ d| }|| }t ||¡ || }t ||¡ d S )Nr® rN r' r r r )r rh rÝ rà r r, r- ) r. r/ r r r1 rë r4 rU Ú expected_subr r r Ú test_td64arr_add_sub_td64_array s z?TestTimedeltaArraylikeAddSubOps.test_td64arr_add_sub_td64_arrayc C s> |}t ||}tddg|d d}|tjtjfkr<t |¡n|}ttddtddg|d d}ttddtddd g|d}t ||¡}t ||¡}|| }t
|
||¡ t |d
|
|| }t
|
||¡ t |d
|
ttd dtdd d g|d}t ||¡}|| }t
|
||¡ t |d
|
|| }t
|
|| ¡ t |d
|
d S ) Nr rr r r rN )Úhoursr& r )r8 r rÓ éýÿÿÿéüÿÿÿ) r# r r r r r r) r
|
r r, r- r ) r. r/ r! r r" r1 r: r4 rU r r r Útest_td64arr_add_sub_tdi® s2
|
|
|
ÿ
|
z8TestTimedeltaArraylikeAddSubOps.test_td64arr_add_sub_tdiÚtdnatr c C s |}t ttdg}t dgd }t ||¡}t ||¡}|| }t ||¡ || }t ||¡ || }t ||¡ || }t ||¡ d S )Nrä r r )r r r r r, r- )r. r/ r r r1 r4 r rU r r r Ú test_td64arr_add_sub_td64_natÑ s z=TestTimedeltaArraylikeAddSubOps.test_td64arr_add_sub_td64_natc C s` |}t dd}t dddd}t ||¡}t ||¡}|| }t ||¡ || }t ||¡ d S r² ©r r r, r- ©r. r· r/ r rD r4 rU r r r Ú test_td64arr_add_timedeltalikeä s
|
z>TestTimedeltaArraylikeAddSubOps.test_td64arr_add_timedeltalikec C s^ |}t dd}t dd}t ||¡}t ||¡}|| }t ||¡ || }t || ¡ d S r¹ r r r r r Ú test_td64arr_sub_timedeltalikeó s
|
|
z>TestTimedeltaArraylikeAddSubOps.test_td64arr_sub_timedeltalikec
|
sj |}t ||}tddg|d dttjddtjddg|d d |tjtj fkr`t
|
¡n t fdd t t D d
|
|d }t fd d t t D d
|
|d }t |¡t ||¡jtd d}t ||¡jtd d}t t¡ }W 5 Q R X t ||¡ t t¡ }W 5 Q R X t ||¡ t t¡ } W 5 Q R X t | |¡ d S )Nú1 days 00:00:00ú3 days 04:00:00r r r ©Únéþÿÿÿc s g | ]}| | qS r r ©r× r ©r2 r1 r r rÙ s zUTestTimedeltaArraylikeAddSubOps.test_td64arr_add_sub_offset_index.<locals>.<listcomp>r¦ rz c s g | ]}| | qS r r r! r" r r rÙ s F)Úcopy)r# r r r r] ÚMinuter r r r r) rc Úlenr, re rg r¯ r r- )
|
r. r! r/ r r" r4 r r3 Úres2Úres_subr r" r Ú!test_td64arr_add_sub_offset_index s6
|
$ ÿ ÿ zATestTimedeltaArraylikeAddSubOps.test_td64arr_add_sub_offset_indexc s( |}t ddgt tjddtjddg¡ t fddttD dd }t fd
|
dttD dd }t |¡t ||¡
|
t ¡}t t ¡ }W 5 Q R X t ||¡ t t ¡ }W 5 Q R X t ||¡ t ||¡
|
t ¡}t t ¡ }W 5 Q R X t ||¡ d S ) Nr r r r r c s g | ]}| | qS r r r! r" r r rÙ - s zUTestTimedeltaArraylikeAddSubOps.test_td64arr_add_sub_offset_array.<locals>.<listcomp>r¦ r´ c s g | ]}| | qS r r r! r" r r rÙ 0 s )r r) r r r] r$ rc r% r r, re rg r¯ r r- )r. r/ r r4 r r3 r& r' r r" r Ú!test_td64arr_add_sub_offset_array& s. ÿ ÿ zATestTimedeltaArraylikeAddSubOps.test_td64arr_add_sub_offset_arrayc s^ |}|t tjtjfkrtn|}t||}tddg|d dttj ddtj
|
ddg|d d t fdd t t D |t d
|
}t |¡}t ||¡ t ¡}t t¡ | }W 5 Q R X t ||¡ t t¡ | } W 5 Q R X t | |¡ t fd d t t D |t d
|
}
|
t |
|
|¡ t ¡}
|
t t¡ | } W 5 Q R X t | |
|
¡ d S ) Nr r r r r r r c s g | ]}| | qS r r r! r" r r rÙ M s zSTestTimedeltaArraylikeAddSubOps.test_td64arr_with_offset_series.<locals>.<listcomp>)r{ r c s g | ]}| | qS r r r! r" r r rÙ [ s )r r r r r r
|
r# r r r] r$ rc r% rg r, re r¯ r r- ) r. r! r/ r Zbox2r" Z expected_addr r3 r& r Zres3r r" r Ú test_td64arr_with_offset_seriesC s6
|
$ ÿ ÿ z?TestTimedeltaArraylikeAddSubOps.test_td64arr_with_offset_seriesÚoboxc
|
C s t ddg}t ||¡}|t ¡ tjddg}d}tjt|d$ t t
|
¡ || W 5 Q R X W 5 Q R X tjt|d$ t t
|
¡ || W 5 Q R X W 5 Q R X tjt|d$ t t
|
¡ || W 5 Q R X W 5 Q R X tjt|d$ t t
|
¡ || W 5 Q R X W 5 Q R X d S )Nr r r r z/has incorrect type|cannot add the type MonthEndrO ) r r r, r ZMonthEndr¨ rR rS rT r¯ r )r. r+ r/ r1 ZanchoredrV r r r Ú-test_td64arr_addsub_anchored_offset_arraylikec s zMTestTimedeltaArraylikeAddSubOps.test_td64arr_addsub_anchored_offset_arraylikec
|
|
C sB |}|t jkrtjn|}tdddd}t ||¡}t tddt d¡t
|
dg¡}t t ¡ || }W 5 Q R X t tddtd dt
|
d
|
g}t ||¡ t¡}t ||¡ d } tjt| d $ t t ¡ || W 5 Q R X W 5 Q R X t t ¡ || }W 5 Q R X t td td t
|
dg}t ||¡ t¡}t ||¡ d S )Nrr rN rH r¤ r r7 r z
|
2000-01-04r& z
|
2000-01-07z3unsupported operand type|cannot subtract a datelikerO r z
|
2000-01-01)r r r) r* r r r, r r r¨ r r¯ r r re rg r- rR rS rT )
|
r. r/ r r0 r1 rë r2 rU r4 rV r r r Ú!test_td64arr_add_sub_object_array~ s* ÿ zATestTimedeltaArraylikeAddSubOps.test_td64arr_add_sub_object_array)(rX rY rZ rÊ rÒ rã rè rR r[ r\ r r r) rê rì rî rñ rò rø r rû r r
|
r r r
|
r r r r rw r r r r r( r) r* r, r- r r r r rÇ Ô s\ jBU
|
# úþ
|
üø
|
|
#
|
|
rÇ c @ sh e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Ze j
|
j d e dd¡ee dd¡e jeedde jeedde je dd¡gdd ddd Zdd Zdd Zdd Zdd Ze j
|
d dd d g¡e j
|
d d!d"d d#d$d%d&g¡d'd( Zd)d* Zd+d, Z d-d. Z d/d0 Z d1d2 Z e j
|
d3¡d4d5 Z!d6d7 Z"d8d9 Z#d:d; Z$d<d= Z%d>d? Z&d@dA Z'dBdC Z(dDdE Z)e j
|
dFdGdHe *dG¡e *dH¡g¡dIdJ Z+e j
|
dFdGdHe *dG¡e *dH¡g¡dKdL Z,e j
|
j dMe *dNdOdPg¡edNdOdPge-dNdOdPggdQd ddRdS Z.e j
|
j dMe *dNdOdPg¡edNdOdPge-dNdOdPggdTd ddUdV Z/dWdX Z0dYdZ Z1d[d\ Z2d]S )^Ú TestTimedeltaArraylikeMulDivOpsc C sJ t tjddd}t ||¡}|d }t ||¡ d| }t ||¡ d S )NrM r r r )r r) rd r r, r- ©r. r/ r rU r r r Útest_td64arr_mul_int¥ s z4TestTimedeltaArraylikeMulDivOps.test_td64arr_mul_intc C sD t dddd}t ||¡}d}tjt|d || W 5 Q R X d S )NrB r³ r r z@argument must be an integer|cannot use operands with types dtyperO )r r r, rR rS rT )r. r· r/ rD rV r r r Ú%test_td64arr_mul_tdlike_scalar_raises¯ s
|
zETestTimedeltaArraylikeMulDivOps.test_td64arr_mul_tdlike_scalar_raisesc C s\ t jddd}t|}t|d }t ||¡}t ||¡}|t jddd }t ||¡ d S )NrM r r )r) rd r r r, r r- ©r. r/ Zrng5r r4 rU r r r Ú test_tdi_mul_int_array_zerodim¶ s z>TestTimedeltaArraylikeMulDivOps.test_tdi_mul_int_array_zerodimc C sR t jddd}t|}t|d }t ||¡}t ||¡}|| }t ||¡ d S ©NrM r r r )r) rd r r r, r- r2 r r r Útest_tdi_mul_int_arrayÁ s z6TestTimedeltaArraylikeMulDivOps.test_tdi_mul_int_arrayc C s |}|t tjtjfkrtn|}ttjddd}ttjdddd }t ||¡}t ||¡}|ttjddd }t
|
||¡ d S r4 © r r r r r r
|
r r) rd r, r- )r. r/ r r0 r r4 rU r r r Útest_tdi_mul_int_seriesÌ s z7TestTimedeltaArraylikeMulDivOps.test_tdi_mul_int_seriesc C s |}|t tjtjfkrtn|}ttjddd}t ||¡}tjddd}t||d }t ||¡}|t|d }t
|
||¡ d S )NrM r r Úfloat64rå r6 )r. r/ r r0 r Zrng5fr4 rU r r r Útest_tdi_mul_float_seriesÙ s z9TestTimedeltaArraylikeMulDivOps.test_tdi_mul_float_seriesr2 r é c C s
|
t | jS r rý rÿ r r r r ñ r z(TestTimedeltaArraylikeMulDivOps.<lambda>r c C sp |}t dgd }tdd d ¡}t ||¡}t||}t ||¡}|| }t ||¡ || }t ||¡ d S )Nr rC rB r³ )r r r§ r r, r r- ) r. r2 r/ r r1 r4 r0 rU Zcommuter r r Útest_tdi_rmul_arraylikeè s
|
z7TestTimedeltaArraylikeMulDivOps.test_tdi_rmul_arraylikec C s t dddd}t ||¡}tjtdd |t W 5 Q R X tjtdd t| W 5 Q R X t dd ¡}d
|
d d d g¡}tjt|d || W 5 Q R X tjt|d || W 5 Q R X d S )NrB r³ r r rÌ rO zCannot divide NaTType byr Únsú|z0ufunc '(true_divide|divide)' cannot use operandsz cannot perform __r?truediv__z*Cannot divide datetime64 by TimedeltaArray)
|
r r r, rR rS rT r r) rê Újoin)r. r/ rD Zdt64natrV r r r Ú test_td64arr_div_nat_invalid s" üÿz<TestTimedeltaArraylikeMulDivOps.test_td64arr_div_nat_invalidc C s |}|t jkrtjn|}tdd}t ||¡}t d¡}t tjgd ¡}t ||¡}|| }t ||¡ || }t ||¡ d S )NrB r³ r rC )
|
r r r) r* r r r, rw rÎ r- )r. r/ r r0 rD r2 r4 rU r r r Útest_td64arr_div_td64nat s
|
|
z8TestTimedeltaArraylikeMulDivOps.test_td64arr_div_td64natc C sX t tjddd}t ||¡}|d }t ||¡ tjtdd d| W 5 Q R X d S )NrM r r r ú Cannot dividerO © r r) rd r r, r- rR rS rT r/ r r r Útest_td64arr_div_int1 s z4TestTimedeltaArraylikeMulDivOps.test_td64arr_div_intc C s |}|t jkrtjn|}tdddd}tt d¡d d tjdd}t ||¡}t ||¡}|| }t
|
||¡ || }d| }t
|
||¡ d S ) NrB r³ r r rC r é ©r r{ ) r r r) r* r r rd r8 r r, r- ©r. r· r/ r r0 rD r4 rU r r r Ú test_td64arr_div_tdlike_scalar< s z>TestTimedeltaArraylikeMulDivOps.test_td64arr_div_tdlike_scalarÚmrN rC ÚunitrH Úhrq ÚmsÚusr< c
|
sÊ |}|t jkrtjn|}ttddgd }tj|d< |}t ||¡}t fdd|D }t ||¡}|t ¡ } t
|
| |¡ t fdd|D }t ||¡}t ¡| } t
|
| |¡ d S )Né; r7 rN r c s g | ]}|t ¡ qS r )r) rw ©r× ry ©rH rI r r rÙ Z s zPTestTimedeltaArraylikeMulDivOps.test_td64arr_div_td64_scalar.<locals>.<listcomp>c s g | ]}t t ¡| qS r )r r) rw rN rO r r rÙ ` s ) r r r) r* r
|
r rÎ r r, rw r- )
|
r. rH rI r/ r r0 r: Zflatr4 rU r rO r Ú test_td64arr_div_td64_scalarN s
|
z<TestTimedeltaArraylikeMulDivOps.test_td64arr_div_td64_scalarc C s |}|t jkrtjn|}tdtdgdd}tdtjdgtjdd}t
|
||¡}t
|
||¡}|| }t ||¡ || }d| }t ||¡ d S ) NrB r r r rD r9 rE r ) r r r) r* r r r rÎ r8 r r, r- rF r r r Ú'test_td64arr_div_tdlike_scalar_with_nate s zGTestTimedeltaArraylikeMulDivOps.test_td64arr_div_tdlike_scalar_with_natc C s> |}|t jkrtjn|}tdtdg}tdtjdgtjd}t
|
||¡}t
|
||¡}tjdddgdd}|| }t ||¡ |t
|
||¡ }t ||¡ || t ¡ }t || t ¡¡ |t| }t ||¡ d | }|| }t ||¡ t
|
||¡| }t ||¡ | t ¡| }t ||¡ t|| }t ||¡ d S )
|
NrB r rD r9 r r r& zm8[h]r )r r r) r* r r r rÎ r8 r r, r- re rg rb )r. r/ r r0 rD r4 r2 rU r r r Ú test_td64arr_div_td64_ndarrayv s0 z=TestTimedeltaArraylikeMulDivOps.test_td64arr_div_td64_ndarrayc C s¢ t dtdg}ddddg}t ||¡}d}||d d
fD ]b}|t |¡t|fD ]H}tjt |d || W 5 Q R X tjt |d || W 5 Q R X qRq:d S ) NrB r r r rN r& z0Cannot divide vectors|Unable to coerce to SeriesrO )
|
r r r r, r) r r rR rS r )r. r/ rD Z
|
mismatchedrV r r2 r r r Ú test_tdarr_div_length_mismatch s z>TestTimedeltaArraylikeMulDivOps.test_tdarr_div_length_mismatchc C sD t ddd dt¡}tj||dd}tj|d dd |d
|
gtd }tj||dd}|| }td t d d¡|d dgtd }tj||dd}t
|
|t r| ¡ }t ||¡ |tkrÆt
|
|jd tj sÆt|| }tdt d d¡|d dgtd }tj||dd}t
|
|t r| ¡ }t ||¡ |tkr@t
|
|jd tj s@td S )Nr rN r' r F©Z transposer ræ rç r r rå r r< )r r )r Úinsertr r r, r) r rg r rw r~ r r+ r- r r¿ r )r. r/ Úorigr1 r2 r3 r4 r r r Ú!test_td64_div_object_mixed_resultª s(
|
|
zATestTimedeltaArraylikeMulDivOps.test_td64_div_object_mixed_resultc C s |}|t jkrtjn|}tdddgdd}tddd gdd}t ||¡}t ||¡}tjddtjgtjd}t ||¡}|t kr|r|ddg
|
d ¡|ddg< tj t |t jkd
|
d || }W 5 Q R X t ||¡ tj t |t jkd
|
d t |¡| }W 5 Q R X t ||¡ d S ) Niè izd é rÓ r rå r r r F)Zcheck_stacklevel)r r r) r* r
|
r r, rÎ r8 r re Zmaybe_produces_warningÚRuntimeWarningr- Zasarray) r. r/ Zusing_array_managerr r0 rk rl r4 rU r r r Ú&test_td64arr_floordiv_td64arr_with_natÌ s0 ÿ ÿzFTestTimedeltaArraylikeMulDivOps.test_td64arr_floordiv_td64arr_with_natz/ignore:invalid value encountered:RuntimeWarningc C sÄ |}|t jkrtjn|}td}t||tgdd}tj||dd}tddtj g}tj||dd}|| }t
|
||¡ tddtj g}tj||dd}|| }t
|
||¡ | |¡}t
|
||¡ d S )NZ5m3srË r FrT r r ) r r r) r* r r
|
r r r, rÎ r- Ú __rfloordiv__) r. r/ Ú scalar_tdr r0 r Útd1r4 rU r r r Ú test_td64arr_floordiv_tdscalarï s
|
z>TestTimedeltaArraylikeMulDivOps.test_td64arr_floordiv_tdscalarc C s\ t tjddd}t ||¡}|d }t ||¡ d}tjt|d d| W 5 Q R X d S )NrM r r r z@floor_divide cannot use operands|Cannot divide int by Timedelta*rO rB )r. r/ r rU Úpatternr r r Útest_td64arr_floordiv_int
|
s z9TestTimedeltaArraylikeMulDivOps.test_td64arr_floordiv_intc C s¶ t dd}t ||¡}tdddgd }t ||¡}|| }t ||¡ d }|tkrnt|tjrnt }|
|
t ¡}t |¡ t ||}W 5 Q R X t |d |¡ t |d || ¡ d S )Nr ú9 daysz2 Daysú0 DaysrN r r )r r r, r r- r r~ r Z
|
DateOffsetr re rg r¯ Údivmod)r. r/ Ú
|
three_daysr1 rë r4 rU Úwarnr r r Útest_td64arr_mod_tdscalar s
|
|
z9TestTimedeltaArraylikeMulDivOps.test_td64arr_mod_tdscalarc C s¢ t dddd}t ||¡}tddgd }t ||¡}|d }t ||¡ d}tjt|d d| W 5 Q R X t|d}t |d
|
|¡ t |d |d ¡ d S ) Nz1 nsz10 nsrC r' z0 nsrM r zCannot divide int byrO r r ) r r r, r r- rR rS rT rc )r. r/ r1 rë r4 rU rV r r r Útest_td64arr_mod_int1 s
|
z4TestTimedeltaArraylikeMulDivOps.test_td64arr_mod_intc C s t dd}t ||¡}dddgdgd }t|}t ||¡}|| }t ||¡ t||}t |d |¡ t |d || ¡ d S )Nr ra rb z3 Daysé r r )r r r, r r- rc )r. r/ rd r1 rë r4 rU r r r Útest_td64arr_rmod_tdscalarC s
|
|
z:TestTimedeltaArraylikeMulDivOps.test_td64arr_rmod_tdscalarc C sz t tdddgd }tj|jd< t ||¡}d}tjt |d || W 5 Q R X tjt |d || W 5 Q R X d S ©NrM rN rÚ r z(operate|unsupported|cannot|not supportedrO ©
|
r
|
r r) rÎ r¿ r r, rR rS rT )r. r/ r\ r] r_ r r r Ú!test_td64arr_mul_tdscalar_invalidU s zATestTimedeltaArraylikeMulDivOps.test_td64arr_mul_tdscalar_invalidc C s t tjddd}t ||¡}d dddg¡}tjt|d ||d d
W 5 Q R X tjt |d |t
|
d
|
d g¡ W 5 Q R X d S ) NrM r r r= ú$cannot use operands with types dtypez$Cannot multiply with unequal lengthszUnable to coerce to SeriesrO rN r r ) r r) rd r r, r> rR rS rT r r ©r. r/ r rV r r r Ú!test_td64arr_mul_too_short_raisesd s ýÿzATestTimedeltaArraylikeMulDivOps.test_td64arr_mul_too_short_raisesc C sH t tjddd}t ||¡}d}tjt|d || W 5 Q R X d S )NrM r r rm rO )r r) rd r r, rR rS rT rn r r r Ú test_td64arr_mul_td64arr_raisest s
|
z?TestTimedeltaArraylikeMulDivOps.test_td64arr_mul_td64arr_raisesc C sº t dddgdd}t dddgdd}t ||¡}t ||¡}|| }t ||¡ | | }t ||¡ t dddgdd}t ||¡}|d| }t ||¡ d| | }t ||¡ d S ) Nrú r rË r z-59 DaysrÓ z118 Daysr )r
|
r r, r- )r. r/ r rí r4 rU r r r Ú test_td64arr_mul_numeric_scalar~ s
|
|
z?TestTimedeltaArraylikeMulDivOps.test_td64arr_mul_numeric_scalarÚtwor rç c C sv t dddgdd}t dddgdd}t ||¡}t ||¡}|| }t ||¡ tjtdd || W 5 Q R X d S © Nrú r rË r z29.5DrÓ rA rO ©r
|
r r, r- rR rS rT ©r. r/ rr rí r4 rU r r r Ú test_td64arr_div_numeric_scalar s z?TestTimedeltaArraylikeMulDivOps.test_td64arr_div_numeric_scalarc C sv t dddgdd}t dddgdd}t ||¡}t ||¡}|| }t ||¡ tjtdd || W 5 Q R X d S rs rt ru r r r Ú$test_td64arr_floordiv_numeric_scalar¤ s zDTestTimedeltaArraylikeMulDivOps.test_td64arr_floordiv_numeric_scalarr é rX é( c C s
|
t | jS r rý rÿ r r r r µ r c C s| t dddgdd}| |¡}t dddgdd}t ||¡}t||}t ||¡}|| }t ||¡ || }t ||¡ d S )Nrú r rË r z 1180 Daysz 1770 DaysrÓ )r
|
re r r, r r- )r. r/ r r rí r4 r0 rU r r r Ú test_td64arr_rmul_numeric_array² s
|
|
z?TestTimedeltaArraylikeMulDivOps.test_td64arr_rmul_numeric_arrayc C s
|
t | jS r rý rÿ r r r r Ó r c sn t dddgdd |¡t dddgdd}t |¡ t }t ||¡} }t ||¡ d d d
|
d d d dg¡}tjt |d W 5 Q R X t
|
¡ }|t krâ fddt t D }t ||¡ t
|
¡}nT fddt t D }dd |D }|tjkr*t |¡ t
|
¡}n ||t
|
d}t ||¡ tjt |d t
|
¡ W 5 Q R X d S )Nrú r rË r z2.95Dz
|
1D 23H 12mrÓ r= z!true_divide'? cannot use operandszcannot perform __div__zcannot perform __truediv__zunsupported operandrA z-ufunc 'divide' cannot use operands with typesrO c s" g | ]} j d |f | qS )r )r¿ r! ©rí r r r rÙ ö s zRTestTimedeltaArraylikeMulDivOps.test_td64arr_div_numeric_array.<locals>.<listcomp>c s g | ]} | | qS r r r! r{ r r rÙ ù s c S s$ g | ] }|t k r|n
|
t d d¡qS )r r< )r r) rw rN r r r rÙ ú s )r
|
re r r, r r- r> rR rS rT rg r rc r% r )r. r/ r r r4 r0 rU r_ r r{ r Ú test_td64arr_div_numeric_arrayÐ sB
|
|
úÿ
|
ÿ z>TestTimedeltaArraylikeMulDivOps.test_td64arr_div_numeric_arrayc
|
C s¬ |}t ||}tdddddg|d d}tddd d
|
d gtj|d d }tdddd dgd|d }t ||¡}t||}t ||¡}|| } t | |¡ || } t | |¡ d S )NÚ0daysÚ1dayÚ2daysÚ3daysÚ4daysr r r r rN r& rE Z9daysZ16daysrÓ ) r# r r
|
r) r r r, r r- )
|
r. r/ r! r r" r1 r: r4 r0 rU r r r Útest_td64arr_mul_int_series s&
|
ÿ ý
|
z;TestTimedeltaArraylikeMulDivOps.test_td64arr_mul_int_seriesc sÔ |}t dddddg|d dtdd d
|
d d gtj|d d |tjtjfkrT|d n|d }t fddtt D d|d}t
|
|¡t }t
|
||¡} ¡}|t krÄ|tksÐtn t ||¡ d S )Nr} r~ r r r r r ræ rN g @rh g @r rE r c s g | ]}| | qS r r r! ©r: r1 r r rÙ . s zRTestTimedeltaArraylikeMulDivOps.test_float_series_rdiv_td64arr.<locals>.<listcomp>rÓ )r r
|
r) r8 r r r r rc r% r, r Ú __rtruediv__r ÚNotImplementedr r- )r. r/ r! r Zxnamer4 r0 rU r r r Ú test_float_series_rdiv_td64arr$ s& ÿ ý
|
|
z>TestTimedeltaArraylikeMulDivOps.test_float_series_rdiv_td64arrc C s t ttg}t ||¡}tjddgtd}tt dd¡gd td}|tk r^t ||¡ t¡}|| }t
|
||¡ || }t
|
||¡ d S )Nr rç r r r< ) r r r r, r) r rg r rw re r- )r. r/ r1 rk rl r4 rU r r r Ú-test_td64arr_all_nat_div_object_dtype_numeric= s zMTestTimedeltaArraylikeMulDivOps.test_td64arr_all_nat_div_object_dtype_numericN)3rX rY rZ r0 r1 r3 r5 r7 r9 rR r[ r\ r) rd r r rc Zuint64r8 r Z
|
RangeIndexr; r? r@ rC rG rP rQ rR rS rW rZ Úfilterwarningsr^ r` rf rg ri rl ro rp rq r rv rw r
|
rz r| r r r r r r r r. st
|
|
|
û÷
|
&"#
|
|
|
|
|
(ý
|
(ý
|
2 r. c @ s e Zd Zdd ZdS )Ú"TestTimedelta64ArrayLikeArithmeticc C sz t tdddgd }tj|jd< t ||¡}d}tjt |d || W 5 Q R X tjt |d || W 5 Q R X d S rj rk )r. r\ r/ r] r_ r r r Útest_td64arr_pow_invalidT s z;TestTimedelta64ArrayLikeArithmetic.test_td64arr_pow_invalidN)rX rY rZ r r r r r r O s r c sD t d tdddd } t fddtdD }t | |¡ d S )Nr? Z0srä é r' c s6 g | ].} t d ¡| t d¡t|dd qS )z 0.033333333sz 0.000000001srN r )r r¾ rc rÖ ©Ú timestampr r rÙ j s û ÿþz3test_add_timestamp_to_timedelta.<locals>.<listcomp>)r r r rc r r r r r r Ú test_add_timestamp_to_timedeltae s
|
úÿ
|
r )'r r Únumpyr) rR Z pandas.errorsr r Zpandasr r r r r r
|
r r r r r Zpandas._testingZ_testingr Zpandas.core.arraysr Z pandas.tests.arithmetic.commonr r r r r# r$ ri rx r¼ rÇ r. r r r r r r Ú<module> s> 0 kg A N 7
|
