Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/scalar/timedelta/__pycache__/test_arithmetic.cpython-37.pyc
Назад
B �5�gĂ � @ s� d Z ddlmZmZ ddlZddlZddlZddlmZ ddl Z ddl m Z m Z m Z mZ ddlmZ ddlmZ G dd� d�ZG d d � d �ZG d d � d �Zd d� Zdd� ZdS )z+ Tests for scalar Timedelta arithmetic ops � )�datetime� timedeltaN)�OutOfBoundsTimedelta)�NaT� Timedelta� Timestamp�offsets)�opsc @ s� e Zd ZdZej�deddd�edd�e � dd�e � dd�e � d�g�d d � �Z ej�d ed �ed �eddd�e � dd�e � dd� e �� e � d� g�dd� �Zej�dejejg�dd� �Zdd� Zej�dejejg�dd� �Zej�dejejg�dd� �Zej�dejejg�d d � �Zej�dejejg�d d � �Zd d!� Zd"d#� Zd$d%� Z d&d'� Z d(d)� Z d*d+� Z d,d-� Z d.d/� Z!d0d1� Z"d2d3� Z#d4d5� Z$d6d7� Z%ej�dejejg�d8d9� �Z&ej�dejejg�d:d;� �Z'd<d=� Z(d>S )?� TestTimedeltaAdditionSubtractionz\ Tests for Timedelta methods: __add__, __radd__, __sub__, __rsub__ � ten_seconds� �s)�unit)�secondsl d( �nsc C sD t d�}t d�}t d�}|| }||ks,t�|| }||ks@t�d S )Nz20130101 09:01:12.123456z20130101 09:01:22.123456z20130101 09:01:02.123456)r �AssertionError)�selfr �base� expected_add� expected_sub�result� r �c/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/scalar/timedelta/test_arithmetic.py�test_td_add_sub_ten_seconds" s z<TestTimedeltaAdditionSubtraction.test_td_add_sub_ten_seconds�one_day_ten_secsz1 day, 00:00:10z1 days, 00:00:10� )�daysr �Dc C sD t d�}t d�}t d�}|| }||ks,t�|| }||ks@t�d S )Nz20130102 09:01:12.123456z20130103 09:01:22.123456z20130101 09:01:02.123456)r r )r r r r r r r r r �#test_td_add_sub_one_day_ten_seconds8 s zDTestTimedeltaAdditionSubtraction.test_td_add_sub_one_day_ten_seconds�opc C s� t ddd�}||tddd��}|tjkr6t|t�s6t�|tddd�ksJt�||td��}t|t�sft�|td�ksvt�||t�d ��}t|t�s�t�|td �ks�t�||t �}|t ks�t�d S ) Nr �d)r i� r � z2018-01-12 18:09z2018-01-22 18:09z 2018-01-12z 2018-01-22) r r �operator�add� isinstancer r �np� datetime64r )r r �tdr r r r � test_td_add_datetimelike_scalarN s z@TestTimedeltaAdditionSubtraction.test_td_add_datetimelike_scalarc C sb d}t jt|d�� td�tddd� W d Q R X t jt|d�� td�tdd� W d Q R X d S )Nz int too (large|big) to convert)�matchz 1700-01-01i�� r )r )r )�pytest�raises� OverflowErrorr r r r )r �msgr r r � test_td_add_timestamp_overflowe s z?TestTimedeltaAdditionSubtraction.test_td_add_timestamp_overflowc C s@ t ddd�}||t dd��}t|t �s*t�|t dd�ks<t�d S )Nr r )r )r � )r r$ r )r r r' r r r r �test_td_add_tdm s z/TestTimedeltaAdditionSubtraction.test_td_add_tdc C s@ t ddd�}||tdd��}t|t �s*t�|t dd�ks<t�d S )Nr r )r � )r � )r r r$ r )r r r' r r r r �test_td_add_pytimedeltau s z8TestTimedeltaAdditionSubtraction.test_td_add_pytimedeltac C sB t ddd�}||t�dd��}t|t �s,t�|t dd�ks>t�d S )Nr r )r ���r � )r )r r% � timedelta64r$ r )r r r' r r r r �test_td_add_timedelta64| s z8TestTimedeltaAdditionSubtraction.test_td_add_timedelta64c C sB t ddd�}||t�d��}t|t �s*t�|t ddd�ks>t�d S )Nr r )r r5 )r �hours)r r �Hourr$ r )r r r' r r r r �test_td_add_offset� s z3TestTimedeltaAdditionSubtraction.test_td_add_offsetc C s> t ddd�}t ddd�}|| }t|t �s.t�||ks:t�d S )Nr r )r r r )r r$ r )r r' �expectedr r r r �test_td_sub_td� s z/TestTimedeltaAdditionSubtraction.test_td_sub_tdc C sh t ddd�}t ddd�}||�� }t|t �s2t�||ks>t�|�� | }t|t �sXt�||ksdt�d S )Nr r )r r r )r �to_pytimedeltar$ r )r r' r; r r r r �test_td_sub_pytimedelta� s z8TestTimedeltaAdditionSubtraction.test_td_sub_pytimedeltac C sh t ddd�}t ddd�}||�� }t|t �s2t�||ks>t�|�� | }t|t �sXt�||ksdt�d S )Nr r )r r r )r �to_timedelta64r$ r )r r' r; r r r r �test_td_sub_timedelta64� s z8TestTimedeltaAdditionSubtraction.test_td_sub_timedelta64c C s$ t ddd�}|t }|tks t�d S )Nr r )r )r r r )r r' r r r r �test_td_sub_nat� s z0TestTimedeltaAdditionSubtraction.test_td_sub_natc C sB t ddd�}t�d�}|| }|tks*t�|| }|tks>t�d S )Nr r )r r )r r% r6 r r )r r' �td_natr r r r �test_td_sub_td64_nat� s z5TestTimedeltaAdditionSubtraction.test_td_sub_td64_natc C s@ t ddd�}|t�d� }t|t �s(t�|t ddd�ks<t�d S )Nr r )r r �� �h)r r r9 r$ r )r r' r r r r �test_td_sub_offset� s z3TestTimedeltaAdditionSubtraction.test_td_sub_offsetc C s� t ddd�}d}x�ddt�d�t�d�gD ]�}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 q*W d S )Nr r )r zunsupported operand type� g @)r) )r r% �int64�float64r* r+ � TypeError)r r' r- �otherr r r � test_td_add_sub_numeric_raises� s z?TestTimedeltaAdditionSubtraction.test_td_add_sub_numeric_raisesc C s> t ddd�}t| }|tks t�t�d�| }|tks:t�d S )Nr r )r r )r r r r% r& )r r' r r r r �test_td_rsub_nat� s z1TestTimedeltaAdditionSubtraction.test_td_rsub_natc C s< t �d�tddd� }t|t�s$t�|tddd�ks8t�d S )Nr r r )r i���rE )r r9 r r$ r )r r r r r �test_td_rsub_offset� s z4TestTimedeltaAdditionSubtraction.test_td_rsub_offsetc C sH t �td�td�g�}t �td�td�g�}|td� }t�||� d S )Nz 20130101 9:01z 20121230 9:02z 20121231 9:01z 20121229 9:02�1D)r% �arrayr r �tm�assert_numpy_array_equal)r �arr�exp�resr r r �,test_td_sub_timedeltalike_object_dtype_array� s zMTestTimedeltaAdditionSubtraction.test_td_sub_timedeltalike_object_dtype_arrayc C sl t �� }t�|td�t�dd�g�}t�|td� td�t�dd�td� g�}|td� }t�||� d S )NrO rG rE Z0D)r �nowr% rP r r6 rQ rR )r rW rS rT rU r r r �7test_td_sub_mixed_most_timedeltalike_object_dtype_array� s zXTestTimedeltaAdditionSubtraction.test_td_sub_mixed_most_timedeltalike_object_dtype_arrayc C sR t �� }t�|td�t�dd�g�}d}tjt|d�� td�| W d Q R X d S )NrO rG rE zAunsupported operand type\(s\) for \-: 'Timedelta' and 'Timestamp')r) ) r rW r% rP r r6 r* r+ rJ )r rW rS r- r r r �8test_td_rsub_mixed_most_timedeltalike_object_dtype_array� s zYTestTimedeltaAdditionSubtraction.test_td_rsub_mixed_most_timedeltalike_object_dtype_arrayc C sJ t �td�td�g�}t �td�td�g�}||td��}t�||� d S )Nz 20130101 9:01z 20121230 9:02z 20130102 9:01z 20121231 9:02rO )r% rP r r rQ rR )r r rS rT rU r r r �,test_td_add_timedeltalike_object_dtype_array� s zMTestTimedeltaAdditionSubtraction.test_td_add_timedeltalike_object_dtype_arrayc C sR t �� }t�|td�g�}t�|td� td�g�}||td��}t�||� d S )NrO Z2D)r rW r% rP r rQ rR )r r rW rS rT rU r r r �2test_td_add_mixed_timedeltalike_object_dtype_array� s zSTestTimedeltaAdditionSubtraction.test_td_add_mixed_timedeltalike_object_dtype_arrayc C s^ t d�}t�dg�j}t�dg�j}t�|| |� t�|| |� d}tjt|d�� |t � dg� W d Q R X d}tjt|d�� t � dg�| W d Q R X t�dg�j}t�|| |� t�| | |� d}tjt|d�� |t � dg� W d Q R X d }tjt|d�� t � dg�| W d Q R X t�dg�j}t�|t � d g� |� t�t � d g�| |� d }tjt|d�� || W d Q R X tjt|d�� || W d Q R X t�|| t j dgt j d �� t�|| t j dgt j d �� t� d g�j}t� dg�j}t�|| |� t�|| |� t� dg�j}t�| | |� t�|| |� d S )Nz1 dayz2 daysz;unsupported operand type\(s\) for \+: 'Timedelta' and 'int')r) r z�unsupported operand type\(s\) for \+: 'numpy.ndarray' and 'Timedelta'|Concatenation operation is not implemented for NumPy arraysz0 daysz:unsupported operand type\(s\) for -: 'Timedelta' and 'int'zDunsupported operand type\(s\) for -: 'numpy.ndarray' and 'Timedelta'rG z_ufunc '?multiply'? cannot use operands with types dtype\('<m8\[ns\]'\) and dtype\('<m8\[ns\]'\))�dtypez 2000-01-01z 2000-01-02z 1999-12-31) r �pdZ to_timedelta�valuesrQ rR r* r+ rJ r% rP rI � to_datetime)r r' rK r; r- r r r �test_ops_ndarray sJ z1TestTimedeltaAdditionSubtraction.test_ops_ndarrayN))�__name__� __module__� __qualname__�__doc__r* �mark� parametrizer r r% r6 r ZSecondr ZDayr r" r# r Zraddr( r. r0 r3 r7 r: r< r> r@ rA rC rF rL rM rN rV rX rY rZ r[ r` r r r r r sH r c @ s\ e Zd ZdZej�dee� dd�e� d�g�ej�de j e j g�dd� ��Zej�deje�d �ed�g�ej�de j e j g�d d � ��Zej�de j e j g�d d � �Zdd� Zdd� Zdd� Zej�deje�d �ed�g�dd� �Zdd� Zdd� Zdd� Zd d � Zd d � Z d d!� Z d"d#� Z d$d%� Z d&d'� Z d(d)� Z!d*d+� Z"d,d-� Z#d.d/� Z$d0d1� Z%d2d3� Z&d4d5� Z'd6d7� Z(d8d9� Z)d:d;� Z*d<d=� Z+d>d?� Z,d@dA� Z-dBdC� Z.dDdE� Z/dFdG� Z0dHdI� Z1dJdK� Z2dLdM� Z3dNdO� Z4dPdQ� Z5dRdS� Z6dTdU� Z7dVdW� Z8dXdY� Z9ej�de j e j e j:e j;e j<g�ej�dZe�=e>d[�e>d\�g�e�=e>�?� e@d]�g�g�d^d_� ��ZAd`S )a�#TestTimedeltaMultiplicationDivisionz� Tests for Timedelta methods: __mul__, __rmul__, __div__, __rdiv__, __truediv__, __rtruediv__, __floordiv__, __rfloordiv__, __mod__, __rmod__, __divmod__, __rdivmod__ rB r r r c C s` t ddd�}d�dddg�}d�d|� d |� d �d g�}tjt|d �� |||� W d Q R X d S ) Nr r )r �|znumpy.timedelta64ZNaTTyper z'unsupported operand type\(s\) for \*: 'z' and '�'z1ufunc '?multiply'? cannot use operands with types)r) )r �joinr* r+ rJ )r r rB r' Ztypsr- r r r �test_td_mul_natN s z3TestTimedeltaMultiplicationDivision.test_td_mul_nat�nan�NaNc C s& t ddd�}|||�}|tks"t�d S )Nr r )r )r r r )r r rl r' r r r r �test_td_mul_nan_ s z3TestTimedeltaMultiplicationDivision.test_td_mul_nanc C s� t dd�}||d�}|t dd�ks&t�||d�}|t ddd�ksDt�||tj�tksXt�|d |�jd |j ksrt�|d |�jd |j ks�t�d }tjt|d �� ||t d dd�� W d Q R X tjt|d �� |||� W d Q R X d S )N� )�minutesrG r5 g �?� � )rp r ���g �zunsupported operand type)r) i� r ) r r r% rl r �valuer* r+ rJ r )r r r' r r- r r r �test_td_mul_scalarg s z6TestTimedeltaMultiplicationDivision.test_td_mul_scalarc C sd t ddd�}|t�d� }|dks&t�|| dks6t�|t�dd� dksNt�t�|t �s`t�d S ) Nr r )r r �� �< rE rq )r r r9 r r% r6 �isnanr )r r' r r r r � test_td_div_timedeltalike_scalar� s zDTestTimedeltaMultiplicationDivision.test_td_div_timedeltalike_scalarc C s� t d�}|t�dd� }||jd ks*t�|t�dd� }||jd ksLt�|t�dd� }||jksjt�t d�}|t�dd� }|dks�t�|t�dd� }|dks�t�|t�dd� }||jks�t�d S ) Nz1 days 2 hours 3 nsr r l �"R: r i ʚ;r i�m )r r% r6 rt r )r r' r r r r �test_td_div_td64_non_nano� s z=TestTimedeltaMultiplicationDivision.test_td_div_td64_non_nanoc C s` t ddd�}|d }t|t �s"t�|t dd�ks4t�|d }t|t �sJt�|t dd�ks\t�d S )Nr r )r rG � )r )r r$ r )r r' r r r r �test_td_div_numeric_scalar� s z>TestTimedeltaMultiplicationDivision.test_td_div_numeric_scalarc C s8 t ddd�}|| }|tks t�|| }|tks4t�d S )Nr r )r )r r r )r rl r' r r r r �test_td_div_nan� s z3TestTimedeltaMultiplicationDivision.test_td_div_nanc C sB t ddd�}t�d�| }|dks&t�t�dd�| dks>t�d S ) Nr r )r r gq?rw rE g �?)r r r9 r r% r6 )r r' r r r r �!test_td_rdiv_timedeltalike_scalar� s zETestTimedeltaMultiplicationDivision.test_td_rdiv_timedeltalike_scalarc C s� t ddd�}t| }t�|�s"t�d | }t�|�s8t�t�d�| }t�|�sTt�d}tjt|d�� t� d�| W d Q R X d}tjt|d�� tj | W d Q R X d S )Nr r )r r zGunsupported operand type\(s\) for /: 'numpy.datetime64' and 'Timedelta')r) z<unsupported operand type\(s\) for /: 'float' and 'Timedelta') r r r% rx r r6 r* r+ rJ r& rl )r r' r r- r r r �test_td_rdiv_na_scalar� s z:TestTimedeltaMultiplicationDivision.test_td_rdiv_na_scalarc C s� t ddd�}tj|gtd�}|| }tjdgtjd�}t�||� t�d g�}|| }t�tjg�}t�||� tjtjgtd�}d}tj t |d�� || W d Q R X tjtjgtjd�}d}tj t |d�� || W d Q R X d S ) Nr r )r )r\ r z<unsupported operand type\(s\) for /: 'float' and 'Timedelta')r) z$cannot use operands with types dtype) r r% rP �objectrI rQ rR rl r* r+ rJ )r r' rS r r; r- r r r �test_td_rdiv_ndarray� s" z8TestTimedeltaMultiplicationDivision.test_td_rdiv_ndarrayc C sZ t ddd�}t ddd�}|| dks(t�| |�� dks>t�d| |�� dksVt�d S )Nro rq )r8 rp r ���rG )r r r= r? )r r' �scalarr r r �%test_td_floordiv_timedeltalike_scalar� s zITestTimedeltaMultiplicationDivision.test_td_floordiv_timedeltalike_scalarc C sL t ddd�}|tj tks t�t�|t �s0t�t�|t�d� �sHt�d S )Nro rq )r8 rp r )r r% rl r r rx r6 )r r' r r r � test_td_floordiv_null_scalar s z@TestTimedeltaMultiplicationDivision.test_td_floordiv_null_scalarc C s< t ddd�}|t�d� dks"t�|t�d� dks8t�d S )Nro rq )r8 rp r rG �\ )r r r9 r ZMinute)r r' r r r �test_td_floordiv_offsets s z<TestTimedeltaMultiplicationDivision.test_td_floordiv_offsetsc C sL t ddd�}d�dddg�}tjt|d�� |tjd d d � W d Q R X d S ) Nro rq )r8 rp rh z.Invalid dtype datetime64\[D\] for __floordiv__z8'dtype' is an invalid keyword argument for this functionz5ufunc '?floor_divide'? cannot use operands with types)r) z 2016-01-01zdatetime64[us])r\ )r rj r* r+ rJ r% r& )r r' r- r r r � test_td_floordiv_invalid_scalar s zCTestTimedeltaMultiplicationDivision.test_td_floordiv_invalid_scalarc C s~ t ddd�}t ddd�}|d |ks(t�|d |ks8t�|t�d� |ksNt�|t�d� |ksdt�|t�d� |kszt�d S )Nro rq )r8 rp r � rG g @)r r r% rI �int32�uint8)r r' r; r r r � test_td_floordiv_numeric_scalar s zCTestTimedeltaMultiplicationDivision.test_td_floordiv_numeric_scalarc C s� t ddd�}t ddd�}|t�|�� � dks2t�d| t�|�� g� }tjdgtjd�}t�||� d| t�|�� t�d�g� }t�dtj g�}t�||� d S )Nro rq )r8 rp r )r\ r r ) r r% rP r? r rH rQ rR r6 rl )r r' r� rU r; r r r �$test_td_floordiv_timedeltalike_array* s zHTestTimedeltaMultiplicationDivision.test_td_floordiv_timedeltalike_arrayc C s: t ddd�}tjdgtjd�}|| }|jjdks6t�d S )Nro rq )r8 rp r )r\ �m)r r] �Seriesr% rH r\ �kindr )r r' �serrU r r r � test_td_floordiv_numeric_series: s zCTestTimedeltaMultiplicationDivision.test_td_floordiv_numeric_seriesc C s` t ddd�}t ddd�}|�|�dks*t�| �|�� �dksBt�d| �|�� �dks\t�d S )Nro )r8 rp rq r r� rG r )r � __rfloordiv__r r= r? )r r' r� r r r �&test_td_rfloordiv_timedeltalike_scalarD s zJTestTimedeltaMultiplicationDivision.test_td_rfloordiv_timedeltalike_scalarc C s> t ddd�}t�|�t��s t�t�|�t�d���s:t�d S )Nro )r8 rp r )r r% rx r� r r r6 )r r' r r r � test_td_rfloordiv_null_scalarS s zATestTimedeltaMultiplicationDivision.test_td_rfloordiv_null_scalarc C s t �d�tdd� dks t�d S )Nr � )rp rG )r r9 r r )r r r r �test_td_rfloordiv_offsetsZ s z=TestTimedeltaMultiplicationDivision.test_td_rfloordiv_offsetsc C sT t ddd�}t�dd�}|�|�tks*t�d}tjt|d�� || W d Q R X d S )Nro )r8 rp z 2016-01-01�uszHunsupported operand type\(s\) for //: 'numpy.datetime64' and 'Timedelta')r) ) r r% r& r� �NotImplementedr r* r+ rJ )r r' Zdt64r- r r r � test_td_rfloordiv_invalid_scalar^ s zDTestTimedeltaMultiplicationDivision.test_td_rfloordiv_invalid_scalarc C s t ddd�}|�tj�tks t�|�d�tks2t�|�d�tksDt�|�t�d��tks\t�|�t�d��tkstt�|�t�d��tks�t�d}t j t |d�� t�d�| W d Q R X t j t |d�� t�d�| W d Q R X t j t |d�� t�d�| W d Q R X d S ) Nro )r8 rp g @rG g @r1 z9unsupported operand type\(s\) for //: '.*' and 'Timedelta)r) ) r r� r% rl r� r rI r� r� r* r+ rJ )r r' r- r r r � test_td_rfloordiv_numeric_scalarl s zDTestTimedeltaMultiplicationDivision.test_td_rfloordiv_numeric_scalarc C s� t ddd�}t ddd�}|�t�|�� ��dks4t�|�t�d| �� g��}tjdgtjd�}t�||� t�d| �� t� d�g�}|�|�}t�dtj g�}t�||� d S )Nro )r8 rp rq r )r\ r r ) r r� r% rP r? r rH rQ rR r6 rl )r r' r� rU r; rS r r r �%test_td_rfloordiv_timedeltalike_array� s zITestTimedeltaMultiplicationDivision.test_td_rfloordiv_timedeltalike_arrayc C sH t �ddddg�d }d}tjt|d�� |tdd d � W d Q R X d S ) Ni�rPi isPi��tPi vPi ʚ;z Invalid dtype)r) r r )r )r% rP r* r+ rJ r )r �intsr- r r r �test_td_rfloordiv_intarray� s z>TestTimedeltaMultiplicationDivision.test_td_rfloordiv_intarrayc C s^ t ddd�}tjdgtjd�}|�|�}|tks4t�d}tj t |d�� || W d Q R X d S )Nro )r8 rp r )r\ z Invalid dtype)r) ) r r] r� r% rH r� r� r r* r+ rJ )r r' r� rU r- r r r � test_td_rfloordiv_numeric_series� s zDTestTimedeltaMultiplicationDivision.test_td_rfloordiv_numeric_seriesc C s| t dd�}|t dd� }t|t �s&t�|t dd�ks8t�|tdd� }t|t �sTt�|t d�ksdt�|t }|tksxt�d S )N�% )r8 r5 r rw )rp r )r r$ r r r )r r' r r r r �test_mod_timedeltalike� s z:TestTimedeltaMultiplicationDivision.test_mod_timedeltalikec C s* t dd�}|t�dd� }|tks&t�d S )Nr� )r8 r r )r r% r6 r r )r r' r r r r �test_mod_timedelta64_nat� s z<TestTimedeltaMultiplicationDivision.test_mod_timedelta64_natc C s> t dd�}|t�dd� }t|t �s(t�|t dd�ks:t�d S )Nr� )r8 rG rE r )r r% r6 r$ r )r r' r r r r �test_mod_timedelta64� s z8TestTimedeltaMultiplicationDivision.test_mod_timedelta64c C s< t dd�}|t�d� }t|t �s&t�|t dd�ks8t�d S )Nr� )r8 r{ rG )r r r9 r$ r )r r' r r r r �test_mod_offset� s z3TestTimedeltaMultiplicationDivision.test_mod_offsetc C s� t dd�}|d }t|t �s t�|t d�ks0t�|d }t|t �sFt�|t ddd�ksZt�|td� }t|t �stt�|t ddd�ks�t�d S ) Nr� )r8 rG r g ��mBro r/ )rp r )r r$ r �int)r r' r r r r �test_mod_numeric� s z4TestTimedeltaMultiplicationDivision.test_mod_numericc C sZ t dd�}d}tjt|d�� |td� W d Q R X tjt|d�� |g W d Q R X d S )Nr� )r8 zunsupported operand type)r) z 2018-01-22)r r* r+ rJ r )r r' r- r r r �test_mod_invalid� s z4TestTimedeltaMultiplicationDivision.test_mod_invalidc C s< t dd�}tdd�| }t|t �s&t�|t dd�ks8t�d S )Nro )rp rq r )r r r$ r )r r' r r r r �test_rmod_pytimedelta� s z9TestTimedeltaMultiplicationDivision.test_rmod_pytimedeltac C s> t dd�}t�dd�| }t|t �s(t�|t dd�ks:t�d S )Nro )rp r{ r� rG )r r% r6 r$ r )r r' r r r r �test_rmod_timedelta64� s z9TestTimedeltaMultiplicationDivision.test_rmod_timedelta64c C s� t dd�}d}tjt|d�� td�| W d Q R X tjt|d�� d| W d Q R X tjt|d�� d| W d Q R X d}tjt|d�� t�d d g�| W d Q R X d S ) Nro )rp zunsupported operand)r) z 2018-01-22� g 0@zInvalid dtype int� � )r r* r+ rJ r r% rP )r r' r- r r r �test_rmod_invalid� s z5TestTimedeltaMultiplicationDivision.test_rmod_invalidc C s� t ddd�}t|d�}|d t ddd�ks.t�t|d t �s@t�|d t dd �ksVt�|s^t�t|tj�}|d tkszt�|d tks�t�d S ) NrG r5 )r r8 g $���Br r r )r )r8 )r �divmodr r$ r% rl r )r r' r r r r �test_divmod_numeric s z7TestTimedeltaMultiplicationDivision.test_divmod_numericc C s� t ddd�}t|tdd��}|d dks,t�t|d t �s>t�|d t dd�ksTt�t|d�}|d t dd�kstt�t|d t �s�t�|d t d�ks�t�t|t�}t�|d �s�t�|d tks�t�d S ) NrG r5 )r r8 r )r r )r8 �6 )r r� r r r$ r r% rx )r r' r r r r � test_divmod s z/TestTimedeltaMultiplicationDivision.test_divmodc C sX t ddd�}t|t�d��}|d dks,t�t|d t �s>t�|d t dd �ksTt�d S ) NrG r5 )r r8 r4 r i�r r� )r8 )r r� r r9 r r$ )r r' r r r r �test_divmod_offset. s z6TestTimedeltaMultiplicationDivision.test_divmod_offsetc C s< t ddd�}d}tjt|d�� t|td�� W d Q R X d S )NrG r5 )r r8 zAunsupported operand type\(s\) for //: 'Timedelta' and 'Timestamp')r) z 2018-01-22)r r* r+ rJ r� r )r r' r- r r r �test_divmod_invalid7 s z7TestTimedeltaMultiplicationDivision.test_divmod_invalidc C sT t tddd�tdd��}|d dks(t�t|d t�s:t�|d tdd�ksPt�d S )NrG r5 )r r8 r )r r )r8 )r� r r r r$ )r r r r r �test_rdivmod_pytimedelta? s z<TestTimedeltaMultiplicationDivision.test_rdivmod_pytimedeltac C sR t t�d�tdd��}|d dks&t�t|d t�s8t�|d tdd�ksNt�d S )Nr� r4 )r8 r i�r r� )r� r r9 r r r$ )r r r r r �test_rdivmod_offsetF s z7TestTimedeltaMultiplicationDivision.test_rdivmod_offsetc C s� t dd�}d}tjt|d�� ttd�|� W d Q R X tjt|d�� td|� W d Q R X tjt|d�� td|� W d Q R X d}tjt|d�� tt�d d g�|� W d Q R X d S ) Nro )rp zunsupported operand type)r) z 2018-01-22r� g 0@zInvalid dtype intr� r� )r r* r+ rJ r� r r% rP )r r' r- r r r �test_rdivmod_invalidL s z8TestTimedeltaMultiplicationDivision.test_rdivmod_invalidrS z 20130101 9:01z 20121230 9:02rO c C s0 d}t jt|d�� ||td�� W d Q R X d S )Nz7unsupported operand type|cannot use operands with types)r) rO )r* r+ rJ r )r r rS r- r r r �(test_td_op_timedelta_timedeltalike_array` s zLTestTimedeltaMultiplicationDivision.test_td_op_timedelta_timedeltalike_arrayN)Bra rb rc rd r* re rf r r% r6 r" �mulr Zrmulrk rl rI �floatrn ru ry rz r| r} r~ r r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� �truedivZrdivZrsubrP r rW r r� r r r r rg ? sn rg c @ s\ e Zd Zdd� Zdd� Zdd� Zdd� Zejj d d �d d � �Z ej� d ddg�dd� �Z dS )�TestTimedeltaComparisonc C s� |}|d�}|j }t|t�s t�||ks,t�||kr8t�||ksDt�||ksPt�||k r\t�||krht�|d| krxt�|d| ks�t�|d| ks�t�|d| k s�t�|d| kr�t�|d| kr�t�d S )Nrq rG )�deltar$ r r )r Z tick_classes�cls�offr' r r r �test_compare_tickq s z)TestTimedeltaComparison.test_compare_tickc C s� t d�}t d�}tj||gtd�}||k}tjddgtd�}||k�� sLt�tj||g||ggtd�}||k}tjddgddggtd�}|j|jks�t�||k�� s�t�d S )Nz2 daysz3 hours)r\ FT)r r% rP r� �bool�allr �shape)r r' rK rS rU r; r r r � test_comparison_object_array� s z4TestTimedeltaComparison.test_comparison_object_arrayc C sD t d�t d�g}t�|�}|d |k}t�ddg�}t�||� d S )Nz0 days 01:00:00r F)r r% rP rQ rR )r ZperiodsrS r r; r r r � test_compare_timedelta_ndarray� s z6TestTimedeltaComparison.test_compare_timedelta_ndarrayc C s� t �d��d�}t|d �}t jdddddgtd�}||k}t�||� ||k}t�||� ||k}t�|| � ||k}t�|| � d S )Nr{ ztimedelta64[ns]r FT)r\ )r% ZarangeZastyper rP r� rQ rR )r rS r' r; r r r r �test_compare_td64_ndarray� s z1TestTimedeltaComparison.test_compare_td64_ndarrayz'GH#20829 is reverted until after 0.24.0)�reasonc C s� G dd� d�}t d�}|dkr"t�|dkr.t�||� kr<t�||dd�krNt�||dd�k s`t�||dd�k rrt�||dd�ks�t�d S ) z� Make sure non supported operations on Timedelta returns NonImplemented and yields to other operand (GH#20829). c @ s. e Zd Zd dd�Zdd� Zdd� Zdd � ZdS ) zGTestTimedeltaComparison.test_compare_custom_object.<locals>.CustomClassNc S s || _ d S )N)� cmp_result)r r� r r r �__init__� s zPTestTimedeltaComparison.test_compare_custom_object.<locals>.CustomClass.__init__c S s | j d krtS | j S d S )N)r� r� )r r r r �generic_result� s zVTestTimedeltaComparison.test_compare_custom_object.<locals>.CustomClass.generic_resultc S s | � � S )N)r� )r rK r r r �__eq__� s zNTestTimedeltaComparison.test_compare_custom_object.<locals>.CustomClass.__eq__c S s | � � S )N)r� )r rK r r r �__gt__� s zNTestTimedeltaComparison.test_compare_custom_object.<locals>.CustomClass.__gt__)N)ra rb rc r� r� r� r� r r r r � CustomClass� s r� �1s�stringr F)r� TN)r r )r r� �tr r r �test_compare_custom_object� s z2TestTimedeltaComparison.test_compare_custom_object�valr� r c C s� t d�}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 )Nr� z>not supported between instances of 'Timedelta' and '(int|str)')r) )r r* r+ rJ )r r� r� r- r r r �test_compare_unknown_type� s z1TestTimedeltaComparison.test_compare_unknown_typeN) ra rb rc r� r� r� r� r* re �skipr� rf r� r r r r r� p s #r� c C sz G dd� d�} | � }t d�}|�|�tks.t�|�|�tks@t�|�|�tksRt�|�|�tksdt�|�|�tksvt�d S )Nc @ s e Zd ZdS )z&test_ops_notimplemented.<locals>.OtherN)ra rb rc r r r r �Other� s r� z1 day)r �__add__r� r �__sub__� __truediv__�__mul__� __floordiv__)r� rK r' r r r �test_ops_notimplemented� s r� c C s� t d�} x�| dfd| fgD ]x\}}d�dddg�}tjt|d�� || W d Q R X d}tjt|d�� ||k W d Q R X ||kr�t�||kst�qW d S ) Nz1 day�arh zunsupported operand typez3can only concatenate str \(not "Timedelta"\) to strzmust be str, not Timedelta)r) z"not supported between instances of)r rj r* r+ rJ r )r' �left�rightr- r r r �test_ops_error_str� s r� )rd r r r" �numpyr% r* Z pandas.errorsr Zpandasr] r r r r Zpandas._testingZ_testingrQ Z pandas.corer r rg r� r� r� r r r r �<module> s( ' 5t
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка