U
|
ý°dþ. ã @ s d dl Zd dlZd dlmZ d dlmZmZmZm Z m
|
Z
|
m Z m Z m Z d dlmZ d dlmZ G dd deZdd Zej d d
|
d g¡d d ZdS )é N)ÚIncompatibleFrequency)ÚIndexÚNaTÚPeriodÚ PeriodIndexÚSeriesÚ
|
date_rangeÚoffsetsÚ period_range)Ú DatetimeLikec @ sä e Zd ZeZejedddZeje
|
d¡e ddddddd
gd
|
d gd d d Z dd Z dd Zdd Zdd Zdd Zdd Zdd Zd d Zd d Zd!d" Zd#d$ Zd%d& Zd'd( Zd)d* Zd+d, Zd-d. Z d/d0 Z dS )1ÚTestPeriodIndex)Úreturnc C s t ddddS )NÚ20130101é ÚD©ÚperiodsÚfreq©r
|
©Úself© r ú^d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/indexes/period/test_period.pyÚ simple_index s z TestPeriodIndex.simple_indexé
|
r r r NéÿÿÿÿZ index_incZ index_dec)ÚparamsÚidsc C s |j S ©N)Úparam)r Úrequestr r r Úindex s zTestPeriodIndex.indexc C s d S r r r r r r Ú
|
test_where% s zTestPeriodIndex.test_wherec C s, t dddd}td|d}t|ts(td S )NÚAú1/1/2001ú 12/1/2009©r ÚstartÚendé )r! )r
|
r Ú
|
isinstanceÚAssertionError)r r! Zseriesr r r Útest_make_time_series) s z%TestPeriodIndex.test_make_time_seriesc C sÈ t g dd}tjg tjd}t | d¡|¡ t |j|¡ t dtgdd}tjddgtjd}t | d¡|¡ t |j|¡ tjddgtjd}t d tgd
|
d}t | d¡|¡ t |j|¡ d S ) NÚM©r ©ÚdtypeÚi8ú2011-01éì ìûÿÿÿ é: ú
|
2011-01-01r ) r ÚnpÚarrayÚint64ÚtmÚassert_numpy_array_equalÚviewÚasi8r ©r ÚidxÚexpr r r Útest_view_asi8. s z TestPeriodIndex.test_view_asi8c C s. t g dd}tjg td}t |j|¡ t | ¡ |¡ tjg tjd}t |j |¡ t dt
|
gdd}tjt dddt
|
gtd}t |j|¡ t | ¡ |¡ tjddgtjd}t |j |¡ t dt
|
gdd}tjt dddt
|
gtd}t |j|¡ t | ¡ |¡ tjd dgtjd}t |j |¡ d S )
|
Nr- r. r/ r2 r3 r4 r6 r r5 ) r r7 r8 Úobjectr: r; ÚvaluesZto_numpyr9 r= r r r> r r r Ú test_values@ s$ zTestPeriodIndex.test_valuesc C st t dddd}t|dks tt dddd}t|dks<tt dddd}t|d ksZttd
|
d }t |d d }t|d kst|j|jkst|d |ks ttdd}t |dd}t|dksÆt|j|jksÖt|d |ksættdd}t |dd}t|t|kst||k ¡ s$t|j|jks6td}tjt|d t ||d W 5 Q R X tdd }t ||d}d}tjt|d t |d W 5 Q R X t |tdd g}t|d ksÂt|d |ksÔtt t
|
|tdd g¡}t|d ksþt|d |kst|tdd g}d }tjt |d t | W 5 Q R X t
|
|¡}tjt|d t | W 5 Q R X d S ) Nr# r$ r% r& é ÚQé$ r- él z 02-Apr-2005ÚBé )r' r r ú
|
2006-12-31ÚWr ©r( r r Z1wz!start and end must have same freq©Úmatch)r' r( z
|
2005-05-01zOOf the three parameters: start, end, and periods, exactly two must be specified)r' z
|
2005-05-05é Úwz9Input has different freq=W-SUN from PeriodIndex\(freq=B\)) r
|
Úlenr+ r r ÚallÚpytestÚraisesÚ
|
ValueErrorr r7 r8 r )r Úpir' Úi1Úend_intvÚi2ÚmsgÚvalsr r r Útest_period_index_lengthZ sV
|
|
|
|
ÿ
|
z(TestPeriodIndex.test_period_index_lengthc C sä t dddd}| |¡ t dddd}| |¡ t dddd}| |¡ t d d
|
d d}| |¡ t d d
|
d d}| |¡ t d ddd}| |¡ t dddd}| |¡ t dddd}| |¡ tdd}t |dd}| |¡ d S )Nr# r$ z 12/1/2005r& rF z 12/1/2002r- z1/1/2002r z 12/1/2001z6/1/2001rI ÚHz
|
12/31/2001z1/1/2002 23:00ZMinz1/1/2002 00:20ÚSz12/31/2001 00:00:00z12/31/2001 00:05:00rK rL r rM )r
|
Ú_check_all_fieldsr )r rW rY rX r r r Ú test_fields s.
|
|
|
|
|
|
|
ÿ
|
|
zTestPeriodIndex.test_fieldsc
|
C sØ ddddddddd d
|
d d d ddg}t |}t|}|D ]}t||}t|t|ksXtt||D ]\}}t|||ksbtqbt|dkrq6t|j|} t|t| ks¬tt|| D ]\}}t|||ks¶tq¶q6d S )NÚyearÚmonthÚdayÚhourÚminuteÚsecondZ
|
weekofyearÚweekZ dayofweekZ day_of_weekZ dayofyearZ day_of_yearZquarterZqyearZ days_in_monthr )Úlistr ÚgetattrrR r+ ÚzipÚdt)
|
r Z periodindexÚfieldsr ZserÚfieldZ field_idxÚxÚvalZfield_sr r r r` ¸ s: ñ
|
z!TestPeriodIndex._check_all_fieldsc C sô dd }| }| |¡s t| | ¡r,t| | ¡ ¡s>t| | ¡ ¡ ¡ ¡ ¡ ¡s`t| ¡ |¡srt| ¡ }d|_| |¡st| |d d
¡r¤t| | d¡¡r¸t| | d¡¡rÌt| |d ¡rÞt| |d ¡rðtd S )Nc S s t ddddS )Nr# r$ r% r& r r r r r Ú<lambda>Ý ó z*TestPeriodIndex.test_is_.<locals>.<lambda>ZAppler- r# rP r )Úis_r+ r< ÚnameZasfreq)r Z create_indexr! Zind2r r r Útest_is_Ü s "zTestPeriodIndex.test_is_c C sL t dddddgdd}t dddgdd}t | ¡ |¡ | ¡ dksHtd S )NiÐ é× éÙ zA-JUNr. é )r r: Úassert_index_equalÚuniqueZnuniquer+ )r r? Úexpectedr r r Útest_index_uniqueî s z!TestPeriodIndex.test_index_uniquec C s d S r r r r r r Ú
|
test_shiftô s zTestPeriodIndex.test_shiftc C sR t ddd t ddd tdddgdd}tt dddg¡dd}t ||¡ d S )Niüÿÿr# )Úordinalr r r r) )r r r7 r8 r: ry )r Zidx1Zidx2r r r Útest_negative_ordinalsø s
|
z&TestPeriodIndex.test_negative_ordinalsc C sl t dddddgddd}td d d
|
d d gtjdd }t |j|¡ td dd
|
ddgtjdd }t |j|¡ d S )Nr2 z2011-02r z2012-03z2012-04r rt )r rt iÛ r iÜ )r0 rt r) rP rx é )r r r7 r9 r: ry rb rc r> r r r Ú test_pindex_fieldaccessor_nat s ÿz-TestPeriodIndex.test_pindex_fieldaccessor_natc C s t ddddddgdd}td d
|
dd }t ||¡ |jt d ¡ksFt|jdksTttd d dd}t ||¡ |jt d ¡kst|jdkstd S )Nr2 z2011-03z2011-05z2011-07z2011-09z2011-11Z2Mr. z1/1/11z12/31/11)r' r( r rP é ©r' r r ) r r
|
r: ry r r ZMonthEndr+ Zfreqstr)r r{ rW r r r Útest_pindex_multiples
|
s þ z%TestPeriodIndex.test_pindex_multiplesc C s@ t dddd}t|}t|d ts(t|d j|jks<td S )Nz1/1/10r rI r r )r
|
ri r* r r+ r )r r! Úresultr r r Útest_iteration s z TestPeriodIndex.test_iterationc C sf t dddd}|jdd|jg}tddd d
|
g|}t|jjd tsHtt|jj d d t
|
sbtd S ) Nz1/1/2012r Z12Hr r r. r r) rP rx ) r Z to_periodre r r* r! Úlevelsr r+ rC r )r r! Zindex_as_arraysÚsr r r Útest_with_multi_index! s
|
z%TestPeriodIndex.test_with_multi_indexc C sB t dddgdd}| dd ¡}tdd |D }t ||¡ d S )
|
NiÕ rv rw r# r. c S s | j S r ©r~ )ro r r r rq 0 rr z*TestPeriodIndex.test_map.<locals>.<lambda>c S s g | ]
|
}|j qS r r )Ú.0ro r r r Ú
|
<listcomp>1 s z,TestPeriodIndex.test_map.<locals>.<listcomp>)r Úmapr r: ry )r r! r
r@ r r r Útest_map, s zTestPeriodIndex.test_mapc C s8 | j g dd}| ¡ g ks t|jdddgks4td S )Nr# r. T)rt Ú )Ú
|
_index_clsÚformatr+ )r Z empty_idxr r r Útest_format_empty4 s z!TestPeriodIndex.test_format_empty) Ú__name__Ú
|
__module__Ú __qualname__r r rT Zfixturer r r: ZmakePeriodIndexr
|
r! r" r, rA rD r] ra r` ru r| r} r r r r r r r r r r r r s6 þû
|
<"$
|
r c C sr t ddgdd} t d¡}| |¡dks,t| d¡dks>tt ¡ }d}tjt|d | |¡ W 5 Q R X d S )NÚ2000Z2001r r. rP z5Input has different freq=B from PeriodIndex\(freq=D\)rN ) r r ZDayZ_maybe_convert_timedeltar+ Z BusinessDayrT rU rV )rW Úoffsetr[ r r r Ú test_maybe_convert_timedelta; s
|
r r8 TFc
|
C s t ddgdd}| r|j}tj|d |d gtd}t |¡}t ||¡ t |¡}t ||¡ |j}dd tj fD ]8}tj||d}t ||¡ tj||d}t ||¡ qrd
|
D ]^}d }t
|
j t |d tj||d W 5 Q R X t
|
j t |d tj|t t|d W 5 Q R X q°d S ) Nz
|
2000-01-01z
|
2001-01-01r r. r r) r/ r1 r9 )Úfloat64Úint32Zuint64zargument must berN )r Ú_datar7 r8 rB r: r; Zasarrayr= r9 rT rU Ú TypeErrorrj )r8 Úobjr{ r
r0 r[ r r r Útest_dunder_arrayG s(
|
|
r )Únumpyr7 rT Zpandas._libs.tslibs.periodr Zpandasr r r r r r r r
|
Zpandas._testingZ_testingr: Z!pandas.tests.indexes.datetimeliker r r ÚmarkZ parametrizer r r r r Ú<module> s (
|
)
|
