Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/arithmetic/__pycache__/test_timedelta64.cpython-37.pyc
Назад
B �5�g�4 � @ s d dl m Z mZ d dlZd dlZd dlmZmZ d dlZ d dlm Z m Z m Z m Z mZmZmZmZmZ d dlmZ d dlmZmZmZ dd� Zdd � ZG d d � d �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�Z"dd� Z#dS )� )�datetime� timedeltaN)�OutOfBoundsDatetime�PerformanceWarning) � DataFrame� DatetimeIndex�NaT�Series� Timedelta�TimedeltaIndex� Timestamp�offsets�timedelta_range)�assert_invalid_addsub_type�assert_invalid_comparison�get_upcast_boxc C s t �| �}||kst�dS )zT Helper to check the dtype for a Series, Index, or single-column DataFrame. N)�tm� get_dtype�AssertionError)�objZexpected_dtype�dtype� r �^/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/arithmetic/test_timedelta64.py� assert_dtype$ s r c C s8 | t kr|d }n"| tjtjgkr,|d }n|d }|S )Nr � � )r r �to_array�pd�array)�box�names�exnamer r r �get_expected_name- s r" c @ sJ e Zd Zdd� Zej�dedd�edd�edd�� � e � d�g�dd� �Z ej�d d d e �� e �d �e �� �� e �� �� e �� �� g�d d� �Zej�deed��e�d�e�d��ej�e�d��e�ejddd d�j e� ejddd��eejddd��ejddd��e�ej dddd�j ej dddd��e�g �dd� �Z dd� Z dS ) �#TestTimedelta64ArrayLikeComparisonsc C s� |}|t jt jgkr|ntj}tddd�}t�|�� d �}t�||�}||k}t�ddddg�}t�||�}t� ||� d}t j t |d�� |t�d�k W d Q R X d S ) N�2H� )�periodsr TFz Invalid comparison between dtype)�match) r �Indexr �np�ndarrayr Zto_numpyr � box_expected� assert_equal�pytest�raises� TypeError) �self�box_with_arrayr �xbox�tdi�other�res�expected�msgr r r � test_compare_timedelta64_zerodimB s zDTestTimedelta64ArrayLikeComparisons.test_compare_timedelta64_zerodim� td_scalarr )�days� c C sr |}|t jt jgkr|ntj}ttdd�tdd�g�}t�||�}||k}tddg�}t�||�}t� ||� d S )Nr )r: r FT) r r( r r) r* r r r r+ r, )r0 r1 r9 r r2 �ser�actualr6 r r r �!test_compare_timedeltalike_scalarW s zETestTimedelta64ArrayLikeComparisons.test_compare_timedeltalike_scalar�invalidl x Ii �a�UTCc C s, |}t ddd�}t�||�}t|||� d S )Nz1 days� )r& )r r r+ r )r0 r1 r? r �rngr r r r � test_td64_comparisons_invalidl s zATestTimedelta64ArrayLikeComparisons.test_td64_comparisons_invalidr4 rB z 1970-01-01)r& �tz)r& z 1971-01-01�D)�freqr& c C s t ddd�j}t||tj� d S )Nz1 daysrB )r& )r �_datar r r )r0 r4 rC r r r �"test_td64arr_cmp_arraylike_invalid� s zFTestTimedelta64ArrayLikeComparisons.test_td64arr_cmp_arraylike_invalidc C s� t ddd�j}t�ddd|d t�� g�}||k}t�dddd dg�}t�||� ||k}t�|| � d }tj t |d �� ||k W d Q R X tj t |d �� ||k W d Q R X tj t |d �� ||k W d Q R X tj t |d �� ||k W d Q R X d S ) Nz1 days� )r& r r r � FTzDInvalid comparison between|Cannot compare type|not supported between)r' ) r rH r) r r �nowr �assert_numpy_array_equalr- r. r/ )r0 rC r4 �resultr6 r7 r r r � test_td64arr_cmp_mixed_invalid� s zBTestTimedelta64ArrayLikeComparisons.test_td64arr_cmp_mixed_invalidN) �__name__� __module__� __qualname__r8 r- �mark� parametrizer r �to_timedelta64r �Hourr> r rL � to_datetime64� to_pydatetime�daterD �list�ranger) �arange�astypeZfloat32�objectr � date_ranger Z period_rangerI rO r r r r r# = s: " r# c @ s6 e Zd Zej�ddeg�dd� �Zdd� Zdd� Z dS ) � TestTimedelta64ArrayComparisonsr Nc C s, t td�ttd�g�}t tttd�g�}|| }}|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 )Nz1 daysz3 daysFT) r r r r^ r] r) r r rM )r0 r �left�right�lhs�rhsrN r6 r r r � test_comp_nat� s( z-TestTimedelta64ArrayComparisons.test_comp_natc C s� t dtdtddg�}t ddttddg�}t�t�dd�t�dd�t�d �t�d �t�d d�t�dd � t�d d�t�d d � g�}||f||fg}�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 d S )Nz1 dayz1 day 00:00:01z5 day 00:00:03z2 dayz1 day 00:00:02z5 days 00:00:03r rF Znatr �srJ rK TF)r r r) r � timedelta64r rM ) r0 Ztdidx1Ztdidx2�tdarrZcasesZidx1Zidx2rN r6 r r r �test_comparisons_nat� sH z4TestTimedelta64ArrayComparisons.test_comparisons_natc C sb t ddd�}||d k }t�dddgdgd �}t�||� |t|�k}||k}t�||� d S )Nz1 daysrB )r& rK TF� )r r) r r rM rZ )r0 rC rN r6 �expr r r �test_comparisons_coverage s z9TestTimedelta64ArrayComparisons.test_comparisons_coverage) rP rQ rR r- rS rT r^ re ri rl r r r r r` � s 7r` c @ sv e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zej � d d d g�dd� �Z dd� Z dd� Z dd� Zdd� Zdd� ZdS )�!TestTimedelta64ArithmeticUnsortedc C s� t dddddgddd�}x\|d t�|d�gD ]D}t|t �s@t�t ddd d d gddd�}t�||� |jdks.t�q.W x\|d t�|d�gD ]D}t|t �s�t�t d dd ddgddd�}t�||� |jdks�t�q�W xX| t� |�gD ]D}t|t �s�t�t dddddgddd�}t�||� |jdks�t�q�W t dddd dgddd�}x`t |�t� |�gD ]J}t|t ��snt�t dd dd dgd dd�}t�||� |jd k�sZt��qZW d S )Nr$ Z4HZ6HZ8HZ10H�x)rG �namer Z12HZ16HZ20HZ1HZ3HZ5H�Hz-2Hz-4Hz-6Hz-8Hz-10Hz-1HZ0H) r r) �multiply� isinstancer r �assert_index_equalrG �divide�negative�abs�absolute)r0 �idxrN rk r r r �test_ufunc_coercions s. z6TestTimedelta64ArithmeticUnsorted.test_ufunc_coercionsc C s� t dtdgdd�}tjdddd�}td�}td�}d }tjt|d �� || W d Q R X tjt|d �� || W d Q R X d }tjt|d �� || W d Q R X d }tjt|d �� || W d 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 )Nz1 daysz2 days�foo)ro �20130101rK �bar)r& ro z0cannot subtract a datelike from a TimedeltaArray)r' z#unsupported operand type\(s\) for -z((bad|unsupported) operand type for unaryz0 daysz-1 daysz-2 daysF)Z check_names�20121231�20130102rF )rG ro Z20121230) r r r r_ r r r- r. r/ r rs r )r0 r3 �dti�td�dtr7 rN r6 r r r �test_subtraction_ops7 sD z6TestTimedelta64ArithmeticUnsorted.test_subtraction_opsc C s t jddd�}td�}|�� }t jddd��d�}td��d�}td��d�}|�� }td�}dd� } || } td �} | | | � || } td �} | | | � || } td �} | | | � d } tjt| d �� || W d Q R X d } tjt| d �� || W d Q R X d } tjt| d �� || W d Q R X d } tjt| d �� || W d Q R X 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 tjt| d �� || W d Q R X tjt| d �� || W d Q R X tjt| d �� || W d Q R X || } t d dd g�} t � | | � || } t d ddg�} t � | | � || } t d dd g�} t � | | � || } t d ddg�} t � | | � || } td �} | | | � || } t dddgdd�} t � | | � d S )Nr{ rK )r& z US/EasternZCETz1 daysc S s | |ks t �t| t�st �d S )N)r rr r )rN r6 r r r �_checkr s zNTestTimedelta64ArithmeticUnsorted.test_subtraction_ops_with_tz.<locals>._checkz0 dayszBTimestamp subtraction must have the same timezones or no timezones)r' z6can't subtract offset-naive and offset-aware datetimesz2 daysz-1 daysz-2 daysr} r~ )rE ) r r_ r rX Z tz_localizer r- r. r/ r r rs r ) r0 r �tsr� Zdti_tzZts_tzZts_tz2Zdt_tzr� r� rN r6 r7 r r r � test_subtraction_ops_with_tzf 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 ) Nz1 daysz2 daysrz )ro r{ rK r| )r& ro z0 daysz4 daysr} )r r r r_ r rs r )r0 r3 r rN r6 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 d Q R X t j t |d �� |dd� | W d Q R X d}t j t |d �� |t� dddg� W d Q R X || }td tdg�}t�||� || }td tdg�}t�||� || }td �}||k�s�t�|| }td �}||k�s�t�d S )Nz1 daysz2 daysrz )ro r{ rK r| )r& ro r~ Z20130103z3 daysz$cannot add indices of unequal length)r' r r z3Addition/subtraction of integers and integer-arraysr Z20130105)r r r r_ r r r r rs r- r. � ValueErrorr/ � Int64Indexr )r0 r3 r r� r� rN r6 r7 r r r �test_addition_ops� sH z3TestTimedelta64ArithmeticUnsorted.test_addition_opsrG rF �Bc C s� t jdd|d�}|td� }|td� }|�d�}t�||� |dkr�t jj�d�}|j |ksbt �|j |kspt �|j |ks�t �n>|j t jj� d�ks�t �|j d ks�t �|j t jj� d�ks�t �|td� }|td� }t�||� t �dd �}t |�}|t� d� } t|t�d � �} |t�d � } t|t� d� �} | j |j k�sHt �| �d �} t�| | � | j |j k�spt �| �d �} t�| | � d S ) Nz1/1/2000�2 )r& rG r ����inferrF Z2013Z2014i �)r r_ r � _with_freqr rs Ztseriesr �DayrG r Z BusinessDayr rV r r) rg ) r0 rG �indexZshifted�backr6 rN rC rf Zresult1Zresult2Zresult3Zresult4r r r �test_timedelta s8 z0TestTimedelta64ArithmeticUnsorted.test_timedeltac C s| t dtdg�}|j}tjdddd�}|| }|| }t�||� || }|| }t�||� || }|| }t�||� d S )Nz1 daysz2 daysz 1999-12-31rK rF )r& rG )r r r r r_ r rs )r0 r3 Ztdar rN r6 r r r �test_tda_add_sub_index* s z8TestTimedelta64ArithmeticUnsorted.test_tda_add_sub_indexc C sr |}t jdd|d�}|�d �}|| }t�||�}t�||�}t�t�� ||�t� }W d Q R X t� ||� d S )Nz 2016-01-01rK )r& rE ) r r_ r� r r+ �assert_produces_warningr r] r^ r, ) r0 r1 �tz_naive_fixturer r r3 r r4 rN r r r � test_tda_add_dt64_object_array= s z@TestTimedelta64ArithmeticUnsorted.test_tda_add_dt64_object_arrayc C sb t dd�}t dddd�}t�||�}t�||�}|}||7 }t�||� |tjk r^t�||� d S )Nz1 daysz10 daysz1 days 02:00:00z10 days 02:00:00rF )rG )r r r+ r, r r( )r0 � two_hoursr1 rC r6 �orig_rngr r r �test_tdi_iadd_timedeltalikeO s z=TestTimedelta64ArithmeticUnsorted.test_tdi_iadd_timedeltalikec C s^ t dd�}t dd�}t�||�}t�||�}|}||8 }t�||� |tjk rZt�||� d S )Nz1 daysz10 daysz0 days 22:00:00z9 days 22:00:00)r r r+ r, r r( )r0 r� r1 rC r6 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�||� |jdks�t�|d }t ddd dd�}t�||� |jd ks�t�|d }t d dd dd�}t�||� |jd ks�t�| }t d dd dd�}t�||� |jd k�st�t d dd dd�}t|�}tdd dd dgdd�}t�||� |jd k�s`t�d S )Nz2 daysrJ Z2Drn )r& rG ro r z4 daysr z-2 daysZ4Dz1 daysrF z-2Dz0 days)ro )r rG r rs r rv r )r0 rC rN rk r r r �test_tdi_ops_attributeso s6 z9TestTimedelta64ArithmeticUnsorted.test_tdi_ops_attributesN)rP rQ rR ry r� r� r� r� r- rS rT r� r� r� r� r� r� r r r r rm s /T8'rm c @ s e Zd Zdd� Zdd� ZdS )�TestAddSubNaTMaskingc C s� t �ddg�}td�}||�� |�� �d�|�� �d�g}td�}||�� |�� �d�|�� �d�g}x(|| D ] }|| }|d tkspt�qpW d S )Nz24658 days 11:15:00r z 1950-01-01zdatetime64[ns]z datetime64[D]z 1980-01-01r )r � to_timedeltar rX rW r] r r )r0 ZtdinatZtsnegZts_neg_variantsZtsposZts_pos_variants�variantr5 r r r �"test_tdi_add_timestamp_nat_masking� s z7TestAddSubNaTMasking.test_tdi_add_timestamp_nat_maskingc C s� t jtdd�� t�dd�td� W d Q R X t jtdd�� td�t�dd� W d Q R X tjd }d}t jt|d�� t�dgd�td� W d Q R X t jt|d�� td�t�dgd� W d Q R X t jt|d�� t�|g�t d� W d Q R X t jt|d�� t�d |g�t d� W d Q R X t jt|d��& t�|d d g�t�d |d g� W d 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 )NZ10155196800000000000)r' iT� rF �2000r zOverflow in int64 additionz1 daysz5 daysz1 hoursz 7 secondsz4 hoursz4 daysz5 hours) r- r. r r r� r r �value� OverflowErrorr r r rs )r0 Z_NaTr7 rk rN r r r �test_tdi_add_overflow� s6 z*TestAddSubNaTMasking.test_tdi_add_overflowN)rP rQ rR r� r� r r r r r� � s r� c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zej � d e d �e d �� � e d �� � g�d d� �Zdd� Zdd� Zdd� Zej � dddddg�ej � dddg�d d � ��Zej � d d d d!e�d"�g�d#d$� �Zej j d%e�d d"d&g�e�d d"d&g�ed d"d&g�ed d"d&gg�gd'd(� d)�d*d+� �Zd,d-� Zd.d/� Z d0d1� Z d2d3� Z d4d5� Z d6d7� Z d8d9� Z!d:d;� Z"d<d=� Z#d>d?� Z$d@dA� Z%dBdC� Z&dDdE� Z'dFdG� Z(dHdI� Z)ej � dJejejeg�dKdL� �Z*dMdN� Z+dS )O� TestTimedeltaArraylikeAddSubOpsc C s� t �tdg��}t �tdg��}d}tjt|d�� t �ttg�� W d Q R X t �ttgdd��}tdg��t j�}tdg��t j�}tjt|d�� ttg��t j� W d Q R X ttj g��t j�}t �d�}t �d�} t �d�} || } | | ks�t �| | } | |k�st �|| } t � | |� || } t � | |� || } t � | |� || } t � | |� || } t � | |� | | } t � | |� || } t � | |� | | } t � | |� || } t � | |� | | } t � | |� d}tjt|d�� |t j W d Q R X tjt|d�� t j| W d Q R X tjt|d�� |t j W d Q R X tjt|d�� t j | W d Q R X |t } t � | |� |t } t � | |� || } t �| |� || } t �| |� || } t �| |� || } t �| |� || } t �| |� || } t �| |� || } t �| |� || } t �| |� || } t �| |� || } t �| |� d }tjt|d�� |t j W d Q R X tjt|d�� |t j W d Q R X |t } t �| |� |t } t �| |� d S ) Nz00:00:01z00:00:02z?dtype datetime64\[ns\] cannot be converted to timedelta64\[ns\])r' zm8[ns])r r zunsupported operand typez8cannot subtract a datelike from|unsupported operand type)r r� r r- r. r/ r r �applyr� r r �assert_series_equalr) �nanZassert_frame_equal) r0 �s1�s2r7 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�||� |jdksbt�t d |i�}td d � t d�D ��}|jdks�t�|d |d � � }|jdks�t�|d | }|jd ks�t�|d � � }t |t�s�t�|d |d � � } | jdk�s t�| |d }td �td�td�g} t| d d�} t�|| � |d tddd� }tdd � t d�D �d d�} t�|| � |jdk�s�t�tddddd�} |d | } | jdk�s�t�| | } t�|d | � tdd�}|d | } | | } t�| |d � | jd k�st�tddd�}|d | } | | } t�|d | � | jd k�sRt�|d t�tddd�� }|d t�tddd��7 < |d |k�s�t�d S )Nz2012-1-1rK rF )r& rG z2012-1-2g �S$��B�int64ztimedelta64[ns]�Ac S s g | ]}t |d ��qS ))r: )r )�.0�ir r r � <listcomp>R s zNTestTimedeltaArraylikeAddSubOps.test_operators_timedelta64.<locals>.<listcomp>zM8[ns]Z20111230Z20120101Z20120103)ro i� r c S s g | ]}t d | d��qS )i� )r: )r )r� r� r r r r� k s zm8[ns]r% )r: rJ )�minutes�secondsr )r r_ r r� r] r r� r r r r[ �shift�maxrr r r r r) rg )r0 Zv1Zv2�rsZxpZdfr� rN ZmaxaZresultb�valuesr6 �dZresultar� r r r �test_operators_timedelta64H sV z:TestTimedeltaArraylikeAddSubOps.test_operators_timedelta64c C s t ttd�g�}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 ttd�g�� t�d| t ttd�g�� t�|tj |� t�tj| |� t�|d t ttd �g�� t�|d t ttd �g�� t�|tj |� d S ) N�1sztimedelta64[ns])r g �?r g �?z1.5sr z0.5sg @)r r r r r� r) r� )r0 Ztimedelta_seriesZnat_series_dtype_timedeltaZsingle_nat_dtype_timedeltar r r �test_timedelta64_ops_nat� s` z8TestTimedeltaArraylikeAddSubOps.test_timedelta64_ops_natc C sF t ddg�}t�||�}d}tjt|d�� |td� W d Q R X d S )Nz1 dayz2 dayzJcannot subtract a datelike from|Could not operate|cannot perform operation)r' z 2011-01-01)r r r+ r- r. r/ r )r0 r1 rx r7 r r r �!test_td64arr_sub_timestamp_raises� s zATestTimedeltaArraylikeAddSubOps.test_td64arr_sub_timestamp_raisesc C sp |}t d|d�}tddg�}tddg|d�}t�||�}t�||�}|| }t�||� || }t�||� d S )Nz 2011-01-01)rE z1 dayz2 dayz 2011-01-02z 2011-01-03)r r r r r+ r, )r0 r1 r� rE r4 rx r6 rN r r r �test_td64arr_add_timestamp� s z:TestTimedeltaArraylikeAddSubOps.test_td64arr_add_timestampr� z 2012-01-01c C s� t ddd�}tjddd�}t�||�}t�||�}t�|| |� t�|| |� tjdddd�}t�||�}t�|| |� t�|| |� d}tjt|d �� || W d Q R X d S ) Nz1 dayrK )r& z 2012-01-02z 2011-12-31z-1D)r& rG zcannot subtract a datelike)r' ) r r r_ r r+ r, r- r. r/ )r0 r� r1 r3 r6 rh Z expected2r7 r r r �(test_td64arr_add_sub_datetimelike_scalar� s zHTestTimedeltaArraylikeAddSubOps.test_td64arr_add_sub_datetimelike_scalarc C s� t jddd�}||�d� }|j}t|�| }t�||�}t�||�}d}tjt |d�� || W d Q R X || }t� ||� d S )Nz 2016-01-01rK )r& r z cannot subtract a datelike from)r' ) r r_ r� r� r r r+ r- r. r/ r, )r0 r1 r r3 �dtarrr6 r7 rN r r r �test_tdi_sub_dt64_array s z7TestTimedeltaArraylikeAddSubOps.test_tdi_sub_dt64_arrayc C sr t jddd�}||�d� }|j}t|�| }t�||�}t�||�}|| }t�||� || }t�||� d S )Nz 2016-01-01rK )r& r )r r_ r� r� r r r+ r, )r0 r1 r r3 r� r6 rN r r r �test_tdi_add_dt64_array, s z7TestTimedeltaArraylikeAddSubOps.test_tdi_add_dt64_arrayc C s` t �d�}tddd�}tdddg�}t�||�}t�||�}t�|| |� t�|| |� d S )Nr z1 dayrK )r& )r) Z datetime64r r r r+ r, )r0 r1 r4 r3 r6 �tdserr r r � test_td64arr_add_datetime64_nat: s z?TestTimedeltaArraylikeAddSubOps.test_td64arr_add_datetime64_nat�pi_freqrF �W�Qrp �tdi_freqNc C s� t ddg|d�}td�| }|�|�}t�||�}d}tjt|d�� || W d Q R X tjt|d�� ||d W d Q R X d S )Nz1 hoursz2 hours)rG z2018-03-07 17:16:40z(cannot subtract|unsupported operand type)r' r )r r Z to_periodr r+ r- r. r/ )r0 r1 r� r� r3 r �pir7 r r r �test_td64arr_sub_periodlikeJ s z;TestTimedeltaArraylikeAddSubOps.test_td64arr_sub_periodliker4 r@ r g �?r c C s, t dddgdd�}t�||�}t||� d S )Nz59 Daysr zm8[ns])r )r r r+ r )r0 r1 r4 r� rh r r r �*test_td64arr_addsub_numeric_scalar_invalid\ s zJTestTimedeltaArraylikeAddSubOps.test_td64arr_addsub_numeric_scalar_invalid�vecrK c C s t | �jS )N)�typerP )rn r r r �<lambda>v � z(TestTimedeltaArraylikeAddSubOps.<lambda>)�idsc C s6 t dddgdd�}t�||�}|�|�}t||� d S )Nz59 Daysr zm8[ns])r )r r r+ r] r )r0 r1 r� �any_real_dtyper� rh �vectorr r r �'test_td64arr_addsub_numeric_arr_invalidn s zGTestTimedeltaArraylikeAddSubOps.test_td64arr_addsub_numeric_arr_invalidc C sr t dddd�}t�||�}d}t|||� tjt|d�� ||7 }W d Q R X tjt|d�� ||8 }W d Q R X d S )Nz1 days 09:00:00rp rB )rG r& z Addition/subtraction of integers)r' )r r r+ r r- r. r/ )r0 r1 �onerC rh r7 r r r �test_td64arr_add_sub_int� s z8TestTimedeltaArraylikeAddSubOps.test_td64arr_add_sub_intc C sX |}|t jkrtjn|}tdddd�}t�||�}t�dddg|�}d}t|||� d S )Nz1 days 09:00:00rp rK )rG r& r% r z3Addition/subtraction of integers and integer-arrays)r r r) r* r r r+ r )r0 r1 r r2 rC rh r4 r7 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 )Nz1 Dayr z3 Hours� r� � z Addition/subtraction of integers)r r r) r* r r r+ r )r0 r1 r r2 r3 rh r4 r7 r r r �)test_td64arr_addsub_integer_array_no_freq� s zITestTimedeltaArraylikeAddSubOps.test_td64arr_addsub_integer_array_no_freqc C sr |}t jddd�}||�d� }|j}d| }t�||�}t�||�}|| }t�||� || }t�||� d S )Nz 2016-01-01rK )r& r r )r r_ r� r� r r+ r, )r0 r1 r r r3 rh r6 rN r r r �test_td64arr_add_td64_array� s z;TestTimedeltaArraylikeAddSubOps.test_td64arr_add_td64_arrayc C sr |}t jddd�}||�d� }|j}d| }t�||�}t�||�}|| }t�||� || }t�||� d S )Nz 2016-01-01rK )r& r r )r r_ r� r� r r+ r, )r0 r1 r r r3 rh r6 rN r r r �test_td64arr_sub_td64_array� s z;TestTimedeltaArraylikeAddSubOps.test_td64arr_sub_td64_arrayc C s> |}t ||�}tddg|d d�}|tjtjgkr<t�|�n|}ttdd�tdd�g|d d�}ttdd�tddd �g|d�}t� ||�}t� ||�}|| }t� ||� t |d � || }t� ||� t |d � ttd d�tdd d �g|d�}t� ||�}|| }t� ||� t |d � || }t� || � t |d � d S ) Nz0 daysz1 dayr )ro rK )�hoursr% r )r: r� ztimedelta64[ns]������) r" r r r r r r) r r r+ r, r ) r0 r1 r r r! r3 r< r6 rN r r r �test_td64arr_add_sub_tdi� s. z8TestTimedeltaArraylikeAddSubOps.test_td64arr_add_sub_tdic C s� |}t ttd�g�}t�d�}t dgd �}t�||�}t�||�}|| }t�||� || }t�||� || }t�||� || }t�||� d S )Nr� r r )r r r r) rg r r+ r, )r0 r1 r r3 r4 r6 r rN r r r � test_td64arr_add_sub_td64_nat� s z=TestTimedeltaArraylikeAddSubOps.test_td64arr_add_sub_td64_natc C sT |}t ttd�g�}t ttgdd�}t�||�}t�||�}|t }t�||� d S )Nr� ztimedelta64[ns])r )r r r r r+ r, )r0 r1 r r< r6 r5 r r r �test_td64arr_sub_NaT s z4TestTimedeltaArraylikeAddSubOps.test_td64arr_sub_NaTc C s` |}t dd�}t dddd�}t�||�}t�||�}|| }t�||� || }t�||� d S )Nz1 daysz10 daysz1 days 02:00:00z10 days 02:00:00rF )rG )r r r+ r, )r0 r� r1 r rC r6 rN 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 )Nz1 daysz10 daysz0 days 22:00:00z9 days 22:00:00)r r r+ r, )r0 r� r1 r rC r6 rN r r r � test_td64arr_sub_timedeltalike s z>TestTimedeltaArraylikeAddSubOps.test_td64arr_sub_timedeltalikec s� |}t ||�}tddg|d d��tjtjdd�tjdd�g|d d�� |tjtj gkrbt � � �n� � t� �fdd �t t ���D �d |d �}t� �|��t� ||�}t�t�� �� }W d Q R X t�||� t�t�� � � }W d Q R X t�||� d S ) Nz1 days 00:00:00z3 days 04:00:00r )ro r )�n���c s g | ]}�| � | �qS r r )r� r� )r4 r3 r r r� 8 s zQTestTimedeltaArraylikeAddSubOps.test_td64arr_add_offset_index.<locals>.<listcomp>r� )rG ro )r" r r r( r rV �Minuter r r r) r[ �lenr+ r� r r, )r0 r r1 r r! r6 r5 �res2r )r4 r3 r � test_td64arr_add_offset_index. s & $ z=TestTimedeltaArraylikeAddSubOps.test_td64arr_add_offset_indexc s� |}t ddg��t�tjdd�tjdd�g�� t � �fdd�tt���D �dd �}t� �|��t� ||�}t� t �� �� }W d Q R X t� ||� t� t �� � � }W d Q R X t� ||� d S ) Nz1 days 00:00:00z3 days 04:00:00r )r� r� c s g | ]}�| � | �qS r r )r� r� )r4 r3 r r r� N s zQTestTimedeltaArraylikeAddSubOps.test_td64arr_add_offset_array.<locals>.<listcomp>r� )rG ) r r) r r rV r� r[ r� r r+ r� r r, )r0 r1 r r6 r5 r� r )r4 r3 r � test_td64arr_add_offset_arrayG s " z=TestTimedeltaArraylikeAddSubOps.test_td64arr_add_offset_arrayc s� |}|t jtjgkr|ntj}t||�}tddg|d d��tjtjdd�tj dd�g|d d�� t� �fdd �t t ���D �d |d �}t � �|��t � ||�}t � t�� �� }W d Q R X t �||� d S ) Nz1 days 00:00:00z3 days 04:00:00r )ro r )r� r� c s g | ]}�| � | �qS r r )r� r� )r4 r3 r r r� f s zQTestTimedeltaArraylikeAddSubOps.test_td64arr_sub_offset_index.<locals>.<listcomp>r� )rG ro )r r r r r( r" r r rV r� r[ r� r+ r� r r, )r0 r r1 r r2 r! r6 r5 r )r4 r3 r � test_td64arr_sub_offset_index\ s &$ z=TestTimedeltaArraylikeAddSubOps.test_td64arr_sub_offset_indexc s� t ddg��t�tjdd�tjdd�g�� t � �fdd�tt���D �dd �}t� �|��t� ||�}t� t �� �� }W d Q R X t� ||� d S ) Nz1 days 00:00:00z3 days 04:00:00r )r� r� c s g | ]}�| � | �qS r r )r� r� )r4 r3 r r r� v s zQTestTimedeltaArraylikeAddSubOps.test_td64arr_sub_offset_array.<locals>.<listcomp>r� )rG ) r r) r r rV r� r[ r� r r+ r� r r, )r0 r1 r6 r5 r )r4 r3 r � test_td64arr_sub_offset_arrayp s " z=TestTimedeltaArraylikeAddSubOps.test_td64arr_sub_offset_arrayc sP |}|t jtjt jgkr tn|}t||�}tddg|d d��ttj dd�tj dd�g|d d�� t� �fdd �t t ���D �|d�}t� �|�}t� ||�}t�t�� |� }W d Q R X t�||� t�t�� � | } W d Q R X t�| |� t� �fd d �t t ���D �|d�} t� | |�} t�t�� |� } W d Q R X t�| | � d S ) Nz1 days 00:00:00z3 days 04:00:00r )ro r )r� r� c s g | ]}�| � | �qS r r )r� r� )r4 r3 r r r� � s zSTestTimedeltaArraylikeAddSubOps.test_td64arr_with_offset_series.<locals>.<listcomp>c s g | ]}�| � | �qS r r )r� r� )r4 r3 r r r� � s )r r( r r r r r" r r rV r� r[ r� r+ r� r r, ) r0 r r1 r Zbox2r! Z expected_addr r5 r� Z expected_subZres3r )r4 r3 r � test_td64arr_with_offset_series� s& $$ $ z?TestTimedeltaArraylikeAddSubOps.test_td64arr_with_offset_series�oboxc C s t ddg�}t�||�}|t�� tjdd�g�}d}tjt|d��$ t� t �� || W d Q R X W d Q R X tjt|d��$ t� t �� || W d Q R X W d Q R X tjt|d��$ t� t �� || W d Q R X W d Q R X tjt|d��$ t� t �� || W d Q R X W d Q R X d S )Nz1 days 00:00:00z3 days 04:00:00r )r� z/has incorrect type|cannot add the type MonthEnd)r' ) r r r+ r ZMonthEndr� r- r. r/ r� r )r0 r� r1 r3 Zanchoredr7 r r r �-test_td64arr_addsub_anchored_offset_arraylike� s zMTestTimedeltaArraylikeAddSubOps.test_td64arr_addsub_anchored_offset_arraylikec C s: |}|t jkrtjn|}tdddd�}t�||�}t�tdd�t� d�t d�g�}t� t �� || }W d Q R X t � tdd�td d�t d �g�}t�||�}t�||� d } tjt| d ��$ t� t �� || W d Q R X W d Q R X t� t �� || }W d Q R X t � td �td �t d�g�}t�||�}t�||� d S )Nz1 dayrK rF )r& rG r )r: r z 2000-01-04r% z 2000-01-07z3unsupported operand type|cannot subtract a datelike)r' r z 2000-01-01)r r r) r* r r r+ r r r� r r� r r( r, r- r. r/ ) r0 r1 r r2 r3 rh r4 rN r6 r7 r r r �!test_td64arr_add_sub_object_array� s( zATestTimedeltaArraylikeAddSubOps.test_td64arr_add_sub_object_array),rP rQ rR r� r� r� r� r� r- rS rT r rX rW r� r� r� r� r� r) r r� r r( r r r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r� � sZ jBU # r� c @ sn e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Ze j j d e � dd�e�edd��e�edd��e�edd��e�dd�gdd� d�dd� �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,� Zd-d.� Z d/d0� Z d1d2� Z d3d4� Z d5d6� Z d7d8� Z!d9d:� Z"e j j d;e#d d<d=�e$d>�e$d>��%� gd?d� d�d@dA� �Z&dBdC� Z'dDdE� Z(dFdG� Z)dHdI� Z*dJdK� Z+dLdM� Z,dNdO� Z-e j � dPdQdRe �.dQ�e �.dR�g�dSdT� �Z/e j j dUe �.dVdWdXg�e�0dVdWdXg�e1dVdWdXg�gdYd� d�dZd[� �Z2e j j dUe �.dVdWdXg�e�0dVdWdXg�e1dVdWdXg�gd\d� d�d]d^� �Z3d_d`� Z4dadb� Z5dcS )d� TestTimedeltaArraylikeMulDivOpsc C sJ t tjddd��}t�||�}|d }t�||� d| }t�||� d S )NrJ r� )r r )r r) r\ r r+ r, )r0 r1 rx rN r r r �test_td64arr_mul_int� s z4TestTimedeltaArraylikeMulDivOps.test_td64arr_mul_intc C sD t dddd�}t�||�}d}tjt|d�� || W d Q R X d S )Nz1 daysz10 daysrz )ro z@argument must be an integer|cannot use operands with types dtype)r' )r r r+ r- r. r/ )r0 r� r1 rC r7 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 )NrJ r� )r )r) r\ r r r+ r r, )r0 r1 �rng5rx r6 rN 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 )NrJ r� )r r )r) r\ r r r+ r, )r0 r1 r� rx r6 rN r r r �test_tdi_mul_int_array� s z6TestTimedeltaArraylikeMulDivOps.test_tdi_mul_int_arrayc C s� |}|t jtjt jgkr tn|}ttjddd��}ttjddd�d �}t� ||�}t� ||�}|ttjddd�� }t� ||� d S )NrJ r� )r r ) r r( r r r r r r) r\ r+ r, )r0 r1 r r2 rx r6 rN r r r �test_tdi_mul_int_series s z7TestTimedeltaArraylikeMulDivOps.test_tdi_mul_int_seriesc C s� |}|t jtjt jgkr tn|}ttjddd��}t� ||�}tjddd�}t||d �}t� ||�}|t|d � }t� ||� d S )NrJ r� )r �float64g �?) r r( r r r r r r) r\ r+ r, )r0 r1 r r2 rx Zrng5fr6 rN r r r �test_tdi_mul_float_series s z9TestTimedeltaArraylikeMulDivOps.test_tdi_mul_float_seriesr4 r � c C s t | �jS )N)r� rP )rn r r r r� * r� z(TestTimedeltaArraylikeMulDivOps.<lambda>)r� c C sr |}t ||�}tdgd �}tdd�}d |j_t�||�}t�||�}|| }t�||� || }t�||� d S )Nz1 DayrB z1 daysz10 days)r r r rH rG r r+ r, ) r0 r4 r1 r r2 r3 r6 rN Zcommuter r r �test_tdi_rmul_arraylike! s z7TestTimedeltaArraylikeMulDivOps.test_tdi_rmul_arraylikec C sb t dddd�}t�||�}tjtdd�� |t W d Q R X tjtdd�� t| W d Q R X d S )Nz1 daysz10 daysrz )ro zunsupported operand type)r' zCannot divide NaTType by)r r r+ r- r. r/ r )r0 r1 rC 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 )Nz1 daysz10 daysr rB ) r r r) r* r r r+ rg r� r, )r0 r1 r r2 rC r4 r6 rN r r r �test_td64arr_div_td64natI s z8TestTimedeltaArraylikeMulDivOps.test_td64arr_div_td64natc C sX t tjddd��}t�||�}|d }t�||� tjtdd�� d| W d Q R X d S )NrJ r� )r r z Cannot divide)r' ) r r) r\ r r+ r, r- r. r/ )r0 r1 rx rN r r r �test_td64arr_div_int\ s z4TestTimedeltaArraylikeMulDivOps.test_td64arr_div_intc C s� |}|t jkrtjn|}tdddd�}t jt�d�d d dd�}t�||�}t�||�}|| }t� ||� || }d| }t� ||� d S )Nz1 daysz10 daysrz )ro rB r � ) r r r) r* r � Float64Indexr\ r r+ r, )r0 r� r1 r r2 rC r6 rN r r r � test_td64arr_div_tdlike_scalarg s z>TestTimedeltaArraylikeMulDivOps.test_td64arr_div_tdlike_scalar�mrK rB �unitrF �hrf �ms�us�nsc s� |}|t jkrtjn|}tt �dd��}tt �dd��}|| }tj|d< |} t�||�}t� �fdd�| D ��} t�| |�} |t� � �� } t� | | � t� �fdd�| D ��} t�| |�} t� � ��| } t� | | � d S ) Nz 2013-01-01z 2013-01-03z 2013-03-01z 2013-03-03r c s g | ]}|t �� �� �qS r )r) rg )r� rn )r r r r r� � s zPTestTimedeltaArraylikeMulDivOps.test_td64arr_div_td64_scalar.<locals>.<listcomp>c s g | ]}t t�� ���| �qS r )r r) rg )r� rn )r r r r r� � s ) r r r) r* r r_ r� r r+ rg r, ) r0 r r r1 r r2 Z startdateZenddater< Zflatr6 rN r )r r r � test_td64arr_div_td64_scalary s z<TestTimedeltaArraylikeMulDivOps.test_td64arr_div_td64_scalarc C s� |}|t jkrtjn|}tdtdgdd�}t jdtjdgdd�}t� ||�}t� ||�}|| }t� ||� || }d| }t� ||� d S )Nz1 daysz2 daysrz )ro r� r; r ) r r r) r* r r r� r� r r+ r, )r0 r� r1 r r2 rC r6 rN r r r �'test_td64arr_div_tdlike_scalar_with_nat� s zGTestTimedeltaArraylikeMulDivOps.test_td64arr_div_tdlike_scalar_with_natc C s4 |}|t jkrtjn|}tdtdg�}t �dtjdg�}t� ||�}t� ||�}tjdddgdd�}|| }t� ||� |t� ||� }t� ||� ||� t � }t� ||� |t |� }t� ||� d | }|| }t� ||� t� ||�| }t� ||� |� t �| }t� ||� t |�| }t� ||� d S ) Nz1 daysz2 daysr� r; r r% zm8[h])r r )r r r) r* r r r� r� r r+ r, r] r^ rZ )r0 r1 r r2 rC r6 r4 rN r r r � test_td64arr_div_td64_ndarray� s0 z=TestTimedeltaArraylikeMulDivOps.test_td64arr_div_td64_ndarrayc C s� t dtdg�}ddddg}t�||�}d}x|||d d� gD ]h}xb|t�|�t�|�gD ]H}tj t |d�� || W d Q R X tj t |d�� || W d Q R X qXW q<W d S ) Nz1 daysz2 daysr r rK r% z0Cannot divide vectors|Unable to coerce to Series)r' ) r r r r+ r) r r r( r- r. r� )r0 r1 rC Z mismatchedr7 r r4 r r r � test_tdarr_div_length_mismatch� s z>TestTimedeltaArraylikeMulDivOps.test_tdarr_div_length_mismatchc C s� |}|t jkrtjn|}tdddgdd�}tddd gdd�}t�||�}t�||�}tjddtjgtjd�}t�||�}|t kr�|r�|ddg � d �|ddg< || }t� ||� t� |�| }t� ||� d S ) Ni� izd � ztimedelta64[ns])r g �?r r r� ) r r r) r* r r r+ r� r� r r] r, Zasarray) r0 r1 �using_array_managerr r2 ra rb r6 rN r r r �&test_td64arr_floordiv_td64arr_with_nat� s zFTestTimedeltaArraylikeMulDivOps.test_td64arr_floordiv_td64arr_with_natc C s� |}|t jkrtjn|}ttddd�gd �}tj|jd< tddtjg�}tj ||dd�}tj ||dd�}|| }t� ||� d S )NrJ rK )r� r� r r F)� transpose) r r r) r* r r r� �ilocr r+ r, )r0 r1 � scalar_tdr r2 �td1r6 rN r r r � test_td64arr_floordiv_tdscalar� s z>TestTimedeltaArraylikeMulDivOps.test_td64arr_floordiv_tdscalarc C s� |}|t jkrtjn|}ttddd�gd �}tj|jd< tddtjg�}tj ||dd�}tj ||dd�}|| }t� ||� d S )NrJ rK )r� r� r r F)r ) r r r) r* r r r� r r r+ r, )r0 r1 r r r2 r r6 rN r r r � test_td64arr_rfloordiv_tdscalar s z?TestTimedeltaArraylikeMulDivOps.test_td64arr_rfloordiv_tdscalarc C s� |}|t jkrtjn|}ttddd�gd �}tj|jd< tddtjg�}tj ||dd�}tj ||dd�}|� |�}t� ||� d S )NrJ rK )r� r� r r F)r ) r r r) r* r r r� r r r+ � __rfloordiv__r, )r0 r1 r r r2 r r6 rN r r r �(test_td64arr_rfloordiv_tdscalar_explicit s zHTestTimedeltaArraylikeMulDivOps.test_td64arr_rfloordiv_tdscalar_explicitc C s\ t tjddd��}t�||�}|d }t�||� d}tjt|d�� d| W d Q R X d S )NrJ r� )r r z@floor_divide cannot use operands|Cannot divide int by Timedelta*)r' ) r r) r\ r r+ r, r- r. r/ )r0 r1 rx rN �patternr r r �test_td64arr_floordiv_int* s z9TestTimedeltaArraylikeMulDivOps.test_td64arr_floordiv_intc C sr |}|t jkrtjn|}tdddd�}t jt�d�d d dd�}t�||�}t�||�}|| }t� ||� d S )Nz1 daysz10 daysrz )ro rB r r� ) r r r) r* r r� r\ r r+ r, )r0 r� r1 r r2 r3 r6 rN r r r �#test_td64arr_floordiv_tdlike_scalar4 s zCTestTimedeltaArraylikeMulDivOps.test_td64arr_floordiv_tdlike_scalarr rj )r� r� Z10m7sc C s t | �jS )N)r� rP )rn r r r r� I r� c C s� |}|t jkrtjn|}tddtgd d�}t �ddtjg�}tj ||dd�}tj ||dd�}|� |�}t� ||� t �ddtjg�}tj ||dd�}|| }t� ||� d S )Nz00:05:03)rG g @F)r g ) r r r) r* r r r( r� r r+ r r, )r0 r r1 r r2 r3 r6 r5 r r r �$test_td64arr_rfloordiv_tdlike_scalarB s zDTestTimedeltaArraylikeMulDivOps.test_td64arr_rfloordiv_tdlike_scalarc C s� t dd�}t�||�}tdddgd �}t�||�}|| }t�||� d }|tkrdt|tj�rdt }t� |�� t ||�}W d Q R X t�|d |� t�|d || � d S )Nz1 Dayz9 daysz2 Daysz0 DaysrK r r ) r r r+ r r, r rr r Z DateOffsetr r� �divmod)r0 r1 � three_daysr3 rh r6 rN �warnr r r �test_td64arr_mod_tdscalard s z9TestTimedeltaArraylikeMulDivOps.test_td64arr_mod_tdscalarc C s� t dddd�}t�||�}tddgd �}t�||�}|d }t�||� d}tjt|d �� d| W d Q R X t|d�}t�|d |� t�|d |d � d S ) Nz1 nsz10 nsrB )r& z0 nsrJ r zCannot divide int by)r' r r ) r r r+ r r, r- r. r/ r )r0 r1 r3 rh r6 rN r7 r r r �test_td64arr_mod_intx s z4TestTimedeltaArraylikeMulDivOps.test_td64arr_mod_intc C s� t dd�}t�||�}dddgdgd }t|�}t�||�}|| }t�||� t||�}t�|d |� t�|d || � d S )Nz1 Dayz9 daysz0 Daysz3 Days� r r )r r r+ r r, r )r0 r1 r r3 rh r6 rN r r r �test_td64arr_rmod_tdscalar� s z:TestTimedeltaArraylikeMulDivOps.test_td64arr_rmod_tdscalarc C sz t tddd�gd �}tj|jd< t�||�}d}tjt |d�� || W d Q R X tjt |d�� || W d Q R X d S )NrJ rK )r� r� r z(operate|unsupported|cannot|not supported)r' ) r r r) r� r r r+ r- r. r/ )r0 r1 r r r r r r �!test_td64arr_mul_tdscalar_invalid� s zATestTimedeltaArraylikeMulDivOps.test_td64arr_mul_tdscalar_invalidc C s| t tjddd��}t�||�}d}tjt|d�� ||d d� W d Q R X tjt|d�� |t� ddg� W d Q R X d S ) NrJ r� )r zdcannot use operands with types dtype|Cannot multiply with unequal lengths|Unable to coerce to Series)r' rK r r ) r r) r\ r r+ r- r. r/ r� r )r0 r1 rx r7 r r r �!test_td64arr_mul_too_short_raises� s zATestTimedeltaArraylikeMulDivOps.test_td64arr_mul_too_short_raisesc C sH t tjddd��}t�||�}d}tjt|d�� || W d Q R X d S )NrJ r� )r z$cannot use operands with types dtype)r' )r r) r\ r r+ r- r. r/ )r0 r1 rx r7 r r r � test_td64arr_mul_td64arr_raises� 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 ) Nz59 Daysr zm8[ns])r z-59 Daysztimedelta64[ns]z118 Daysr )r r r+ r, )r0 r1 r� r� r6 rN r r r � test_td64arr_mul_numeric_scalar� s z?TestTimedeltaArraylikeMulDivOps.test_td64arr_mul_numeric_scalar�twor g @c C sv t dddgdd�}t dddgdd�}t�||�}t�||�}|| }t�||� tjtdd�� || W d Q R X d S ) Nz59 Daysr zm8[ns])r z29.5Dztimedelta64[ns]z Cannot divide)r' )r r r+ r, r- r. r/ )r0 r1 r$ r� r6 rN r r r � test_td64arr_div_numeric_scalar� s z?TestTimedeltaArraylikeMulDivOps.test_td64arr_div_numeric_scalarr� � r �( c C s t | �jS )N)r� rP )rn r r r r� � r� c C s| t ||�}tdddgdd�}|�|�}tdddgdd�}t�||�}t�||�}|| }t�||� || }t�||� d S )Nz59 Daysr zm8[ns])r z 1180 Daysz 1770 Daysztimedelta64[ns])r r r] r r+ r, )r0 r1 r� r� r2 r� r6 rN r r r � test_td64arr_rmul_numeric_array� s z?TestTimedeltaArraylikeMulDivOps.test_td64arr_rmul_numeric_arrayc C s t | �jS )N)r� rP )rn r r r r� r� c sn t |��}tdddgdd�� ��|��tdddgdd�}t�� |�� t�||�}� � }t�||� d}tjt|d �� �� W d Q R X t �t j ��sB� ��t � }|t kṙ �fd d �tt���D �}n � �fd d �tt� ��D �}t � |�}t�||�}t�|�dk�st�|�r6|t k�r6tttgt d�|d < t�||� tjt|d �� ��t �� W d Q R X d S )Nz59 Daysr zm8[ns])r z2.95Dz 1D 23H 12mztimedelta64[ns]zutrue_divide'? cannot use operands|cannot perform __div__|cannot perform __truediv__|unsupported operand|Cannot divide)r' c s" g | ]}� j d |f �| �qS )r )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 r r� ( s r )r r r] r r+ r, r- r. r/ rr r r( r^ r r[ r� r r r ) r0 r1 r� r� r r2 r6 rN r r )r� r� r � test_td64arr_div_numeric_array s0 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 kr�| t ks�t �n t�| |� d S )N�0days�1day�2days�3days�4daysr )ro r r rK r% )r ro Z9daysZ16daysztimedelta64[ns]) r" r r r) r� r r+ r r, �__rmul__r �NotImplementedr ) r0 r1 r �requestr r! r3 r< r6 r2 rN r r r �test_td64arr_mul_int_series: s$ z;TestTimedeltaArraylikeMulDivOps.test_td64arr_mul_int_seriesc s� |}t dddddg|d d��tdd d d d gtj|d d�� |tjtjgkrT|d n|d }t� �fdd�tt � ��D �d|d�}t |� �}t� �|��t� ||�}� � ��}|t kr�|tks�t�n t�||� d S )Nr* r+ r, r- r. r )ro g �?rK g @r g @r )r ro r c s g | ]}�| � | �qS r r )r� r� )r< r3 r r r� g s zRTestTimedeltaArraylikeMulDivOps.test_float_series_rdiv_td64arr.<locals>.<listcomp>ztimedelta64[ns])r r r) r� r r r r r[ r� r r+ � __rtruediv__r r0 r r, )r0 r1 r r Zxnamer6 r2 rN r )r< r3 r � test_float_series_rdiv_td64arr[ s z>TestTimedeltaArraylikeMulDivOps.test_float_series_rdiv_td64arrN)6rP rQ rR r� r� r� r� r� r� r- rS rT r) r\ r r� r[ Z UInt64Indexr� Z RangeIndexr� r� r� r� r� r r r r r r r r r r r r rU r r r r r r! r" r# r r% r( r r( r) r2 r4 r r r r r� � sj && ***6!r� c @ s e Zd Zdd� ZdS )�"TestTimedelta64ArrayLikeArithmeticc C sz t tddd�gd �}tj|jd< t�||�}d}tjt |d�� || W d Q R X tjt |d�� || W d Q R X d S )NrJ rK )r� r� r z(operate|unsupported|cannot|not supported)r' ) r r r) r� r r r+ r- r. r/ )r0 r r1 r r r r r �test_td64arr_pow_invalid~ s z;TestTimedelta64ArrayLikeArithmetic.test_td64arr_pow_invalidN)rP rQ rR r6 r r r r r5 y s r5 c sD t �� � � tdddd� } t� fdd�td�D ��}t�| |� d S )NZ0sr� � )r& c s6 g | ].}� t �d �| t �d�t|d�d �qS )z 0.033333333sz 0.000000001srK r )r r� r )r� r� )� timestampr r r� � s z3test_add_timestamp_to_timedelta.<locals>.<listcomp>)r rL r r r[ r rs )rN r6 r )r8 r � test_add_timestamp_to_timedelta� s r9 )$r r �numpyr) r- Z pandas.errorsr r Zpandasr r r r r r r r r r Zpandas._testingZ_testingr Z pandas.tests.arithmetic.commonr r r r r" r# r` rm r� r� r� r5 r9 r r r r �<module> s> , pg D (
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка