U
|
¬ý°dMæ ã @ 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 dlmZmZm Z G dd d Z G d
|
d d Z G d d d Z G dd dZ G dd dZ!G dd dZ"G dd dZ#G dd dZ$dS )é N)ÚIncompatibleFrequencyÚPeriodÚ TimestampÚ to_offset)ÚPerformanceWarning)Ú PeriodIndexÚSeriesÚ TimedeltaÚTimedeltaIndexÚ period_range)Úops)ÚTimedeltaArray)Úassert_invalid_addsub_typeÚassert_invalid_comparisonÚget_upcast_boxc @ s e Zd Zej ddedddg¡dd Zdd Zej d d
|
e d e
|
d d dddddg¡dd Z ej de j dd dje jdd dje d ¡e d ¡ ej¡eed ddddge dd¡e dd¡ e¡e ddddg¡g ¡d d Zej d eejd d g¡d!d" ZdS )#Ú TestPeriodArrayLikeComparisonsÚotherÚ2017ÚD©Úfreqc C s^ t dddgdd}t ||¡}t||d}t dddg¡}t ||¡}||k}t ||¡ d S )Nr Z2018r r TF©r ÚtmÚ box_expectedr ÚnpÚarrayÚ assert_equal)Úselfr Úbox_with_arrayÚidxÚxboxÚexpectedÚresult© r# úZd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/arithmetic/test_period.pyÚtest_eq_scalar+ s z-TestPeriodArrayLikeComparisons.test_eq_scalarc C sl t ddd}t | ¡ d ¡}t ||¡}t||d}||k}t ddddg¡}t ||¡}t ||¡ d S )NÚ2000é ©Úperiodsr TF)r r r Zto_numpyr r r r )r r Úpir r r" r! r# r# r$ Útest_compare_zerodim8 s z3TestPeriodArrayLikeComparisons.test_compare_zerodimÚscalarZfooz
|
2021-01-01r' ©Údaysé g #@iÐ FNc 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_scalarF s z:TestPeriodArrayLikeComparisons.test_compare_invalid_scalarr& r( Z1DiÑ iÒ iÓ iÔ c C s( t ddd}t ||¡}t||| d S r0 r1 )r r r r* r2 r# r# r$ Ú test_compare_invalid_listlikeZ 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 ||¡}||}t||d}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 r5 r* r2 r r r! r" r# r# r$ Útest_compare_object_dtypen sV z8TestPeriodArrayLikeComparisons.test_compare_object_dtype)Ú__name__Ú
|
__module__Ú __qualname__ÚpytestÚmarkÚ parametrizer r% r+ r r r3 ÚpdÚ
|
date_ranger Ztimedelta_ranger Úaranger7 Úfloat64ÚlistÚranger8 ÚIndexr4 r? r# r# r# r$ r & s@
|
øþ
|
|
|
|
öþ
|
r c @ s e Zd Zdd Zdd Zej ddddg¡d d
|
Zej ddddg¡d d Z ej ddddg¡d d Z
|
ej ddddg¡dd Z ej ddddg¡dd Z ej de dg¡dd ZdS )ÚTestPeriodIndexComparisonsc C s~ t dddd}|d }||k }|j|jd k }t ||¡ |j dd¡|k }t || dd¡¡ |t |¡k }t ||¡ d S )Nú2007-01é ÚM©r) r é
|
é )r Úvaluesr Úassert_numpy_array_equalZreshaper r )r r Úperr" Úexpr# r# r$ Útest_pi_cmp_period« s z-TestPeriodIndexComparisons.test_pi_cmp_periodc s t dddd}|d fdd|D }t ||¡}t| d}t ||¡}| k}t ||¡ |d | k} fd
|
d|D }t ||¡}t ||¡ d S ) Nz
|
2000-01-01rR r rQ é c s g | ] }| kqS r# r# ©Ú.0r: ©Úvalr# r$ Ú
|
<listcomp>À s zKTestPeriodIndexComparisons.test_parr_cmp_period_scalar2.<locals>.<listcomp>Tr= c s g | ] }| kqS r# r# rZ r\ r# r$ r^ Ë s )r r r r r )r r r* r! Úserr r" r# r\ r$ Ú test_parr_cmp_period_scalar2¼ s z7TestPeriodIndexComparisons.test_parr_cmp_period_scalar2r rP Ú2MÚ3Mc C s° t ddddg|d}t ||¡}td|d}t||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 )Nú2011-01ú2011-02ú2011-03ú2011-04r TF)r r r r r r r r )r r r ÚbaserV r rW r# r# r$ Útest_parr_cmp_period_scalarÏ s8 z6TestPeriodIndexComparisons.test_parr_cmp_period_scalarc C sX t ddddg|d}t ||¡}t ddddg|d}t||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 ) Nrc rd re rf r ú2011-05TFr )r r r rg r r rW 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
|
dk W 5 Q R X tjt|d td d
|
d|k W 5 Q R X t d d d d gd
|
d}d}|tjtj fkr¬|n|}tjt|d ||k W 5 Q R X d| d}tjt|d |td ddk W 5 Q R X tjt|d td dd|k W 5 Q R X t d d d d gdd}d}|tjtj fkr`|n|}tjt|d ||k W 5 Q R X d S )Nrc rd re rf r ú)Invalid comparison between dtype=period\[z \] and Period©ÚmatchÚ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 rC ÚraisesÚ TypeErrorr Úto_arrayrF r )r r r rg Úmsgr 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}|d }||k}t ddddg¡}t ||¡ ||k }t ||¡ |tjk}t ddddg¡}t ||¡ tj|k}t ||¡ |tjk}t ddddg¡}t ||¡ tj|k}t ||¡ t ddd dg|d}||k }t ddddg¡}t ||¡ ||k}t ddddg¡}t ||¡ ||k}t ddddg¡}t ||¡ ||k}t ddddg¡}t ||¡ ||k}t ddddg¡}t ||¡ d S )
|
Nrc rd ÚNaTri r é FTrf )r r r r rU rF ry )r r Úidx1rV r" rW Zidx2r# r# r$ Útest_pi_cmp_nat; sB
|
|
|
|
z*TestPeriodIndexComparisons.test_pi_cmp_natc C s t ddddg|d}t ddddgdd}d| d }tjt|d
|
||k W 5 Q R X ||k}tjd d d d gtd }t ||¡ d S ) Nrc rd ry ri r rf rs rk z\] and PeriodArrayrl F©Údtype) r rC rt ru r r Úboolr rU )r r r{ Zdiffrw r" r! r# r# r$ Ú&test_pi_cmp_nat_mismatched_freq_raisesg s zATestPeriodIndexComparisons.test_pi_cmp_nat_mismatched_freq_raisesr~ Nc C s2 t tdtjtdg}t tjtjtdg}|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 )Nú
|
2011-01-01ú
|
2011-01-03FT) r r rF ry r7 r r r rU )r r~ ÚleftÚrightr" r! r# r# r$ Ú test_comp_natu s(
|
|
z(TestPeriodIndexComparisons.test_comp_nat)r@ rA rB rX r` rC rD rE rh rj rx r| r r8 r
r# r# r# r$ rM ¨ s
|
%
|
"
|
"
|
+
|
rM c @ s e Zd Zdd ZdS )ÚTestPeriodSeriesComparisonsc C s$ t tdddtdddtdddtdddg}t tdddtd ddtdddtd
|
ddg}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 ) Nrn ro r rd rP rq rf rp rc ri FT)r r r Úassert_series_equal)r rg r_ rW r# r# r$ Ú(test_cmp_series_period_series_mixed_freq s4
|
|
|
|
üÿ
|
|
|
|
üÿ zDTestPeriodSeriesComparisons.test_cmp_series_period_series_mixed_freqN)r@ rA rB r r# r# r# r$ r s r c @ s( e Zd ZdZdd Zdd Zdd ZdS ) Ú*TestPeriodIndexSeriesComparisonConsistencyú2Test PeriodIndex and Period Series Ops consistencyc C s` t |}||}t|tjtjfs&tt ||¡ t |}||}t ||j
|
d}t ||¡ d S ©N©Úname) r Ú
|
isinstancerF rL r ZndarrayÚAssertionErrorr r r r r )r rT Úfuncr! r r" ÚsrW r# r# r$ Ú_checkÀ s z1TestPeriodIndexSeriesComparisonConsistency._checkc sP t ddddgddd}|d fd d
|
}tjd d d d gtjd }| |||¡ fdd
|
}| |||¡ fdd
|
}tjd d d d gtjd }| |||¡ fdd
|
}| |||¡ fdd
|
}tjd d d d gtjd }| |||¡ fdd
|
}tjd d d d gtjd }| |||¡ fdd
|
}tjd d d d gtjd }| |||¡ d S )Nrc rd re rf rP r ©r r rS c s | kS r6 r# r9 ©rV r# r$ r; Ö r< zPTestPeriodIndexSeriesComparisonConsistency.test_pi_comp_period.<locals>.<lambda>FTr} c s | kS r6 r# r9 r r# r$ r; Ù r< c s | kS r6 r# r9 r r# r$ r; Ü r< c s | kS r6 r# r9 r r# r$ r; ß r< c s | kS r6 r# r9 r r# r$ r; â r< c s | kS r6 r# r9 r r# r$ r; æ r< c s | kS r6 r# r9 r r# r$ r; ê r< ©r r r Zbool_r ©r r ÚfrW r# r r$ Útest_pi_comp_periodÐ s2
|
ÿ z>TestPeriodIndexSeriesComparisonConsistency.test_pi_comp_periodc s t ddddgddd}|d fd d
|
}tjd d d d gtjd }| |||¡ fdd
|
}| |||¡ dd
|
}tjd d d d gtjd }| |||¡ dd
|
}| |||¡ fdd
|
}tjd d d d gtjd }| |||¡ fdd
|
}| |||¡ dd
|
}tjd d d d gtjd }| |||¡ dd
|
}| |||¡ fdd
|
}tjd d d d gtjd }| |||¡ fdd
|
}tjd d d d gtjd }| |||¡ dd
|
}tjd d d d gtjd }| |||¡ dd
|
}tjd d d d gtjd }| |||¡ d S )Nrc ry re rf rP r r rS c s | kS r6 r# r9 r r# r$ r; ô r< zTTestPeriodIndexSeriesComparisonConsistency.test_pi_comp_period_nat.<locals>.<lambda>FTr} c s | kS r6 r# r9 r r# r$ r; ÷ r< c S s
|
| t jkS r6 ©rF ry r9 r# r# r$ r; ú r< c S s
|
t j| kS r6 r r9 r# r# r$ r; ý r< c s | kS r6 r# r9 r r# r$ r; r< c s | kS r6 r# r9 r r# r$ r; r< c S s
|
| t jkS r6 r r9 r# r# r$ r; r< c S s
|
t j| kS r6 r r9 r# r# r$ r; r< c s | kS r6 r# r9 r r# r$ r; r< c s | k S r6 r# r9 r r# r$ r; r< c S s
|
| t jkS r6 r r9 r# r# r$ r; r< c S s
|
t j| kS r6 r r9 r# r# r$ r; r< r r r# r r$ Útest_pi_comp_period_natî sL
|
ÿ zBTestPeriodIndexSeriesComparisonConsistency.test_pi_comp_period_natN)r@ rA rB Ú__doc__r r r r# r# r# r$ r » s r c @ s e Zd Zdd ZdS )ÚTestPeriodFrameArithmeticc C s t tdddtdddgtdddtdddgd¡}|d jd ksJt|d
|
jd ks\ttd dd}|j}t tjd | d | gtdtjd| d| gtdd¡}t
|
|| |¡ t
|
|| d| ¡ t tdddtdddgtdddtdddgd¡}|d jd kst|d
|
jd ks.tt tjd| d| gtdtjd| d| gtdd¡}t
|
|| |¡ t
|
|| d| ¡ d S )Nz2015-01rP r z2015-02ú2014-01z2014-02)ro ÚBro z Period[M]r z2015-03rS rz r} é é r> z2015-05z2015-06r' é ) rF Ú DataFramer r~ r r r r r8 r Zassert_frame_equal)r ZdfÚpÚoffrW Zdf2r# r# r$ Útest_ops_frame_period" s< þÿ þÿþÿþÿz/TestPeriodFrameArithmetic.test_ops_frame_periodN)r@ rA rB r¥ r# r# r# r$ r ! s r c @ s, 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 deded ¡ ed ¡ e jdd dde jdd dd e jdd d dje jdd d d jd e d d!d"g¡g ¡d#d$ Zd%d& Zd'd( Zej d)d*d+d,dg¡ej d-ddg¡d.d/ Z ej d0eje j g¡d1d2 Z ej d0eje j g¡d3d4 Z d5d6 Z d7d8 Z!ej d9d:ejd:ej"d;g¡d<d= Z#d>d? Z$d@dA Z%ej dBdCdDg¡dEdF Z&dGdH Z'ej dIeje j g¡ej dJe(j)e*j+g¡dKdL Z,ej dIeje j g¡dMdN Z-dOdP Z.ej dQdRdSdTdUdVdWdXg¡dYdZ Z/d[d\ Z0d]d^ Z1d_d` Z2dadb Z3dcdd Z4dedf Z5dgdh Z6didj Z7dkdl Z8dmdn Z9ej dBdCdDg¡dodp Z:ej dejdqgdr dsd;e; <dqgdr ¡g¡dtdu Z=dvdw Z>dxdy Z?dS )zÚTestPeriodIndexArithmeticc C st t dddd}t dddd}t ||¡}d}tjt|d || W 5 Q R X tjt|d ||7 }W 5 Q R X d S )Nú1/1/2000r r= ©r r) ú1/6/2000z/unsupported operand type\(s\) for \+: .* and .*rl )r r r rC rt ru )r r Úrngr rw r# r# r$ Ú test_parr_add_iadd_parr_raisesS s z8TestPeriodIndexArithmetic.test_parr_add_iadd_parr_raisesc C sb t dddd}t dddd}|j}t d| gd ¡}|| }t ||¡ ||8 }t ||¡ d S )Nr§ r r= r¨ r© éûÿÿÿ)r r rF rL r Úassert_index_equal)r rª r r¤ r! r" r# r# r$ Útest_pi_sub_isub_pid 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 )Nr§ r r= r¨ rz r )
|
r ÚinsertrF ry Úequalsr r rL r r )r rª r r" r¤ r! r# r# r$ Útest_pi_sub_pi_with_natt s &z1TestPeriodIndexArithmetic.test_pi_sub_pi_with_natc C s^ t dddd}t dddd}t ||¡}t ||¡}d}tjt|d || W 5 Q R X d S ) Nr§ r r= r¨ r© ÚHz;Input has different freq=[HD] from PeriodArray\(freq=[DH]\)rl )r r r rC rt r )r r Zbox_with_array2rª r rw r# r# r$ Ú test_parr_sub_pi_mismatched_freq~ s z:TestPeriodIndexArithmetic.test_parr_sub_pi_mismatched_freqÚnrz rS rY r' c C sd d}d}t |g||d}t |g||d}t |g|jjdt |g|jjd }t || |¡ d S )NÚ19910905Ú19920406r )r r rg 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|jdt |g|jd } t |
|
| ¡ d S )NrY rµ r¶ Ú normalizeFr )r rg r r ) r ÚoffsetZkwd_namer´ Úkwdsr· r¸ r r¹ rº r" r! r# r# r$ Útest_sub_n_gt_1_offsets s ÿz1TestPeriodIndexArithmetic.test_sub_n_gt_1_offsetsr ú
|
2016-01-01r² rQ zEurope/Brussels)r) ÚtzÚSz
|
Asia/Tokyog
ëQ¸ @g @g @g @c C sÄ t dddd}t ||¡}d dddd g¡}t||| tjt|d
|
|| W 5 Q R X tjt|d
|
|| W 5 Q R X tjt|d
|
|| W 5 Q R X tjt|d
|
|| W 5 Q R X d S ) Nr§ r rY r¨ ú|z!(:?cannot add PeriodArray and .*)z&(:?cannot subtract .* from (:?a\s)?.*)z3(:?unsupported operand type\(s\) for \+: .* and .*)z1unsupported operand type\(s\) for [+-]: .* and .*rl )r r r Újoinr rC rt ru )r r r rª rw 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 5 Q R X tjt|d || W 5 Q R X tjt|d || W 5 Q R X d}tjt|d || W 5 Q R X d S ) Nr§ ÚQrY r¨ ú-1 DayzCCannot add or subtract timedelta64\[ns\] dtype from period\[Q-DEC\]rl z/cannot subtract PeriodArray from TimedeltaArray)r r
|
rT rC rt ru )r rª ÚtdiÚtdarrrw 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 5 Q R X tjt|d || W 5 Q R X d S )
|
Nr§ Z90DrY r¨ rÊ z
|
12/31/1999z1/2/2000úcannot subtract .* from .*rl )r r
|
rT r r rC rt ru )r rª rË rÌ r! r" rw 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 jtjfkr|nt j}tddg|d}td| }| |¡}t ||¡} |dkrÊ|| }
|
| d¡| |¡} t | |¡} t
|
|
|
| ¡ |d | }
|
|d d¡| |¡} t | |¡} t
|
|
|
| ¡ n¤|dkr"d } t j t | d
|
|| W 5 Q R X t j t | d
|
|d | W 5 Q R X nLd } t j t| d
|
|| W 5 Q R X t j t| d
|
|d | W 5 Q R X d S ) Nz1 hoursz2 hoursr z2018-03-07 17:16:40r² rÅ r r únCannot add/subtract timedelta-like from PeriodArray that is not an integer multiple of the PeriodArray's freq.rl z"Cannot add or subtract timedelta64)rF r r rv rL r
|
r Z to_periodr Z to_timestampr rC rt r ru ) r r rÒ rÐ Úboxr rË Zdtir* Ztd64objr" r! rw r# r# r$ Útest_parr_sub_td64array s6
|
|
ÿz1TestPeriodIndexArithmetic.test_parr_sub_td64arrayrÔ c
|
C s@ t tdtdg}|tjjdddtjjdddg}t tdtdg t¡}t t ¡ || }W 5 Q R X t
|
||¡ t t ¡ || }W 5 Q R X t
|
||¡ t tjj dd tjjdd g¡}d
|
}tjt|d $ t t ¡ || W 5 Q R X W 5 Q R X tjt|d $ t t ¡ || W 5 Q R X W 5 Q R X d S ) NÚ2015Q1Ú2016Q2rz é ©r´ r½ éþÿÿÿZ2015Q2Z2015Q4©r´ z1Input cannot be converted to Period\(freq=Q-DEC\)rl )r r rF ÚoffsetsÚ
|
QuarterEndr7 r8 r Úassert_produces_warningr r r r ÚHourZMinuterC rt r ) r rÔ r* Zoffsr! ÚresZres2Z
|
unanchoredrw r# r# r$ Útest_pi_add_offset_array? s* þÿ " z2TestPeriodIndexArithmetic.test_pi_add_offset_arrayc
|
s t tdtdg|tjjdddtjjdddg t fddttD }| t¡}t
|
t ¡ }W 5 Q R X t ||¡ |tj ¡ tjjd d
|
g}d }tjt|d $ t
|
t ¡ | W 5 Q R X W 5 Q R X tjt|d $ t
|
t ¡ | W 5 Q R X W 5 Q R X d S ) NrÖ r× rz rØ rÙ rÚ c s g | ]}| | qS r# r# )r[ r´ ©r r* r# r$ r^ i s zFTestPeriodIndexArithmetic.test_pi_sub_offset_array.<locals>.<listcomp>rS rÛ z6Input has different freq=-1M from Period\(freq=Q-DEC\)rl )r r rF rÜ rÝ rK Úlenr7 r8 r rÞ r r ÚMonthEndÚDayrC rt r )r rÔ r! rà Zanchoredrw r# râ r$ Útest_pi_sub_offset_array^ s& þÿ
|
z2TestPeriodIndexArithmetic.test_pi_sub_offset_arrayc C sH t dddd}|| }t dddd}t ||¡ ||7 }t ||¡ d S )Nú2000-01-01 09:00r² rR r¨ z2000-01-01 10:00©r r r ©r Zonerª 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, ...
|
rç r² rR r¨ z2000-01-01 08:00Nrè ré r# r# r$ Útest_pi_sub_isub_int
s z.TestPeriodIndexArithmetic.test_pi_sub_isub_intÚfiver= r} c C s. t ddd}|| }|| }t ||¡ d S )NrN é2 r( rè )r rì rª r" rW r# r# r$ Útest_pi_sub_intlike s
|
z-TestPeriodIndexArithmetic.test_pi_sub_intlikec C sx t dddd}t d¡}|| }t dd t||D ¡}t ||¡ || }t dd t||D ¡}t ||¡ d S )Nrà rR Ú2DrQ c S s g | ]\}}|| qS r# r# ©r[ r: Úyr# r# r$ r^ s zQTestPeriodIndexArithmetic.test_pi_add_sub_int_array_freqn_gt1.<locals>.<listcomp>c S s g | ]\}}|| qS r# r# rð r# r# r$ r^ £ s )r r rH rF rL Úzipr r )r r* Zarrr" r! r# r# r$ Ú#test_pi_add_sub_int_array_freqn_gt1 s
|
z=TestPeriodIndexArithmetic.test_pi_add_sub_int_array_freqn_gt1c 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 ) Nrr Ú2024ro r r= Z2009Ú2019r ú2016-12rP z2013-08z2016-07)r rF 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 )Nú2016-01ra r z2016-03©rú )r r r r r r )r r rú rV 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 )Nrû ra r z2016-04rb )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_holderÚopc C sN t tdtdg}|ddg}|||}t tdtdg}t ||¡ d S )NrÖ ry r' r> 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 tdtdg}|ddg}|| }t tdtdg}t ||¡ d}tjt|d || W 5 Q R X d S )NrÖ ry r' r> Z2014Q1ú+bad operand type for unary -: 'PeriodArray'rl )r r r r rC rt ru )r rÿ r* r r" r! rw r# r# r$ Útest_pi_sub_intarrayè s z.TestPeriodIndexArithmetic.test_pi_sub_intarrayc C sÊ |}t dddd}t ||¡}tdddgdd}t ||¡}|| }t ||¡ || }t ||¡ td d
|
d gdd}t ||¡}|| }t ||¡ d d dg¡}tjt|d || W 5 Q R X d S )Nú
|
2014-05-01rY rï rQ ú
|
2014-05-04z
|
2014-05-06z
|
2014-05-08r ú
|
2014-04-28z
|
2014-04-30z
|
2014-05-02rÆ r ú4cannot subtract PeriodArray from timedelta64\[[hD]\]rl ) r r r r r rÇ rC rt ru )r Ú
|
three_daysr r rª r! r" rw r# r# r$ Ú&test_parr_add_timedeltalike_minute_gt1ú s( þÿz@TestPeriodIndexArithmetic.test_parr_add_timedeltalike_minute_gt1ÚfreqstrZ5nsZ5usZ5msZ5sZ5TZ5hZ5dc
|
C sÒ |}t dd|d}|d }t ||¡}t || d|d}t ||¡}|| }t ||¡ || }t ||¡ t || d|d}t ||¡}|| }t ||¡ d ddg¡} tjt| d || W 5 Q R X d S ) Nr é rQ r rÆ r r rl )r r r r rÇ rC rt ru )
|
r r r
|
r r rª Úfirstr! r" rw r# r# r$ Ú$test_parr_add_timedeltalike_tick_gt1 s* þÿz>TestPeriodIndexArithmetic.test_parr_add_timedeltalike_tick_gt1c C sL |}t dddd}t dddd}|| }t ||¡ ||7 }t ||¡ d S )Nr ú
|
2014-05-15r r r z
|
2014-05-18rè ©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 )Nr r r r r z
|
2014-05-12rè r r# r# r$ Ú$test_pi_sub_isub_timedeltalike_dailyG s z>TestPeriodIndexArithmetic.test_pi_sub_isub_timedeltalike_dailyc C s¸ |}t dddd}t ||¡}d ddg¡}tjt|d || W 5 Q R X tjt|d ||7 }W 5 Q R X tjt|d || W 5 Q R X tjt|d ||8 }W 5 Q R X d S ) Nr r r r rÆ z7Input has different freq(=.+)? from Period.*?\(freq=D\)rÓ rl ©r r r rÇ rC rt r )r Z not_dailyr r rª rw r# r# r$ Ú3test_parr_add_sub_timedeltalike_freq_mismatch_dailyS s üÿ zMTestPeriodIndexArithmetic.test_parr_add_sub_timedeltalike_freq_mismatch_dailyc C sL |}t dddd}t dddd}|| }t ||¡ ||7 }t ||¡ d S )Nú2014-01-01 10:00ú2014-01-05 10:00r² r z2014-01-01 12:00z2014-01-05 12:00rè ©r Z two_hoursr rª r! r" r# r# r$ Ú%test_pi_add_iadd_timedeltalike_hourlyl s z?TestPeriodIndexArithmetic.test_pi_add_iadd_timedeltalike_hourlyc C st |}t dddd}t ||¡}d ddg¡}tjt|d || W 5 Q R X tjt|d ||7 }W 5 Q R X d S ) Nr r r² r rÆ z7Input has different freq(=.+)? from Period.*?\(freq=H\)rÓ rl r )r Z
|
not_hourlyr r rª rw r# r# r$ Ú2test_parr_add_timedeltalike_mismatched_freq_hourlyw s üÿ
|
zLTestPeriodIndexArithmetic.test_parr_add_timedeltalike_mismatched_freq_hourlyc C sL |}t dddd}t dddd}|| }t ||¡ ||8 }t ||¡ d S )Nr r r² r z2014-01-01 08:00z2014-01-05 08:00rè r r# r# r$ Ú%test_pi_sub_isub_timedeltalike_hourly s z?TestPeriodIndexArithmetic.test_pi_sub_isub_timedeltalike_hourlyc C sX t dddd}|tj d¡ }t dddd}t ||¡ |tj d¡7 }t ||¡ d S )Nrr rô ro r r= rõ Z2029)r rF rÜ r÷ r r rø r# r# r$ Ú"test_add_iadd_timedeltalike_annual s z<TestPeriodIndexArithmetic.test_add_iadd_timedeltalike_annualc C s¢ |}t dddd}d}tjt|d || W 5 Q R X tjt|d ||7 }W 5 Q R X tjt|d || W 5 Q R X tjt|d ||8 }W 5 Q R X d S )Nrr rô ro r z;Input has different freq(=.+)? from Period.*?\(freq=A-DEC\)rl ©r rC rt r ©r Zmismatched_freqr rª rw r# r# r$ Ú2test_pi_add_sub_timedeltalike_freq_mismatch_annual¢ 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 )Nr rö rP r z2014-06z2017-05r= )r rF 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 5 Q R X tjt|d ||7 }W 5 Q R X tjt|d || W 5 Q R X tjt|d ||8 }W 5 Q R X d S )Nr rö rP r z7Input has different freq(=.+)? from Period.*?\(freq=M\)rl r r 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 5 Q R X d S )
|
Nú
|
1994-04-01r/ Ú19DrQ ry r rü rÎ rl )
|
r r Ú timedelta64r r r r rC rt ru ) r r rú r* r r! Úobjr" rw r# r# r$ Útest_parr_add_sub_td64_natÆ s
|
z4TestPeriodIndexArithmetic.test_parr_add_sub_td64_natry r/ zm8[ns]c C s< t dddd}tdgd dd}t ||¡}t ||¡}|| }t ||¡ || }t ||¡ || }t ||¡ d}tjt|d || W 5 Q R X | ¡ }t
|
d d
|
¡|d < t|d gdgd dd}t ||¡}|| }t ||¡ || }t ||¡ || }t ||¡ tjt|d || W 5 Q R X d S ) Nr r/ r! rQ ry r rÎ rl r Únsé ) r r r r r rC rt ru Úcopyr r" )r r r r* r! r# r" rw r# r# r$ Ú!test_parr_add_sub_tdt64_nat_arrayÚ s2 z;TestPeriodIndexArithmetic.test_parr_add_sub_tdt64_nat_arrayc C s2 t ddd}|j}|| }|| }t ||¡ d S )Nú
|
2000-12-31rY r( )r r r r )r r* r2 r" r! r# r# r$ Útest_parr_add_sub_index s
|
z1TestPeriodIndexArithmetic.test_parr_add_sub_indexc C s¾ t dddd}|j}t tddtj d¡dg¡}t t ¡ || }W 5 Q R X t
|
dd d
|
gdd j t ¡}t ||¡ t t ¡ || }W 5 Q R X t
|
d gd dd j t ¡}t ||¡ d S ) Nr) rY r rQ rz r- rS z
|
2001-01-01z
|
2001-01-03z
|
2001-01-05r z
|
2000-12-30)r r r r rF rÜ rå r rÞ r r Ú_datar7 r8 r )r r* r2 r r" r! r# r# r$ Ú test_parr_add_sub_object_array s ÿþ z8TestPeriodIndexArithmetic.test_parr_add_sub_object_array)@r@ rA rB r« r® r± r³ rC rD rE r» rF rÜ r÷ rÝ rä ZWeekr r Z to_pydatetimeZ to_datetime64rG r+ r r rÈ rÍ rÏ rÕ rL rá ræ rê rë Úint64rî ró rù rý rþ ÚoperatorÚaddr Zraddr 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¦ L s
|
|
|
|
|
|
|
üþ
|
|
ôþ
|
/
|
|
|
|
|
!
|
|
|
|
þþ
|
# r¦ c @ s e Zd Zdd Zdd ZdS )ÚTestPeriodSeriesArithmeticc C s® t tdddtdddgdd}|jdks.tt tdddtd ddgdd}t ||¡}|tjkrv|jdk ¡ svtt ||¡}|| }t
|
||¡ || }t
|
||¡ d S )
|
Nú
|
2015-01-01r r ú
|
2015-01-02Úxxxr ú Period[D]ú
|
2015-01-04ú
|
2015-01-05) r r r~ r r r rF r¢ ZdtypesÚallr )r r r r_ r! r# r" r# r# r$ Ú"test_parr_add_timedeltalike_scalar# s" þþ
|
z=TestPeriodSeriesArithmetic.test_parr_add_timedeltalike_scalarc C sî t tdddtdddgdd}|jdks.ttddd}|j}t d | d
|
| gdtd }t || |¡ t || d | ¡ t td ddtdddgdd}|jdks¬tt d| d| gdtd }t || |¡ t || d | ¡ d S )Nr1 r r r2 r3 r r4 z
|
2015-01-10r/ r& )r r~ r> r6 r5 r' rS )r r r~ r r r8 r r )r r_ rV r¤ r! Ús2r# r# r$ Útest_ops_series_period< s$ þ þz1TestPeriodSeriesArithmetic.test_ops_series_periodN)r@ rA rB r8 r: r# r# r# r$ r0 " s r0 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" Zd#S )$Ú TestPeriodIndexSeriesMethodsr c C sJ t |}||}t ||¡ t|}||}t||jd}t ||¡ d S r )r r r r r r )r rT r r! r r" r_ rW r# r# r$ r Y 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 )Nrc rd re rf rP r r ri ú2011-06c S s | d S ©NrS r# r9 r# r# r$ r; m r< z:TestPeriodIndexSeriesMethods.test_pi_ops.<locals>.<lambda>c S s d| S r= r# r9 r# r# r$ r; n r< rS c S s | d S r= r# r9 r# r# r$ r; p r< r r rz rY r r> rÚ éýÿÿÿ)r r r r rF rL r r )r r r! r" r¤ rW r# r# r$ Ú test_pi_opsd s(
|
ÿ
|
ÿ& &z(TestPeriodIndexSeriesMethods.test_pi_opsÚngÚstrg ø?r c C s | | S r6 r# ©r# r@ r# r# r$ r; r< z%TestPeriodIndexSeriesMethods.<lambda>c C s || S r6 r# rB r# r# r$ r; r< c C s | | S r6 r# rB r# r# r$ r; r< c C s || S r6 r# rB r# r# r$ r; r< c C s t | |¡S r6 ©r r/ rB r# r# r$ r; r< c C s t || ¡S r6 rC rB r# r# r$ r; r< c C s t | |¡S r6 ©r ÚsubtractrB r# r# r$ r;
r< c C s t || ¡S r6 rD rB r# r# r$ r; r< c C s\ t ddddgddd}t ||¡}d d d
|
d d g¡}tjt|d ||| W 5 Q R X d S )Nrc rd re rf rP r r rÆ z unsupported operand type\(s\)zcan only concatenatez must be strzobject to str implicitlyrl )r r r rÇ rC rt ru )r r@ r r r r# rw r# r# r$ Útest_parr_ops_errors{ 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 )Nrc rd ry rf rP r r re r< c S s | d S r= r# r9 r# r# r$ r; ¢ r< z>TestPeriodIndexSeriesMethods.test_pi_ops_nat.<locals>.<lambda>c S s d| S r= r# r9 r# r# r$ r; £ r< c S s t | d¡S r= rC r9 r# r# r$ r; ¤ r< rS c S s | d S r= r# r9 r# r# r$ r; ¦ r< c S s t | d¡S r= rD r9 r# r# r$ r; § r< ra z2011-07ú2011-08z2011-10c S s | d S ©NrY r# r9 r# r# r$ r; ± r< c S s d| S rH r# r9 r# r# r$ r; ² r< c S s t | d¡S rH rC r9 r# r# r$ r; ³ r< rY c S s | d S rH r# r9 r# r# r$ r; µ r< c S s t | d¡S rH rD r9 r# r# r$ r; ¶ r< ©r r )r r 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 )Nrc rd ry rf rP r r c S s | t ddddg¡ S ©Nrz rS rY r' ©r r r9 r# r# r$ r; ¼ r< zDTestPeriodIndexSeriesMethods.test_pi_ops_array_int.<locals>.<lambda>rG c S s t | t ddddg¡¡S )Nr' r> rz rS )r r/ r r9 r# r# r$ r; Â r< ri r< c S s | t ddddg¡ S rK rL r9 r# r# r$ r; È r< z2010-12c S s t | t ddddg¡¡S )NrY rS rÚ )r rE r r9 r# r# r$ r; Î r< z2010-10rI r r# r# r$ Útest_pi_ops_array_int¸ sB
|
ÿ
|
ÿ
|
ÿ
|
ÿ
|
ÿ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 )Nr ú
|
2011-02-01ú
|
2011-03-01ú
|
2011-04-01r r r c S s | t j ¡ S r6 ©rF rÜ rå r9 r# r# r$ r; Ú r< 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 r= rQ r9 r# r# r$ r; â r< r z
|
2011-02-03z
|
2011-03-03z
|
2011-04-03c S s | t j d¡ S r= rQ r9 r# r# r$ r; ê r< z
|
2010-12-30z
|
2011-01-30z
|
2011-02-27z
|
2011-03-30rI r r# r# r$ Útest_pi_ops_offsetÔ s4
|
ý
|
ý
|
ý
|
ýz/TestPeriodIndexSeriesMethods.test_pi_ops_offsetc
|
C s² t ddddgddd}t|}d}||fD ]}tjt|d |tj d
|
¡ W 5 Q R X tjt|d tj d
|
¡| W 5 Q R X tjt|d |tj d
|
¡ W 5 Q R X q*d S ) Nr rN rO rP r r r zmCannot add/subtract timedelta-like from PeriodArray that is not an integer multiple of the PeriodArray's freqrl rS )r r rC rt r rF rÜ rß )r r r_ rw r# 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|jks*t tddd | }t ||¡ |j|jksXt d S )Nrc rd re rf rP r r ú2012-01r éôÿÿÿiõÿÿÿéöÿÿÿé÷ÿÿÿr rØ é rR r/ ry ) r r r rF rL r r r rE r
|
Únanr ©r r r" r¤ rW 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 )
|
Nrc rd ry rf rP r r r' r )r r
|
rF ry r r )r r rW 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 )Nrc ry re rf rP r r rT r rU rV rW r rØ rR r/ ) r r r rF rL ry r r r
|
r rY rZ r# r# r$ Útest_pi_sub_period_nat/ s
|
ÿ$ $ z3TestPeriodIndexSeriesMethods.test_pi_sub_period_natN)r@ rA rB r r r? rC rD rE rF rJ rM rR rS r[ r\ r] r# r# r# r$ r; V s0 øþ r; )%r. Únumpyr rC Zpandas._libs.tslibsr r r r Z pandas.errorsr ZpandasrF r r r r
|
r Zpandas._testingZ_testingr Z pandas.corer Zpandas.core.arraysr Z pandas.tests.arithmetic.commonr r r r rM r r r r¦ r0 r; r# r# r# r$ Ú<module> s2
|
l(f+ [4
|
