bMhA ~ d dl mZ d dlZd dlZd dlmZ d dlmZm Z m
Z
m Z m Z G d d Z
G d d ZdS ) ) timedeltaN)IncompatibleFrequency)NaTPeriod Timedelta Timestampoffsetsc
< e Zd Zd Zd Zd Zd Zd Zej
dg d d Z ej
d d
d g ej
dg d ej
d e j
d
fe j dfe j dfe j dfg d Zd Zd Zej
dg d d Zej
dg d d Zd Zd ZdS ) TestPeriodArithmeticc t j d }d ddg }t j t | 5 |dz d d d n # 1 swxY w Y d}t j t | 5 |t d z d d d n # 1 swxY w Y t j t | 5 |t j d z d d d d S # 1 swxY w Y d S )Nns|z)Python int too large to convert to C longzint too big to convertmatch zvalue too large)
r max to_periodjoinpytestraises
OverflowErrorr r Nano)selfpermsgs j/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/pandas/tests/scalar/period/test_arithmetic.pytest_add_overflow_raisesz-TestPeriodArithmetic.test_add_overflow_raises s m%%d++hh;(
]=
4
4
4 !GG
]=
4
4
4 )A,,
]=
4
4
4 " " ',q// ! ! " " " " " " " " " " " " " " " " " "s6 A$$A(+A( B++B/2B/C66C:=C:c ~ t dddd }t dddd }|dz |k sJ d|z |k sJ d S )ND r freqyearmonthday r )r per1per2s r test_period_add_integerz,TestPeriodArithmetic.test_period_add_integer& s^ 3T:::3T:::ax4 4x4 c t dddd }t dddd }d g d }t j t | 5 |d z d d d n # 1 swxY w Y t j t | 5 d |z d d d n # 1 swxY w Y t j t | 5 ||z d d d d S # 1 swxY w Y d S )
Nr r r r! r& r )z unsupported operand type\(s\)zcan only concatenate strzmust be str, not Periodr str)r r r r TypeError)r r( r) r s r test_period_add_invalidz,TestPeriodArithmetic.test_period_add_invalid, s 3T:::3T:::hh
]9C
0
0
0 5LL
]9C
0
0
0 DLL
]9C
0
0
0 4KK s6 A++A/2A/B##B'*B' C C #C c t dd t dd }}||z
}|d|j z k sJ d}t j t | 5 |t dd z
d d d d S # 1 swxY w Y d S )
N2011Yr" 2007 z4Input has different freq=M from Period\(freq=Y-DEC\)r z2007-01Mr r" r r r )r leftrightresultr s r test_period_sub_period_annualz2TestPeriodArithmetic.test_period_sub_period_annual? s V#...vC0H0H0He UZ ''''E
]0
<
<
< / / 6)#... . . / / / / / / / / / / / / / / / / / /s A77A;>A;c $ t dd }t dd }|j }||z
d|z k sJ ||z
d|z k sJ d}t j t | 5 |t d d
z
d d d d S # 1 swxY w Y d S ) N
2011-01-01r r3 z
2011-01-15i z0Input has different freq=M from Period\(freq=D\)r z2011-02r6 r7 )r r( r) offr s r test_period_sub_periodz+TestPeriodArithmetic.test_period_sub_periodH s l---l---id{cCi''''d{b3h&&&&A
]0
<
<
< / / 6)#... . . / / / / / / / / / / / / / / / / / /s #BB B n)r r& r5 c & t d || }t d || }t t | |j j t t | |j j z
}||z
|k sJ d S )N19910905r3 19920406)r r- r" base)r tick_classesrA p1p2expecteds r test_sub_n_gt_1_ticksz*TestPeriodArithmetic.test_sub_n_gt_1_ticksT s J\\!__
5
5
5
J\\!__
5
5
5#b'' 555 GG"',9
9
9
RH$$$$$$r+ normalizeTFzoffset, kwd_namer$
startingMonthNweekdayc ||dini }d}d}t | |||fi | }t | |||fi | } t || j j t ||j j z
}
| |z
|
k sJ d S )NrB rD rE r3 )r r" rF ) r offsetkwd_namerA rL kwdsp1_dp2_drH rI rJ s r test_sub_n_gt_1_offsetsz,TestPeriodArithmetic.test_sub_n_gt_1_offsets` s !) 4!}}"
Dvva ; ;d ; ;
<
<
<
Dvva ; ;d ; ;
<
<
<$RW\222VDrw|5T5T5TTRH$$$$$$r+ c p dD ]0}t d| }t d| }|t j d z |k sJ t j d |z |k sJ t j d t j d t j t
j dd t d fD ]t}d }t j
t |
5 ||z d d d n # 1 swxY w Y t j
t |
5 ||z d d d n # 1 swxY w Y u2d D ]}t d | }t d
| }|t j d z |k sJ t j d |z |k sJ t d| }|t j d z |k sJ t j d |z |k sJ d
ddg }t j d t j d t j t
j dd t d fD ]r}t j
t |
5 ||z d d d n # 1 swxY w Y t j
t |
5 ||z d d d n # 1 swxY w Y sdD ]}t d| }t d| }|t j d z |k sJ t j d |z |k sJ t d| }|t j d z |k sJ t j d |z |k sJ t d| }|t
j dd z |k sJ t
j dd |z |k sJ t d| }|t
j dd z |k sJ t
j dd |z |k sJ t d | }|t d z |k sJ t d |z |k sJ t d| }|t d z |k sJ t d |z |k sJ d
ddg }t j d t j d t j t
j d d! t d" fD ]r}t j
t |
5 ||z d d d n # 1 swxY w Y t j
t |
5 ||z d d d n # 1 swxY w Y sd#D ]}t d$| }t d%| }|t j d z |k sJ t j d |z |k sJ t d&| }|t j d' z |k sJ t j d' |z |k sJ d(}t d&| }|t
j d'd! z |k sJ t
j d'd! |z |k sJ t d)| }|t
j d*d z |k sJ t
j d*d |z |k sJ t d+| }|t d,- z |k sJ t d,- |z |k sJ t d.| }|t d d/0 z |k sJ t d d/0 |z |k sJ d
ddg }t j d t j d t j t
j d1d t d"d23 fD ]r}t j
t |
5 ||z d d d n # 1 swxY w Y t j
t |
5 ||z d d d n # 1 swxY w Y sd S )4Nr2 2Y3Yr1 r3 2013r& r m r z>C'3.... D111C b)))S0000<##c)S0000 D111C 3///36666>!S))C/36666 D111C 37773>>>>>)S11C73>>>> D111C2 &#---- R==3&#---- D111C,,,,3333 2&&&,3333((.9 C !!$$"1%% q#&&###
]#8DDD #II ]#8DDD #II
& - - D+$777C+$777CQ '3....;q>>C'3....+$777C a (C////<??S(C////2C+$777C 3///36666>!S))C/36666+$777C c222c9999>$,,s2c9999+$777C3////36666 S)))C/36666+$777C377773>>>> !S111C73>>>>((.9 C !!$$"1%% tS))B///
]#8DDD #II ]#8DDD #II
G- - s C//C3 6C3 D''D+ .D+ 1JJ
J )J;;J? J? $T66T: =T: U..U2 5U2 _00_4 7_4 `((`, /`, c
d ddg }dD ]}t d| }|t j d z
t d| k sJ t j d t j d t j t j d
d t d
fD ]:}t j t | 5 ||z
d d d n # 1 swxY w Y ;d
D ]}t d| }|t j
d z
t d| k sJ |t j
d z
t d| k sJ t j d t j d t j t j d
d t d
fD ]:}t j t | 5 ||z
d d d n # 1 swxY w Y ;dD ]}t d| }|t j d z
t d| k sJ |t j d z
t d| k sJ |t j dd z
t d| k sJ |t j dd z
t d| k sJ |t d z
t d | k sJ |t d z
t d| k sJ t j d t j d t j t j d d t d! fD ]:}t j t | 5 ||z
d d d n # 1 swxY w Y ;d"D ]}t d#| }|t j d z
t d$| k sJ |t j d% z
t d&| k sJ |t j d%d z
t d&| k sJ |t j d'd z
t d(| k sJ |t d)* z
t d+| k sJ |t d d,- z
t d.| k sJ t j d t j d t j t j d/d t d!d01 fD ]:}t j t | 5 ||z
d d d n # 1 swxY w Y ;d S )2Nr ra rb rW r1 r3 r& 2009r r[ r r r\ r_ 2011-01r` z2010-03rc rf rg z
2011-03-27ri z
2011-03-31rm rk rl rn rj ro rp r5 rr rs rt rw z2011-03-30 09:00rB z2011-04-01 06:00rx z2011-04-01 08:00ry rz z2011-04-01 07:00r| r} z2011-03-28 06:00r r r )r r r r r r r r r r r r r r r r )r r r" r r? s r test_period_sub_offsetz+TestPeriodArithmetic.test_period_sub_offset s hh*5
& Dd+++C +++vf4/H/H/HHHHH !!$$"1%% sC((#
]#8DDD #II
&
D...C)!,,,yt0L0L0LLLLL)"--- 1M1M1MMMMM !!$$"1%% sC((#
]#8DDD #II
& D D111CQ '6,T+J+J+JJJJJ b)))VLt-L-L-LLLLL 3///6,T3R3R3RRRRR 37776 4< < < 2 && D*I*I*IIIII,,,,|$0O0O0OOOOO !!$$"1%% q#&&###
]#8DDD #II
& D+$777CQ '62D4+P+P+PPPPP a (F3ED,Q,Q,QQQQQ 3///6:LSW3X3X3XXXXX c222f"7 7 7 3////6:LSW3X3X3XXXXX377776"< < <
!!$$"1%% tS))B///
]#8DDD #II
sH C$$C( +C( G**G. 1G. NN "N =UU U r" r\ c t d| }t |z
t u sJ |t z
t u sJ t |z t u sJ |t z t u sJ d S )Nr r3 )r r )r r" r s r test_period_addsub_natz+TestPeriodArithmetic.test_period_addsub_natZ st YT*** SyC SyC SyC SyC r+ unit)r
usmsrl mc t dd }t j d| }||z t u sJ ||z t u sJ ||z
t u sJ t j t d 5 ||z
d d d d S # 1 swxY w Y d S )Nz
2022-06-01r r zunsupported operandr )r r r r r r r. )r r r nats r test_period_add_sub_td64_natz1TestPeriodArithmetic.test_period_add_sub_td64_natg s \3''nUD))SyC SyC SyC
]9,A
B
B
B #II s +A>>BBc t dd }|t j z }t dd }||k sJ |t j d z
}t dd }||k sJ d}t j t
| 5 |t j d z d d d n # 1 swxY w Y t j t
| 5 |t j d z
d d d d S # 1 swxY w Y d S ) Nrf r r3 rh r& rm z-Input cannot be converted to Period\(freq=D\)r )r r r r r r r )r r r: r r s r test_period_ops_offsetz+TestPeriodArithmetic.test_period_ops_offsett s \,,,w{}}$\,,,}}}}w{1~~%\,,,}}}}>
]0
<
<
< " " ',q// ! ! " " " " " " " " " " " " " " " ]0
<
<
< " " ',q// ! ! " " " " " " " " " " " " " " " " " "s$ B22B69B6C==DDc 0 t d }t dd }d}t j t | 5 ||z d d d n # 1 swxY w Y d}t j t | 5 ||z d d d d S # 1 swxY w Y d S )N2017r6 r3 z>unsupported operand type\(s\) for \+: 'Timestamp' and 'Period'r z>unsupported operand type\(s\) for \+: 'Period' and 'Timestamp')r r r r r. )r tsr r s r test_period_add_timestamp_raisesz5TestPeriodArithmetic.test_period_add_timestamp_raises s#
v
V#&&&O
]9C
0
0
0 HH P
]9C
0
0
0 "HH s# AAA8B BB)__name__
__module__ __qualname__r r* r/ r; r@ r mark parametrizerK r r
QuarterEndr WeekrU r r r r r r r+ r r r s " " "( &/ / /
/
/
/ [ S,,,// % % 0/ % [ [4-88 [ S,,,// [
_g &
1
t $
\9 %
%
% 0/ 98
%N N N `Q Q Q f [ V%6%6%677
87
[ V%A%A%ABB
CB
" " "" r+ r c
e Zd Zd Zd Zd Zd Zd Zej
d e j d df e j e
d d
d ff d Zd
S )TestPeriodComparisonsc t dd }t dd }||k rJ ||k sJ ||k sJ ||k sJ ||k rJ ||k rJ d S )N2000-01r6 z2000-02r' )r janfebs r test_period_comparison_same_freqz6TestPeriodComparisons.test_period_comparison_same_freq sr Y$$Y$$#::::czzzzSyyyyczzzz9999#::::::r+ c t dd }t dd }||k sJ ||k sJ ||k sJ ||k rJ ||k rJ d S Nr r6 r' )r r8 r9 s r 3test_period_comparison_same_period_different_objectzITestPeriodComparisons.test_period_comparison_same_period_different_object se i%%y#&&u}}}}u}}}}u}}}}%<<<<%<<<<< t dd }|dk rJ |dk sJ d}d| d| }|dfd|ffD ]\ }}t j t | 5 ||k d d d n # 1 swxY w Y t j t | 5 ||k d d d n # 1 swxY w Y t j t | 5 ||k d d d n # 1 swxY w Y t j t | 5 ||k d d d n # 1 swxY w Y d S )Nr r6 r z'(Period|int)'z#not supported between instances of z and r )r r r r. )r r
int_or_perr r8 r9 s r #test_period_comparison_invalid_typez9TestPeriodComparisons.test_period_comparison_invalid_type s< Y$$!8888axxxx%
QJQQZQQ !Hq#h/ KD% y444
u
y444
y444
u
y444
sH A&&A* -A* B B# &B# CC C >DD D c t dd }t d }t |f|t ft |f|t ffD ]5\ }}||k rJ ||k rJ ||k rJ ||k sJ ||k rJ ||k rJ 6d S )Nr= r r3 )r r r )r r r r8 r9 s r test_period_comparison_natz0TestPeriodComparisons.test_period_comparison_nat s \,,,
|
$
$ #J
#J
"I
I
% %KD% e||||e||||u}}}}5====u}}}}u}}}}} % %r+ zzerodim_arr, expectedr Fr r6 Tc N t dd }||k |u sJ ||k |u sJ d S r r' )r zerodim_arrrJ r s r (test_period_comparison_numpy_zerodim_arrz>TestPeriodComparisons.test_period_comparison_numpy_zerodim_arr sE
Y$${"x////s"x//////r+ N)r r r r r r r r r r r r arrayr r r r+ r r r s $% % %$ [
"(1++u 3)?)? @ @$ GH 0 0 0 0 0r+ r )datetimer numpyr r pandas._libs.tslibs.periodr pandasr r r r r r r r r+ r r s
< < < < < < D S0 S0 S0 S0 S0 S0 S0 S0 S0 S0r+