U
|
®ý°d¦Ô ã @ s d dl mZm Z mZ d dlZd dlZd dlmZmZ d dl
|
m Z m Z d dl mZ d dlmZ d dlmZmZ d dlZd dlmZmZmZmZmZ d dlm Z G d d
|
d
|
Z G d d d Z G d d dZ G dd dZ!G dd dZ"G dd dZ#dd Z$dd Z%dd Z&dd Z'dS ) é )ÚdateÚdatetimeÚ timedeltaN)ÚiNaTÚperiod)ÚDAYSÚMONTHS)ÚOutOfBoundsDatetime)ÚDateParseError)ÚINVALID_FREQ_ERR_MSGÚIncompatibleFrequency)ÚNaTÚPeriodÚ TimedeltaÚ TimestampÚoffsetsc @ sj e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z dd Z ej de¡dd Zej de¡dd Zej d e¡ej d ed d ¡d d! Zd"d# Zd$d% Zej d&d'd(d)d*g¡d+d, Zd-d. Zd/d0 Zd1d2 Z ej d d3d4d5g¡ej d6d7d8d9d:g¡ej d;d<d=d>d?d@dAdBdCg¡dDdE Z ej d6edF¡dGdH Z dIS )JÚTestPeriodConstructionc C sZ t d¡}d}tjt|d t| W 5 Q R X tjt|d t|dd W 5 Q R X d S )Nzm8[ns]ú2Value must be Period, string, integer, or datetime©ÚmatchÚD©Úfreq)r Zto_numpyÚpytestÚraisesÚ
|
ValueErrorr )ÚselfÚtdÚmsg© r ú]d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/scalar/period/test_period.pyÚtest_from_td64nat_raises% s
|
z/TestPeriodConstruction.test_from_td64nat_raisesc C s t ddd}t d}||ks tt ddd}t d}t ddd}||ksLt||ksXtt ddd}t ddd}||ks|t||kstt d ¡}t t ¡ d d}t d
|
¡}||ks¸t||ksÄtt d ¡}t t ¡ d d}t t ¡ ¡}||ksøt||kstt d d d}t d dd}||ks,tt dddd d}t dd d}||ksVtt ddddd}||ksttt d}t tddddddddd}||ks¦tt ddd}||ksÀtt d}t tddddddd d d}||ksòtt dd d}||ks td }tjt|d t d d! W 5 Q R X d"}tjt|d t d#d$d W 5 Q R X tjt d%d t d d&d W 5 Q R X d S )'Nú1/1/2005ÚMr zJan 2005Ú2005ÚAÚaÚmÚQÚqr Ú1982ÚminZMINéÕ é é ©ÚyearÚmonthÚdayr z3/1/2005Údú2007-01-01 09:00:00.001é× é r éè ÚLú2007-01-01 09:00:00.00101éò ÚUz"Must supply freq for ordinal valuer éý )ÚordinalzInvalid frequency: Xz2007-1-1ÚXzpass as a string instead)ÚMinr. )
|
r ÚAssertionErrorÚnowr r ÚDayr r r Ú TypeError)r Úi1Úi2Úi3Úi4Úi5Úexpectedr r r r Útest_construction0 s`
|
|
|
z(TestPeriodConstruction.test_constructionc C sT t d}t|dd}| ¡ }||ks(t|j}t|dd}| ¡ }|j|ksPtd S )Nz 2022-04-20 09:23:24.123456789Únsr )r r Ú to_timestampr@ Zasm8)r ÚtsÚperÚrtZdt64Úper2Zrt2r r r Ú&test_construction_from_timestamp_nanosv s z=TestPeriodConstruction.test_construction_from_timestamp_nanosc C sª t ddd}t ddd}|| d¡ks*tt ddd}|| d¡ksHtt ddd}|| d¡ksftt ddd}||ks~tt dd d
|
dd }t ddd}||ks¦td S ) Nú3/10/12ÚBr r z3/11/12z3/12/12ÚbéÜ r- é
|
r/ )r Úasfreqr@ )r rD rE rF r r r Útest_construction_bday
s z-TestPeriodConstruction.test_construction_bdayc C s, t dddd}t ddd}||ks&tt dddd}t ddd}||ksLtt d }t dddd}t d
|
}||ksvt||kstt d }||kstt d }||ksªtt d }||ks¾tt d}||ksÒtt d}||ksætt d}||ksútt d}|jdkstt d}||ks(td S )Nr, r. r( ©r0 Úquarterr r" r r- z9/1/2005Z2005Q1Z2005q1Z05Q1Z05q1Z1Q2005Z1q2005Z1Q05Z1q05Z4Q1984iÀ Z4q1984)r r@ r0 )r rD rE rF Úlowerr r r Útest_construction_quarter s6 z0TestPeriodConstruction.test_construction_quarterc C s& t ddd}t ddd}||ks$tt ddd}||ks<tt ddd}||ksTtt ddd}|jdksntt tdd d dd}t ddd}||kstt tdd d dd}t tdd d dd}t t d
|
¡dd}t d dd}t d dd}||ksøt||kst||kst||ks"td S ) Nú2007-01r# r Ú200701r< ©r= r éI r5 r. ú
|
2007-01-01ú2007-01-01 00:00:00ú2007-01-01 00:00:00.000)r r@ r0 r r ÚnpÚ
|
datetime64©r rI rD rE rF rG rH r r r Útest_construction_month¹ s* z.TestPeriodConstruction.test_construction_monthc C sv t dt ¡ dt dddks tt dt ¡ dt dddks@tt dt ¡ dt dddks`tt dt ¡ dt dddkstt dt ¡ dt dddks tt d d
|
tjd d d t d d
|
dd ksÈtt d dtjd d d t d ddd ksðtt d dd
|
t ¡ dt d dd
|
ddkstt dddt ¡ dt dddddksDtt ddd}t d dd
|
t d¡d|ksptt d dd
|
dd|kstt dddt d¡dt dddddks¶tt dt ¡ dt dddksØtt dt ¡ d}t ddd}||kst|j dkst|j dks"tt t
|
dd
|
d
|
dd}t ddd}||ksPtt t dd
|
d
|
dd}t t
|
dd
|
d
|
dd}t t d ¡dd}t d dd}t d dd}||ks°t||ks¾t||ksÌt||ksÚtt d }t t
|
dd
|
d
|
d d!d!d"d#d}||ks tt d d#d}||ks&tt d$}t t
|
dd
|
d
|
d d!d!d%d&d}||ksXtt d$d&d}||ksrtd S )'Nr" r r# r$ r% rR rS r r, r. é )Ú startingMonthrY r( é r- r/ rU rV z
|
2005-03-01Ú3DZ3Br< r_ r` r5 r^ ra rb rc r4 r6 r r7 r8 r9 r: r; )r r ÚMonthEndr@ ÚYearEndZ BusinessDayrB Ú
|
QuarterEndZBDayr0 r r rd re rf r r r Ú test_period_constructor_offsetsÕ s ÿ ÿ
|
ÿ þ
|
ÿ þ ÿ ÿ ÿ" z6TestPeriodConstruction.test_period_constructor_offsetsc C sx d}t jt|d tt ¡ W 5 Q R X t jt|d tt ¡ ¡ W 5 Q R X d}t jt|d tddd W 5 Q R X d}t jt|d tddd W 5 Q R X d }t jt|d td
|
d dd W 5 Q R X d }t jt|d td d W 5 Q R X d}t jt|d tdd W 5 Q R X d}t jt|d tdd W 5 Q R X d}t jt|d tdd W 5 Q R X d S )Nz#Must supply freq for datetime valuer r gù?r r zOrdinal must be an integerr_ z?Only value or ordinal but not both should be given but not bothrj r. )r= Úvaluer z%If value is None, freq cannot be None)r1 z1^Given date string "-2000" not likely a datetime$z-2000r% z day is out of range for monthÚ0z/Unknown datetime string format, unable to parsez 1/1/-2000)r r r r r rA r r
|
©r r r r r Útest_invalid_arguments s4 z-TestPeriodConstruction.test_invalid_argumentsc C sf t ddd}t dddd|ks"tt d tks2tt ddd}t |d d}t d
|
d d}||ksbtd S ) Nr] Ú2Mr r5 r. )r0 r1 r ra r r% Ú2007)r r@ r )r rI ÚpÚresultÚexpr r r Útest_constructor_corner9 s z.TestPeriodConstruction.test_constructor_cornerc C s t d}|jdkstt d}|jdks,tt d}|jdksBtt d}|jdksXtt d }|jd
|
ksntt d }|jd kstt d }|jd kstd S )Nra r z 2007-01-01 07ÚHz2007-01-01 07:10ÚTz2007-01-01 07:10:15ÚSz2007-01-01 07:10:15.123r8 z2007-01-01 07:10:15.123000r; z2007-01-01 07:10:15.123400)r r r@ ©r rv r r r Útest_constructor_infer_freqE s z2TestPeriodConstruction.test_constructor_infer_freqc C sÎ t ddd}t ddd}|j|jks(t|jdks6t|jdksDt|jt d¡ksXt|jt ¡ ksjt|d j|jd kstd| j|jd kst|d j|jd ks²td | j|jd ksÊtd S )
|
NÚ1989Ú2Ar r% z2A-DECzA-DECrj r. éÿÿÿÿ)r r= r@ Úfreqstrr r rm )r Zresult1Zresult2r r r Útest_multiples] s z%TestPeriodConstruction.test_multiplesr1 c C sv d| }t d|d}dt|ks&t|jddd}t ||d}||ksLt|jddd}t ||d}||ksrtd S )NzQ-Z1989Q3r r Úend©Úhowrk )r Ústrr@ rL ©r r1 r rx Zstamprv r r r Útest_period_cons_quarterlyk s
|
z1TestPeriodConstruction.test_period_cons_quarterlyc C s\ d| }t d|d}|jdddtdd }t ||d}||d ksJtt|t sXtd S )
|
NzA-r r r r r
é )Údaysr. )r rL r r@ Ú
|
isinstancer r r r Útest_period_cons_annualy s
|
z.TestPeriodConstruction.test_period_cons_annualr2 ÚnumrV é c C sP d| }d| }t ||d}t |dd |¡}||ks>tt|t sLtd S )Nz2011-02-zW-r r )r rW r@ r )r r r2 Zdaystrr rw rI r r r Útest_period_cons_weekly s
|
|
z.TestPeriodConstruction.test_period_cons_weeklyc C sZ t d}|jjdkstt d}|jjdks0td}tjt|d t d W 5 Q R X d S )Nz2017-01-23/2017-01-29úW-SUNz2017-01-24/2017-01-30zW-MONz%Could not parse as weekly-freq Periodr z2016-01-23/2017-01-29)r r r r@ r r r )r rN r r r r Ú test_parse_week_str_roundstrip s z5TestPeriodConstruction.test_parse_week_str_roundstripc C s: t ddd}t j|jdd}||ks(tt|t s6td S ©Nú2011-01r# r )r Z _from_ordinalr= r@ r ©r rv Úresr r r Útest_period_from_ordinal s z/TestPeriodConstruction.test_period_from_ordinalr r% r# r rz c C sT t d|d}|tkstt dd| d}|tks4tt dd| d}|tksPtd S )Nr r Ú2Ú3©r r r@ ©r r rN r r r Ú'test_construct_from_nat_string_and_freq¢ s z>TestPeriodConstruction.test_construct_from_nat_string_and_freqc C s t ddd}|tkstt tdd}|tks0tt tdd}|tksHtt tdd}|tks`tt d}|tksttt t}|tkstd S )NÚnatr r r rk Ú1D1Hr )r r r@ r r} r r r Útest_period_cons_nat s z+TestPeriodConstruction.test_period_cons_natc C s6 t ddd}t ddd}|j|jks(t|jt d¡ks<t|jdksJt|jt ¡ ks\t|jdksjt|d }|j|d jkst|j|jkst|jdks¤t|d }|j|d jksÀt|j|jksÐt|jdksÞtd}tjt |d t dd d W 5 Q R X d
|
}tjt |d t dd d W 5 Q R X d S ) Nr Ú3Mr r# r- r. z;Frequency must be positive, because it represents span: -3Mr z-3Mz:Frequency must be positive, because it represents span: 0MZ0M)
|
r r= r@ r r rl r r r r )r Úp1Úp2rw r r r r Útest_period_cons_multÀ s* z,TestPeriodConstruction.test_period_cons_multc C s t dddt dddt dddft dddt dddt dddfg}|D ]\}}}|j|jksdt|j|jkstt|jt d¡kst|jd kst|jt d¡ksªt|jd ks¸t|jt ¡ ksÊt|jdksØt|d }|j|d jksôt|j|jkst|jd kst|d }|j|d jks4t|j|jksFt|jd ksVt|d }|j|d jkstt|j|jkst|jd kst|d }|j|d jks´t|j|jksÆt|jd ksHtqHd
|
}tjt |d t dd d W 5 Q R X tjt |d t dd d W 5 Q R X tjt |d t dd d W 5 Q R X tjt |d t dd d W 5 Q R X d}tjt |d t ddd W 5 Q R X tjt |d t ddd W 5 Q R X d}tjt |d t ddd W 5 Q R X d}tjt |d t ddd W 5 Q R X d S )Nr r r Ú1H1Drz r. r_ é Z25Hz<Frequency must be positive, because it represents span: -25Hr z-1D1Hz-1H1Dz:Frequency must be positive, because it represents span: 0DZ0D0HzInvalid frequency: 1W1DZ1W1DzInvalid frequency: 1D1WZ1D1W)
|
r r= r@ r r ÚHourr r r r )r rv r¡ r¢ Zp3rw r r r r Útest_period_cons_combinedÞ sl
|
|
|
ý
|
|
|
ýú z0TestPeriodConstruction.test_period_cons_combinedz 1970/01/01 z 2020-12-31 z 1981/09/13 Úhourz00:00:00z00:00:01z23:59:59z12:00:59zsec_float, expected)z
|
.000000001r. )z
|
.000000999éç )z
|
.123456789i )z
|
.999999999r© )z
|
.999999000r )z .999999001123r. )z.999999001123456r. )z.999999001123456789r. c C s t || | jj|ks td S ©N)r Ú
|
start_timeÚ
|
nanosecondr@ )r r2 r¨ Z sec_floatrI r r r Ú"test_period_constructor_nanosecond% s z9TestPeriodConstruction.test_period_constructor_nanosecondé c C s" t d| dd}|j|ks td S )Ni ' Z1Hr_ )r r¨ r@ )r r¨ rv r r r Útest_period_large_ordinal: s z0TestPeriodConstruction.test_period_large_ordinalN) Ú__name__Ú
|
__module__Ú __qualname__r! rJ rQ rX r\ rg ro rs ry r~ r r ÚmarkÚ parametrizer r r r Úranger r r r r r£ r§ r r¯ r r r r r $ sT F# E
|
|
|
|
|
G÷þr c @ s e Zd Zdd Zdd Zdd Zdd Zd d
|
Zej
|
d d d dddddddg ¡ej
|
ddddg¡dd Z dd Z d d Z d d Zd!d" Zd#d$ ZdS )%ÚTestPeriodMethodsc C s" t d}t |¡}||ks td S )NZ2000Q1)r ÚtmZround_trip_pickler@ )r rv Znew_pr r r Útest_round_tripC s
|
z!TestPeriodMethods.test_round_tripc C s t tdddt tdddks$tt tdddt tdddksHtt tdddt tdddksltt tdddt tdddkstd S ) Nr r# r ú
|
2011-01-01r r rt ú2011-02)Úhashr r@ )r r r r Ú test_hashH s $$$zTestPeriodMethods.test_hashc C s t ddd}|jddtdks$ttdtdd }|jd
|
d|ksJtt dd d}|jddtdksnttd tdd }|jd
|
d|kstd S ) Nr r# r r| r
r¹ z
|
2011-02-01r. rK ÚEr ú
|
2011-04-01)r rL r r@ r )r rv rI r r r Útest_to_timestamp_multT s z(TestPeriodMethods.test_to_timestamp_multc C s t ddd}|jdd}dddg}|D ]0}||jd |dks@t||jd
|
|dks&tq&|jd d}d d dg}|D ]0}||jd |dkst||jd
|
|dksrtqrdddddd dddg }dd }|D ]T}t d|d}| ¡ |¡} | |ksðt|j|jddkst|j||ksÆtqÆt ddd}|jdd d} tdddtdd }
|
| |
|
ksZt|jdd d} | |
|
ksvt|jd d d} tdddtdd }
|
| |
|
ks¨t|jd d d} | |
|
ksÄt|jd d} tdddtdd }
|
| |
|
ksôttd dd}
|
|jdd d} | |
|
ks t|jd d d} | |
|
ks8t|jdd d} | |
|
ksTt|jdd d} | |
|
kspt|jd d d} | |
|
kstd S )!Nr* r% r r| r
ÚsZStarTZBEGInr rk r½ Úer ZFINIsHr( r# ÚWrS rz r? c S s2 | j dkr | jtddd S t| | j jjd S )NrS r. r )r Ú nanoseconds)r r« r r Ú_value)rv r r r Ú_exp s
|
z0TestPeriodMethods.test_to_timestamp.<locals>._exZ1985i r. rK Ú3Hr{ Z2TiÁ ÚstartZ5S) r rL r@ Z to_periodr« Úend_timer r r ) r rv Ústart_tsÚaliasesr& Zend_tsZfrom_lstrÅ Zfcoderw rI r r r Útest_to_timestamp_ sX
|
|
z#TestPeriodMethods.test_to_timestampc C s: t dd}|jddd}tdtdd }||ks6td S )Nú
|
1990-01-05rS r½ r
ú
|
1990-01-06r. ©rà )r rL r r r@ ©r rN rw rI r r r Ú test_to_timestamp_business_end s
|
z0TestPeriodMethods.test_to_timestamp_business_endz ts, expected)z1970-01-01 00:00:00r )z1970-01-01 00:00:00.000001r. )z1970-01-01 00:00:00.00001rV )z1970-01-01 00:00:00.499i8 )z1999-12-31 23:59:59.999iX> )z1999-12-31 23:59:59.999999i?B )z2050-12-31 23:59:59.5i ¡ )z2050-12-31 23:59:59.500001i!¡ )z2050-12-31 23:59:59.123456i@â r NÚusrK c C s" t |j|dj}||ks td S )Nr )r rL Ú microsecondr@ )r rM rI r rw r r r Ú test_to_timestamp_microsecond¥ s z/TestPeriodMethods.test_to_timestamp_microsecondc C s4 t d}dt|kstt d}dt|ks0td S )NzJan-2000ú2000-01z
|
2000-12-15©r Úreprr@ r} r r r Ú test_repr¼ s zTestPeriodMethods.test_reprc C s$ t ddd}ttt|ks td S )Nr r# r )r rÖ r r@ r} r r r Ú test_repr_natà s z TestPeriodMethods.test_repr_natc C s t d}t|dkstd S )Nz2000-01-01 12:15:02.123z&Period('2000-01-01 12:15:02.123', 'L')rÕ r} r r r Útest_millisecond_reprÇ s z'TestPeriodMethods.test_millisecond_reprc C s t d}t|dkstd S )Nz2000-01-01 12:15:02.123567z)Period('2000-01-01 12:15:02.123567', 'U')rÕ r} r r r Útest_microsecond_reprÌ s z'TestPeriodMethods.test_microsecond_reprc C s4 t ddd}| d¡}|dks"tt|ts0td S )Nz2000-1-1 12:34:12r| r z%Y-%m-%d %H:%M:%Sz2000-01-01 12:34:12)r Ústrftimer@ r r r r r r Ú test_strftimeÑ s
|
z TestPeriodMethods.test_strftime)r° r± r² r¸ r¼ r¿ rË rÐ r r³ r´ rÓ r× rØ rÙ rÚ rÜ r r r r r¶ B s2 ?÷þ r¶ c @ s* e Zd ZdZej dddddg¡dd Zd d
|
Zd d Z d d Z
|
e dd Z ej de jdfe jdfg¡ej dddg¡dd Zej de jdfe jdfg¡ej dddg¡dd Zdd Zd d Zd d Zd!d" Zd#d$ Zd%d& Zd'd( Zd)d* Zd+d, Zd-d. Zd/d0 Z d1d2 Z d3d4 Z d5S )6ÚTestPeriodPropertiesz5Test properties such as year, month, weekday, etc....r r% r# r rz c C sl t d|d}|jstt|jts&tt d|d}|jr<tt d|d}|jsRtt d|d}|jrhtd S )Nz2000-01-01 00:00:00r z1999-01-01 00:00:00z2004-01-01 00:00:00z2100-01-01 00:00:00)r Z is_leap_yearr@ r Úbool)r r rv r r r Útest_is_leap_yearÜ s
|
|
|
z&TestPeriodProperties.test_is_leap_yearc C s¦ t ddd}|jdkst|jdks(tt|t s6tt ddd}|jdksPt|jdks^tt|t sltt ddd}|jdkst|jd kstt|t s¢td S )
|
Nr úQ-DECr_ é± é éþÿÿÿr- r# é )r r0 r@ rZ r r1 r} r r r Ú test_quarterly_negative_ordinalsì s z5TestPeriodProperties.test_quarterly_negative_ordinalsc C s0 t ddd}|jt ¡ ks t|jdks,td S )Nr* r? r r{ )r r r ÚMinuter@ r )r rD r r r Ú test_freq_strü s z"TestPeriodProperties.test_freq_strc C s ddddddgddd d
|
d gd d ddddgdddddddgdddd gd d d d d!gd"d#d$gd%d&d'gd(d)d*gd+ }t }| ¡ D ]\}}|D ]P}tjt|d, td-|d. W 5 Q R X tjt|d, td/|d0 W 5 Q R X qtd-|d.}td/|d0}t|tsütt|tsxtqxd S )1NZMTHZMONTHZMONTHLYZMthr1 ZmonthlyZBUSZBUSINESSZ
|
BUSINESSLYZWEEKDAYZbusZDAYZDLYZDAILYrB ZDlyZDailyZHRZHOURZHRLYZHOURLYÚhrr¦ ZHRlyÚminuteZMINUTEZMINUTELYZminutelyÚsecZSECZSECONDZSECONDLYÚsecondZ MILLISECONDZ MILLISECONDLYZ millisecondZ MICROSECONDZ MICROSECONDLYrÒ Z
|
NANOSECONDZ NANOSECONDLYr¬ ) r# rS r rz r{ r| r8 r; ÚNr z2016-03-01 09:00r r. r_ )r Úitemsr r r r r r@ )r Zcasesr rx Úfreqsr r¡ r¢ r r r Útest_period_deprecated_freq s*
|
÷ z0TestPeriodProperties.test_period_deprecated_freqc C s& t | j| j| j| j| j| j| ddS )NrÑ )r0 r1 r2 r¨ ré rë r )r r0 r1 r2 r¨ ré rë )ÚboundÚoffsetr r r Ú_period_constructor s ùz(TestPeriodProperties._period_constructorz bound, offsetr r. Úperiod_propertyr« rÈ c C s4 t ||¡}tjtdd t|| W 5 Q R X d S )NzOut of bounds nanosecondr )rÝ rò r r r Úgetattr)r rð rñ ró r r r r Ú$test_outer_bounds_start_and_end_time( s z9TestPeriodProperties.test_outer_bounds_start_and_end_timec C sn t || ¡}| ¡ jdd}t||jdd|ks8t||tddd d¡}t|| d¡|ksjtd S )Nr| r r. )Úunit)rÝ rò rL Úroundrô r@ r Úfloor)r rð rñ ró r rI r r r Ú$test_inner_bounds_start_and_end_time0 s
|
z9TestPeriodProperties.test_inner_bounds_start_and_end_timec C s dddddddg}t dd d }|D ] }td
|
|d }|j|ks"tq"td
|
d d jt dd d ks`ttd
|
dd jt dddks~td S )Nr% r( r# r rz r{ r| rU r. Ú2012r rS rj r iÛ rh é )r r r« r@ )r Zfreq_lstÚxpÚfrv r r r Útest_start_time: s z$TestPeriodProperties.test_start_timec C s t ddd}dd }|ddd}||jks.tt ddd}|d d
|
d}||jksTtt dd d}|d d d}||jksztt dd d}|d dd }||jks tt ddd}|d ddd}||jksÈtt ddd}|d dd}||jksîtt ddd}|d dd }||jkstt ddd}|d dd}||jks>tt ddd}|d dd d}||jkshtt ddd}|d dd d}||jkstd S )Nrú r% r c W s t t t| d¡jd S ©NrK r. ©r r Zas_unitrÄ ©Úargsr r r rÅ F s z/TestPeriodProperties.test_end_time.<locals>._exéÝ r. r( rU râ r# rj r rz rS r- r Z15Dé r r¤ ©r rÈ r@ )r rv rÅ rü r r r Ú test_end_timeC s> z"TestPeriodProperties.test_end_timec C s2 t dd}|j}tdtdd }||ks.td S )NrÌ rS rÍ r. rÎ )r rÈ r r r@ rÏ r r r Ú test_end_time_business_fridayq s
|
z2TestPeriodProperties.test_end_time_business_fridayc C s0 dd }t dd}|ddd}|j|ks,td S )Nc W s t t t| d¡jd S rÿ r r r r r rÅ z s z;TestPeriodProperties.test_anchor_week_end_time.<locals>._exz2013-1-1zW-SATr r. é r )r rÅ rv rü r r r Útest_anchor_week_end_timey s
|
z.TestPeriodProperties.test_anchor_week_end_timec C s t ddd}|jdkstd S )Nr% r5 )r r0 )r r0 r@ )r Za_dater r r Útest_properties_annually s z-TestPeriodProperties.test_properties_annuallyc C st t dddd}t dddd}t dddd}tdD ]<}|||fD ],}|| jdksVt|| j|d ks@tq@q2d S )Nrà r5 r. )r r0 rZ zQ-JANzQ-JUNr- )r rµ Zqyearr@ rZ )r Z
|
qedec_dateZ
|
qejan_dateZ
|
qejun_dateÚxZqdr r r Útest_properties_quarterly s z.TestPeriodProperties.test_properties_quarterlyc C sö t dddd}tdD ]Ú}|| }|jdks0td|d krHdkr\n n|jdksÞtnd|d krtdkrn n|jd ksÞtnVd
|
|d kr d kr´n n|jdksÞtn*d |d krÌd krÞn n|jdksÞt|j|d kstqd S )Nr# r5 r. )r r0 r1 rä r- râ r rj é r6 rV rh )r rµ r0 r@ rZ r1 )r Zm_dater Zm_ival_xr r r Útest_properties_monthly s z,TestPeriodProperties.test_properties_monthlyc C s t ddddd}|jdks t|jdks,t|jdks:t|jdksHt|d jdksZt|jdkshtt ddd ddjd
|
kstd S ) Nr r5 r. r ©r r0 r1 r2 é4 é rU rj é )r r0 r@ rZ r1 ÚweekÚ days_in_month)r Úw_dater r r Útest_properties_weekly¡ s z+TestPeriodProperties.test_properties_weeklyc C s¸ t ddddd}|jdks t|jdks,t|jdks:t|jdksHt|d jdksZt|jdkshtt ddd dd}|jd
|
kstt}tj t
|
|d t d dddd W 5 Q R X d S ) Nr r5 r. r r r r rU rj r r ZWK) r r0 r@ rZ r1 r r r r r r )r r rx r r r r Ú test_properties_weekly_legacy s z2TestPeriodProperties.test_properties_weekly_legacyc C s t ddddd}|jdks t|jdks,t|jdks:t|jdksHt|jdksVt|jdksdt|jdksrtt dddddjd kstt d
|
dddd}|jdksªt|jdks¸t|jdksÆt|jdksÔt|jdksât|jdksðt|jdksþtt d
|
ddddjd kstd S ) NrS r5 r. r r r rU rj r r ) r r0 r@ rZ r1 r2 ÚweekdayÚ dayofyearr )r Zb_dateZd_dater r r Útest_properties_daily¾ s$ z*TestPeriodProperties.test_properties_dailyc C s t dddddd}t dddddd}||fD ]}|jdks>t|jdksLt|jdksZt|jdksht|jdksvt|jdkst|jdkst|j dks tt ddd dddj d
|
ks,tq,d S ) Nrz r5 r. r )r r0 r1 r2 r¨ Ú2Hr rU rj r )
|
r r0 r@ rZ r1 r2 r r r¨ r )r Zh_date1Zh_date2Zh_dater r r Útest_properties_hourlyÖ s ÿz+TestPeriodProperties.test_properties_hourlyc C s¦ t ddddddd}|jdks"t|jdks0t|jdks>t|jdksLt|jdksZt|jdksht|jdksvt|j dkstt ddd ddddj d
|
ks¢td S ) Nr? r5 r. r )r r0 r1 r2 r¨ ré r r rU rj r )
|
r rZ r@ r1 r2 r r r¨ ré r )r Zt_dater r r Útest_properties_minutelyè s ÿÿz-TestPeriodProperties.test_properties_minutelyc C sÆ t dddddddd}|jdks$t|jdks2t|jdks@t|jdksNt|jdks\t|jdksjt|jdksxt|j dkst|j
|
dkst|j dks¢tt ddddddddj d ksÂtd S )
|
Nr? r5 r. r )r r0 r1 r2 r¨ ré rë r rU rj r ) r r0 r@ rZ r1 r2 r r r¨ ré rë r )r Zs_dater r r Útest_properties_secondlyù s> ÿ ÿýÿz-TestPeriodProperties.test_properties_secondlyN) r° r± r² Ú__doc__r r³ r´ rß rå rç rï Ú staticmethodrò r r+ Úmaxrõ rù rþ r r r r
|
r r r r r r r r r r r r rÝ Ù s6
|
|
. rÝ c @ s e Zd Zdd ZdS )ÚTestPeriodFieldc C s6 d}t jt|d t dt d¡d¡ W 5 Q R X d S )Nz@Buffer dtype mismatch, expected 'const int64_t' but got 'double'r r r. r )r r r Ú libperiodZget_period_field_arrrd Úemptyrr r r r Ú2test_get_period_field_array_raises_on_out_of_range s zBTestPeriodField.test_get_period_field_array_raises_on_out_of_rangeN)r° r± r² r% r r r r r" s r" c @ sp e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Ze j
|
d e d¡dfe edd¡dff¡dd ZdS )ÚTestPeriodComparisonsc C sT t dd}t dd}||ks t||ks,t||ks8t||k rDt||krPtd S ©NrÔ r# ©r r@ )r ÚleftÚrightr r r Ú,test_comparison_same_period_different_object s
|
|
zBTestPeriodComparisons.test_comparison_same_period_different_objectc C s` t dd}t dd}||kr t||ks,t||k s8t||ksDt||krPt||kr\td S )NrÔ r# ú2000-02r( )r ÚjanÚfebr r r Útest_comparison_same_freq$ s
|
|
z/TestPeriodComparisons.test_comparison_same_freqc C s¼ t dd}t dd}||kr t||ks,td}tjt|d ||k W 5 Q R X tjt|d ||k W 5 Q R X tjt|d ||k W 5 Q R X tjt|d ||k W 5 Q R X d S )NrÔ r# z
|
2012-01-01r z0Input has different freq=D from Period\(freq=M\)r )r r@ r r r )r r- r2 r r r r Ú test_comparison_mismatched_freq/ s
|
|
z5TestPeriodComparisons.test_comparison_mismatched_freqc
|
C sÜ t dd}|dkrt|dks"td}d| d| }|dfd|ffD ]\}}tjt|d ||k W 5 Q R X tjt|d ||k W 5 Q R X tjt|d ||k W 5 Q R X tjt|d ||k W 5 Q R X qFd S )NrÔ r# r. z'(Period|int)'z#not supported between instances of z and r )r r@ r r rC )r r- Z
|
int_or_perr r) r* r r r Ú test_comparison_invalid_type? s
|
z2TestPeriodComparisons.test_comparison_invalid_typec C sF t dd}t dd}t dd}|||g}|||g}t||ksBtd S )NrÔ r# r, z2000-03)r Úsortedr@ )r r- r. ÚmarZperiodsZcorrectPeriodsr r r Útest_sort_periodsQ s
|
|
|
|
|
z'TestPeriodComparisons.test_sort_periodsc C s t ddd}td}t|f|tft|f|tffD ]P\}}||k rDt||krPt||kr\t||ksht||krtt||kr0tq0d S )Nr¹ r r )r r r r@ )r rv Útr) r* r r r Útest_period_cmp_natY s ü z)TestPeriodComparisons.test_period_cmp_natzzerodim_arr, expectedr FrÔ r# Tc C s. t dd}||k|kst||k|ks*td S r' r( )r Z zerodim_arrrI rv r r r Ú!test_comparison_numpy_zerodim_arrk s
|
z7TestPeriodComparisons.test_comparison_numpy_zerodim_arrN)r° r± r² r+ r/ r0 r1 r4 r6 r r³ r´ rd Úarrayr r7 r r r r r& s þr& c
|
@ sP e Zd Zej ddddddg¡dd Zd d
|
Zd d Zd d Z dd Z
|
dd dd dd gZ dddgZ ejjde e dejjde e ddd Z d d Zej d d d!d"d#g¡d$d% Zej d&d'd(g¡ej d d d!d"d#g¡ej d)ejd*fejd+fejd,fejd-fg¡d.d/ Zd0d1 Zd2d3 Zej d4d5d6d7g¡d8d9 Zd:d; Zd,S )<ÚTestArithmeticrö rK rÑ ÚmsrÀ r' c C sl t dd}t d|¡}|| tks&t|| tks6t|| tksFttjtdd || W 5 Q R X d S )Nz
|
2022-06-01r r úunsupported operandr )r rd Ú timedelta64r r@ r r rC )r rö rN r r r r Útest_add_sub_td64_natw s
|
z$TestArithmetic.test_add_sub_td64_natc C sf t dddt ddd }}|| }|d|j ks4td}tjt|d |t dd d W 5 Q R X d S )
|
NÚ2011r% r ru râ z4Input has different freq=M from Period\(freq=A-DEC\)r r] r# ©r r r@ r r r )r r) r* rw r r r r Útest_sub_delta s z TestArithmetic.test_sub_deltac C sD t ddddd}t ddddd}|d |ks0td| |ks@td S )Nr éØ r. r rj r( )r Úper1rP r r r Útest_add_integer s z TestArithmetic.test_add_integerc C sP t ddd}|t tks tt| tks,t|t tks<tt| tksLtd S r r r} r r r Útest_add_sub_nat s
|
z TestArithmetic.test_add_sub_natc C s t ddddd}t ddddd}d ddd g¡}tjt|d
|
|d W 5 Q R X tjt|d
|
d | W 5 Q R X tjt|d
|
|| W 5 Q R X d S ) Nr rA r. r rj ú|z unsupported operand type\(s\)zcan only concatenate strzmust be str, not Periodr r )r Újoinr r rC )r rB rP r r r r Útest_add_invalid s ýÿz TestArithmetic.test_add_invalidc C s | S rª r ©r r r r Ú<lambda>® ó zTestArithmetic.<lambda>c C s t | g¡S rª )ÚpdÚSeriesrH r r r rI ® rJ c C s t | g¡S rª )rK ÚIndexrH r r r rI ® rJ ÚidentityrL rM Úlbox)ÚidsÚrboxc C sª t d}tddd}d ddddd g¡}tjt|d
|
|||| W 5 Q R X tjt|d
|
|||| W 5 Q R X tjt|d
|
|||| W 5 Q R X d S ) NZ2017r# r rE z
|
cannot addr; zcan only operate on azincompatible typez ufunc add cannot use operandsr )r r rF r r rC )r rQ rO rM rN r r r r Útest_add_timestamp_raises± s ûÿ z(TestArithmetic.test_add_timestamp_raisesc C sx t ddd}t ddd}|j}|| d| ks2t|| d| ksFtd}tjt|d |t d d
|
d W 5 Q R X d S ) Nr¹ r r z
|
2011-01-15iòÿÿÿé z0Input has different freq=M from Period\(freq=D\)r rº r# r? )r rB rP Úoffr r r r Útest_subÌ s zTestArithmetic.test_subÚnr. rj r- râ c C s\ t d||d}t d||d}t t||jjdt t||jjd }|| |ksXtd S )NÚ19910905r Ú19920406)r r r Úbaser@ )r Z tick_classesrV r¡ r¢ rI r r r Útest_sub_n_gt_1_ticksØ s ÿz$TestArithmetic.test_sub_n_gt_1_ticksÚ normalizeTFzoffset, kwd_namer1 ri Nr c C s| |d k r|dini }d}d}t ||||f|d}t ||||f|d} t || jjdt ||jjd }
|
| | |
|
ksxtd S )Nr- rW rX r )r r rY r@ ) r rñ Zkwd_namerV r[ ÚkwdsZp1_dZp2_dr¡ r¢ rI r r r Útest_sub_n_gt_1_offsetsä s z&TestArithmetic.test_sub_n_gt_1_offsetsc C s dD ]À}t d|d}t d|d}|t d¡ |ks6tt d¡| |ksLtt d¡t d¡t ¡ t dd¡t dfD ]L}d }t
|
j t |d
|
|| W 5 Q R X t
|
j t |d
|
|| W 5 Q R X qvqd D ]
|
}t d |d}t d |d}|t d¡ |ksþtt d¡| |kstt d|d}|t d¡ |ks:tt d¡| |ksRtd ddg¡}t d¡t d¡t ¡ t dd¡t dfD ]J}t
|
j t |d
|
|| W 5 Q R X t
|
j t |d
|
|| W 5 Q R X qqÊdD ]}t d|d}t d|d}|t d¡ |kstt d¡| |ks*tt d|d}|t d¡ |ksNtt d¡| |ksftt d|d}|t dd¡ |kstt dd¡| |ks¦tt d|d}|t dd¡ |ksÌtt dd¡| |ksætt d |d}|t d |kstt d | |ks tt d|d}|t d d |ksBtt d d | |ksZtd ddg¡}t d¡t d¡t ¡ t d d!¡t d"d fD ]J}t
|
j t |d
|
|| W 5 Q R X t
|
j t |d
|
|| W 5 Q R X qqÜd#D ]}t d$|d}t d%|d}|t d¡ |ks tt d¡| |ks6tt d&|d}|t d'¡ |ksZtt d'¡| |ksrtd(}t d&|d}|t d'd!¡ |kstt d'd!¡| |ks¶tt d)|d}|t d*d¡ |ksÜtt d*d¡| |ksött d+|d}|t d,d- |kstt d,d-| |ks2tt d.|d}|t d d/d0 |ksXtt d d/d0| |ksrtd ddg¡}t d¡t d¡t ¡ t d1d¡t d"d2d3fD ]J}t
|
j t |d
|
|| W 5 Q R X t
|
j t |d
|
|| W 5 Q R X q®qèd S )4N©r% r Z3Ar> r Z2013rj r. ém r z<Input has different freq|Input cannot be converted to Periodr ©r# rt r ú2011-03z2011-05z2012-03rh rE úInput has different freqú#Input cannot be converted to Period©r Z2Drk r¾ z
|
2011-04-06é ú
|
2011-04-02r® ú
|
2011-04-03éQ rÀ ú
|
2011-03-30rã é0 ©Úhoursrâ Úhé ©rz r rÆ ú2011-04-01 09:00z2011-04-03 09:00z2011-04-01 12:00r- z cannot use operands with typesz2011-04-01 10:00é z2011-04-01 11:00éx ©Úminutesz2011-04-05 12:00é´ ©r rt é r ©rl rt )r r rm r@ Ú YearBeginÚ
|
MonthBeginræ rd r< r r r r rl rF rB r¦ )r r rv rx Úor r r r Útest_add_offsetû sÜ
|
û
|
þÿ
|
û
|
þÿ
|
û
|
þÿ
|
|
ûz TestArithmetic.test_add_offsetc C s¨ d ddg¡}dD ]}td|d}|t d¡ td|dks@tt d¡t d ¡t ¡ t d
|
d ¡t
|
d
|
fD ]&}t j t |d || W 5 Q R X qjqd D ] }td|d}|t d¡ td|dksÆt|t d¡ td|dksätt d¡t d ¡t ¡ t d
|
d ¡t
|
d
|
fD ](}t j t |d || W 5 Q R X qqdD ]*}td|d}|t d¡ td|dkspt|t d¡ td|dkst|t dd ¡ td|dks²t|t dd¡ td|dksÔt|t
|
d td |dksòt|t
|
d d td|dkstt d¡t d ¡t ¡ t d d ¡t
|
d!d fD ](}t j t |d || W 5 Q R X q>q>d"D ]0}td#|d}|t d¡ td$|dks¢t|t d%¡ td&|dksÂt|t d%d ¡ td&|dksät|t d'd¡ td(|dkst|t
|
d)d* td+|dks&t|t
|
d d,d- td.|dksHtt d¡t d ¡t ¡ t d/d¡t
|
d!d0d1fD ](}t j t |d || W 5 Q R X qvqpd S )2NrE rb rc r^ r> r rj Z2009r. r_ r r r` ra r rh z2010-03rd r¾ re z
|
2011-03-27r® z
|
2011-03-31ri rh rÀ rã rg rj rk râ rm rn ro rp z2011-03-30 09:00r- z2011-04-01 06:00rq z2011-04-01 08:00rr rs z2011-04-01 07:00ru rv z2011-03-28 06:00rw r rx )rF r r rm r@ ry rz ræ rd r< r r r r rl rB r¦ )r r r rv r{ r r r Útest_sub_offset s þÿ
|
û
|
û
|
""
|
û
|
" ÿ ÿ
|
|
ûz TestArithmetic.test_sub_offsetr r# rt r c C sP t d|d}t| tks t|t tks,tt| tks<t|t tksLtd S )Nr r r r r r r Útest_period_addsub_natÜ s
|
z%TestArithmetic.test_period_addsub_natc C s® t ddd}|t ¡ }t ddd}||ks0t|t d¡ }t ddd}||ksVtd}tjt|d |t d¡ W 5 Q R X tjt|d |t d¡ W 5 Q R X d S ) Nr¾ r r rf rj ri z-Input cannot be converted to Period\(freq=D\)r )r r rB r@ r r r r¦ )r rv rw rx r r r r Útest_period_ops_offsetè s z%TestArithmetic.test_period_ops_offset)r° r± r² r r³ r´ r= r@ rC rD rG ZboxesrP rR rU rZ r rm rn rl ZWeekr] r| r} r~ r r r r r r9 v s>
|
|
|
üþ Q
|
r9 c C sZ d} t d}tjt| d d|_W 5 Q R X |j}tjt| d d| |_W 5 Q R X d S )Nz not writableZ2014Q1r rS rj )r r r ÚAttributeErrorr= r )r rN r r r r Útest_period_immutableú s r c C s* t dd} | jdkst| jdks&td S )Nz
|
0001-01-07r r. r )r r0 r@ r2 )rB r r r Útest_small_year_parsing s
|
r c C s dddddddg} t ddd }| D ]}t| |¡ q"| D ]&}t d|d }t| |jd
|
ks:tq:t dd d }t| t dd d }t| d S ) Nr% r# r( r rz r{ r| r r_ rá rS r )r rÖ rW r0 r@ )rî r r r r r Útest_negone_ordinals s r c C s. d} t jt| d tddd W 5 Q R X d S )Nz3Invalid frequency: <WeekOfMonth: week=0, weekday=0>r z
|
2012-01-02zWOM-1MONr )r r r r )r r r r Ú$test_invalid_frequency_error_message s r )(r r r Únumpyrd r Zpandas._libs.tslibsr r r# Z pandas._libs.tslibs.ccalendarr r Z pandas._libs.tslibs.np_datetimer Zpandas._libs.tslibs.parsingr
|
Zpandas._libs.tslibs.periodr r ZpandasrK r r r r r Zpandas._testingZ_testingr· r r¶ rÝ r" r& r9 r r r r r r r r Ú<module> s< " :^
|
