Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/arithmetic/__pycache__/test_datetime64.cpython-37.pyc
Назад
B �5�g�d � @ 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 dlmZ d dlZd dlmZmZmZmZmZmZm Z m Z m Z d dl m Z! d d l"m#Z#m$Z$ d d l%m&Z& d d l'm(Z(m)Z)m*Z* G d d � d �Z+G dd� d�Z,G dd� d�Z-G dd� d�Z.G dd� d�Z/G dd� d�Z0G dd� d�Z1G dd� d�Z2e j3�4d d d d g�e j3�4d d d d!g�d"d#� ��Z5d$d%� Z6dS )&� )�datetime�time� timedelta)�product�starmapN)�localize_pydatetime)� shift_months)�np_datetime64_compat)�PerformanceWarning) � DateOffset� DatetimeIndex�NaT�Period�Series� Timedelta�TimedeltaIndex� Timestamp� date_range)� DatetimeArray�TimedeltaArray)� roperator)�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�deed��e �d�e �d��e j�e �d��e�ejddd�je �ejddd��eejddd��ejddd��e�ejdddd�jejdddd��e�g �dd� �Zdd� Zdd� Zd S ) �"TestDatetime64ArrayLikeComparisonsc C s� |}|}|t jt jgkr |ntj}tdd|d�}t�|�� d �}t�||�}||k} t�dddg�} t�| |�} t� | | � d S )N�20130101� )�periods�tzr TF) �pd�Index�array�np�ndarrayr Zto_numpy�tm� box_expected� assert_equal) �self�tz_naive_fixture�box_with_arrayr �box�xbox�dti�other�dtarr�result�expected� r1 �]/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/arithmetic/test_datetime64.py�test_compare_zerodim: s z7TestDatetime64ArrayLikeComparisons.test_compare_zerodimr- �foo����c g @� )�daysi� � Nc C s. |}t dd|d�}t�||�}t|||� d S )Nz1/1/2000� )r r )r r$ r% r )r' r- r( r) r �rngr. r1 r1 r2 � test_dt64arr_cmp_scalar_invalidK s zBTestDatetime64ArrayLikeComparisons.test_dt64arr_cmp_scalar_invalidr: Z1ns)r z1 Dayz 1971-01-01�D)�freqr c C s( |}t ddd|d�j}t||tj� d S )Nz 1970-01-01�nsr: )r> r r )r �_datar r$ �to_array)r' r- r( r �dtar1 r1 r2 �"test_dt64arr_cmp_arraylike_invalide s zETestDatetime64ArrayLikeComparisons.test_dt64arr_cmp_arraylike_invalidc C s |}t ddd|d�j}t�ddd|d tdd �g�}||k}t�d d d d d g�}t�||� ||k}t�|| � d }tjt |d �� ||k W d Q R X tjt |d �� ||k W d Q R X tjt |d �� ||k W d Q R X tjt |d �� ||k W d Q R X d S )Nz 1970-01-01�h� )r> r r r r9 r7 r )r8 FTzDInvalid comparison between|Cannot compare type|not supported between)�match) r r@ r" r! r r$ �assert_numpy_array_equal�pytest�raises� TypeError)r' r( r rB r- r/ r0 �msgr1 r1 r2 � test_dt64arr_cmp_mixed_invalid| s" zATestDatetime64ArrayLikeComparisons.test_dt64arr_cmp_mixed_invalidc C sz |}|}|t jt jgkr |ntj}t�|�}t|tg�}t � ||�}tddgtj d�} t � | |�} ||k} t � | | � d S )NTF)�dtype) r r r! r" r# r �nowr r r$ r% Zbool_r& ) r' r( r) r r* r+ �ts�ser�objr0 r/ r1 r1 r2 �test_dt64arr_nat_comparison� s z>TestDatetime64ArrayLikeComparisons.test_dt64arr_nat_comparison)�__name__� __module__� __qualname__r3 rH �mark� parametrize�objectr r �dater" �nanr< �list�rangeZarange�astypeZfloat32r �timedelta_ranger! Z period_rangerC rL rR r1 r1 r1 r2 r 5 s4 " r c @ s� e Zd Zej�ded�eed�geeed�gfed�eed�geeed�gfe ddd�ee d dd�geee d dd�gfg�ej�d d d g�ej�d de g�ej�de j e d d d g�fe je d d d g�fe je d d d g�fe je d d d g�fe je d d d g�fe je d d d g�fg�dd� ����Zdd� Zej�ded�eed�ged�eed�ge ddd�ee d dd�gg�ej�d de g�dd� ��Zej�deddd�eddd�g�d d � �Zej�d d d d!d"g�d#d$� �Zd%d&� ZdS )'� TestDatetime64SeriesComparison�pairz 2011-01-01z 2011-01-03z1 daysz3 daysz2011-01�M)r> z2011-03�reverseTFrM Nz op, expectedc C sL |}|\}} |r| | }} t ||d�} || |d�} || | �} t�| |� d S )N)rM )r r$ �assert_series_equal) r' rM �index_or_seriesrb r` �opr0 r* �l�r�left�rightr/ r1 r1 r2 �test_nat_comparisons� s ' z3TestDatetime64SeriesComparison.test_nat_comparisonsc C sJ |}t td��}t tdd|d��}t�||�}t�||�}t|||� d S )NrE �20010101)r r )r r\ r r$ r% r )r' r( r) r rP Zser2r1 r1 r2 �test_comparison_invalid� s z6TestDatetime64SeriesComparison.test_comparison_invalid�datac C s� |}|t jkr|tkrd S |tjtjgkr.|ntj}t||d�}t � ||�}dddg}t � ||�}|tjkr�|tkr�tj|dd�}t � |t k|� t � t |k|� dddg}t � ||�}|tjkr�|tkr�tj|dd�}t � |t k|� t � t |k|� dddg}t � ||�}|tjk�r2|tk�r2tj|dd�}t � |t k |� t � t |k|� t � |t k|� t � t |k|� t � |t k|� t � t |k |� t � |t k|� t � t |k|� d S )N)rM F�boolT) r$ rA rX r r r! r" r# r r% r& r )r' rM rm r) r* r+ rh r0 r1 r1 r2 �test_nat_comparisons_scalar� s< z:TestDatetime64SeriesComparison.test_nat_comparisons_scalar�vali� r9 � rE c s> t tddd��}|� k}t � fdd�|D ��}t�||� d S )Nz1/1/2000r: )r c s g | ] }|� k�qS r1 r1 )�.0�x)rp r1 r2 � <listcomp> s zQTestDatetime64SeriesComparison.test_series_comparison_scalars.<locals>.<listcomp>)r r r$ rc )r' rp Zseriesr/ r0 r1 )rp r2 � test_series_comparison_scalars s z=TestDatetime64SeriesComparison.test_series_comparison_scalarsz left,right)�lt�gt)�le�ge)�eqrz )�ner{ c C s� t tddd�dd�}|jdd�}td�|d < td�|d < tt|�}tt|�}||td ��}|td �|�}t�||� ||td��}|td�|�}t�||� ||td ��}|td �|�}t�||� ||td��}|td�|�}t�||� d S ) Nrk r: )r �dates)�nameT)�deep�natr r Z20010109)r r �copyr �getattr�operatorr$ rc ) r' rh ri rP Zs_natZleft_fZright_fr0 r/ r1 r1 r2 � test_timestamp_compare_series" s$ z<TestDatetime64SeriesComparison.test_timestamp_compare_seriesc C s� |t jt jgkr|ntj}ttd�td�tg�}t� ||�}||k}t� dddg|�}t� ||� |t j krpt nd }t� |�� ||d k}W d Q R X t� dddg|�}t� ||� t� |�� ||d k}W d Q R X t� dddg|�}t� ||� ||k}t� dddg|�}t� ||� t� |�� ||d k}W d Q R X t� dddg|�}t� ||� t� |�� ||d k}W d Q R X t� dddg|�}t� ||� d S )Nz2000-01-29 01:59:00z 2000-01-30FTr r7 )r r r! r" r# r r r r$ r% r& � DataFrame� FutureWarning�assert_produces_warning)r' r) r+ rP r/ r0 �warnr1 r1 r2 � test_dt64arr_timestamp_equalityE s4 z>TestDatetime64SeriesComparison.test_dt64arr_timestamp_equality)rS rT rU rH rV rW r r r r rX r� rz r r{ rv rw ry rx rj rl ro r ru r� r� r1 r1 r1 r2 r_ � s: & **"r_ c @ s� e Zd Zej�dejejej ej ej ej g�dd� �Z ej�deddd�ed�e�d�g�dd � �Zej�d d eg�d d � �Zdd� Zej�dejejej ej ej ej g�dd� �Zej�dejejej ej ej ej g�dd� �Zej�dejejej ej ej ej g�ej�deddd�ed�e�d�g�ej�d�ej�d�dd� ����Zej�dejejej ej ej ej g�dd� �Zdd� Zd d � Z ejjde j d d d �e j d d d �� � e j d d d �j �!d!�gd"d#� d$�d%d&� �Z"d'd(� Z#d S ))� TestDatetimeIndexComparisonsre c C sd t �d�}|t|�d }t|��� }t�|�}|||�}|||�}t|tj�sTt �t � ||� d S )N�d r7 ) r$ Z makeDateIndex�lenr � to_datetime64r" r! � isinstancer# �AssertionErrorrG )r' re �index�elementZarrZ arr_resultZ index_resultr1 r1 r2 �test_comparatorss s z-TestDatetimeIndexComparisons.test_comparatorsr- i� r9 z 2016-01-01c C s� |}t dd|d�}|d k r6t|tj�r*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 )Nz 2016-01-01r7 )r r TF) r r� r" � datetime64r �tzinfor! r$ rG )r' r- r( r r, r/ r0 r1 r1 r2 �test_dti_cmp_datetimelike� s* z6TestDatetimeIndexComparisons.test_dti_cmp_datetimelikerM Nc C s� |t jkr|tkrd S |tjtjgkr*|ntj}tt d�t t d�g�}tt t t d�g�}t � ||�}t � ||�}|| }}|tkr�|� t�|� t� }}||k}t�dddg�} t � | |�} t � || � ||k}t�dddg�} t � | |�} t � || � t�dddg�} t � | |�} t � |t k| � t � t |k| � t�dddg�} t � | |�} t � |t k| � t � t |k| � t�dddg�} t � | |�} t � |t k | � t � t |k| � d S )Nz 2011-01-01z 2011-01-03FT)r$ rA rX r r r! r" r# r r r r% r] r& ) r' rM r) r+ rh ri �lhs�rhsr/ r0 r1 r1 r2 �test_dti_cmp_nat� s< z-TestDatetimeIndexComparisons.test_dti_cmp_natc C s� t �dtjdtjddg�}t �ddtjtjddg�}tdtdtd d g�}td dttd d g�}t�td �td�td�t�d�td�td�g�}||f||f||fg}t � d ��� �x|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 W d Q R X t � d ��� x�|tjf|tfgD ]�\}} || k } t�ddddddg�} t � | | � || k} t � | | � || k} t � | | � || k} t � | | � || k} t � | | � || k} t�ddddddg�} t � | | � �q�W W d Q R X t � d ���. �x$|df|t ddd�fgD �]\}} || 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 W d Q R X d S )Ng �?g @g @g @g @g @z 2014-01-01z 2014-03-01z 2014-05-01z 2014-07-01z 2014-02-01z 2014-06-01z2014-02-01 00:00Zz2014-03-01 00:00Zr z2014-06-01 00:00Zz2014-07-01 00:00ZTFr i� r9 ) r r r" rZ r r r! r r� r$ r� rG r ) r' Zfidx1Zfidx2Zdidx1Zdidx2�darrZcasesZidx1Zidx2r/ r0 rp r1 r1 r2 �+test_dti_cmp_nat_behaves_like_float_cmp_nan� s� & zHTestDatetimeIndexComparisons.test_dti_cmp_nat_behaves_like_float_cmp_nanc C sp |}t ddd�}|�d�}t�||�}t�||�}|tjkrFdd� }nt}|tjtj gk�r\d}t j t |d�� |||� W d Q R X t j t |d�� ||||�� W d Q R X t j t |d�� ||t j||�td �� W d Q R X t j t |d�� |||� W d Q R X t j t |d�� ||||�� W d Q R X t j t |d�� ||t j||�td �� W d Q R X t �||k��spt�t �|||�k��s�t�t �||�|k��s�t�t �t j||�td �|k��s�t�t �|t j||�td �k��s�t�t �||k��s�t�t �|||�k��st�t �||�|k��s(t�t �t j||�td �|k��sJt�t �|t j||�td �k��slt�d S ) Nz 2016-01-01� )r z US/Pacificc S s | � t�j�� d S )Nr )r] rX �values�tolist)rs r1 r1 r2 �<lambda>A � zQTestDatetimeIndexComparisons.test_comparison_tzawareness_compat.<locals>.<lambda>z^Invalid comparison between dtype=datetime64\[ns.*\] and (Timestamp|DatetimeArray|list|ndarray))rF )rM )r � tz_localizer$ r% r r� r[ r� rz r{ rH rI rJ r" r! rX �allr� )r' re r) r* �dr�dzr� rK r1 r1 r2 �"test_comparison_tzawareness_compat2 s@ """""z?TestDatetimeIndexComparisons.test_comparison_tzawareness_compatc C s t ddd�}|�d�}t�||�}t�||�}td�}tddd�}t�||k�sTt�d}|tj tj gkr�t j t |d �� |||� W d Q R X t�||k�s�t�|tj tj gkr�t j t |d �� |||� W d Q R X |tj tj gk�rt j t |d �� |||� W d Q R X d S ) Nz 2016-01-01r� )r z US/Pacificz2000-03-14 01:59zEurope/Amsterdam)r zAInvalid comparison between dtype=datetime64\[ns.*\] and Timestamp)rF )r r� r$ r% r r" r� r� r� rz r{ rH rI rJ )r' re r) r� r� rO Zts_tzrK r1 r1 r2 �*test_comparison_tzawareness_compat_scalarsf s$ zGTestDatetimeIndexComparisons.test_comparison_tzawareness_compat_scalarsz*ignore:elementwise comp:DeprecationWarningz.ignore:Converting timezone-aware:FutureWarningc C s� |}|}t dd|d�}|tjtjgkr*|ntj}t�||�} |tj tj gkr�|tj k} tj| | gt d�} t�| |�} || |�} t� | | � ||| �} t� | | � nXdt |�j� �} tjt| d�� || |� W d Q R X tjt| d�� ||| � W d Q R X d S )Nz 2016-01-01r7 )r r )rM z:Invalid comparison between dtype=datetime64\[ns, .*\] and )rF )r r r r! r" r# r$ r% r� rz r{ rn r& �typerS rH rI rJ )r' re r- �tz_aware_fixturer) r* r r, r+ r. Zexboolr0 r/ rK r1 r1 r2 �"test_scalar_comparison_tzawareness� s$ z?TestDatetimeIndexComparisons.test_scalar_comparison_tzawarenessc C sd t dtdtddg�}t�|tjkgt|� �}||t�}t�||� ||� d�t�}t�||� d S )Nz 2014-01-01z 2014-03-01z 2014-05-01z 2014-07-01z US/Pacific) r r r" r! r� r{ r� r$ rG r� )r' re r, r0 r/ r1 r1 r2 � 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 )Nz1/1/2000r: )r r TF� )r r" r! r$ rG )r' r( r r; r- r/ r0 r1 r1 r2 �test_dti_cmp_str� s* z-TestDatetimeIndexComparisons.test_dti_cmp_strc C s0 t ddd�}|t|�k}||k}t�||� d S )Nz1/1/2000r: )r )r r[ r$ rG )r' r; r/ r0 r1 r1 r2 �test_dti_cmp_list� s z.TestDatetimeIndexComparisons.test_dti_cmp_listZ1Dr: )r zm8[ns]c C s t | �jS )N)r� rS )rs r1 r1 r2 r� � r� z%TestDatetimeIndexComparisons.<lambda>)�idsc C s� t dddd�}||k}t�dgd �}t�||� ||k}t�dgd �}t�||� d}tjt|d�� ||k W d Q R X tjt|d�� ||k W d Q R X tjt|d�� ||k W d Q R X tjt|d�� ||k W d Q R X d S ) Nz 2000-01-01r: z Asia/Tokyo)r r FTzInvalid comparison between)rF )r r" r! r$ rG rH rI rJ )r' r- r, r/ r0 rK r1 r1 r2 � 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 d Q R X d S ) Nz 2000-01-01r: z Asia/Tokyo)r r �OTrE r9 )r8 FzB>=' not supported between instances of 'Timestamp' and 'Timedelta')rF ) r r] r" r! r$ rG r� r[ r rH rI rJ )r' r, r- r/ r0 rK r1 r1 r2 �test_dti_cmp_object_dtype s & z6TestDatetimeIndexComparisons.test_dti_cmp_object_dtype)$rS rT rU rH rV rW r� rz r{ rw rv ry rx r� r r r" r� r� rX r� r� r� r� �filterwarningsr� r� r� r� r r^ Z to_series�asi8�viewr� r� r1 r1 r1 r2 r� p sD && .`&2& & r� c @ s> e Zd Zejjdd� �Zdd� Zdd� Zdd� Z d d � Z d d � Z d d� Z ej� ded�ed��� ed��� g�dd� �Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zej� d#d$e�d%d&g�e d'd(d)�e d*d+d,�g�ej� d-d.d(g�d/d0� ��Z ej� d1d(d2d3d4g�ej� d-d.d(g�d5d6� ��Z d7d8� Z d.S )9�TestDatetime64Arithmeticc C sP |}t dd|d�}t dd|d�}t�||�}t�||�}|| }t�||� d S )Nz 2000-01-01z 2000-02-01)r z2000-01-01 02:00z2000-02-01 02:00)r r$ r% r& )r' r( � two_hoursr) r r; r0 r/ r1 r1 r2 �%test_dt64arr_add_timedeltalike_scalar) s z>TestDatetime64Arithmetic.test_dt64arr_add_timedeltalike_scalarc C sP |}t dd|d�}t dd|d�}t�||�}t�||�}||7 }t�||� d S )Nz 2000-01-01z 2000-02-01)r z2000-01-01 02:00z2000-02-01 02:00)r r$ r% r& )r' r( r� r) r r; r0 r1 r1 r2 �&test_dt64arr_iadd_timedeltalike_scalar9 s z?TestDatetime64Arithmetic.test_dt64arr_iadd_timedeltalike_scalarc C sP |}t dd|d�}t dd|d�}t�||�}t�||�}|| }t�||� d S )Nz 2000-01-01z 2000-02-01)r z1999-12-31 22:00z2000-01-31 22:00)r r$ r% r& )r' r( r� r) r r; r0 r/ r1 r1 r2 �%test_dt64arr_sub_timedeltalike_scalarG s z>TestDatetime64Arithmetic.test_dt64arr_sub_timedeltalike_scalarc C sP |}t dd|d�}t dd|d�}t�||�}t�||�}||8 }t�||� d S )Nz 2000-01-01z 2000-02-01)r z1999-12-31 22:00z2000-01-31 22:00)r r$ r% r& )r' r( r� r) r r; r0 r1 r1 r2 �&test_dt64arr_isub_timedeltalike_scalarU s z?TestDatetime64Arithmetic.test_dt64arr_isub_timedeltalike_scalarc C s� t td�td�g�}t td�td�g�}t�||�}t�||�}|t�dd� }t�||� t�dd�| }t�||� t td�td�g�}t�||�}|t�d d � }t�||� t�d d �| }t�||� d S ) Nz 20130101 9:01z 20130101 9:02z20130101 9:01:01z20130101 9:02:01r9 �sz20130101 9:01:00.005z20130101 9:02:00.005rE �ms)r r r$ r% r" � timedelta64r& )r' r) rP r0 r. r/ r1 r1 r2 � test_dt64arr_add_td64_scalard s z5TestDatetime64Arithmetic.test_dt64arr_add_td64_scalarc C s� |}t dd|dd�}t�d�}tdgd |d�}t�||�}t�||�}|| }t�||� || }t�||� || }t�||� d} tjt | d�� || W d Q R X d S ) Nz 1994-04-01r� ZQS)r r r> r )r zcannot subtract)rF ) r r" r� r r$ r% r& rH rI rJ ) r' r) r( r r, r- r0 rQ r/ rK r1 r1 r2 � 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 d Q R X d S ) Nz 2016-01-01r )r r z-1 Dayz 2015-12-31z 2016-01-02z 2016-01-04z8cannot subtract|(bad|unsupported) operand type for unary)rF ) r r r� r$ r% r& rH rI rJ ) r' r( r) r r, �tdiZtdarrr0 r. r/ rK r1 r1 r2 � test_dt64arr_add_sub_td64ndarray� s$ z9TestDatetime64Arithmetic.test_dt64arr_add_sub_td64ndarrayrO z 2013-01-01c C sP t ddd��d �}t�||�}tdddg�}t�||�}|| }t�||� d S )Nz 2013-01-01r )r z0 Daysz1 Dayz2 Days)r � _with_freqr$ r% r r& )r' r) rO �idxr0 r/ r1 r1 r2 �test_dt64arr_sub_dtscalar� s z2TestDatetime64Arithmetic.test_dt64arr_sub_dtscalarc C s~ t �d�}|jdkst�tddd��d �}t�||�}tdddg�}t�||�}|| }t� ||� || }t� || � d S ) Nz 2013-01-01z datetime64[D]r r )r z0 Daysz1 Dayz2 Days) r" r� rM r� r r� r$ r% r r& )r' r) Zdt64r, r. r0 r/ r1 r1 r2 �"test_dt64arr_sub_datetime64_not_ns� s z;TestDatetime64Arithmetic.test_dt64arr_sub_datetime64_not_nsc C s| t ddddd�}|�d �}|d }t�||�}tt�dd�t�dd�g�}t�||�}t�|| |� t�|| | � d S )Nz 2014-03-17r7 r= z US/Eastern)r r> r r r9 )r r� r$ r% r r" r� r& )r' r) rP rO Z delta_seriesr0 r1 r1 r2 �test_dt64arr_sub_timestamp� s z3TestDatetime64Arithmetic.test_dt64arr_sub_timestampc C s� t ttd�g�}t�||�}|t }tttgdd�}t�||�}t�||� |�d�}t�||�}|t }tttgdd�}t�||�}t�||� d S )N�19900315ztimedelta64[ns])rM z Asia/Tokyo)r r r r$ r% r r& r� )r' r) r, rP r/ r0 �dti_tzZser_tzr1 r1 r2 �test_dt64arr_sub_NaT� s z-TestDatetime64Arithmetic.test_dt64arr_sub_NaTc C sb t dd|d�}|| }t�||�}t�||�}t�t�� ||�t� }W d Q R X t�||� d S )Nz 2016-01-01r )r r )r r$ r% r� r r] rX r& )r' r) r( r, r0 rQ r/ r1 r1 r2 �!test_dt64arr_sub_dt64object_array� s z:TestDatetime64Arithmetic.test_dt64arr_sub_dt64object_arrayc C sT t ddd d�}|j}t�||�}|| }|| }t�||� || }t�||� d S )Nz 2016-01-01r )r r )r r� r$ r% r& )r' r) r, �dt64valsr. r0 r/ r1 r1 r2 �"test_dt64arr_naive_sub_dt64ndarray s z;TestDatetime64Arithmetic.test_dt64arr_naive_sub_dt64ndarrayc C sp |}t dd|d�}|j}t�||�}d}tjt|d�� || W d Q R X tjt|d�� || W d Q R X d S )Nz 2016-01-01r )r r z+subtraction must have the same timezones or)rF )r r� r$ r% rH rI rJ )r' r� r) r r, r� r. rK r1 r1 r2 �)test_dt64arr_aware_sub_dt64ndarray_raises s zBTestDatetime64Arithmetic.test_dt64arr_aware_sub_dt64ndarray_raisesc C sp |}t dd|d�}|j}t�||�}d}tjt|d�� || W d Q R X tjt|d�� || W d Q R X d S )Nz 2016-01-01r )r r z cannot add)rF )r r� r$ r% rH rI rJ )r' r( r) r r, r� r. rK r1 r1 r2 �#test_dt64arr_add_dt64ndarray_raises! s z<TestDatetime64Arithmetic.test_dt64arr_add_dt64ndarray_raisesc C sl t ddg�}t�||�}d}tjt|d�� |td� W d Q R X tjt|d�� td�| W d Q R X d S )Nz 2011-01-01z 2011-01-02z cannot add)rF )r r$ r% rH rI rJ r )r' r) r� rK r1 r1 r2 �!test_dt64arr_add_timestamp_raises. s z:TestDatetime64Arithmetic.test_dt64arr_add_timestamp_raisesr- g ��Q� @g @g @z 2011-01-01r= )r> r9 r7 r �dti_freqNc C s@ t ddg|d�}t�||�}d�ddddd g�}t|||� d S ) Nz 2011-01-01z 2011-01-02)r> �|zunsupported operand typezcannot (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% �joinr )r' r� r- r) r, r. rK r1 r1 r2 � test_dt64arr_add_sub_invalid; s z5TestDatetime64Arithmetic.test_dt64arr_add_sub_invalid�pi_freq�W�Q�Hc C sT t ddg|d�}|�|�}t�||�}t�||�}d�ddddg�} t||| � d S ) Nz 2011-01-01z 2011-01-02)r> r� zcannot (add|subtract)zunsupported operandzdescriptor.*requireszufunc.*cannot use operands)r Z to_periodr$ r% r� r ) r' r� r� r) Zbox_with_array2r, �pir. ZparrrK r1 r1 r2 �test_dt64arr_add_sub_parrU s z2TestDatetime64Arithmetic.test_dt64arr_add_sub_parrc C s |}t dd|d�}dd� td�D �}t�||�}t�||�}tjdd��� t�dt� d }tj t |d �� || W d Q R X d � d d g�}tj t |d �� || W d Q R X d}tj t |d �� || W d Q R X d � ddg�}tj t |d �� || W d Q R X W d Q R X d S )Nz 2012-01-01r )r r c S s g | ]}t |||��qS r1 )r )rr �ir1 r1 r2 rt p s zTTestDatetime64Arithmetic.test_dt64arr_addsub_time_objects_raises.<locals>.<listcomp>T)�record�ignorezTunsupported operand type\(s\) for -: '(Timestamp|DatetimeArray)' and 'datetime.time')rF r� z*cannot subtract DatetimeArray from ndarrayz`ufunc (subtract|'subtract') cannot use operands with types dtype\('O'\) and dtype\('<M8\[ns\]'\)zUunsupported operand type\(s\) for \+: '(Timestamp|DatetimeArray)' and 'datetime.time'zVufunc (add|'add') cannot use operands with types dtype\('O'\) and dtype\('<M8\[ns\]'\)) r r\ r$ r% �warnings�catch_warnings� simplefilterr rH rI rJ r� )r' r) r( r Zobj1�obj2rK r1 r1 r2 �'test_dt64arr_addsub_time_objects_raisesj s. z@TestDatetime64Arithmetic.test_dt64arr_addsub_time_objects_raises)!rS rT rU rH rV Zarm_slowr� r� r� r� r� r� r� rW r � to_pydatetimer� r� r� r� r� r� r� r� r� r� r" r! r r r� r� r� r1 r1 r1 r2 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/d2d3�fd4d5d/d2d6�fd7d/d/d8�fd9d:d;d<ifd:ddifg!�ej�d=d>d?g�ej�d@d5dg�dAdB� ���Z dCdD� Z ej�dEe �ej�� ejjd/dF�g�e �ejjd1dG�ej�� g�e �ejjd1dG�ejjd1dG�g�g�ej�dHejejejg�ej�dId>d?g�dJdK� ���Zej�dLdMed dNdO�edP�edQ�edR�edS�gdTfdMed dU�edV�edW�edX�edY�gdZfd[ed dNdO�ed\�ed]�ed^�ed_�gdTfd[ed dU�ed`�eda�edb�edc�gddfg�dedf� �ZdTS )g�"TestDatetime64DateOffsetArithmeticc C s| t td�td�g�}t td�td�g�}t�||�}t�||�}|tj�d� }t�||� tj�d�| }t�||� d S )Nz 20130101 9:01z 20130101 9:02z20130101 9:01:05z20130101 9:02:05rE )r r r$ r% r �offsets�Secondr& )r' r) rP r0 r/ �result2r1 r1 r2 �'test_dt64arr_series_add_tick_DateOffset� s zJTestDatetime64DateOffsetArithmetic.test_dt64arr_series_add_tick_DateOffsetc C s� t td�td�g�}t td�td�g�}t�||�}t�||�}|tj�d� }t�||� tj�d� | }t�||� d}tj t |d�� tj�d�| W d Q R X d S )Nz 20130101 9:01z 20130101 9:02z20130101 9:00:55z20130101 9:01:55rE z((bad|unsupported) operand type for unary)rF ) r r r$ r% r r� r� r& rH rI rJ )r' r) rP r0 r/ r� rK r1 r1 r2 �'test_dt64arr_series_sub_tick_DateOffset� s zJTestDatetime64DateOffsetArithmetic.test_dt64arr_series_sub_tick_DateOffset�cls_name�Day�Hour�Minuter� �MilliZMicroZNanoc C sT t td�td�g�}t�||�}ttj|�}||d� |d�| ||d� d S )Nz 20130101 9:01z 20130101 9:02rE )r r r$ r% r� r r� )r' r� r) rP � offset_clsr1 r1 r2 �*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 ksJt�t dd|dd �}td d d gd|d �}t�||�}t�||�}|tj�d� }t�||� |t � dd� }t�||� |t dd� }t�||� d S )Nz US/Pacificz 2012-11-01r )r r rE r z2010-11-01 00:00r� )r r r> z2010-11-01 05:00z2010-11-01 06:00z2010-11-01 07:00)r> r rD )�hours) r r r� r� r� r r$ r% r& r" r� r )r' r� r) r r| �offsetr0 r1 r1 r2 �test_dti_add_tick_tzaware� s$ z<TestDatetime64DateOffsetArithmetic.test_dti_add_tick_tzawarec s� t td�td�td�td�td�td�td�td�g�}t�||�}|tjkrX|jd n|}d d d d dddg}�x0t|�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 d Q R X qzW d S )Nz2000-01-05 00:15:00z2000-01-31 00:23:00z 2000-01-01z 2000-03-31z 2000-02-29z 2000-12-31z 2000-05-15z 2001-06-15r )�yearsr7 )�monthsrE )r8 r )r� rE )�minutesr: )�secondsr7 )� microsecondsrE c s g | ] }|� �qS r1 r1 )rr rs )�offr1 r2 rt s zaTestDatetime64DateOffsetArithmetic.test_dt64arr_add_sub_relativedelta_offsets.<locals>.<listcomp>c s g | ] }|� �qS r1 r1 )rr rs )r� r1 r2 rt s r9 c s g | ] }|� �qS r1 r1 )rr rs )r� r1 r2 rt s c s g | ] }|� �qS r1 r1 )rr rs )r� r1 r2 rt # s z((bad|unsupported) operand type for unary)rF )r r r$ r% r r� �iloc� enumerater r& �dictrH rI rJ ) r' r) �vec� vec_itemsZrelative_kwargsr� �unit�valuer0 rK r1 )r� r2 �*test_dt64arr_add_sub_relativedelta_offsets� sH zMTestDatetime64DateOffsetArithmetic.test_dt64arr_add_sub_relativedelta_offsets�cls_and_kwargsZ YearBegin�monthrE ZYearEndZ MonthBegin�MonthEndZ SemiMonthEndZSemiMonthBeginZWeek�weekdayr r� Z BusinessDayZBDayZ QuarterEndZ QuarterBeginZCustomBusinessDayZCDayZ CBMonthEndZ CBMonthBeginZ BMonthBeginZ BMonthEndZ BusinessHourZ BYearBeginZBYearEndZ BQuarterBegin�LastWeekOfMonthr7 � FY5253Quarterr9 Znearest)Zqtr_with_extra_week� startingMonthr � variation�FY5253r )r r r � WeekOfMonth)r �weekZEasterr �dayrq � normalizeTF�nc s� t |t�r|\}}n|}i }|dkr0|dkr0d S ttd�td�td�td�td�td�td �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 d Q R X W d Q R X d S )Nr )r r r r z2000-01-05 00:15:00z2000-01-31 00:23:00z 2000-01-01z 2000-03-31z 2000-02-29z 2000-12-31z 2000-05-15z 2001-06-15T)r� r� r c s g | ] }|� �qS r1 r1 )rr rs )r� r1 r2 rt � s zWTestDatetime64DateOffsetArithmetic.test_dt64arr_add_sub_DateOffsets.<locals>.<listcomp>c s g | ] }|� �qS r1 r1 )rr rs )r� r1 r2 rt � s c s g | ] }� | �qS r1 r1 )rr rs )r� r1 r2 rt � s z((bad|unsupported) operand type for unary)rF )r� �tupler r r$ r% r r� r� r� r� r� r� r� r r& rH rI rJ ) r' r) r r r� r� �kwargsr� r� r� r0 rK r1 )r� r2 � test_dt64arr_add_sub_DateOffsets/ sB 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 )Nz 2000-01-01z 2000-01-31�a)r} r9 )r� z 2001-01-01z 2001-01-31z 1999-01-01z 1999-01-31z2000-01-15 00:15:00z US/Central)r z 2000-02-15z2000-01-16 00:15:00z 2000-02-16z2000-01-31 00:15:00z 2000-02-29) r r$ r% r r� r& r r r r� r� r )r' r) r� r/ r� �expr1 r1 r2 � test_dt64arr_add_sub_DateOffset� sP zBTestDatetime64DateOffsetArithmetic.test_dt64arr_add_sub_DateOffsetr- )r )r� re � box_otherc s� |}t dd|d�� t�� |�}t�tj�� tjjdd�g��t � ��fdd�t t � ��D ��}t�||�}|rzt��|��t� t �� �|��} W d Q R X t�| |� d S )Nz 2017-01-01r7 )r r )r c s g | ]}�� | �| ��qS r1 r1 )rr r )r, re r- r1 r2 rt � s zXTestDatetime64DateOffsetArithmetic.test_dt64arr_add_sub_offset_array.<locals>.<listcomp>)r r$ r% r" r! r r� r r� r r\ r� r� r r& ) r' r( r) r re r- r r. r0 �resr1 )r, re r- r2 �!test_dt64arr_add_sub_offset_array� s " zDTestDatetime64DateOffsetArithmetic.test_dt64arr_add_sub_offset_arrayzop, offset, exp, exp_freq�__add__r: )r� r8 z 2014-04-11z 2015-04-11z 2016-04-11z 2017-04-11N)r� 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 F)r )r r$ r% r� r r& ) r' re r� r Zexp_freqr� r) r rY Zmthr/ r0 r1 r1 r2 �#test_dti_add_sub_nonzero_mth_offset� s 5 zFTestDatetime64DateOffsetArithmetic.test_dti_add_sub_nonzero_mth_offset) rS rT rU r� r� rH rV rW r� r� r� r r r" r! r r� r r� r r� �addr �radd�subr r r r1 r1 r1 r2 r� � s� 8 @: r� c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) � TestDatetime64OverflowHandlingc C s^ t td�g�}t tg�}t�||�}t�||�}ttg�}t�||�}|| }t�||� d S )Nz 1969-12-31)r r r r$ r% r r& )r' r) rh ri r0 r/ r1 r1 r2 �test_dt64_overflow_masking= s z9TestDatetime64OverflowHandling.test_dt64_overflow_maskingc C sH t d�}td�}tdddd�}t|�}d}tjt|d�� || W d Q R X tjt|d�� || W d Q R X tjt|d�� || W d Q R X tjt|d�� || W d 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-30Z100Yrq )r> r zOverflow in int64 addition)rF r5 z 2004-10-03z 2104-10-04z 2204-10-04r zdatetime64[ns])rM r9 z 91279 Daysztimedelta64[ns]) r r r r rH rI � OverflowErrorr r� r$ rc )r' �dt�tdr, rP rK r0 r r1 r1 r2 � test_dt64_series_arith_overflowK s4 z>TestDatetime64OverflowHandling.test_dt64_series_arith_overflowc C sN t �dtjg�}t �dtjg�}td�}||�� |�� �d�|�� �d�g}td�}||�� |�� �d�|�� �d�g}d}x.|D ]&}tj t |d�� || W d Q R X q�W tjj |j } x&|D ] }|| } | d j | ks�t �q�W tjj |j } x&|D ] }|| } | d j | ks�t �q�W x0|D ](}tj t |d�� || W d Q R X �q W d S ) NrN z 1950-01-01zdatetime64[ns]z datetime64[D]z 1980-01-01zOverflow in int64 addition)rF r9 ) r � to_datetimer �max�minr� r� r] rH rI r r� r� ) r' �dtimax�dtiminZtsnegZts_neg_variantsZtsposZts_pos_variantsrK �variantr0 r r1 r1 r2 �)test_datetimeindex_sub_timestamp_overflowk s6 zHTestDatetime64OverflowHandling.test_datetimeindex_sub_timestamp_overflowc C sd t �dtjg�}t �dtjg�}t �ddg�}t �ddg�}tjj|d j }|| }|d j|ksht�tjj|d j }|| }|d j|ks�t�d}tjt |d�� || W d Q R X tjt |d�� || W d Q R X t �tjg�}|t j t d� } tjt |d�� | | W d Q R X t �tjg�} | t j t d� } tjt |d�� | | W d Q R X d S )NrN z 1950-01-01z 1980-01-01r9 zOverflow in int64 addition)rF Z1us) r r$ r r% r& r� r� rH rI r r ) r' r' r( Zts_negZts_posr0 r/ rK Ztmin�t1Ztmax�t2r1 r1 r2 �-test_datetimeindex_sub_datetimeindex_overflow� s. zLTestDatetime64OverflowHandling.test_datetimeindex_sub_datetimeindex_overflowN)rS rT rU r r# r* r- r1 r1 r1 r2 r : s &r c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z e j �deed�ed�g�eeed�g�eeegdd�g�e j �ddde�d�g�dd� ��Ze j �d d d d d g�d!d"� �Zd#d$� Zd%S )&� TestTimestampSeriesArithmeticc C sn t dd�}t dd�}t�||| � t�||| � t�||| � d}tjt|d�� || W d Q R X d S )NzM8[ns])rM zm8[ns]zcannot subtract)rF )r r$ rc rH rI rJ )r' r �brK r1 r1 r2 �test_empty_series_add_sub� s z7TestTimestampSeriesArithmetic.test_empty_series_add_subc C s� t tddd�gd �}tj|jd< t td�td�td�g�}tj|jd< t td�td �td �g�}|| || || || || || || d S ) NrE r )r� r� r7 �20111230�20120101�20120103�20111231�20120102�20120104)r r r" rZ r� r )r' �td1�dt1�dt2r1 r1 r2 �test_operators_datetimelike� s$ z9TestTimestampSeriesArithmetic.test_operators_datetimelikec C sV t tdddddd��}tddddd�}t|g�}t�t�|| ��}|jd ksRt�d S ) Ni� r9 � � � r r� � ztimedelta64[ns]) r r r r Z to_timedeltar" �absrM r� )r' rO r! rP r/ r1 r1 r2 � test_dt64ser_sub_datetime_dtype� s z=TestTimestampSeriesArithmetic.test_dt64ser_sub_datetime_dtypec s |� � fdd�}t tddd�gd �}tj|jd< t td�td�td �g�}tj|jd< t td �td �td �g�}� d kr�|||� � dkr�|||� d}t tdd|d�dd�}|�� }tj|jd< t tj dddd��}|�� }tj|jd< � dkr�|||� d S )Nc s2 t | � d �}tjtdd�� ||� W d Q R X d S )Nz%operate|[cC]annot|unsupported operand)rF )r� rH rI rJ )Zget_serZtest_serre )�op_strr1 r2 �check� s zPTestTimestampSeriesArithmetic.test_operators_datetimelike_invalid.<locals>.checkrE r )r� r� r7 r1 r2 r3 r4 r5 r6 )r �__rsub__)r �__radd__r z US/Easternz2000-01-01 09:00:00)r r r4 )r} z 1 days 1 minr� )r r> r9 )r rD r rC ) r r r" rZ r� r r r� r r^ )r' Zall_arithmetic_operatorsrB r7 r8 r9 r �td2r1 )rA r2 �#test_operators_datetimelike_invalid� s, zATestTimestampSeriesArithmetic.test_operators_datetimelike_invalidc C sl t tddd�g�}t tddd�g�}|| }t td�g�}t�||� || }t td�g�}t�||� d S )Nz 2016-02-10zAmerica/Sao_Paulo)r z 2016-02-08Z2daysz-2days)r r r r$ rc )r' �s1�s2r/ r0 r1 r1 r2 �test_sub_single_tz s z0TestTimestampSeriesArithmetic.test_sub_single_tzc C sT t dddd�}t|�}ttdgd ��}|| }t�||� || }t�||� d S )Nz 1999-09-30r: z US/Pacific)r r Z0days)r r r r$ rc )r' r, rP r0 r r1 r1 r2 � test_dt64tz_series_sub_dtitz* s z:TestTimestampSeriesArithmetic.test_dt64tz_series_sub_dtitzc C sh t tddddtjd�tg�}tddddtjd�}t td�tg�}t�|| |� t�|t|� |� d S )Ni� � � � )r� r> z1 days) r r �pytz�utcr r r$ rc r )r' r� r! r r1 r1 r2 �test_sub_datetime_compat6 s z6TestTimestampSeriesArithmetic.test_sub_datetime_compatc C s� t td�td�g�}|tj�d� }tj�d�| }t td�td�g�}t�||� t�||� |tj�d� tj�d� }t td�td�g�}t�||� d S )Nz 20130101 9:01z 20130101 9:02rE z20130101 9:01:00.005z20130101 9:02:00.005z20130101 9:06:00.005z20130101 9:07:00.005)r r r r� r� r$ rc r� )r' r� r/ r� r0 r1 r1 r2 �*test_dt64_series_add_mixed_tick_DateOffset> s zHTestTimestampSeriesArithmetic.test_dt64_series_add_mixed_tick_DateOffsetc C s� t ttd�g�}t ttgdd�}t tgdd�}t�t | |� d}tjt|d�� | | W d Q R X t�t | |� tjt|d�� | | W d Q R X t�|t |� t�t| |� t�|t |� t�t| |� d S )Nr� zdatetime64[ns])rM zUnary negative expects)rF )r r r r$ rc rH rI rJ )r' �datetime_series�nat_series_dtype_timestamp�single_nat_dtype_datetimerK r1 r1 r2 �test_datetime64_ops_natQ s& z5TestTimestampSeriesArithmetic.test_datetime64_ops_nat� dt64_seriesr� zdatetime64[ns])rM �oner9 g �?c C s� d}t jt|d�� || W d Q R X t jt|d�� || W d Q R X t jt|d�� || W d Q R X t jt|d�� || W d Q R X d S )Nz&cannot perform .* with this index type)rF )rH rI rJ )r' rW rV rK r1 r1 r2 �!test_dt64_mul_div_numeric_invalidv s z?TestTimestampSeriesArithmetic.test_dt64_mul_div_numeric_invalidre r rD r rC c C s� |}t dddg|d�}t|�}tdddgdd �}t||�}d �d d g�}tjt|d �� |d� W d Q R X tjt|d �� ||� W d Q R X tjt|d �� |t�|�� W d Q R X tjt|d �� |t � |�� W d Q R X d S )Nz 2016-01-02z 2016-02-03r )r � r= �( Zuint8)rM r� z3Addition/subtraction of integers and integer-arraysz cannot subtract .* from ndarray)rF r9 ) r r r� r� rH rI rJ r" r! r r ) r' r( re r r, rP r- �methodrK r1 r1 r2 � test_dt64_series_add_intlike� s z:TestTimestampSeriesArithmetic.test_dt64_series_add_intlikec C s� d}t tdd|d�dd�}|�� }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 d Q R X ||d }|j � d �|d j � |�}t �||� tjt|d�� |d | W d Q R X || }|j � d �| j � |�}t �||� || }|j � d �| j � |�}t �||� || }|j � d �| j � |�}t �||� || }|j � d �| j � |�}t �||� d}tjt|d�� || W d Q R X tjt|d�� || W d Q R X d S )Nz US/Easternz2000-01-01 09:00:00rE )r r r4 )r} r7 z 1 days 1 minr� )r r> r9 r z((bad|unsupported) operand type for unary)rF zcannot (add|subtract))r r r� r" rZ r� r r^ Z_valuesr> r� r! r� r$ rc rH rI rJ ) r' r r8 r9 r7 rE r/ r rK r1 r1 r2 �*test_operators_datetimelike_with_timezones� s` zHTestTimestampSeriesArithmetic.test_operators_datetimelike_with_timezonesN)rS rT rU r0 r: r@ rF rI rJ rP rQ rU rH rV rW r r r r" r! rX r\ r] r1 r1 r1 r2 r. � s" " 2 %$ r. c @ s� e Zd Zdd� Zej�dddg�ej�deje j g�dd� ��Z ej�dd d d d g�ej�deje j g�d d� ��Z ej�deje j g�dd� �Z dd� Zdd� Zdd� Zdd� Zejjdeddd�ed d g�ed d g��d �e�d �ed �gd d � d!�ej�d"d#d g�d$d%� ��Zd&d'� Zd(d)� Zej�d*ejej g�d+d,� �Z d-d.� Z d/d0� Z d1d2� Z ej�d*eje!j"ej g�d3d4� �Z#ej�d5e j ejg�d6d7� �Z$d#S )8�TestDatetimeIndexArithmeticc C s� |}t ddd|d�}d}tjt|d�� || W d Q R X tjt|d�� ||7 }W d Q R X tjt|d�� || W d Q R X tjt|d�� ||8 }W d Q R X d S )Nz2000-01-01 09:00r� r: )r> r r z Addition/subtraction of integers)rF )r rH rI rJ )r' r( rW r r; rK r1 r1 r2 �test_dti_addsub_int� s z/TestDatetimeIndexArithmetic.test_dti_addsub_intr> r� r= � int_holderc C s. t dd|d�}|ddg�}d}t|||� d S )Nz 2016-01-01r7 )r r> rq r5 zCAddition/subtraction of integers|cannot subtract DatetimeArray from)r r )r' r` r> r, r- rK r1 r1 r2 �test_dti_add_intarray_tick s z6TestDatetimeIndexArithmetic.test_dti_add_intarray_tickr� ra ZMSr� c C s. t dd|d�}|ddg�}d}t|||� d S )Nz 2016-01-01r7 )r r> rq r5 zCAddition/subtraction of integers|cannot subtract DatetimeArray from)r r )r' r` r> r, r- rK r1 r1 r2 � test_dti_add_intarray_non_tick s z:TestDatetimeIndexArithmetic.test_dti_add_intarray_non_tickc C s: t dddg�}|dddg�}d�dd g�}t|||� d S ) Nz 2016-01-01r z2017-04-05 06:07:08r� rq r5 r� z"cannot subtract DatetimeArray fromz Addition/subtraction of integers)r r� r )r' r` r, r- rK r1 r1 r2 � test_dti_add_intarray_no_freq s z9TestDatetimeIndexArithmetic.test_dti_add_intarray_no_freqc C s� |}t td|d�gd �}tjddd�}tdd|d�}|�d �}|| }t�||� || }t�||� ||j }t�||� |j| }t�||� d S )Nz 2017-01-01)r r: z0 days)r )r r ) r r r r^ r r� r$ �assert_index_equalr� )r' r( r r, r� r0 r/ r1 r1 r2 �test_dti_add_tdi! s z,TestDatetimeIndexArithmetic.test_dti_add_tdic C s� |}t td|d�gd �}tjddd�}tdd|d�}|�d �}t td|d�gd �}||7 }t�||� tjddd�}||7 }t�||� t td|d�gd �}||j7 }t�||� tjddd�}||7 }t�||� d S )Nz 2017-01-01)r r: z0 days)r )r r ) r r r r^ r r� r$ rd r� )r' r( r r, r� r0 r/ r1 r1 r2 �test_dti_iadd_tdi7 s" z-TestDatetimeIndexArithmetic.test_dti_iadd_tdic C s� |}t td|d�gd �}tjddd�}tdd|dd�}|�d �}|| }t�||� d}tj t |d �� || W d Q R X ||j }t�||� d }tj t |d �� |j | W d Q R X d S ) Nz 2017-01-01)r r: z0 days)r z-1D)r r r> z cannot subtract .*TimedeltaArray)rF z"cannot subtract DatetimeArray from) r r r r^ r r� r$ rd rH rI rJ r� )r' r( r r, r� r0 r/ rK r1 r1 r2 �test_dti_sub_tdiQ s z,TestDatetimeIndexArithmetic.test_dti_sub_tdic C s� |}t td|d�gd �}tjddd�}tdd|dd�}|�d �}t td|d�gd �}||8 }t�||� d}tj t |d �� ||8 }W d Q R X t td|d�gd �}||j 8 }t�||� d � d d d g�}tj t |d �� | j |8 _ W d Q R X d S )Nz 2017-01-01)r r: z0 days)r z-1D)r r r> z(cannot subtract .* from a TimedeltaArray)rF r� z,cannot perform __neg__ with this index type:z-ufunc subtract cannot use operands with typesz"cannot subtract DatetimeArray from) r r r r^ r r� r$ rd rH rI rJ r� r� )r' r( r r, r� r0 r/ rK r1 r1 r2 �test_dti_isub_tdii s( z-TestDatetimeIndexArithmetic.test_dti_isub_tdi�addendi� r9 z 2011-01-01z 2011-01-02z US/Easternc C s t | �jS )N)r� rS )rs r1 r1 r2 r� � r� z$TestDatetimeIndexArithmetic.<lambda>)r� r Nc C sj t ddg��|�}t�||�}d}tjt|d�� || W d Q R X tjt|d�� || W d Q R X d S )Nz 2011-01-01z 2011-01-02z cannot add DatetimeArray and)rF )r r� r$ r% rH rI rJ )r' r) ri r r, r. rK r1 r1 r2 � test_add_datetimelike_and_dtarr� s z;TestDatetimeIndexArithmetic.test_add_datetimelike_and_dtarrc C sp t dd|d�}|j}|| }|| }t�||� |}|| }|| }t�||� || }|| }t�||� d S )Nr r )r r )r r! r$ rd )r' r( r, rB r/ r0 r� r1 r1 r2 �test_dta_add_sub_index� s z2TestDatetimeIndexArithmetic.test_dta_add_sub_indexc C sl t ddd�}t ddd��d�}t ddd��d�}tdddg�}|| }t�||� || }t�||� d}tjt|d�� || W d Q R X tjt|d�� || W d Q R X tjt|d�� || W d Q R X ||8 }t�||� t ddd�}t dd d�}d }tjt|d�� || W d Q R X t d t j d g�}t d d t j g�}tdt j t j g�}|| }t�||� d S )Nr r )r z US/Eastern�UTCr z9DatetimeArray subtraction must have the same timezones or)rF rq z$cannot add indices of unequal lengthz 2012-01-01z 2012-01-03z 2012-01-02z1 days) r r� r r$ rd rH rI rJ � ValueErrorr r" rZ ) r' r, r� Zdti_tz2r0 r/ rK Zdti1Zdti2r1 r1 r2 �test_sub_dti_dti� s6 z,TestDatetimeIndexArithmetic.test_sub_dti_dtire c s� t td�td�td�td�g�}ddddd g� � fd d �}xfttd �gd � D ]P\}}}}}||||||�} t|||||d�} ||| �} ||| �} t�| | � qNW d S )NZ20130301z20130228 23:00:00z20130228 22:00:00z20130228 21:00:00r= rD �mr� �usc s t �ttt jt| � ����S )N)r" �sumr[ r r� �zip)�args)� intervalsr1 r2 r� � s z_TestDatetimeIndexArithmetic.test_timedelta64_equal_timedelta_supported_ops.<locals>.timedelta64r7 rE )r8 r� r� r� r� )r r r r\ r r$ rc ) r' re rP r� �drD ro r� rp ZnptdZpytdr� r� r1 )rt r2 �.test_timedelta64_equal_timedelta_supported_ops� s " zJTestDatetimeIndexArithmetic.test_timedelta64_equal_timedelta_supported_opsc C sJ t ttd�g�}t ttd�g�}t ttgdd�}t ttgdd�}t tgdd�}t tgdd�}t�|| |� t�|| |� t�| | |� t�|| |� t�|| |� t�| | |� d}tjt|d�� || W d Q R X t�|| |� t�|| |� t�|| |� t�|| |� t�|| |� t�|| |� d S )NZ1sr� ztimedelta64[ns])rM zdatetime64[ns]zcannot subtract a datelike)rF ) r r r r r$ rc rH rI rJ )r' Ztimedelta_seriesrR Znat_series_dtype_timedeltarS rT Zsingle_nat_dtype_timedeltarK r1 r1 r2 �)test_ops_nat_mixed_datetime64_timedelta64� sT zETestDatetimeIndexArithmetic.test_ops_nat_mixed_datetime64_timedelta64c C s� t ddddd�}t�dd�}t ddddd�}xD|| t�||�gD ],}t|t�sTt�t�||� |j dksBt�qBW t d dddd�}xD|| t� ||�gD ],}t|t�s�t�t�||� |j dks�t�q�W |� d �}t� t�dd�t�d d�t�dd�g�}tdd d gdd �}x<|| t�||�gD ]$}t�||� |j |j k�s"t��q"W td ddgdd �}xL|| t� ||�gD ]4}t|t��s�t�t�||� |j |j k�srt��qrW d S )Nz 2011-01-01r Z2Drs )r r> r} r9 r= z 2011-01-02z 2010-12-31r7 z 2011-01-05z 2011-01-08)r} ) r r" r� r r� r r� r$ rd r> �subtractr� r! )r' r� �deltar r/ r1 r1 r2 �test_ufunc_coercionsD s0 $ z0TestDatetimeIndexArithmetic.test_ufunc_coercionsc C s� |}t ddg||d d�}ttdd�gd ||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} rE )r� r7 r9 )r� r} ) r r r r} rM r� r$ rc r� rd ) r' r( �namesr r� rP r0 r/ r� Zresult3Zresult4r1 r1 r2 �test_dti_add_seriesg s z/TestDatetimeIndexArithmetic.test_dti_add_seriesc s� t j}|}|}tdd||d d�� |t j�� t jjdd�g|d d��t|��}t�t �� �� ��} W d Q R X t � ��fdd �t t � ��D �|d d d �} t� | |�} t�| | � d S ) Nz 2017-01-01r7 r )r r r} )r r9 )r} c s g | ]}�� | �| ��qS r1 r1 )rr r )r, re r- r1 r2 rt � s zPTestDatetimeIndexArithmetic.test_dti_addsub_offset_arraylike.<locals>.<listcomp>Zinfer)r} r> )r r r r� r r� r r$ r� r r r\ r� r% r& ) r' r( r{ re rd r* � other_boxr r+ r r0 r1 )r, re r- r2 � test_dti_addsub_offset_arraylike~ s $ * z<TestDatetimeIndexArithmetic.test_dti_addsub_offset_arrayliker} c C s� |}t dd|d�}t�||�}|tj�� tdd�g�}t||�}tddg|d�} t�| |�} t� t �� || } W d Q R X t� | | � td d g|d�} t�| |�} t� t �� || } W d Q R X t� | | � d S ) Nz 2017-01-01r7 )r r rq )r8 z 2017-01-31z 2017-01-06)r z 2016-12-31z 2016-12-29) r r$ r% r r� r r r r r� r r& ) r' r( r) r} r r, r. r- r+ r0 r/ r1 r1 r2 � test_dti_addsub_object_arraylike� s z<TestDatetimeIndexArithmetic.test_dti_addsub_object_arraylike)%rS rT rU r_ rH rV rW r" r! r r ra rb rc re rf rg rh r r r� r� r rj rk rn r� r r rv rw rz r| r r r~ r r1 r1 r1 r2 r^ � s6 % . E#"r^ r� r5 r9 r� ���r7 c sj t td�td�td�td�td�g�}t t|j�d � ��}� �fdd�|D �}t |�}t�||� d S ) Nz2000-01-05 00:15:00z2000-01-31 00:23:00z 2000-01-01z 2000-02-29z 2000-12-31rM c s g | ]}|t jj�� d � �qS ))r� r� )r r� r )rr rs )r� r� r1 r2 rt � s z%test_shift_months.<locals>.<listcomp>)r r r r� r$ rd )r� r� r, �actual�rawr0 r1 )r� r� r2 �test_shift_months� s r� c C s4 t dddd�} | j�d�}t�dd� td�D ��}|j|jksBt�t� t �� || }W d Q R X t� t ��. |d d �df |d d �df �d d �}W d Q R X t |t �s�t�|j d ks�t�t�|j|j� t� t �� ||�t� }W d Q R X t |t�s�t�|jdk�s t�|j d k�s t�|jdk�� �s0t�d S ) Nz 1994-02-13Z2Wrq )r> r )rq r9 c S s g | ]}t j�|�g�qS r1 )r r� r� )rr r r1 r1 r2 rt � s z7test_dt64arr_addsub_object_dtype_2d.<locals>.<listcomp>r r5 r9 )r r@ Zreshaper" r! r\ �shaper� r$ r� r r� r r> rG r] rX r r� r� )r, rB r- r/ r0 r� r1 r1 r2 �#test_dt64arr_addsub_object_dtype_2d� s" 2 r� )7r r r � itertoolsr r r� r� �numpyr" rH rN Z pandas._libs.tslibs.conversionr Zpandas._libs.tslibs.offsetsr Z pandas.compatr Z pandas.errorsr Zpandasr r r r r r r r r r Zpandas._testingZ_testingr$ Zpandas.core.arraysr r Zpandas.core.opsr Z pandas.tests.arithmetic.commonr r r r r_ r� r� r� r r. r^ rV rW r� r� r1 r1 r1 r2 �<module> sT , p L 5 z 7 I
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка