U
|
¬ý°d:g ã @ s d dl m Z mZmZ d dlmZmZ d dlZd dlZd dlZ d dl
|
Z
|
d dl Z d dl m Z d dlmZ d dlmZ d dlZd dlmZmZmZmZmZmZmZmZm Z d dl m Z d dl m!Z! d d l"m#Z#m$Z$m%Z%m&Z& G d
|
d d Z'G d d d Z(G dd dZ)G dd dZ*G dd dZ+G dd dZ,G dd dZ-G dd dZ.e
|
j/ 0ddd d g¡e
|
j/ 0d d d d g¡d d! Z1d"d# Z2d$d% Z3d&d' Z4d(d) Z5dS )*é )ÚdatetimeÚtimeÚ timedelta)ÚproductÚstarmapN)Úlocalize_pydatetime)Ú shift_months)ÚPerformanceWarning) Ú
|
DateOffsetÚ DatetimeIndexÚNaTÚPeriodÚSeriesÚ TimedeltaÚTimedeltaIndexÚ TimestampÚ
|
date_range)Ú roperator)Úassert_cannot_addÚassert_invalid_addsub_typeÚassert_invalid_comparisonÚget_upcast_boxc @ s e Zd Zdd Zej ddddde edd e d
|
d d
|
¡ d e j g ¡d d Z ej deede d¡e d¡ e j¡e d¡ e¡ejdddje ejddd¡eejdddejddd e¡ejddddjejdddd e¡g
|
¡dd Zdd Zdd Zd S ) Ú"TestDatetime64ArrayLikeComparisonsc C st |}|}t dd|d}t | ¡ d ¡}t ||¡}t||d}||k} t dddg¡}
|
t |
|
|¡}
|
t | |
|
¡ d S )NÚ20130101é ©ÚperiodsÚtzr TF)r ÚnpÚarrayZto_numpyÚtmÚ box_expectedr Ú assert_equal) ÚselfÚtz_naive_fixtureÚbox_with_arrayr ÚboxÚdtiÚotherÚdtarrÚxboxÚresultÚexpected© r- ú^d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/arithmetic/test_datetime64.pyÚtest_compare_zerodim6 s z7TestDatetime64ArrayLikeComparisons.test_compare_zerodimr( Úfooéÿÿÿÿéc g @é ©ÚdaysiÑ é Nc C s. |}t dd|d}t ||¡}t||| d S )Nú1/1/2000é
|
r )r r r! r )r# r( r$ r% r Úrngr) r- r- r. Ú test_dt64arr_cmp_scalar_invalidE s zBTestDatetime64ArrayLikeComparisons.test_dt64arr_cmp_scalar_invalidr8 Z1ns©r ú1 Dayz
|
1971-01-01ÚD©Úfreqr c C s2 |}t ddd|dj}t ||¡}t||| d S )Nú
|
1970-01-01Únsr8 ©r? r r )r Ú_datar r! r )r# r( r$ r% r ÚdtaÚobjr- r- r. Ú"test_dt64arr_cmp_arraylike_invalid_ s zETestDatetime64ArrayLikeComparisons.test_dt64arr_cmp_arraylike_invalidc C s |}t ddd|dj}t ddd|d tdd g¡}||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 )Nr@ Úhé rB r r6 r3 r r4 FTzDInvalid comparison between|Cannot compare type|not supported between©Úmatch)
|
r rC r r r r Úassert_numpy_array_equalÚpytestÚraisesÚ TypeError)r# r$ r rD r( r+ r, Úmsgr- r- r. Ú test_dt64arr_cmp_mixed_invalidx s" zATestDatetime64ArrayLikeComparisons.test_dt64arr_cmp_mixed_invalidc C sn |}|}t d|d}t|tg}t ||¡}t||d}tddgtjd} t | |¡} ||k}
|
t |
|
| ¡ d S )Nz
|
2021-01-01©r TF©Údtype) r r r r r! r r Zbool_r" ) r# r$ r% r r& ÚtsÚserrE r* r, r+ r- r- r. Útest_dt64arr_nat_comparison s z>TestDatetime64ArrayLikeComparisons.test_dt64arr_nat_comparison)Ú__name__Ú
|
__module__Ú __qualname__r/ rL ÚmarkÚ parametrizeÚobjectr r Údater Únanr: ÚlistÚrangeZarangeÚastypeÚfloat32ÚpdÚtimedelta_ranger Z period_rangerF rP rV r- r- r- r. r 1 s@ óþ
|
|
õþ
|
r c @ sT e Zd Zej dedeedgeeedgfedeedgeeedgfe dddee d ddgeee d ddgfg¡ej d
|
d d g¡ej d de
|
g¡ej de j e d d d gfe je d d d gfe je d d d gfe je d d d gfe je d d d gfe je d d d gfg¡dd Zej dedeedgedeedge dddee d ddgg¡ej d de
|
g¡dd Zej dedddedddg¡dd Zej d d d d d g¡d!d" Zd#d$ Zej d%ed&ed'dde d(d)¡g¡ej d*e jd d d d gfe jd d d d gfe j d d d d gfe jd d d d gfg¡d+d, ZdS )-Ú TestDatetime64SeriesComparisonÚpairú
|
2011-01-01ú
|
2011-01-03ú1 daysz3 daysz2011-01ÚM©r? z2011-03ÚreverseTFrS Nz op, expectedc C sL |}|\}} |r| | }} t ||d}
|
|| |d} ||
|
| } t | |¡ d S )NrR )r r Úassert_series_equal) r# rS Úindex_or_seriesrl rf Úopr, r& ÚlhsÚrhsÚleftÚrightr+ r- r- r. Útest_nat_comparisons¤ s '
|
|
z3TestDatetime64SeriesComparison.test_nat_comparisonsÚdatac C s |}t ||d}t ||¡}t|td}dddg}t ||¡}|tjkr^|tkr^tj|dd}t |tk|¡ t t|k|¡ dddg}t ||¡}|tjkr´|tkr´tj|dd}t |tk|¡ t t|k|¡ dddg}t ||¡}|tjkr|tkrtj|dd}t |tk |¡ t t|k|¡ t |tk|¡ t t|k|¡ t |tk|¡ t t|k |¡ t |tk|¡ t t|k|¡ d S )NrR TFÚbool) r r r! r r rc r r\ r" )r# rS ru r% r& rr r* r, r- r- r. Útest_nat_comparisons_scalarØ s8
|
|
|
|
z:TestDatetime64SeriesComparison.test_nat_comparisons_scalarÚvaliÐ r6 é rH c s> t tddd}| k}t fdd|D }t ||¡ d S )Nr7 r8 r; c s g | ] }| kqS r- r- ©Ú.0Úx©rx r- r. Ú
|
<listcomp>
|
s zQTestDatetime64SeriesComparison.test_series_comparison_scalars.<locals>.<listcomp>)r r r rm )r# rx Zseriesr+ r, r- r} r. Ú test_series_comparison_scalars s z=TestDatetime64SeriesComparison.test_series_comparison_scalarsz
|
left,right)ÚltÚgt)ÚleÚge)Úeqr )Úner
c C sè t tddddd}|jdd}td|d < td|d
|
< tt|}tt|}||td }|td |}t ||¡ ||td}|td|}t ||¡ ||td }|td |}t ||¡ ||t}|t|}t ||¡ d S ) NZ20010101r8 r; Údates©ÚnameT)ÚdeepÚnatr r Z20010109) r r Úcopyr ÚgetattrÚoperatorr rm r ) r# rr rs rU Zs_natZleft_fZright_fr, r+ r- r- r. Ú test_timestamp_compare_series s$
|
|
|
|
z<TestDatetime64SeriesComparison.test_timestamp_compare_seriesc C sö |}t tdtdtg}t ||¡}t||d}||k}t dddg|¡}t ||¡ |tjkrt j
|
t dd ||d k W 5 Q R X n*||d k}t dddg|¡}t ||¡ |tjkræt j
|
t dd ||d k W 5 Q R X n*||d k}t dddg|¡}t ||¡ ||k}t dddg|¡}t ||¡ |tjkrjt j
|
t dd ||d k W 5 Q R X n*||d k}t dddg|¡}t ||¡ |tjkrÈt j
|
t dd ||d k W 5 Q R X n*||d k}t dddg|¡}t ||¡ d S ) Nz2000-01-29 01:59:00z
|
2000-01-30TFz not alignedrI r r3 ) r r r r r! r r" rc Ú DataFramerL rM Ú
|
ValueError)r# r% r& rU r* r+ r, r- r- r. Ú test_dt64arr_timestamp_equality0 sD
|
|
z>TestDatetime64SeriesComparison.test_dt64arr_timestamp_equalityÚ datetimeliker iÝ z2013-01-01T00:00rA z op,expectedc C sH t tdtdtjtdgdd}|||}t |dd}t ||¡ d S )NÚ20120101r Z20130103ÚAr )r r r r^ r rm )r# r ro r, rU r+ r- r- r. Ú!test_dt64_compare_datetime_scalari s üù
|
z@TestDatetime64SeriesComparison.test_dt64_compare_datetime_scalar) rW rX rY rL rZ r[ r r r r r\ r r r r
r r r r rt rw r r r r r Ú
|
datetime64r r- r- r- r. re ¡ sv þ þþ÷þúþ ýþ $
|
|
ÿ
|
9
|
|
ýþüþ re c @ s e Zd Zdd Zej dedddede
|
d¡g¡dd Z ej d d
|
e g¡d d Z d d Zdd Zdd Zej dedddede
|
d¡g¡ej d¡dd Zdd Zdd Zdd Zejjdejd d d ejd d d ¡ ejd d d j d ¡gd d! d"d#d$ Zd%d& Z d
|
S )'Ú TestDatetimeIndexComparisonsc C sd t d¡}|t|d }t| ¡ }t |¡}|||}|||}t|tjsTt t
|
||¡ d S )Néd r3 ) r Z makeDateIndexÚlenr Ú to_datetime64r r Ú
|
isinstanceZndarrayÚAssertionErrorrK )r# Ú comparison_opÚindexÚelementZarrZ
|
arr_resultZ index_resultr- r- r. Útest_comparators s
|
|
|
|
z-TestDatetimeIndexComparisons.test_comparatorsr( éà r6 ú
|
2016-01-01c C sä |}t dd|d}|d k r6t|tjr*d S t||j}||k}t ddg¡}t ||¡ ||k}t ddg¡}t ||¡ ||k}t ddg¡}t ||¡ ||k }t ddg¡}t ||¡ ||k}t ddg¡}t ||¡ d S )Nr¢ r3 r TF) r r r r r Útzinfor r rK )r# r( r$ r r' r+ r, r- r- r. Útest_dti_cmp_datetimelike s* z6TestDatetimeIndexComparisons.test_dti_cmp_datetimelikerS Nc
|
C s t tdttdg}t tttdg}t ||¡}t ||¡}t||d}|| }}|tkrt| t¡| t¡ }}||k}t dddg¡} t | |¡} t
|
|| ¡ ||k}t dddg¡} t | |¡} t
|
|| ¡ t dddg¡} t | |¡} t
|
|tk| ¡ t
|
t|k| ¡ t dddg¡} t | |¡} t
|
|tk| ¡ t
|
t|k| ¡ t dddg¡} t | |¡} t
|
|tk | ¡ t
|
t|k| ¡ d S )Nrg rh TF) r r r r r! r r\ ra r r r" )
|
r# rS r% rr rs r* rp rq r+ r, r- r- r. Útest_dti_cmp_natº s8
|
z-TestDatetimeIndexComparisons.test_dti_cmp_natc
|
C sæ t dtjdtjddg¡}t ddtjtjddg¡}tdtdtd d
|
g}td dttd d
|
g}t t d ¡t d¡t d¡t d¡t d¡t d¡g¡}||f||f||fg}t d ¡ |D ]\}}||k } t ddddddg¡}
|
t
|
| |
|
¡ ||k} t ddddddg¡}
|
t
|
| |
|
¡ ||k} t ddddddg¡}
|
t
|
| |
|
¡ ||k} t ddddddg¡}
|
t
|
| |
|
¡ ||k} t ddddddg¡}
|
t
|
| |
|
¡ ||k} t ddddddg¡}
|
t
|
| |
|
¡ q¾W 5 Q R X t d ¡È |tjf|tffD ]®\}} || k } t ddddddg¡}
|
t
|
| |
|
¡ || k} t
|
| |
|
¡ || k} t
|
| |
|
¡ || k} t
|
| |
|
¡ || k} t
|
| |
|
¡ || k} t ddddddg¡}
|
t
|
| |
|
¡ qîW 5 Q R X t d ¡( |df|t dddffD ]\}} || k } t ddddddg¡}
|
t
|
| |
|
¡ || k} t ddddddg¡}
|
t
|
| |
|
¡ || k} t ddddddg¡}
|
t
|
| |
|
¡ || k} t ddddddg¡}
|
t
|
| |
|
¡ || k} t ddddddg¡}
|
t
|
| |
|
¡ || k} t ddddddg¡}
|
t
|
| |
|
¡ qÎW 5 Q R X d S )Nç ð?ç @g @g @ç @g @ú
|
2014-01-01ú
|
2014-03-01ú
|
2014-05-01ú
|
2014-07-01z
|
2014-02-01z
|
2014-06-01z2014-02-01 00:00z2014-03-01 00:00r z2014-06-01 00:00z2014-07-01 00:00TFr iÞ r6 ) rc ÚIndexr r^ r r r r r Úassert_produces_warningrK r ) r# Zfidx1Zfidx2Zdidx1Zdidx2ÚdarrZcasesZidx1Zidx2r+ r, rx r- r- r. Ú+test_dti_cmp_nat_behaves_like_float_cmp_nanà s ÿÿúÿ " zHTestDatetimeIndexComparisons.test_dti_cmp_nat_behaves_like_float_cmp_nanc C st |}|}t ddd}| d¡}t ||¡}t ||¡}|tjkrJdd }nt}|tjtj fkr`d}t
|
j t |d ||| W 5 Q R X t
|
j t |d |||| W 5 Q R X t
|
j t |d ||t j||td W 5 Q R X t
|
j t |d ||| W 5 Q R X t
|
j t |d |||| W 5 Q R X t
|
j t |d ||t j||td W 5 Q R X t ||k¡sttt |||k¡stt |||k¡s¤tt t j||td |k¡sÆtt |t j||td k¡sètt ||k¡sütt |||k¡stt |||k¡s,tt t j||td |k¡sNtt |t j||td k¡sptd S )
|
Nr¢ é r; ú
|
US/Pacificc S s | t¡j ¡ d S )Nr )ra r\ ÚvaluesÚtolist©r| r- r- r. Ú<lambda>K ó zQTestDatetimeIndexComparisons.test_comparison_tzawareness_compat.<locals>.<lambda>z^Invalid comparison between dtype=datetime64\[ns.*\] and (Timestamp|DatetimeArray|list|ndarray)rI rR )r Ú tz_localizer r! rc r r_ r r r
rL rM rN r r r\ Úallr ) r# r r% ro r& ÚdrÚdzr´ rO r- r- r. Ú"test_comparison_tzawareness_compat? sD
|
|
|
ÿ"""""z?TestDatetimeIndexComparisons.test_comparison_tzawareness_compatc C s |}t ddd}| d¡}t ||¡}t ||¡}td}tddd}t ||k¡sXtd}|tj tj
|
fkrt j t |d ||| W 5 Q R X t ||k¡s¢t|tj tj
|
fkrÖt j t |d ||| W 5 Q R X |tj tj
|
fkr t j t |d ||| W 5 Q R X d S )
|
Nr¢ r± r; r² z2000-03-14 01:59zEurope/AmsterdamrQ zAInvalid comparison between dtype=datetime64\[ns.*\] and TimestamprI )r r¸ r r! r r r¹ r r r r
rL rM rN ) r# r r% ro rº r» rT Zts_tzrO r- r- r. Ú*test_comparison_tzawareness_compat_scalarsp s&
|
zGTestDatetimeIndexComparisons.test_comparison_tzawareness_compat_scalarsz*ignore:elementwise comp:DeprecationWarningc C sð |}|}t dd|d}t ||¡}t||d} |tjtjfkr|tjk}
|
tj|
|
|
|
gt d} t | | ¡} |||} t
|
| | ¡ |||} t
|
| | ¡ nXdt |j } t jt| d ||| W 5 Q R X t jt| d ||| W 5 Q R X d S )Nr¢ r3 r TrR z:Invalid comparison between dtype=datetime64\[ns, .*\] and rI )r r r! r r r r
r r rv r" ÚtyperW rL rM rN )r# r r( Útz_aware_fixturer% ro r r' r) r* Zexboolr, r+ rO r- r- r. Ú"test_scalar_comparison_tzawareness s&
|
|
|
ÿz?TestDatetimeIndexComparisons.test_scalar_comparison_tzawarenessc C sh |}t dtdtddg}t |tjkgt| ¡}||t}t ||¡ || d¡t}t ||¡ d S )Nr© rª r« r¬ r² )
|
r r r r r r
r r rK r¸ )r# r ro r' r, r+ r- r- r. Ú test_nat_comparison_tzawareness³ s ÿ
|
z<TestDatetimeIndexComparisons.test_nat_comparison_tzawarenessc C s
|
|}t dd|d}d}||k}t dgdgd ¡}t ||¡ ||k}t dgdgd ¡}t ||¡ ||k }t dgd ¡}t ||¡ ||k}t dgdgd ¡}t ||¡ ||k}t dgdgd ¡}t ||¡ ||k}t dgd ¡}t ||¡ d S )Nr7 r8 r TFé )r r r r rK )r# r$ r r9 r( r+ r, r- r- r. Útest_dti_cmp_str s* z-TestDatetimeIndexComparisons.test_dti_cmp_strc C s0 t ddd}|t|k}||k}t ||¡ d S )Nr7 r8 r; )r r_ r rK )r# r9 r+ r, r- r- r. Útest_dti_cmp_listá s z.TestDatetimeIndexComparisons.test_dti_cmp_listZ1Dr8 r; úm8[ns]c C s
|
t | jS ©N)r¾ rW rµ r- r- r. r¶ ï r· z%TestDatetimeIndexComparisons.<lambda>)Úidsc C sæ t dddd}||k}t dgd ¡}t ||¡ ||k}t dgd ¡}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 ) Nú
|
2000-01-01r8 ú
|
Asia/Tokyor FTzInvalid comparison betweenrI )r r r r rK rL rM rN )r# r( r' r+ r, rO r- r- r. Ú test_dti_cmp_tdi_tzawarenessè s z9TestDatetimeIndexComparisons.test_dti_cmp_tdi_tzawarenessc C sØ t dddd}| d¡}||k}t dgd ¡}t ||¡ | d ¡}||k}t ||¡ t t|d d
tdd gd ¡}||k}t dgd d
|
gd ¡}t ||¡ d }t j
|
t |d ||k W 5 Q R X d S ) NrÈ r8 rÉ r ÚOTrH r6 r4 FzB>=' not supported between instances of 'Timestamp' and 'Timedelta'rI ) r ra r r r rK r¸ r_ r rL rM rN )r# r' r( r+ r, rO r- r- r. Útest_dti_cmp_object_dtype s
|
|
& z6TestDatetimeIndexComparisons.test_dti_cmp_object_dtype) rW rX rY r rL rZ r[ r r r r r¤ r\ r¥ r° r¼ r½ ÚfilterwarningsrÀ rÁ rà rÄ rc rd Z to_seriesÚasi8ÚviewrÊ rÌ r- r- r- r. r s< þ
|
|
%_1 þ
|
ýù
|
r c @ sÆ e Zd Zejjdd Zdd Zdd Zdd Z d d
|
Z
|
d d Z ej d e de d ¡ e d ¡ e dd¡g¡dd Zdd Zdd Zdd Zdd Zdd Zd d Zej d d dd d!d"d#d$d%g¡ej d&d%d'g¡d(d) Zej d*d+e d,d-g¡ed.dd/e d0d1d2g¡ej d3d%dg¡d4d5 Z ej d6dd d#d g¡ej d3d%dg¡d7d8 Z d9d: Z ej d;e e d<e d<ge e!e d<ge e!e!gd=d>g¡ej d?d0d@e d0¡g¡dAdB Z"d%S )CÚTestDatetime64Arithmeticc C sx |}t dd|d}t dd|d}t ||¡}t ||¡}|| }t ||¡ || }t ||¡ ||7 }t ||¡ d S )NrÈ ú
|
2000-02-01rQ z2000-01-01 02:00z2000-02-01 02:00©r r r! r" ©r# r$ Z two_hoursr% r r9 r, r+ r- r- r. Ú%test_dt64arr_add_timedeltalike_scalar* s z>TestDatetime64Arithmetic.test_dt64arr_add_timedeltalike_scalarc C sd |}t dd|d}t dd|d}t ||¡}t ||¡}|| }t ||¡ ||8 }t ||¡ d S )NrÈ rÑ rQ z1999-12-31 22:00z2000-01-31 22:00rÒ rÓ r- r- r. Ú%test_dt64arr_sub_timedeltalike_scalar@ s z>TestDatetime64Arithmetic.test_dt64arr_sub_timedeltalike_scalarc C sÜ t ddd d¡}t ||¡}td d¡}td}|| }tddd g}t ||¡}t ||¡ || }td
|
d d g}t ||¡}t ||¡ d }tjt |d || W 5 Q R X tjt |d || W 5 Q R X d S )Nr r r; ú
|
US/EasternÚCETi5)3ú0 days 06:00:00z1 days 06:00:00z2 days 06:00:00ú-1 days +18:00:00z-2 days +18:00:00z-3 days +18:00:00ú;Cannot subtract tz-naive and tz-aware datetime-like objectsrI )
|
r r¸ r r! r r r" rL rM rN ©r# r% Út1Út2Ztnaiver+ r, rO r- r- r. Ú.test_dt64_array_sub_dt_with_different_timezoneQ s* ÿ ÿ zGTestDatetime64Arithmetic.test_dt64_array_sub_dt_with_different_timezonec C sð t ddd d¡}t ||¡}t ddd d¡}t ||¡}t ddd}|| }tdddg}t ||¡}t ||¡ || }tdddg}t ||¡}t ||¡ d}tjt|d || W 5 Q R X tjt|d || W 5 Q R X d S )
|
Nr r r; rÖ r× rØ rÙ rÚ rI ) r r¸ r r! r r" rL rM rN rÛ r- r- r. Ú6test_dt64_array_sub_dt64_array_with_different_timezonel s, ÿ ÿ zOTestDatetime64Arithmetic.test_dt64_array_sub_dt64_array_with_different_timezonec
|
C s® |}t dd|dd}t d¡}tdgd |d}t ||¡}t ||¡}|| }t ||¡ || }t ||¡ || }t ||¡ d} tjt | d || W 5 Q R X d S ) Nz
|
1994-04-01r ZQS©r r r? r rQ úcannot subtractrI )
|
r r Ú timedelta64r r r! r" rL rM rN )
|
r# r% r$ r r' r( r, rE r+ rO r- r- r. Ú test_dt64arr_add_sub_td64_nat s
|
z6TestDatetime64Arithmetic.test_dt64arr_add_sub_td64_natc C sÐ |}t dd|d}tdddg}|j}t ddd|d}t ||¡}t ||¡}|| } t | |¡ || } t | |¡ t ddd|d}t ||¡}|| } t | |¡ d}
|
tjt|
|
d || W 5 Q R X d S )
|
Nr¢ r r z-1 Dayz
|
2015-12-31z
|
2016-01-02z
|
2016-01-04z8cannot subtract|(bad|unsupported) operand type for unaryrI ) r r r³ r r! r" rL rM rN ) r# r$ r% r r' ÚtdiZtdarrr, r) r+ rO r- r- r. Ú test_dt64arr_add_sub_td64ndarray s$ z9TestDatetime64Arithmetic.test_dt64arr_add_sub_td64ndarrayrT ú
|
2013-01-01r= c C st t ddd d ¡}t ||¡}tdddg}t ||¡}|| }t ||¡ || }t || ¡ t || ¡ d S )Nræ r r; z0 Daysr< z2 Days)r Ú
|
_with_freqr r! r r" )r# r% rT Úidxr, r+ r- r- r. Útest_dt64arr_sub_dtscalar¹ s z2TestDatetime64Arithmetic.test_dt64arr_sub_dtscalarc C s| t ddddd}| d ¡}|d }t ||¡}tt dd¡t dd¡g}t ||¡}t || |¡ t || | ¡ d S )Nz
|
2014-03-17r3 r= rÖ ©r r? r r r6 )r rç r r! r r râ r" )r# r% rU rT Z delta_seriesr, r- r- r. Ú"test_dt64arr_sub_timestamp_tzawareÓ s
|
z;TestDatetime64Arithmetic.test_dt64arr_sub_timestamp_tzawarec C s t ttdg}t ||¡}|t }tttgdd}t ||¡}t ||¡ | d¡}t ||¡}|t }tttgdd}t ||¡}t ||¡ d S )NÚ19900315útimedelta64[ns]rR rÉ )r r r r r! r r" r¸ )r# r% r' rU r+ r, Údti_tzZser_tzr- r- r. Útest_dt64arr_sub_NaTà s
|
z-TestDatetime64Arithmetic.test_dt64arr_sub_NaTc C sh t dd|d}|| }t ||¡}t ||¡ t¡}t t¡ || t¡ }W 5 Q R X t ||¡ d S ©Nr¢ r r )r r r! ra r\ r® r r" )r# r% r$ r' r, rE r+ r- r- r. Ú!test_dt64arr_sub_dt64object_arrayõ s z:TestDatetime64Arithmetic.test_dt64arr_sub_dt64object_arrayc C sT t ddd d}|j}t ||¡}|| }|| }t ||¡ || }t ||¡ d S rð )r r³ r r! r" )r# r% r' Údt64valsr) r, r+ r- r- r. Ú"test_dt64arr_naive_sub_dt64ndarray s z;TestDatetime64Arithmetic.test_dt64arr_naive_sub_dt64ndarrayc C sp |}t dd|d}|j}t ||¡}d}tjt|d || W 5 Q R X tjt|d || W 5 Q R X d S )Nr¢ r r z.Cannot subtract tz-naive and tz-aware datetimerI )r r³ r r! rL rM rN )r# r¿ r% r r' rò r) rO r- r- r. Ú)test_dt64arr_aware_sub_dt64ndarray_raises s zBTestDatetime64Arithmetic.test_dt64arr_aware_sub_dt64ndarray_raisesc C sÄ |}t dd|d}|d kr&| d¡}n
|
| d ¡}t ||¡}t||j t|| t|| t||d t||d ¡ t||d ¡ t||d t||d ¡ t|t dd¡ d S )Nr¢ r r rÖ r rg r= )
|
r r¸ r r! r r³ Ú to_pydatetimer r r )r# r$ r% r r' Údti2r) r- r- r. Ú test_dt64arr_add_dtlike_raises s
|
|
|
z7TestDatetime64Arithmetic.test_dt64arr_add_dtlike_raisesr? ÚHÚWrj ZMSÚQÚBNrS Zuint8c
|
C s |}|t jkrd S |d kr,tddg|d}ntdd||d}||}t ddg¡}|d k rd| |¡}d d
|
d d d dg¡} t|d| t|t d¡| t|tjdtj d| t||| t|t |¡| t|t |¡| t|t
|
|¡| t|t |¡| t|t || d S )Nr z2017-04-05 06:07:08rQ r¢ r3 rê ry r1 ú|z Addition/subtraction of integersz"cannot subtract DatetimeArray fromz(can only perform ops with numeric valuesz%unsupported operand type.*Categoricalz:unsupported operand type\(s\) for -: 'int' and 'Timestamp'r6 r rR ) rc r r r r r ra Újoinr Úint64Z Categoricalr r )
|
r# rS r% r? r$ r r' rE r( rO r- r- r. Útest_dt64arr_addsub_intlike9 s6
|
|
úÿ
|
z4TestDatetime64Arithmetic.test_dt64arr_addsub_intliker( g
ëQ¸ @r¨ r§ rg rk r6 r3 r Údti_freqc C s@ t ddg|d}t ||¡}d ddddd g¡}t||| d S )
|
Nrg ú
|
2011-01-02rk rü zunsupported operand typeúcannot (add|subtract)z cannot use operands with typesz7ufunc '?(add|subtract)'? cannot use operands with typesz;Concatenation operation is not implemented for NumPy arrays)r r r! rý r )r# r r( r% r' r) rO r- r- r. Ú test_dt64arr_add_sub_invalidb s ûÿ z5TestDatetime64Arithmetic.test_dt64arr_add_sub_invalidÚpi_freqc
|
C sT t ddg|d}| |¡}t ||¡}t ||¡}d ddddg¡} t||| d S ) Nrg r rk rü r úunsupported operandzdescriptor.*requireszufunc.*cannot use operands)r Z to_periodr r! rý r )
|
r# r r r% Zbox_with_array2r' Úpir) ZparrrO r- r- r. Útest_dt64arr_add_sub_parr| s
|
üÿz2TestDatetime64Arithmetic.test_dt64arr_add_sub_parrc C s |}t dd|d}dd tdD }t ||¡}t ||¡}d ddg¡}tjd d
|
t d t¡ t |||d W 5 Q R X d S ) Nú
|
2012-01-01r r c S s g | ]}t |||qS r- )r )r{ Úir- r- r. r~ s zTTestDatetime64Arithmetic.test_dt64arr_addsub_time_objects_raises.<locals>.<listcomp>rü r z*cannot subtract DatetimeArray from ndarrayT©ÚrecordÚignore)rO )
|
r r` r r! rý ÚwarningsÚcatch_warningsÚ simplefilterr r )r# r% r$ r Zobj1Úobj2rO r- r- r. Ú'test_dt64arr_addsub_time_objects_raises s þÿ z@TestDatetime64Arithmetic.test_dt64arr_addsub_time_objects_raisesÚ dt64_seriesrì údatetime64[ns]rR Úoner¦ c C s t ||¡}d}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 d S )Nz&cannot perform .* with this index typerI )r r! rL rM rN )r# r r r% rE rO r- r- r. Ú!test_dt64_mul_div_numeric_invalid® s
|
z:TestDatetime64Arithmetic.test_dt64_mul_div_numeric_invalid)#rW rX rY rL rZ Zarm_slowrÔ 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Ð # s`
|
|
|
|
úþ
|
'
|
|
úþ ýþrÐ c & @ s e Zd Zdd Zdd Zej ddddd d
|
d d g¡d d Zdd Z dd Z
|
ej dddddifddddifddddd d d d ifd d d d ifd d!d"d#d$d%d&d'd(d)d*d+d,d-d.d d/ifd0d1d1d/d2d3fd4d5d/d2d6fd7d/d/d8fd9d:d;d<ifd:ddifg!¡ej d=d>d?g¡ej d@d5dg¡dAdB Z dCdD Z ej dEe ej ¡ ejjd/dFg¡e ejjd1dGej ¡ g¡e ejjd1dGejjd1dGg¡g¡ej dHejejejg¡ej dId>d?g¡dJdK Zej dLdMed dNdOedPedQedRedSgdTfdMed dUedVedWedXedYgdZfd[ed dNdOed\ed]ed^ed_gdTfd[ed dUed`edaedbedcgddfg¡dedf ZdTS )gÚ"TestDatetime64DateOffsetArithmeticc C s| t tdtdg}t tdtdg}t ||¡}t ||¡}|tj d¡ }t ||¡ tj d¡| }t ||¡ d S )Nú 20130101 9:01ú 20130101 9:02z20130101 9:01:05z20130101 9:02:05rH )r r r r! rc ÚoffsetsÚSecondr" )r# r% rU r, r+ Úresult2r- r- r. Ú'test_dt64arr_series_add_tick_DateOffsetÎ s ÿ zJTestDatetime64DateOffsetArithmetic.test_dt64arr_series_add_tick_DateOffsetc C s¬ t tdtdg}t tdtdg}t ||¡}t ||¡}|tj d¡ }t ||¡ tj d¡ | }t ||¡ d}tj t
|
|d tj d¡| W 5 Q R X d S )Nr r z20130101 9:00:55z20130101 9:01:55rH ú((bad|unsupported) operand type for unaryrI ) r r r r! rc r r r" rL rM rN )r# r% rU r, r+ r rO r- r- r. Ú'test_dt64arr_series_sub_tick_DateOffsetß s ÿ zJTestDatetime64DateOffsetArithmetic.test_dt64arr_series_sub_tick_DateOffsetÚcls_nameÚDayÚHourÚMinuter ÚMilliZMicroZNanoc C sT t tdtdg}t ||¡}ttj|}||d |d| ||d d S )Nr r rH )r r r r! r rc r )r# r r% rU Ú
|
offset_clsr- r- r. Ú*test_dt64arr_add_sub_tick_DateOffset_smokeó s zMTestDatetime64DateOffsetArithmetic.test_dt64arr_add_sub_tick_DateOffset_smokec
|
|
C s |}|dkrJt dd|d}|tj d¡ }|d tj d¡ |d ksJtt dd|dd }td
|
d d gd|d }t ||¡}t ||¡}tj d¡t dd¡t
|
ddfD ]p}|| }t ||¡ || }t ||¡ || }t ||¡ d ddg¡} t jt| d || W 5 Q R X q¦d S )Nr² z
|
2012-11-01r r rH r z2010-11-01 00:00rø rà z2010-11-01 05:00z2010-11-01 06:00z2010-11-01 07:00)r? r rG )Úhoursrü z bad operand type for unary -z cannot subtract DatetimeArrayrI )r rc r r! r r r r! r râ r r" rý rL rM rN )
|
r# r¿ r% r r Úoffsetr, ZscalarZ roundtriprO r- r- r. Útest_dti_add_tick_tzaware s2 ý $ ÿz<TestDatetime64DateOffsetArithmetic.test_dti_add_tick_tzawarec
|
|
s t tdtdtdtdtdtdtdtdg}t ||¡}|tjkrX|jd n|}d
|
d d d dddg}t|D ]"\}\}}tf ||i t fdd|D }t ||¡}t || ¡ t fdd|D }t ||¡}t || ¡ tf t
|
|d |d
t fdd|D }t ||¡}t || ¡ t fdd|D }t ||¡}t || ¡ d} t j t | d | W 5 Q R X qvd S )Nú2000-01-05 00:15:00ú2000-01-31 00:23:00rÈ ú
|
2000-03-31ú
|
2000-02-29ú
|
2000-12-31ú
|
2000-05-15ú
|
2001-06-15r )Úyearsr3 )ÚmonthsrH )r5 r )r& rH )Úminutesr8 )Úsecondsr3 )Ú microsecondsrH c s g | ] }| qS r- r- rz ©Úoffr- r. r~ C s zaTestDatetime64DateOffsetArithmetic.test_dt64arr_add_sub_relativedelta_offsets.<locals>.<listcomp>c s g | ] }| qS r- r- rz r5 r- r. r~ G s r6 c s g | ] }| qS r- r- rz r5 r- r. r~ M s c s g | ] }| qS r- r- rz r5 r- r. r~ Q s r rI )r r r r! rc r ÚilocÚ enumerater
|
r" ÚdictrL rM rN )
|
r# r% ÚvecÚ vec_itemsZrelative_kwargsr ÚunitÚvaluer, rO r- r5 r. Ú*test_dt64arr_add_sub_relativedelta_offsets% sN øÿ ù zMTestDatetime64DateOffsetArithmetic.test_dt64arr_add_sub_relativedelta_offsetsÚcls_and_kwargsZ YearBeginÚmonthrH ZYearEndZ
|
MonthBeginÚMonthEndZ SemiMonthEndZSemiMonthBeginZWeekÚweekdayr r± Z BusinessDayZBDayZ
|
QuarterEndZ QuarterBeginZCustomBusinessDayZCDayZ
|
CBMonthEndZ CBMonthBeginZ BMonthBeginZ BMonthEndZ BusinessHourZ
|
BYearBeginZBYearEndZ BQuarterBeginÚLastWeekOfMonthr3 Ú FY5253Quarterr6 Znearest)Zqtr_with_extra_weekÚ startingMonthrB Ú variationÚFY5253r )rB rE rF Ú WeekOfMonth)rB ÚweekZEasterr
|
Údayry Ú normalizeTFÚnc
|
s t |tr|\}}n|}i }|dkr0|dkr0d S ttdtdtdtdtdtdtd td
|
g}t ||¡}|tjkr|jd n|}t tj
|
|} t j d d â t d t¡ | |fd|i| t fdd|D }
|
t |
|
|¡}
|
t |
|
| ¡ t fdd|D }
|
t |
|
|¡}
|
t |
|
| ¡ t fdd|D }
|
t |
|
|¡}
|
t |
|
| ¡ d} tjt| d | W 5 Q R X W 5 Q R X d S )Nr )rH rC rD rG r) r* rÈ r+ r, r- r. r/ Tr
|
r rK c s g | ] }| qS r- r- rz ©r' r- r. r~ ½ s zWTestDatetime64DateOffsetArithmetic.test_dt64arr_add_sub_DateOffsets.<locals>.<listcomp>c s g | ] }| qS r- r- rz rM r- r. r~ Á s c s g | ] } | qS r- r- rz rM r- r. r~ Å s r rI )r Útupler r r r! rc r r7 r r r r r r r" rL rM rN ) r# r% rL rK r? r Úkwargsr: r; r$ r, rO r- rM r. Ú test_dt64arr_add_sub_DateOffsets] sF 6
|
|
øÿ zCTestDatetime64DateOffsetArithmetic.test_dt64arr_add_sub_DateOffsetsc C sÄ t dddd}t ||¡}|tdd }tdd| }t dddd d ¡}t ||¡}t ||¡ t ||¡ |tdd }t d d
|
dd d ¡}t ||¡}t ||¡ ttd d d tdd d gdd}t ||¡}|tj
|
¡ }tj
|
¡ | }ttdd d tdd d gdd}t ||¡}t ||¡ t ||¡ ttd d d tdd d gdd}t ||¡}|tj ¡ }tj ¡ | }ttdd d tdd d gdd}t ||¡}t ||¡ t ||¡ d S )NrÈ z
|
2000-01-31Úar r6 ©r0 z
|
2001-01-01z
|
2001-01-31z
|
1999-01-01z
|
1999-01-31z2000-01-15 00:15:00z
|
US/CentralrQ z
|
2000-02-15z2000-01-16 00:15:00z
|
2000-02-16z2000-01-31 00:15:00r, ) r r r! r
|
rç r" r r rc r r rA )r# r% Úsr+ r Úexpr- r- r. Ú test_dt64arr_add_sub_DateOffsetÌ s`
|
|
þû
|
|
þû
|
|
þû
|
|
þû zBTestDatetime64DateOffsetArithmetic.test_dt64arr_add_sub_DateOffsetr( ©rL rR ro Ú box_otherc
|
sÒ |}t dd|d t |¡}t tj ¡ tjjddg¡t fddt
|
t D }t ||¡ t ¡}|r¢t |¡|tjkr¢tjkr¢tj|t d}t t¡ |} W 5 Q R X t | |¡ d S )Nú
|
2017-01-01r3 r rV c s g | ]} | | qS r- r- ©r{ rL ©r' ro r( r- r. r~ s zXTestDatetime64DateOffsetArithmetic.test_dt64arr_add_sub_offset_array.<locals>.<listcomp>rR )r r r! r r rc r rA r r r` r ra r\ r Úraddr® r r" )
|
r# r$ r% rW ro r( r r) r, Úresr- rZ r. Ú!test_dt64arr_add_sub_offset_array s " zDTestDatetime64DateOffsetArithmetic.test_dt64arr_add_sub_offset_arrayzop, offset, exp, exp_freqÚ__add__r8 )r1 r5 z
|
2014-04-11z
|
2015-04-11z
|
2016-04-11z
|
2017-04-11N)r1 z
|
2014-04-01z
|
2015-04-01z
|
2016-04-01z
|
2017-04-01zAS-APRÚ__sub__z
|
2013-09-21z
|
2014-09-21z
|
2015-09-21z
|
2016-09-21z
|
2013-10-01z
|
2014-10-01z
|
2015-10-01z
|
2016-10-01zAS-OCTc C s^ |}t ddd|d}t ||d¡}t||} | |}
|
t||d} t | |d¡} t |
|
| ¡ d S )Nz 01 Jan 2014z 01 Jan 2017ZAS)ÚstartÚendr? r FrQ )r r r! r r r" ) r# ro r' rT Zexp_freqr¿ r% r r] Zmthr+ r, r- r- r. Ú#test_dti_add_sub_nonzero_mth_offset+ s 5
|
zFTestDatetime64DateOffsetArithmetic.test_dti_add_sub_nonzero_mth_offset) rW rX rY r r rL rZ r[ r% r( r> rP rU r r rc r rA r r
|
r Úaddr r[ Úsubr] r rb r- r- r- r. r É sÚ ÿ
|
$8
|
|
|
|
|
üþ
|
|
×þ.?: ÿýþ
|
|
ü÷ ü÷
|
ü÷ ü÷Þþ1r c @ s, e Zd Zdd Zdd Zdd Zdd Zd S )
|
Ú TestDatetime64OverflowHandlingc C s^ t tdg}t tg}t ||¡}t ||¡}ttg}t ||¡}|| }t ||¡ d S )Nz
|
1969-12-31)r r r r r! r r" )r# r% rr rs r, r+ r- r- r. Útest_dt64_overflow_maskingn s
|
|
z9TestDatetime64OverflowHandling.test_dt64_overflow_maskingc C sH t d}td}tdddd}t|}d}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 t|jd < td
|
d d d gdd}|| }t
|
||¡ || }t
|
||¡ t|jdd
< tdd d d gdd}|| }t
|
||¡ || }t
|
|| ¡ d S )Nz
|
1700-01-31z
|
20000 Daysz
|
1949-09-30Z100Yry r> úOverflow in int64 additionrI r1 z
|
2004-10-03z
|
2104-10-04z
|
2204-10-04r r rR r6 z
|
91279 Daysrí ) r r r r rL rM Ú OverflowErrorr r7 r rm )r# ÚdtÚtdr' rU rO r, r\ r- r- r. Ú test_dt64_series_arith_overflow| s8
|
|
ÿ z>TestDatetime64OverflowHandling.test_dt64_series_arith_overflowc
|
C sJ t dtjg¡}t dtjg¡}td d¡}|| ¡ | ¡ d¡| ¡ d¡g}td d¡}|| ¡ | ¡ d¡| ¡ d¡g}d}|D ]&}t j
|
t |d || W 5 Q R X qtjj |j } |D ] }|| }
|
|
|
d j | ksÆt qÆtjj |j } |D ] }|| }
|
|
|
d j | ksøt qø|D ](}t j
|
t |d || W 5 Q R X q d S )
|
Nú2021-12-28 17:19ú
|
1950-01-01rA r z datetime64[D]ú
|
1980-01-01rg rI r6 )rc Ú to_datetimer ÚmaxÚminZas_unitrõ r ra rL rM rh Ú_valuer ) r# ÚdtimaxÚdtiminZtsnegZts_neg_variantsZtsposZts_pos_variantsrO Úvariantr, r\ r- r- r. Ú)test_datetimeindex_sub_timestamp_overflow s: ü üzHTestDatetime64OverflowHandling.test_datetimeindex_sub_timestamp_overflowc C sd t dtjg¡}t dtjg¡}t ddg¡}t ddg¡}tjj|d j }|| }|d j|kshttjj|d j }|| }|d j|kstd}tjt |d || W 5 Q R X tjt |d || W 5 Q R X t tjg¡}|t
|
j t
|
d } tjt |d | | W 5 Q R X t tjg¡}
|
|
|
t
|
j t
|
d } tjt |d |
|
| W 5 Q R X d S )Nrl rm rn r6 rg rI Z1us) rc ro r rp rq rr r rL rM rh r ) r# rs rt Zts_negZts_posr, r+ rO ZtminrÜ ZtmaxrÝ r- r- r. Ú-test_datetimeindex_sub_datetimeindex_overflow s. zLTestDatetime64OverflowHandling.test_datetimeindex_sub_datetimeindex_overflowN)rW rX rY rf rk rv rw r- r- r- r. re k s &re c @ s e Zd Zdd Zdd Zdd Zej de de d e
|
ge d
|
e d e d gd dgge de d e
|
ge ddde ddde
|
gddd gge ddde ddde
|
ge ddde
|
e dddgddd ggg¡dd Z dd Z dd Zd d Zd d Zd d! Zd"d# Zd$S )%Ú TestTimestampSeriesArithmeticc C s~ t dd}t dd}||}||}t ||| ¡ t ||| ¡ t ||| ¡ d}tjt|d || W 5 Q R X d S )NzM8[ns]rR rÅ rá rI )r r r" rL rM rN )r# r% rQ ÚbrO r- r- r. Útest_empty_series_add_subæ s
|
|
z7TestTimestampSeriesArithmetic.test_empty_series_add_subc C s t tdddgd }tj|jd< t tdtdtdg}tj|jd< t tdtd td
|
g}|| || || || || || || d S ) NrH r ©r2 r3 r3 Ú20111230r Z20120103Ú20111231Ú20120102Ú20120104)r r r r^ r7 r )r# Útd1Údt1Údt2r- r- r. Útest_operators_datetimelikeó s, ýÿ ýÿz9TestTimestampSeriesArithmetic.test_operators_datetimelikec C sV t tdddddd}tddddd}t|g}t t || ¡¡}|jd ksRtd S )
|
NiÉ r6 é é é r r± é rí ) r r r rc Z to_timedeltar ÚabsrS r )r# rT ri rU r+ r- r- r. Ú test_dt64ser_sub_datetime_dtype s
|
|
z=TestTimestampSeriesArithmetic.test_dt64ser_sub_datetime_dtypezleft, right, op_failr| r r} r~ r r_ Ú__rsub__rH r r{ r^ Ú__radd__rÖ rQ c C sX |}t |}t |}t||d }||krLtjtdd || W 5 Q R X n|| d S )Nz%operate|[cC]annot|unsupported operandrI )r r rL rM rN ) r# rr rs Zop_failZall_arithmetic_operatorsZop_strZarg1Zarg2ro r- r- r. Ú#test_operators_datetimelike_invalid s ÿzATestTimestampSeriesArithmetic.test_operators_datetimelike_invalidc C sl t tdddg}t tdddg}|| }t tdg}t ||¡ || }t tdg}t ||¡ d S )Nz
|
2016-02-10zAmerica/Sao_PaulorQ z
|
2016-02-08Z2daysz-2days)r r r r rm )r# Ús1Ús2r+ r, r- r- r. Útest_sub_single_tzM s z0TestTimestampSeriesArithmetic.test_sub_single_tzc C sT t dddd}t|}ttdgd }|| }t ||¡ || }t ||¡ d S )Nz
|
1999-09-30r8 r² r Z0days)r r r r rm )r# r' rU r, r\ r- r- r. Ú test_dt64tz_series_sub_dtitzX s z:TestTimestampSeriesArithmetic.test_dt64tz_series_sub_dtitzc C sh t tddddtjdtg}tddddtjd}t tdtg}t || |¡ t |t| |¡ d S )Nr¡ é é é )r£ r ri ) r r ÚpytzÚutcr r r rm r )r# rS ri rT r- r- r. Útest_sub_datetime_compatd s
|
z6TestTimestampSeriesArithmetic.test_sub_datetime_compatc C s t tdtdg}|tj d¡ }tj d¡| }t tdtdg}t ||¡ t ||¡ |tj d¡ tj d¡ }t tdtdg}t ||¡ d S )Nr r rH z20130101 9:01:00.005z20130101 9:02:00.005z20130101 9:06:00.005z20130101 9:07:00.005)r r rc r r# r rm r" )r# rS r+ r r, r- r- r. Ú*test_dt64_series_add_mixed_tick_DateOffsetl s ÿ ÿzHTestTimestampSeriesArithmetic.test_dt64_series_add_mixed_tick_DateOffsetc C sâ t ttdg}t ttgdd}t tgdd}t t | |¡ d}tjt|d | | W 5 Q R X t t | |¡ tjt|d | | W 5 Q R X t |t |¡ t t| |¡ t |t |¡ t t| |¡ d S )Nrì r rR z-bad operand type for unary -: 'DatetimeArray'rI )r r r r rm rL rM rN )r# Údatetime_seriesÚnat_series_dtype_timestampÚsingle_nat_dtype_datetimerO r- r- r. Útest_datetime64_ops_nat s: ÿ ÿ ÿ ÿ ÿz5TestTimestampSeriesArithmetic.test_datetime64_ops_natc C sê d}t tdd|ddd}| ¡ }tj|jd< t tjddd d
|
}| ¡ }tj|jd < |jj d ksft
|
||d }|j d ¡|d j |¡}t ||¡ ||d }|j d ¡|d j |¡}t ||¡ |d | }|j d ¡|d j |¡}t ||¡ |d | }|j d ¡|d j |¡}t ||¡ ||d }|j d ¡|d j |¡}t ||¡ d }tjt|d |d | W 5 Q R X ||d }|j d ¡|d j |¡}t ||¡ tjt|d |d | W 5 Q R X || }|j d ¡| j |¡}t ||¡ || }|j d ¡| j |¡}t ||¡ || }|j d ¡| j |¡}t ||¡ || }|j d ¡| j |¡}t ||¡ d}tjt|d || W 5 Q R X tjt|d || W 5 Q R X d S )NrÖ z2000-01-01 09:00:00rH r r0 r r3 z 1 days 1 minrø )r r? r6 r r rI r )r r r r r^ r7 rc rd Ú_valuesr? r ri r¸ r rm rL rM rN ) r# r r r r Ztd2r+ rT rO r- r- r. Ú*test_operators_datetimelike_with_timezones£ s` zHTestTimestampSeriesArithmetic.test_operators_datetimelike_with_timezonesN)rW rX rY rz r r rL rZ r[ r r r r r r r r r r r- r- r- r. rx å s: ! ýý
|
|
ýùõþ
|
$rx c @ s² e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Ze j
|
d e j e jg¡dd Zdd Zdd Zdd Ze j
|
d e j eje jg¡dd Ze j
|
dejejg¡dd ZdS ) ÚTestDatetimeIndexArithmeticc C s |}t td|dgd }tjddd}tdd|d}| d ¡}|| }t ||¡ || }t ||¡ ||j }t ||¡ |j| }t ||¡ d S ©NrX rQ r8 ú0 daysr; r © r r rc rd r rç r Úassert_index_equalr³ ©r# r$ r r' rä r, r+ r- r- r. Útest_dti_add_tdiæ s
|
|
|
z,TestDatetimeIndexArithmetic.test_dti_add_tdic C sÞ |}t td|dgd }tjddd}tdd|d}| d ¡}t td|dgd }||7 }t ||¡ tjddd}||7 }t ||¡ t td|dgd }||j7 }t ||¡ tjddd}||7 }t ||¡ d S r r¡ r£ r- r- r. Útest_dti_iadd_tdiü s"
|
|
z-TestDatetimeIndexArithmetic.test_dti_iadd_tdic C s¾ |}t td|dgd }tjddd}tdd|dd}| d ¡}|| }t ||¡ d}tj t
|
|d || W 5 Q R X ||j }t ||¡ d
|
}tj t
|
|d |j | W 5 Q R X d S ) NrX rQ r8 r r; ú-1Drà z cannot subtract .*TimedeltaArrayrI ú0cannot subtract a datelike from a TimedeltaArray) r r rc rd r rç r r¢ rL rM rN r³ )r# r$ r r' rä r, r+ rO r- r- r. Útest_dti_sub_tdi s
|
|
z,TestDatetimeIndexArithmetic.test_dti_sub_tdic
|
C sZ |}t td|dgd }tjddd}tdd|dd}| d ¡}t td|dgd }||8 }t ||¡ |j ¡ }||8 }t
|
||j¡ |j ¡ }t j |||d t
|
||j¡ d } t jt| d
|
||8 }W 5 Q R X t td|dgd }||j8 }t ||¡ t jt| d
|
| j|8 _W 5 Q R X t jt| d
|
| j|8 _W 5 Q R X d S ) NrX rQ r8 r r; r¦ rà )Úoutr§ rI )r r rc rd r rç r r¢ rC r Zassert_datetime_array_equalr ÚsubtractrL rM rN r³ r )
|
r# r$ r r' rä r, r+ rD r© rO r- r- r. Útest_dti_isub_tdi. s0
|
|
|
|
z-TestDatetimeIndexArithmetic.test_dti_isub_tdic C sp t dd|d}|j}|| }|| }t ||¡ |}|| }|| }t ||¡ || }|| }t ||¡ d S )Nr r r )r r r r¢ )r# r$ r' rD r+ r, rä r- r- r. Útest_dta_add_sub_indexZ s z2TestDatetimeIndexArithmetic.test_dta_add_sub_indexc C s8 t ddd}t ddd d¡}tdddg}|| }t ||¡ || }t ||¡ d}tjt|d || W 5 Q R X tjt|d || W 5 Q R X ||8 }t ||¡ t ddd}t ddd}d }tjt|d || W 5 Q R X t d
|
t
|
j d g}t d d t
|
j g}td t
|
j t
|
j g}|| }t ||¡ d S )Nr r r; rÖ r rÚ rI ry z$cannot add indices of unequal lengthr z
|
2012-01-03z
|
2012-01-02ri ) r r¸ r r r¢ rL rM rN r r r r^ )r# r' rî r, r+ rO Zdti1rö r- r- r. Útest_sub_dti_dtik s0 z,TestDatetimeIndexArithmetic.test_sub_dti_dtiro c sª t tdtdtdtdg}||}ddddd g fd
|
d }ttd gd D ]P\}}}} }
|
||||| |
|
} t|||| |
|
d} ||| } ||| }t | |¡ qTd S )NZ20130301z20130228 23:00:00z20130228 22:00:00z20130228 21:00:00r= rG ÚmrS Úusc s t ttt jt| ¡S rÆ )r Úsumr_ r râ Úzip)Úargs©Z intervalsr- r. râ £ s z_TestDatetimeIndexArithmetic.test_timedelta64_equal_timedelta_supported_ops.<locals>.timedelta64r3 rH )r5 r& r2 r3 r4 )r r r r` r r r" )r# ro r% rU rE râ ÚdrG r® rS r¯ ZnptdZpytdrp rq r- r³ r. Ú.test_timedelta64_equal_timedelta_supported_ops s üÿ
|
|
zJTestDatetimeIndexArithmetic.test_timedelta64_equal_timedelta_supported_opsc C sJ t ttdg}t ttdg}t ttgdd}t ttgdd}t tgdd}t tgdd}t || |¡ t || |¡ t | | |¡ t || |¡ t || |¡ t | | |¡ d}tjt|d || W 5 Q R X t || |¡ t || |¡ t || |¡ t || |¡ t || |¡ t || |¡ d S )NZ1srì rí rR r zcannot subtract a datelikerI ) r r r r r rm rL rM rN )r# Ztimedelta_seriesr Znat_series_dtype_timedeltar r Zsingle_nat_dtype_timedeltarO r- r- r. Ú)test_ops_nat_mixed_datetime64_timedelta64¯ sr ÿ ÿ ÿþþþþþþþþþzETestDatetimeIndexArithmetic.test_ops_nat_mixed_datetime64_timedelta64c C s t ddddd}t dd¡}t ddddd}|| t ||¡fD ],}t|tsRtt ||¡ |j dks@tq@t d dddd}|| t
|
||¡fD ],}t|ts¤tt ||¡ |j dkstq| d ¡}t t dd¡t d
|
d¡t dd¡g¡}tdd d gdd }|| t ||¡fD ]$}t ||¡ |j |j kstqtd ddgdd }|| t
|
||¡fD ]4}t|tsxtt ||¡ |j |j ksdtqdd S )Nrg r Z2Dr| )r r? r r6 r= r z
|
2010-12-31r3 z
|
2011-01-05z
|
2011-01-08r ) r r râ rc r r r r r¢ r? rª rç r )r# rè ÚdeltarT r+ r- r- r. Útest_ufunc_coercionsô s2
|
ÿ z0TestDatetimeIndexArithmetic.test_ufunc_coercionsc C sÔ |}t ddg||d d}ttddgd ||d d }t|tdd ||d d }|d |_|j|jksnt|| }t ||¡ || }t ||¡ |tdd }|j| } t | |¡ ||j }
|
t |
|
|¡ d S )
|
Nz2016-06-28 05:30z2016-06-28 05:31r )r r rH )r3 r3 r6 )r r )
|
r r r r rS r r rm r³ r¢ ) r# r$ Únamesr r rU r, r+ r Zresult3Zresult4r- r- r. Útest_dti_add_series s& ÿ
|
|
|
z/TestDatetimeIndexArithmetic.test_dti_add_seriesc
|
s¸ |}|}t dd||d d |tj ¡ tjjddg|d dt }t t¡ }W 5 Q R X t fdd t
|
t D |d d
|
d } t | |¡ t¡} t || ¡ d S ) NrX r3 r )r r r rV r6 r c s g | ]} | | qS r- r- rY rZ r- r. r~ ? s zPTestDatetimeIndexArithmetic.test_dti_addsub_offset_arraylike.<locals>.<listcomp>Zinfer)r r? )r rc r rA r r r r® r r r` r r! ra r\ r" )
|
r# r$ r¹ ro rn Ú other_boxr r* r\ r, r- rZ r. Ú test_dti_addsub_offset_arraylike. s $
|
ÿz<TestDatetimeIndexArithmetic.test_dti_addsub_offset_arrayliker» c C sÜ |}t dd|d}t ||¡}|tj ¡ tddg}t||}tddg|d} t | |¡ t
|
¡} t t ¡ || }
|
W 5 Q R X t |
|
| ¡ td d
|
g|d} t | |¡ t
|
¡} t t ¡ || }
|
W 5 Q R X t |
|
| ¡ d S ) NrX r3 r ry r4 z
|
2017-01-31z
|
2017-01-06rQ z
|
2016-12-31z
|
2016-12-29)r r r! rc r rA r r r ra r\ r® r r" ) r# r$ r% r» r r' r) r( r* r, r+ r- r- r. Ú test_dti_addsub_object_arraylikeD s
|
z<TestDatetimeIndexArithmetic.test_dti_addsub_object_arraylikeN)rW rX rY r¤ r¥ r¨ r« r¬ r rL rZ r[ r rc rd rµ r¶ r¸ rº r r[ r¼ rc r r r r½ r- r- r- r. r â s ,*
|
E#
|
r r0 r1 r6 r1 éþÿÿÿr3 c sj t tdtdtdtdtdg}t t|jd } fdd|D }t |}t ||¡ d S ) Nr) r* rÈ r, r- r c s g | ]}|t jj d qS ))r0 r1 )rc r r
|
rz ©r1 r0 r- r. r~ l s z%test_shift_months.<locals>.<listcomp>)r r r rÎ r r¢ )r0 r1 r' ÚactualÚrawr, r- r¿ r. Útest_shift_months^ s ûÿ r c C sú t dddd} | j d¡}t dd tdD ¡}|j|jksBtt t
|
¡ || }W 5 Q R X t t
|
¡. |d d
df |d d
df d d
|
¡}W 5 Q R X t ||¡ t t
|
¡ || t ¡ }W 5 Q R X |jdksÜttd d | ¡ D sötd S ) Nz
|
1994-02-13Z2Wry r> )ry r6 c S s g | ]}t j |¡gqS r- )rc r r rY r- r- r. r~ w s z7test_dt64arr_addsub_object_dtype_2d.<locals>.<listcomp>r r1 r6 c s s | ]}|j d kV qdS )r N)rr )r{ rj r- r- r. Ú <genexpr> s z6test_dt64arr_addsub_object_dtype_2d.<locals>.<genexpr>)r rC Zreshaper r r` Úshaper r r® r rK ra r\ r¹ Zravel)r' rD r( r+ r, r r- r- r. Ú#test_dt64arr_addsub_object_dtype_2dq s 2 rÅ c C sr t dddgdd} | t dd¡ }t tgd d d}t ||¡ | t dd¡ }t tgd dd}t ||¡ d S )
|
Nì ÈCñE|ì |<Ø ì ({?M5údatetime64[ms]rR r Úmsr ztimedelta64[ms]©r r r r r rm râ ©rU r+ r, r- r- r. Ú(test_non_nano_dt64_addsub_np_nat_scalars s rÍ c C sn t dddgdd} | t d¡ }t tgd dd}t ||¡ | t d¡ }t tgd d d}t ||¡ d S )
|
NrÆ rÇ rÈ rÉ rR r r rí r rË rÌ r- r- r. Ú1test_non_nano_dt64_addsub_np_nat_scalars_unitless s rÎ c C sr t dddgdd} | t dd¡ }t tgd d d}t ||¡ | t dd¡ }t tgd dd}t ||¡ d S )
|
Ni,0 iËZ iÛ z datetime64[s]rR r r= r ztimedelta64[s]rË rÌ r- r- r. Ú9test_non_nano_dt64_addsub_np_nat_scalars_unsupported_unit¢ s rÏ )6r r r Ú itertoolsr r r r Únumpyr rL r Z pandas._libs.tslibs.conversionr Zpandas._libs.tslibs.offsetsr Z pandas.errorsr Zpandasrc r
|
r r r r r r r r Zpandas._testingZ_testingr Zpandas.core.opsr Z pandas.tests.arithmetic.commonr r r r r re r rÐ r re rx r rZ r[ r rÅ rÍ rÎ rÏ r- r- r- r. Ú<module> sT , p k ) %z ~ ~
|
