ehA p d dl mZ d dlZd dlZd dlmZ d dlmZm Z m
Z
m Z m Z G d d Z
G d d Zy) ) timedeltaN)IncompatibleFrequency)NaTPeriod Timedelta Timestampoffsetsc
J e Zd Zd Zd Zd Zd Zd Zej j dg d d Z ej j d d
d g ej j dg d ej j d e j d
fe j dfe j dfe j dfg d Zd Zd Zej j dg d d Zej j dg d d Zd Zd Zy) TestPeriodArithmeticc t j j d }dj ddg }t j
t | 5 |dz d d d d}t j
t | 5 |t d z d d d t j
t | 5 |t j d z d d d y # 1 sw Y zxY w# 1 sw Y RxY w# 1 sw Y y xY w)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 h/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/pandas/tests/scalar/period/test_arithmetic.pytest_add_overflow_raisesz-TestPeriodArithmetic.test_add_overflow_raises s mm%%d+hh;(
]]=
4 !G
]]=
4 )A,
]]=
4 " ',,q/ ! " "
" "s$ C
9C+C%
CC"%C.c h t dddd }t dddd }|dz |k( sJ d|z |k( sJ y )ND r freqyearmonthday r )r per1per2s r test_period_add_integerz,TestPeriodArithmetic.test_period_add_integer& sF 3T:3T:ax4 4x4 c t dddd }t dddd }dj g d }t j t | 5 |d z d d d t j t | 5 d |z d d d t j t | 5 ||z d d d y # 1 sw Y \xY w# 1 sw Y ?xY w# 1 sw Y y xY w)
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 5L
]]9C
0 DL
]]9C
0 4K s$
B.6B: C.B7:CCc t dd t dd }}||z
}|d|j z k( sJ d}t j t | 5 |t dd z
d d d y # 1 sw Y y xY w)
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? sp V#.vC0He UZZ '''E
]]0
< / 6)#. . / / /s A++A4c 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 y # 1 sw Y y xY w) 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-iid{cCi'''d{b3h&&&A
]]0
< / 6)#. . / / /s A88Bn)r r& r5 c t d || }t d || }t t | |j j t t | |j j z
}||z
|k( sJ y )N19910905r3 19920406)r r- r" base)r tick_classesrA p1p2expecteds r test_sub_n_gt_1_ticksz*TestPeriodArithmetic.test_sub_n_gt_1_ticksT sm J\!_
5
J\!_
5#b' 5 G"'',,9
RH$$$r+ normalizeTFzoffset, kwd_namer$
startingMonthNweekdayc
||dini }d}d}t | |||fi | }t | |||fi | } t || j j t ||j j z
}
| |z
|
k( sJ y )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!}"
Dva ;d ;
<
Dva ;d ;
<$RWW\\2VDrww||5TTRH$$$r+ c j dD ] }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 ]V }d }t j t |
5 ||z d d d t j t |
5 ||z d d d X d D ]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 dj ddg }t j d t j d t j
t
j dd t d fD ]T }t j t |
5 ||z d d d t j t |
5 ||z d d d V g dD ]_ }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 dj ddg }t j d t j d t j
t
j d d! t d" fD ]T }t j t |
5 ||z d d d t j t |
5 ||z d d d V b d#D ]f }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 dj ddg }t j d t j d t j
t
j d1d t d"d23 fD ]T }t j t |
5 ||z d d d t j t |
5 ||z d d d V i y # 1 sw Y txY w# 1 sw Y xY w# 1 sw Y $xY w# 1 sw Y ]xY w# 1 sw Y xY w# 1 sw Y xY w# 1 sw Y xY w# 1 sw Y xY w)4Nr2 2Y3Yr1 r3 2013r& r m r zC'3. .. D1C b))S0 00<<#c)S0 00 D1C 3//36 66>>!S)C/36 66 D1C 3773> >>>>)S1C73> >> D1C2 &#- -- R=3&#- -- D1C,,3 33 2&,3 33((.9C !!!$""1% q#&#
]]#8D #I ]]#8D #I
E, \ & - D+$7C+$7CQ '3. ..;;q>C'3. ..+$7C a (C/ //<<?S(C/ //2C+$7C 3//36 66>>!S)C/36 66+$7C c22c9 99>>$,s2c9 99+$7C3//36 66 S)C/36 66+$7C3773> >> !S1C73> >>((.9C !!!$""1% tS)B/
]]#8D #I ]]#8D #I
G- c 6 Z Z s` ] ] ])*]6&^^^ ;^)] ]& )]3 6^ ^
^ ^& )^2 c dj 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 - 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 - dD ]z }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 - } 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 - y # 1 sw Y ,xY w# 1 sw Y KxY w# 1 sw Y xY w# 1 sw Y bxY w)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/HH HH !!!$""1% sC(#
]]#8D #I
&
D.C))!,,yt0LL LL))"-- 1MM MM !!!$""1% sC(#
]]#8D #I
& D D1CQ '6,T+JJ JJ b))VLt-LL LL 3//6,T3RR RR 3776 4<
2 && D*II II,,|$0OO OO !!!$""1% q#&#
]]#8D #I
* & D+$7CQ '62D4+PP PP a (F3ED,QQ QQ 3//6:LSW3XX XX c22f"7
3//6:LSW3XX XX3776"<
!!!$""1% tS)B/
]]#8D #I
Q , . s0 S 6S5S&8S3 S S# &S0 3S< 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 y )Nr r3 )r r )r r" r s r test_period_addsub_natz+TestPeriodArithmetic.test_period_addsub_natZ s\ 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 y # 1 sw Y y xY w)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'nnUD)SyC SyC SyC
]]9,A
B #I s %A44A=c 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 t j t
| 5 |t j d z
d d d y # 1 sw Y FxY w# 1 sw Y y xY w) 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 C >C, C),C5c t d }t dd }d}t j t | 5 ||z d d d d}t j t | 5 ||z d d d y # 1 sw Y 5xY w# 1 sw Y y xY w)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 H P
]]9C
0 "H s A0!A<0A9<B)__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-8 [[ S,/ [[
__g &
1
t $
\\9 %
% 0 9
%N `Q f [[ V%67
8
[[ V%AB
C
"" r+ r c
e Zd Zd Zd Zd Zd Zd Zej j d e j d df e j e
d d
d ff d Zy
)TestPeriodComparisonsc t dd }t dd }||k( rJ ||k7 sJ ||k sJ ||k sJ ||kD rJ ||k\ rJ y )N2000-01r6 z2000-02r' )r janfebs r test_period_comparison_same_freqz6TestPeriodComparisons.test_period_comparison_same_freq sd Y$Y$#: ~czzSyyczz9 }#: ~:r+ c z t dd }t dd }||k( sJ ||k\ sJ ||k sJ ||k rJ ||kD rJ y Nr r6 r' )r r8 r9 s r 3test_period_comparison_same_period_different_objectzITestPeriodComparisons.test_period_comparison_same_period_different_object sZ i%y#&u} }u} }u} }%< %< C# &C/ 2C; >D c t dd }t d }t |f|t ft |f|t ffD ]/ \ }}||k rJ ||kD rJ ||k( rJ ||k7 sJ ||k rJ ||k\ s/J y )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 @ t dd }||k( |u sJ ||k( |u sJ y r r' )r zerodim_arrrJ r s r (test_period_comparison_numpy_zerodim_arrz>TestPeriodComparisons.test_period_comparison_numpy_zerodim_arr s6
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 so $%$ [[
"((1+u 3)? @$ GH0 0r+ r )datetimer numpyr r pandas._libs.tslibs.periodr pandasr r r r r r r r r+ r r s4
< D S0 S0r+