Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/arithmetic/__pycache__/test_period.cpython-37.pyc
Назад
B �5�g�� � @ s d dl Z d dlZd dlZd dlmZmZmZmZ d dl m Z d dl Z d dl m Z mZmZmZmZ d dlmZ d dlmZ d dlmZ d dlmZ G dd � d �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"dS )� N)�IncompatibleFrequency�Period� Timestamp� to_offset)�PerformanceWarning)� PeriodIndex�Series� Timedelta�TimedeltaIndex� period_range)�ops)�TimedeltaArray)�assert_invalid_comparisonc @ s� e Zd Zdd� Zej�dde�� e dd�ddg�d d � �Z ej�d e j d dd �j e jddd �j e�d�e�d��ej�eed��g�dd� �Zej�deej dd� g�dd� �ZdS )� TestPeriodArrayLikeComparisonsc C sz |t jt jgkr|ntj}tddd�}t�|�� d �}t�||�}||k}t�ddddg�}t�||�}t� ||� d S )N�2000� )�periodsr TF) �pd�Index�array�np�ndarrayr Zto_numpy�tm� box_expected� assert_equal)�self�box_with_array�xbox�pi�other�result�expected� r" �Y/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/arithmetic/test_period.py�test_compare_zerodim' s z3TestPeriodArrayLikeComparisons.test_compare_zerodim�scalarZfoor )�days� g #@c C s( t ddd�}t�||�}t|||� d S )Nr r )r )r r r r )r r r% r �parrr" r" r# �test_compare_invalid_scalar6 s z:TestPeriodArrayLikeComparisons.test_compare_invalid_scalarr r )r Z1Dc C s( t ddd�}t�||�}t|||� d S )Nr r )r )r r r r )r r r r r( r" r" r# � test_compare_invalid_listlike? s z<TestPeriodArrayLikeComparisons.test_compare_invalid_listlike� other_boxc C s | � t�S )N)�astype�object)�xr" r" r# �<lambda>N � z'TestPeriodArrayLikeComparisons.<lambda>c C s& t ddd�}t�||�}|tjtjgkr.tjn|}||�}t�dddddg�}t�||�}||k}t�||� ||k}t�||� ||k}t�||� ||k}t�|| � ||k }t�|| � ||k}t�|| � ||d d d� �}t�dddddg�}t�||�}||k}t�||� t�dddddg�}t�||�}||k}t�||� t�dddddg�}t�||�}||k}t�||� t�dddddg�}t�||�}||k}t�||� t�dddddg�}t�||�}||k }t�||� t�dddddg�}t�||�}||k}t�||� d S )Nr � )r T���F) r r r r r r r r r ) r r r+ r r( r r r! r r" r" r# �test_compare_object_dtypeN sV z8TestPeriodArrayLikeComparisons.test_compare_object_dtypeN)�__name__� __module__� __qualname__r$ �pytest�mark� parametrizer �nowr r) r � date_ranger Ztimedelta_ranger Zaranger, Zfloat64�list�ranger* r3 r" r" r" r# r " s $r c @ s2 e Zd Zej�ddeddd�g�dd� �Zej�dddddge� dddg�ej dddge d�e � dddg�g�d d � �Z d d � Zd d� Zej�ddddg�dd� �Zej�ddddg�dd� �Zej�ddddg�dd� �Zej�ddddg�dd� �Zej�ddddg�dd � �Zej�d e d g�d d � �Zd S )!�TestPeriodIndexComparisonsr �2017�D)�freqc C s: t dddgdd�}t�dddg�}||k}t�||� d S )Nr? �2018r@ )rA TF)r r r r �assert_numpy_array_equal)r r �idxr! r r" r" r# �test_eq� s z"TestPeriodIndexComparisons.test_eqi� )�dtypec C s� t dddgdd�}t�dddg�}||k}t�||� d�ddg�}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? rB r@ )rA F�|z5not supported between instances of 'Period' and 'int'z1Invalid comparison between dtype=period\[D\] and )�match) r r r r rC �joinr7 �raises� TypeError)r r rD r! r �msgr" r" r# �test_eq_integer_disallowed� s z5TestPeriodIndexComparisons.test_eq_integer_disallowedc C s: t dddd�}||d k }|j|jd k }t�||� d S )Nz2007-01� �M)r rA � )r �valuesr rC )r rD r �expr" r" r# �test_pi_cmp_period� s z-TestPeriodIndexComparisons.test_pi_cmp_periodc s� |t jt jgkr|ntj}tdddd�}tddd�� � fdd�|D �}t�||�}t�||�}|� k}t� ||� |d � |� k}� fd d�|D �}t�||�}t� ||� d S ) Nz 2000-01-01rP r@ )r rA z 2000-01-04)rA c s g | ] }|� k�qS r" r" )�.0r. )�valr" r# � <listcomp>� s zKTestPeriodIndexComparisons.test_parr_cmp_period_scalar2.<locals>.<listcomp>r1 c s g | ] }|� k�qS r" r" )rT r. )rU r" r# rV � s ) r r r r r r r r r r )r r r r r! �serr r" )rU r# � test_parr_cmp_period_scalar2� s z7TestPeriodIndexComparisons.test_parr_cmp_period_scalar2rA rO �2M�3Mc C s� |t jt jgkrtjn|}tddddg|d�}t�||�}td|d�}t�ddddg�}t�||�}t� ||k|� t� ||k|� t�ddddg�}t�||�}t� ||k|� t� ||k|� t�ddddg�}t�||�}t� ||k|� t� ||k |� t�ddddg�}t�||�}t� ||k |� t� ||k|� t�ddddg�}t�||�}t� ||k|� t� ||k|� t�ddddg�}t�||�}t� ||k|� t� ||k|� d S )Nz2011-01z2011-02z2011-03z2011-04)rA FT) r r r r r r r r r r )r rA r r �base�perrR r" r" r# �test_parr_cmp_period_scalar� s8 z6TestPeriodIndexComparisons.test_parr_cmp_period_scalarc C sf |t jt jgkrtjn|}tddddg|d�}t�||�}tddddg|d�}t�ddddg�}t�||�}t�||k|� t�ddddg�}t�||�}t�||k|� t�ddddg�}t�||�}t�||k|� t�ddddg�}t�||�}t�||k |� t�ddddg�}t�||�}t�||k|� t�ddddg�}t�||�}t�||k|� d S ) Nz2011-01z2011-02z2011-03z2011-04)rA z2011-05FT) r r r r r r r r r )r rA r r r[ rD rR r" r" r# �test_parr_cmp_pi� s, z+TestPeriodIndexComparisons.test_parr_cmp_pic C s� t ddddg|d�}t�||�}d|� d�}tjt|d�� |td d d�k W d Q R X tjt|d�� td d d�|k W d Q R X t d d d d gd d�}d}|tjtj gkr�|n|}tjt|d�� ||k W d Q R X d|� d�}tjt|d�� |td dd�k W d Q R X tjt|d�� td dd�|k W d Q R X t d d d d gdd�}d}|tjtj gk�r`|n|}tjt|d�� ||k W d Q R X d S )Nz2011-01z2011-02z2011-03z2011-04)rA z)Invalid comparison between dtype=period\[z \] and Period)rH �2011�A�2012�2013�2014z@Invalid comparison between dtype=period\[A-DEC\] and PeriodArray�4Mz=Invalid comparison between dtype=period\[4M\] and PeriodArray) r r r r7 rJ rK r �to_arrayr r )r rA r r[ rL rD Zrev_msgZidx_msgr" r" r# � test_parr_cmp_pi_mismatched_freq s, z;TestPeriodIndexComparisons.test_parr_cmp_pi_mismatched_freqc C s� t ddddg|d�}|td|d�k}t�ddddg�}t�||� td|d�|k }t�||� |td|d�k}t�ddddg�}t�||� td|d�|k}t�||� |td|d�k}t�ddddg�}t�||� td|d�|k}t�||� t ddddg|d�}||k }t�ddddg�}t�||� ||k}t�ddddg�}t�||� ||k}t�ddddg�}t�||� ||k}t�ddddg�}t�||� ||k}t�ddddg�}t�||� d S ) Nz2011-01z2011-02�NaTz2011-05)rA FTz2011-04)r r r r r rC )r rA �idx1r rR Zidx2r" r" r# �test_pi_cmp_nat? s@ z*TestPeriodIndexComparisons.test_pi_cmp_natc C s� t ddddg|d�}t ddddgdd�}d|� d �}tjt|d �� ||k W d Q R X ||k}tjd d d d gtd �}t�||� d S ) Nz2011-01z2011-02rg z2011-05)rA z2011-04rd z)Invalid comparison between dtype=period\[z\] and PeriodArray)rH F)rF ) r r7 rJ rK r r �boolr rC )r rA rh �diffrL r r! r" r" r# �&test_pi_cmp_nat_mismatched_freq_raisesj s zATestPeriodIndexComparisons.test_pi_cmp_nat_mismatched_freq_raisesrF Nc C s2 t td�tjtd�g�}t tjtjtd�g�}|d k rJ|�|�}|�|�}||k}t�dddg�}t�||� ||k}t�dddg�}t�||� t�dddg�}t�|tjk|� t�tj|k|� t�dddg�}t�|tjk|� t�tj|k|� t�dddg�}t�|tjk |� t�tj|k|� d S )Nz 2011-01-01z 2011-01-03FT) r r r rg r, r r r rC )r rF �left�rightr r! r" r" r# � test_comp_natx s( z(TestPeriodIndexComparisons.test_comp_nat)r4 r5 r6 r7 r8 r9 r rE r r r- r r rM rS rX r] r^ rf ri rl ro r" r" r" r# r> � s " ' # # + r> c @ s e Zd Zdd� ZdS )�TestPeriodSeriesComparisonsc C s$ t tddd�tddd�tddd�tddd�g�}t tddd�td dd�tddd�td dd�g�}t d d d d g�}t�||k|� t d d d d g�}t�||k|� t d d d d g�}t�||k|� t d d d d g�}t�||k |� t d d d d g�}t�||k|� t d d d d g�}t�||k|� d S ) Nr_ r` )rA z2011-02rO rb z2011-04ra z2011-01z2011-05FT)r r r �assert_series_equal)r r[ rW rR r" r" r# �(test_cmp_series_period_series_mixed_freq� s, zDTestPeriodSeriesComparisons.test_cmp_series_period_series_mixed_freqN)r4 r5 r6 rr r" r" r" r# rp � s rp c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �*TestPeriodIndexSeriesComparisonConsistencyz2Test PeriodIndex and Period Series Ops consistencyc C s` t |�}||�}t|tjtjf�s&t�t�||� t |�}||�}t ||j d�}t� ||� d S )N)�name) r � isinstancer r r r �AssertionErrorr r r rt rq )r rQ �funcr! rD r �srR r" r" r# �_check� s z1TestPeriodIndexSeriesComparisonConsistency._checkc C s, t ddddgddd�}dd � }tjd d d d gtjd �}| �|||� d d � }| �|||� dd � }tjd d d d gtjd �}| �|||� dd � }| �|||� dd � }tjd d d d gtjd �}| �|||� dd � }tjd d d d gtjd �}| �|||� dd � }tjd d d d gtjd �}| �|||� d S )Nz2011-01z2011-02z2011-03z2011-04rO rD )rA rt c S s | t ddd�kS )Nz2011-03rO )rA )r )r. r" r" r# r/ � r0 zPTestPeriodIndexSeriesComparisonConsistency.test_pi_comp_period.<locals>.<lambda>FT)rF c S s t ddd�| kS )Nz2011-03rO )rA )r )r. r" r" r# r/ � r0 c S s | t ddd�kS )Nz2011-03rO )rA )r )r. r" r" r# r/ � r0 c S s t ddd�| kS )Nz2011-03rO )rA )r )r. r" r" r# r/ � r0 c S s t ddd�| kS )Nz2011-03rO )rA )r )r. r" r" r# r/ � r0 c S s | t ddd�kS )Nz2011-03rO )rA )r )r. r" r" r# r/ � r0 c S s t ddd�| kS )Nz2011-03rO )rA )r )r. r" r" r# r/ � r0 )r r r �bool_ry )r rD �frR r" r" r# �test_pi_comp_period� s* z>TestPeriodIndexSeriesComparisonConsistency.test_pi_comp_periodc C s� t ddddgddd�}dd � }tjd d d d gtjd �}| �|||� d d � }| �|||� dd � }tjd d d d gtjd �}| �|||� dd � }| �|||� dd � }tjd d d d gtjd �}| �|||� dd � }| �|||� dd � }tjd d d d gtjd �}| �|||� dd � }| �|||� dd � }tjd d d d gtjd �}| �|||� dd � }tjd d d d gtjd �}| �|||� dd � }tjd d d d gtjd �}| �|||� dd � }tjd d d d gtjd �}| �|||� d S )Nz2011-01rg z2011-03z2011-04rO rD )rA rt c S s | t ddd�kS )Nz2011-03rO )rA )r )r. r" r" r# r/ � r0 zTTestPeriodIndexSeriesComparisonConsistency.test_pi_comp_period_nat.<locals>.<lambda>FT)rF c S s t ddd�| kS )Nz2011-03rO )rA )r )r. r" r" r# r/ � r0 c S s | t jkS )N)r rg )r. r" r" r# r/ � r0 c S s t j| kS )N)r rg )r. r" r" r# r/ � r0 c S s | t ddd�kS )Nz2011-03rO )rA )r )r. r" r" r# r/ r0 c S s t ddd�| kS )Nz2011-03rO )rA )r )r. r" r" r# r/ r0 c S s | t jkS )N)r rg )r. r" r" r# r/ r0 c S s t j| kS )N)r rg )r. r" r" r# r/ r0 c S s t ddd�| kS )Nz2011-03rO )rA )r )r. r" r" r# r/ r0 c S s | t ddd�k S )Nz2011-03rO )rA )r )r. r" r" r# r/ r0 c S s | t jkS )N)r rg )r. r" r" r# r/ r0 c S s t j| kS )N)r rg )r. r" r" r# r/ r0 )r r r rz ry )r rD r{ rR r" r" r# �test_pi_comp_period_nat� sD zBTestPeriodIndexSeriesComparisonConsistency.test_pi_comp_period_natN)r4 r5 r6 �__doc__ry r| r} r" r" r" r# rs � s rs c @ s e Zd Zdd� ZdS )�TestPeriodFrameArithmeticc C s� t �tddd�tddd�gtddd�tddd�gd��}|d jd ksJt�|d jd ks\t�td dd�}|j}t �tjd | d | gtd�tjd| d| gtd�d��}t � || |� t � || d| � t �tddd�tddd�gtddd�tddd�gd��}|d jd k�st�|d jd k�s.t�t �tjd| d| gtd�tjd| d| gtd�d��}t � || |� t � || d| � d S )Nz2015-01rO )rA z2015-02z2014-01z2014-02)r` �Br` z Period[M]r� z2015-03� � )rF � � r2 z2015-05z2015-06r � ) r Z DataFramer rF rv rA r r r- r Zassert_frame_equal)r Zdf�p�offrR Zdf2r" r" r# �test_ops_frame_period# s, z/TestPeriodFrameArithmetic.test_ops_frame_periodN)r4 r5 r6 r� r" r" r" r# r " s r c @ sV e Zd Zdd� Zdd� Zdd� Zdd� Zej� d d d d d g�dd� �Z ej� d d d d d g�ej� de j j dfe j jdfe j jdfe j jdfg�dd� ��Zej� dde�ddg�g�ej� dejejejejg�d d � ��Zej� de� � e� � � � e� � � � e j d d d d �e j d d d!d"�e j d d d#d �j e j d d d$d"�j g�d%d&� �Z!d'd(� Z"d)d*� Z#ej� d+d,d-d.d g�ej� d/dd g�d0d1� ��Z$ej� d2eje j%g�d3d4� �Z&ej� d2eje j%g�d5d6� �Z'd7d8� Z(d9d:� Z)ej� d;d<ejd<ej*d=�g�d>d?� �Z+d@dA� Z,ej� dBdCdDg�dEdF� �Z-dGdH� Z.ej� dIeje j%g�ej� dejejg�dJdK� ��Z/ej� dIeje j%g�dLdM� �Z0dNdO� Z1ej� dPdQdRdSdTdUdVdWg�dXdY� �Z2dZd[� Z3d\d]� Z4d^d_� Z5d`da� Z6dbdc� Z7ddde� Z8dfdg� Z9dhdi� Z:djdk� Z;dldm� Z<ej� dBdCdDg�dndo� �Z=ej� dejdpgdq drd=�e>�?dpgdq �g�dsdt� �Z@dudv� ZAdwdx� ZBdS )y�TestPeriodIndexArithmeticc C st t dddd�}t dddd�}t�||�}d}tjt|d�� || W d Q R X tjt|d�� ||7 }W d Q R X d S )Nz1/1/2000r@ r1 )rA r z1/6/2000z/unsupported operand type\(s\) for \+: .* and .*)rH )r r r r7 rJ rK )r r �rngr rL r" r" r# � test_parr_add_iadd_parr_raisesT s z8TestPeriodIndexArithmetic.test_parr_add_iadd_parr_raisesc C sb t dddd�}t dddd�}|j}t�d| gd �}|| }t�||� ||8 }t�||� d S )Nz1/1/2000r@ r1 )rA r z1/6/2000���)r rA r r r �assert_index_equal)r r� r r� r! r r" r" r# �test_pi_sub_isub_pie s z-TestPeriodIndexArithmetic.test_pi_sub_isub_pic C s� t dddd�}|dd � �dtj�}|dd � �|dd � �sBt�|| }|j}t�tjd| d| d| d| g�}t� ||� d S )Nz1/1/2000r@ r1 )rA r r� r ) r �insertr rg �equalsrv rA r r r� )r r� r r r� r! r" r" r# �test_pi_sub_pi_with_natu s &z1TestPeriodIndexArithmetic.test_pi_sub_pi_with_natc C sR t dddd�}t dddd�}t�||�}d}tjt|d�� || W d Q R X d S ) Nz1/1/2000r@ r1 )rA r z1/6/2000�Hz;Input has different freq=[HD] from PeriodArray\(freq=[DH]\))rH )r r r r7 rJ r )r r r� r rL r" r" r# � test_parr_sub_pi_mismatched_freq s z:TestPeriodIndexArithmetic.test_parr_sub_pi_mismatched_freq�nr� r� � r c C sd d}d}t |g||�d�}t |g||�d�}t |g|jjd�t |g|jjd� }t�|| |� d S )N�19910905�19920406)rA )r rA r[ r r� )r Z tick_classesr� �p1_d�p2_d�p1�p2r! r" r" r# �test_sub_n_gt_1_ticks� s z/TestPeriodIndexArithmetic.test_sub_n_gt_1_tickszoffset, kwd_name�month� startingMonthN�weekdayc C s� |d k r|dini }d}d}||fddi|��}t |g|d�}t |g|d�} | | } t |g|jd�t |g|jd� } t�| | � d S )Nr� r� r� � normalizeF)rA )r r[ r r� ) r �offsetZkwd_namer� �kwdsr� r� rA r� r� r r! r" r" r# �test_sub_n_gt_1_offsets� s z1TestPeriodIndexArithmetic.test_sub_n_gt_1_offsetsr g ��Q� @g @g @�opc C sT t jddgdd�}|�d�}t�||�}d}tjt|d�� |||� W d Q R X d S )Nz 2011-01-01z 2011-01-02r@ )rA zmunsupported operand type\(s\) for [+-]: .* and .*|Concatenation operation is not implemented for NumPy arrays)rH )r Z DatetimeIndex� to_periodr r r7 rJ rK )r r� r r �dtir rL r" r" r# � test_parr_add_sub_float_raises� s z8TestPeriodIndexArithmetic.test_parr_add_sub_float_raisesz 2016-01-01r� )r rA zEurope/Brussels)r �tz�Sz Asia/Tokyoc C s� t dddd�}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 d S )Nz1/1/2000r@ r� )rA r z|(:?cannot add PeriodArray and .*)|(:?cannot subtract .* from (:?a\s)?.*)|(:?unsupported operand type\(s\) for \+: .* and .*))rH )r r r r7 rJ rK )r r r r� rL r" r" r# �test_parr_add_sub_invalid� s z3TestPeriodIndexArithmetic.test_parr_add_sub_invalidc C s� t dddd�}tdddg�}|j}d}tjt|d�� || W d Q R X tjt|d�� || W d Q R X tjt|d�� || W d Q R X d}tjt|d�� || W d Q R X d S ) Nz1/1/2000�Qr� )rA r z-1 DayzCCannot add or subtract timedelta64\[ns\] dtype from period\[Q-DEC\])rH z2cannot subtract PeriodArray from timedelta64\[ns\])r r rQ r7 rJ rK )r r� �tdi�tdarrrL r" r" r# �*test_pi_add_sub_td64_array_non_tick_raises� s zDTestPeriodIndexArithmetic.test_pi_add_sub_td64_array_non_tick_raisesc C s t dddd�}tdddg�}|j}t dddd�}|| }t�||� || }t�||� || }t�||� || }t�||� t dddd�}|| }t�||� || }t�||� d}tjt|d �� || W d Q R X tjt|d �� || W d Q R X d S ) Nz1/1/2000Z90Dr� )rA r z-1 Dayz 12/31/1999z1/2/2000zcannot subtract .* from .*)rH )r r rQ r r� r7 rJ rK )r r� r� r� r! r rL r" r" r# � test_pi_add_sub_td64_array_tick� s, z9TestPeriodIndexArithmetic.test_pi_add_sub_td64_array_tick�pi_freqr@ �Wr� �tdi_freqc C sr |}|t jtjgkr|nt j}tddg|d�}td�| }|�|�}t�||�} |dkr�|| } |� d�| �|�} t�| |�} t� | | � |d | } |d � d�| �|�} t�| |�} t� | | � n�|dk�r"d } t j t | d �� || W d Q R X t j t | d �� |d | W d Q R X nLd } t j t| d �� || W d Q R X t j t| d �� |d | W d Q R X d S ) Nz1 hoursz2 hours)rA z2018-03-07 17:16:40r� r� r r@ z.Input has different freq=None from PeriodArray)rH z"Cannot add or subtract timedelta64)r r r re r r r r� r Z to_timestampr r7 rJ r rK ) r r r� r� �boxr r� r� r Ztd64objr r! rL r" r" r# �test_parr_sub_td64array s4 z1TestPeriodIndexArithmetic.test_parr_sub_td64arrayr� c C s: t td�td�g�}|tjjddd�tjjddd�g�}t td�td�g�}t�t�� || }W d Q R X t�||� t�t�� || }W d Q R X t�||� t � tjj dd �tjj dd �g�}d }t jt|d ��$ t�t�� || W d Q R X W d Q R X t jt|d ��$ t�t�� || W d Q R X W d Q R X d S ) N�2015Q1�2016Q2r� � )r� r� ���Z2015Q2Z2015Q4)r� z1Input cannot be converted to Period\(freq=Q-DEC\))rH )r r r �offsets� QuarterEndr �assert_produces_warningr r� r r �HourZMinuter7 rJ r ) r r� r Zoffsr! �resZres2Z unanchoredrL r" r" r# �test_pi_add_offset_arrayD s& " z2TestPeriodIndexArithmetic.test_pi_add_offset_arrayc s t td�td�g��|tjjddd�tjjddd�g�� t � �fdd�tt���D ��}t�t �� �� }W d Q R X t� ||� |tj� � tjj d d �g�}d }t jt|d ��$ t�t �� �| W d Q R X W d Q R X t jt|d ��$ t�t �� |� W d Q R X W d Q R X d S ) Nr� r� r� r� )r� r� r� c s g | ]}�| � | �qS r" r" )rT r� )r r r" r# rV n s zFTestPeriodIndexArithmetic.test_pi_sub_offset_array.<locals>.<listcomp>r� )r� z6Input has different freq=-1M from Period\(freq=Q-DEC\))rH )r r r r� r� r= �lenr r� r r� �MonthEnd�Dayr7 rJ r )r r� r! r� ZanchoredrL r" )r r r# �test_pi_sub_offset_arrayc s z2TestPeriodIndexArithmetic.test_pi_sub_offset_arrayc C sH t dddd�}|| }t dddd�}t�||� ||7 }t�||� d S )Nz2000-01-01 09:00r� rP )rA r z2000-01-01 10:00)r r r� )r �oner� r r! r" r" r# �test_pi_add_iadd_int� s z.TestPeriodIndexArithmetic.test_pi_add_iadd_intc C sH t dddd�}|| }t dddd�}t�||� ||8 }t�||� dS )z� PeriodIndex.__sub__ and __isub__ with several representations of the integer 1, e.g. int, np.int64, np.uint8, ... z2000-01-01 09:00r� rP )rA r z2000-01-01 08:00N)r r r� )r r� r� r r! r" r" r# �test_pi_sub_isub_int� s z.TestPeriodIndexArithmetic.test_pi_sub_isub_int�fiver1 )rF c C s. t ddd�}|| }|| }t�||� d S )Nz2007-01�2 )r )r r r� )r r� r� r rR r" r" r# �test_pi_sub_intlike� s z-TestPeriodIndexArithmetic.test_pi_sub_intlikec C s� t dddd�}|tj�d� }t dddd�}t�||� |tj�d�8 }t�||� t dd d d�}|tj�d� }t d d d d�}t�||� |tj�d�8 }t�||� d S ) Nrc �2024r` )rA r1 Z2009�2019z2014-01z2016-12rO z2013-08z2016-07)r r r� �YearEndr r� r� )r r� r r! r" r" r# �test_pi_sub_isub_offset� s z1TestPeriodIndexArithmetic.test_pi_sub_isub_offset� transposeTFc C st t ddd�}t|g�}tdgdd�}tj|||d�}tj|||d�}||j }t�||� |j| }t�||� d S )Nz2016-01rY )rA z2016-03)r� )r r r r rA r )r r r� r\ r r! r r" r" r# �test_pi_add_offset_n_gt1� s z2TestPeriodIndexArithmetic.test_pi_add_offset_n_gt1c C sh t dgdd�}t dgdd�}t�||�}t�||�}|td� }t�||� td�| }t�||� d S )Nz2016-01rY )rA z2016-04rZ )r r r r r )r r r r! r r" r" r# �&test_pi_add_offset_n_gt1_not_divisible� s z@TestPeriodIndexArithmetic.test_pi_add_offset_n_gt1_not_divisible� int_holderc C sN t td�td�g�}|ddg�}|||�}t td�td�g�}t�||� d S )Nr� rg r r2 Z2016Q1)r r r r� )r r� r� r r r r! r" r" r# �test_pi_add_intarray� s z.TestPeriodIndexArithmetic.test_pi_add_intarrayc C sr t td�td�g�}|ddg�}|| }t td�td�g�}t�||� d}tjt|d�� || W d Q R X d S )Nr� rg r r2 Z2014Q1z+bad operand type for unary -: 'PeriodArray')rH )r r r r� r7 rJ rK )r r� r r r r! rL r" r" r# �test_pi_sub_intarray� s z.TestPeriodIndexArithmetic.test_pi_sub_intarrayc C s� |}t dddd�}tdddgdd�}|| }t�||� || }t�||� td d d gdd�}|| }t�||� d }tjt|d �� || W d Q R X d S )Nz 2014-05-01r� Z2D)r rA z 2014-05-04z 2014-05-06z 2014-05-08)rA z 2014-04-28z 2014-04-30z 2014-05-02zh(:?bad operand type for unary -: 'PeriodArray')|(:?cannot subtract PeriodArray from timedelta64\[[hD]\]))rH )r r r r� r7 rJ rK )r � three_daysr r� r! r rL r" r" r# �$test_pi_add_timedeltalike_minute_gt1� s z>TestPeriodIndexArithmetic.test_pi_add_timedeltalike_minute_gt1�freqstrZ5nsZ5usZ5msZ5sZ5TZ5hZ5dc C s� |}t dd|d�}t |d | d|d�}|| }t�||� || }t�||� t |d | d|d�}|| }t�||� d}tjt|d�� || W d Q R X d S )Nz 2014-05-01� )r rA r zh(:?bad operand type for unary -: 'PeriodArray')|(:?cannot subtract PeriodArray from timedelta64\[[hD]\]))rH )r r r� r7 rJ rK )r r� r� r r� r! r rL r" r" r# �"test_pi_add_timedeltalike_tick_gt1 s z<TestPeriodIndexArithmetic.test_pi_add_timedeltalike_tick_gt1c C sL |}t dddd�}t dddd�}|| }t�||� ||7 }t�||� d S )Nz 2014-05-01z 2014-05-15r@ )rA z 2014-05-04z 2014-05-18)r r r� )r r� r r� r! r r" r" r# �$test_pi_add_iadd_timedeltalike_daily' s z>TestPeriodIndexArithmetic.test_pi_add_iadd_timedeltalike_dailyc C sL |}t dddd�}t dddd�}|| }t�||� ||8 }t�||� d S )Nz 2014-05-01z 2014-05-15r@ )rA z 2014-04-28z 2014-05-12)r r r� )r r� r r� r! r r" r" r# �$test_pi_sub_isub_timedeltalike_daily3 s z>TestPeriodIndexArithmetic.test_pi_sub_isub_timedeltalike_dailyc C s� |}t dddd�}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 )Nz 2014-05-01z 2014-05-15r@ )rA z7Input has different freq(=.+)? from Period.*?\(freq=D\))rH )r r7 rJ r )r Z not_dailyr r� rL r" r" r# �1test_pi_add_sub_timedeltalike_freq_mismatch_daily? s zKTestPeriodIndexArithmetic.test_pi_add_sub_timedeltalike_freq_mismatch_dailyc C sL |}t dddd�}t dddd�}|| }t�||� ||7 }t�||� d S )Nz2014-01-01 10:00z2014-01-05 10:00r� )rA z2014-01-01 12:00z2014-01-05 12:00)r r r� )r � two_hoursr r� r! r r" r" r# �%test_pi_add_iadd_timedeltalike_hourlyL s z?TestPeriodIndexArithmetic.test_pi_add_iadd_timedeltalike_hourlyc C s^ |}t dddd�}d}tjt|d�� || W d Q R X tjt|d�� ||7 }W d Q R X d S )Nz2014-01-01 10:00z2014-01-05 10:00r� )rA z7Input has different freq(=.+)? from Period.*?\(freq=H\))rH )r r7 rJ r )r Z not_hourlyr r� rL r" r" r# �0test_pi_add_timedeltalike_mismatched_freq_hourlyW s zJTestPeriodIndexArithmetic.test_pi_add_timedeltalike_mismatched_freq_hourlyc C sL |}t dddd�}t dddd�}|| }t�||� ||8 }t�||� d S )Nz2014-01-01 10:00z2014-01-05 10:00r� )rA z2014-01-01 08:00z2014-01-05 08:00)r r r� )r r� r r� r! r r" r" r# �%test_pi_sub_isub_timedeltalike_hourlyb s z?TestPeriodIndexArithmetic.test_pi_sub_isub_timedeltalike_hourlyc C sX t dddd�}|tj�d� }t dddd�}t�||� |tj�d�7 }t�||� d S )Nrc r� r` )rA r1 r� Z2029)r r r� r� r r� )r r� r r! r" r" r# �"test_add_iadd_timedeltalike_annualm s z<TestPeriodIndexArithmetic.test_add_iadd_timedeltalike_annualc C s� |}t dddd�}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 )Nrc r� r` )rA z;Input has different freq(=.+)? from Period.*?\(freq=A-DEC\))rH )r r7 rJ r )r �mismatched_freqr r� rL r" r" r# �2test_pi_add_sub_timedeltalike_freq_mismatch_annualw s zLTestPeriodIndexArithmetic.test_pi_add_sub_timedeltalike_freq_mismatch_annualc C sX t dddd�}t dddd�}|tj�d� }t�||� |tj�d�7 }t�||� d S )Nz2014-01z2016-12rO )rA z2014-06z2017-05r1 )r r r� r� r r� )r r� r! r r" r" r# � test_pi_add_iadd_timedeltalike_M� s z:TestPeriodIndexArithmetic.test_pi_add_iadd_timedeltalike_Mc C s� |}t dddd�}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 )Nz2014-01z2016-12rO )rA z7Input has different freq(=.+)? from Period.*?\(freq=M\))rH )r r7 rJ r )r r� r r� rL r" r" r# �3test_pi_add_sub_timedeltalike_freq_mismatch_monthly� s zMTestPeriodIndexArithmetic.test_pi_add_sub_timedeltalike_freq_mismatch_monthlyc C s� t dddd�}t�d�}tdgd dd�}tj|||d�}tj|||d�}|| }t�||� || }t�||� || }t�||� d}tjt |d �� || W d Q R X d S ) Nz 1994-04-01r' �19D)r rA rg )rA )r� zcannot subtract .* from .*)rH ) r r Z timedelta64r r r r r7 rJ rK ) r r r� r r r! �objr rL r" r" r# �test_parr_add_sub_td64_nat� s z4TestPeriodIndexArithmetic.test_parr_add_sub_td64_natrg r' zm8[ns]c C s� t dddd�}tdgd dd�}t�||�}t�||�}|| }t�||� || }t�||� || }t�||� d}tjt|d�� || W d Q R X d S ) Nz 1994-04-01r' r� )r rA rg )rA zcannot subtract .* from .*)rH )r r r r r r7 rJ rK )r r r r r! r� r rL r" r" r# �!test_parr_add_sub_tdt64_nat_array� s z;TestPeriodIndexArithmetic.test_parr_add_sub_tdt64_nat_arrayc C s2 t ddd�}|j}|| }|| }t�||� d S )Nz 2000-12-31r� )r )r r r r� )r r r( r r! r" r" r# �test_parr_add_sub_index� s z1TestPeriodIndexArithmetic.test_parr_add_sub_indexc C s� t dddd�}|j}t�tdd�tj�d�dg�}t�t �� || }W d Q R X t dd d gdd �j}t� ||� t�t �� || }W d Q R X t d gd dd �j}t� ||� d S ) Nz 2000-12-31r� r@ )r rA r� )r& r� z 2001-01-01z 2001-01-03z 2001-01-05)rA z 2000-12-30) r r r r r r� r� r r� r r r )r r r( r r r! r" r" r# � test_parr_add_sub_object_array� s z8TestPeriodIndexArithmetic.test_parr_add_sub_object_array)Cr4 r5 r6 r� r� r� r� r7 r8 r9 r� r r� r� r� r� ZWeekr� r r �operator�addr Zradd�subZrsubr� r r: Z to_pydatetimeZ to_datetime64r; �_datar� r� r� r� r r� r� r� r� Zint64r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r Z_from_sequencer� r� r� r" r" r" r# r� M sr ( , & $ r� c @ s e Zd Zdd� Zdd� ZdS )�TestPeriodSeriesArithmeticc C s� t tddd�tddd�gdd�}|jdks.t�t tddd�tddd�gdd�}|td � }t�||� td �| }t�||� |tjj � � }t�||� tjj � � | }t�||� d S ) Nz 2015-01-01r@ )rA z 2015-01-02�xxx)rt z Period[D]z 2015-01-03z1 days) r r rF rv r r rq r Ztseriesr� r� )r rW r! r r" r" r# �test_ops_series_timedelta� s z4TestPeriodSeriesArithmetic.test_ops_series_timedeltac C s� t tddd�tddd�gdd�}|jdks.t�tddd�}|j}t d | d | gdtd �}t�|| |� t�|| d | � t td dd�tddd�gdd�}|jdks�t�t d| d| gdtd �}t�|| |� t�|| d | � d S )Nz 2015-01-01r@ )rA z 2015-01-02r� )rt z Period[D]z 2015-01-10r' � )rt rF r2 z 2015-01-05z 2015-01-04r r� )r r rF rv rA r- r rq )r rW r\ r� r! �s2r" r" r# �test_ops_series_period s z1TestPeriodSeriesArithmetic.test_ops_series_periodN)r4 r5 r6 r� r� r" r" r" r# r� � s r� c @ s� e Zd ZdZdd� Zdd� Zej�dddg�ej�d d d � d d � d d � dd � dd � dd � dd � dd � g�dd� ��Z dd� Z dd� Z dd� Z dd � Z d d � Zd d � Zd!d"� Zej�d#d$d%d&d'd(g�d)d*� �Zd(S )+� TestPeriodIndexSeriesMethodsz2Test PeriodIndex and Period Series Ops consistencyc C sJ t |�}||�}t�||� t|�}||�}t||jd�}t�||� d S )N)rt )r r r r rt rq )r rQ rw r! rD r rW rR r" r" r# ry s z#TestPeriodIndexSeriesMethods._checkc C s� t ddddgddd�}t dddd gddd�}| �|d d � |� | �|d d � |� | �|d dd � |� |tddd� }|j}tjd| d| d | d| gdd�}t�||� tddd�| }tjd| d| d| d| gdd�}t�||� d S )Nz2011-01z2011-02z2011-03z2011-04rO rD )rA rt z2011-05z2011-06c S s | d S )Nr� r" )r. r" r" r# r/ 3 r0 z:TestPeriodIndexSeriesMethods.test_pi_ops.<locals>.<lambda>c S s d| S )Nr� r" )r. r" r" r# r/ 4 r0 r� c S s | d S )Nr� r" )r. r" r" r# r/ 6 r0 )rA r r� r� )rt r2 r� ���)r ry r rA r r r r� )r rD r! r r� rR r" r" r# � test_pi_ops* s & &z(TestPeriodIndexSeriesMethods.test_pi_ops�ng�strg �?rw c C s | | S )Nr" )r� r� r" r" r# r/ E r0 z%TestPeriodIndexSeriesMethods.<lambda>c C s || S )Nr" )r� r� r" r" r# r/ F r0 c C s | | S )Nr" )r� r� r" r" r# r/ G r0 c C s || S )Nr" )r� r� r" r" r# r/ H r0 c C s t �| |�S )N)r r� )r� r� r" r" r# r/ I r0 c C s t �|| �S )N)r r� )r� r� r" r" r# r/ J r0 c C s t �| |�S )N)r �subtract)r� r� r" r" r# r/ K r0 c C s t �|| �S )N)r r� )r� r� r" r" r# r/ L r0 c C sN t ddddgddd�}t�||�}d}tjt|d �� |||� W d Q R X d S ) Nz2011-01z2011-02z2011-03z2011-04rO rD )rA rt zWunsupported operand type\(s\)|can only concatenate|must be str|object to str implicitly)rH )r r r r7 rJ rK )r r� rw r rD r� rL r" r" r# �test_parr_ops_errorsA s z1TestPeriodIndexSeriesMethods.test_parr_ops_errorsc C s t ddddgddd�}t dddd gddd�}| �|d d � |� | �|d d � |� | �|d d � |� | �|d dd � |� | �|d dd � |� t ddddgddd�}t ddddgddd�}| �|dd � |� | �|dd � |� | �|dd � |� | �|d dd � |� | �|d dd � |� d S )Nz2011-01z2011-02rg z2011-04rO rD )rA rt z2011-03z2011-06c S s | d S )Nr� r" )r. r" r" r# r/ d r0 z>TestPeriodIndexSeriesMethods.test_pi_ops_nat.<locals>.<lambda>c S s d| S )Nr� r" )r. r" r" r# r/ e r0 c S s t �| d�S )Nr� )r r� )r. r" r" r# r/ f r0 r� c S s | d S )Nr� r" )r. r" r" r# r/ h r0 c S s t �| d�S )Nr� )r r� )r. r" r" r# r/ i r0 rY z2011-07z2011-08z2011-10c S s | d S )Nr� r" )r. r" r" r# r/ s r0 c S s d| S )Nr� r" )r. r" r" r# r/ t r0 c S s t �| d�S )Nr� )r r� )r. r" r" r# r/ u r0 r� c S s | d S )Nr� r" )r. r" r" r# r/ w r0 c S s t �| d�S )Nr� )r r� )r. r" r" r# r/ x r0 )r ry )r rD r! r" r" r# �test_pi_ops_nat\ s$ z,TestPeriodIndexSeriesMethods.test_pi_ops_natc C s� t ddddgddd�}dd � }t dddd gddd�}| �|||� d d � }t d ddd gddd�}| �|||� dd � }t ddddgddd�}| �|||� dd � }t dddd gddd�}| �|||� d S )Nz2011-01z2011-02rg z2011-04rO rD )rA rt c S s | t �ddddg� S )Nr� r� r� r )r r )r. r" r" r# r/ r0 zDTestPeriodIndexSeriesMethods.test_pi_ops_array_int.<locals>.<lambda>z2011-08c S s t �| t �ddddg��S )Nr r2 r� r� )r r� r )r. r" r" r# r/ � r0 z2011-05z2011-06c S s | t �ddddg� S )Nr� r� r� r )r r )r. r" r" r# r/ � r0 z2010-12c S s t �| t �ddddg��S )Nr� r� r� )r r� r )r. r" r" r# r/ � r0 z2010-10)r ry )r rD r{ rR r" r" r# �test_pi_ops_array_intz s$ z2TestPeriodIndexSeriesMethods.test_pi_ops_array_intc C s� t ddddgddd�}dd � }t d d d d gddd�}| �|||� dd � }t ddddgddd�}| �|||� dd � }t ddddgddd�}| �|||� d S )Nz 2011-01-01z 2011-02-01z 2011-03-01z 2011-04-01r@ rD )rA rt c S s | t j�� S )N)r r� r� )r. r" r" r# r/ � r0 zATestPeriodIndexSeriesMethods.test_pi_ops_offset.<locals>.<lambda>z 2011-01-02z 2011-02-02z 2011-03-02z 2011-04-02c S s | t j�d� S )Nr� )r r� r� )r. r" r" r# r/ � r0 z 2011-01-03z 2011-02-03z 2011-03-03z 2011-04-03c S s | t j�d� S )Nr� )r r� r� )r. r" r" r# r/ � r0 z 2010-12-30z 2011-01-30z 2011-02-27z 2011-03-30)r ry )r rD r{ rR r" r" r# �test_pi_ops_offset� s, z/TestPeriodIndexSeriesMethods.test_pi_ops_offsetc C s� t ddddgddd�}t|�}x�||gD ]�}d}tjt|d �� |tj�d � W d Q R X tjt|d �� tj�d �| W d Q R X d }tjt|d �� |tj�d � W d Q R X q(W d S ) Nz 2011-01-01z 2011-02-01z 2011-03-01z 2011-04-01r@ rD )rA rt z4Input has different freq=2H from Period.*?\(freq=D\))rH r� z5Input has different freq=-2H from Period.*?\(freq=D\))r r r7 rJ r r r� r� )r rD rW r� rL r" r" r# �test_pi_offset_errors� s z2TestPeriodIndexSeriesMethods.test_pi_offset_errorsc C s\ t ddddgddd�}|tddd � }|j}tjd | d | d | d | gdd�}t�||� t�|tddd ��}t�||� tddd �| }tjd| d| d| d| gdd�}t�||� t�tddd �|�}t�||� t tj tj tj tj gdd�}|tddd � }t�||� |j|jk�s*t �tddd �| }t�||� |j|jk�sXt �d S )Nz2011-01z2011-02z2011-03z2011-04rO rD )rA rt z2012-01)rA i�i����i����i����)rt r� � rP r' rg ) r r rA r r r r� r r� r �nanrv )r rD r r� rR r" r" r# �test_pi_sub_period� s( & & z/TestPeriodIndexSeriesMethods.test_pi_sub_periodc C sR t ddddgddd�}ttjgd dd �}t�tj| |� t�|tj |� d S ) Nz2011-01z2011-02rg z2011-04rO rD )rA rt r )rt )r r r rg r r� )r rD rR r" r" r# �test_pi_sub_pdnat� s z.TestPeriodIndexSeriesMethods.test_pi_sub_pdnatc C s� t ddddgddd�}|tddd � }|j}tjd | tjd | d | gdd �}t�||� tddd �| }tjd| tjd| d| gdd �}t�||� tt j t j t j t j gdd �}t�|tddd � |� t�tddd �| |� d S )Nz2011-01rg z2011-03z2011-04rO rD )rA rt z2012-01)rA i�i����i����)rt r� rP r' ) r r rA r r rg r r� r r r )r rD r r� rR r" r" r# �test_pi_sub_period_nat� s $ $ z3TestPeriodIndexSeriesMethods.test_pi_sub_period_nat�scalars�aFr� g �?Nc C s< t ddg�}t td�td�gdd�}||k}t�||� d S )NFr� Z2020z period[A-DEC])rF )r r r rq )r r r! rx r r" r" r# �test_comparison_operations s z7TestPeriodIndexSeriesMethods.test_comparison_operations)r4 r5 r6 r~ ry r� r7 r8 r9 r� r� r� r� r� r r r r r" r" r" r# r� s, r� )#r� �numpyr r7 Zpandas._libs.tslibsr r r r Z pandas.errorsr Zpandasr r r r r r Zpandas._testingZ_testingr Z pandas.corer Zpandas.core.arraysr Z pandas.tests.arithmetic.commonr r r> rp rs r r� r� r� r" r" r" r# �<module> s2 g (d+ 4
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.03 |
proxy
|
phpinfo
|
Настройка