U
|
ý°déo ã @ s d dl m Z d dlZd dlZd dlZd dlmZ d dlm Z d dl
|
Z d dl
|
m Z m Z mZmZmZmZmZmZmZ d dlmZ edddZedd
Ze ed
|
¡Z ej ee d ¡e d ¡eed e e d dd
|
g¡e dddg¡e jdddgdde j !e¡g ddd Z"G dd dZ#G dd d Z$G d d d Z%G d d d Z&G d!d" d"Z'G d#d$ d$Z(G d%d& d&Z)G d'd( d(Z*dS ))é )ÚdatetimeN)Úperiod)ÚInvalidIndexError) Ú DatetimeIndexÚNaTÚPeriodÚ PeriodIndexÚSeriesÚ TimedeltaÚ
|
date_rangeÚnotnaÚ period_rangeú
|
2016-01-01é ©Úperiodséÿÿÿÿé ÚUTCÚWé é g @g @g @é é Úu8©Údtype)Úparamsc C s | j S )N)Úparam)Úrequest© r ú`d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/indexes/period/test_indexing.pyÚnon_comparable_idx s r" c @ sn e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z e
|
j j dd Z e
|
j deeg¡dd ZdS )Ú TestGetItemc C s, t ddddd}|j|dd
jks(td S )NZ20010101é
|
ÚDZbob©r ÚfreqÚnamer )r r( ÚAssertionError©ÚselfÚidxr r r! Ú test_getitem_slice_keeps_name0 s z)TestGetItem.test_getitem_slice_keeps_namec C s t ddddd}|fD ]x}|d }|tdddks8t|d }|tdddksTt|dd
}t dd
|
ddd}t ||¡ |j|jkst|jdkst|dd d
}tdd d
|
ddgddd}t ||¡ |j|jksÜt|jdksêt|ddd
}tdddddgddd}t ||¡ |j|jks.t|jdks>t|dd d
}td
|
dd ddgddd}t ||¡ |j|jkst|jdkstqd S )Nú
|
2011-01-01ú
|
2011-01-31r% r, ©r' r( r ©r' r r ú
|
2011-01-05r$ r ú
|
2011-01-03z
|
2011-01-07z
|
2011-01-09iìÿÿÿéûÿÿÿr z
|
2011-01-12z
|
2011-01-15z
|
2011-01-18z
|
2011-01-21z
|
2011-01-24r ú
|
2011-01-04ú
|
2011-01-02)r r r) ÚtmÚassert_index_equalr' r ©r+ Úidx1r, ÚresultÚexpectedr r r! Ú test_getitem4 sL ý ý ý zTestGetItem.test_getitemc C s t ddddd}|dddg }td d
|
d gddd }t ||¡ |d d dddd d dddg
|
}tdd d dgddd }t ||¡ d S )Nú2007-01r$ ÚMÚxr& r r r z2007-02z2007-04z2007-06r0 TFz2007-07)r r r7 r8 )r+ r, r; Úexpr r r! Útest_getitem_indexb s
|
ÿz TestGetItem.test_getitem_indexc C sX t dddd}ttj t|¡|}tjtdd |d W 5 Q R X |d }|j j
|
d k ¡ sbt |dd
|
}t|d ks~t |d d
}t|d kst |dd
}t|d ks¶t |d d
|
}t|dksÒt |d
|
d
}t|dksît |}|d d
}t ||¡ t |dd
|dd
g¡}d}tjt|d |tdd
|
W 5 Q R X d S )Nr> é2 r? ©r r' z^'2006'$©ÚmatchZ2006Z2008iØ Z2009é z2008-1z2009-12Z2008Q1Z2009Q4é$ é r$ z-left slice bound for non-unique label: '2008')r r ÚnpÚrandomZrandnÚlenÚpytestÚraisesÚKeyErrorÚindexÚyearÚallr) r7 Úassert_series_equalÚpdÚconcatÚslice)r+ ÚrngÚtsr; rA Úmsgr r r! Útest_getitem_partialo s. z TestGetItem.test_getitem_partialc C sV t dddd}ttt||d}tddd}tdd d
|
}|||
}t ||¡ d S ) Nú
|
2012-01-01r$ zW-MON©Ústartr r' ©rP iÛ r iÜ r é )r r ÚrangerL r r7 rS )r+ rW rX Zdt1Zdt4Úrsr r r! Útest_getitem_datetime s z!TestGetItem.test_getitem_datetimec C s t dddgdd}|d tdddks*t|d tks:ttdddg|d }|t dks\tt||d }|tddd tdddkst|t tkstd S )
|
Nú2011-01r ú2011-02r? r1 r r r r^ )r r r) r r )r+ r, Úsr r r! Útest_getitem_nat s z TestGetItem.test_getitem_natc C sL t dddd}ttt||d}|jdg }t |tdddg |¡ d S ) Nr[ r$ r% r\ r^ r z
|
2012-01-02r1 )r r r` rL Zilocr7 rS r )r+ rW rX rA r r r! Útest_getitem_list_periods§ s z%TestGetItem.test_getitem_list_periodsc C sÊ t dddd}tdddd}||fD ] }ddddd g}|D ]&}tjtd
|
d || W 5 Q R X q:ttj t |¡|d }t
|
|d |dd
¡ t
|
|d |d d
¡ dD ]}t
|
|| |¡ q®q$d S )Nz2013/01/01 09:00:00ÚSi ©r] r' r Ú2014ú2013/02ú
|
2013/01/02ú 2013/02/01 9Hú2013/02/01 09:00úonly integers, slicesrE r^ z2013/01/01 10:00i iL z 2013/01/01 9H)ú
|
2013/01/01ú2013/01Z2013) r r rM rN Ú
|
IndexErrorr rJ rK ÚrandrL r7 rS )r+ ÚdidxÚpidxr, ÚvaluesÚvalÚserÚdr r r! Útest_getitem_seconds® s" ûz TestGetItem.test_getitem_secondsÚ idx_rangec
|
C sà |dddd}ddddd g}|D ]&}t jtd
|
d || W 5 Q R X q ttj t|¡|d }t |d |dd
¡ t |d |dd
¡ t |d |dd
¡ dd g}|D ]&}t jt
|
|d || W 5 Q R X q´d S )Nrp r% i ri rj rk rl rm rn ro rE r^ rq r é é; im ) rM rN rr r rJ rK rs rL r7 rS rO )r+ r{ r, rv rw rx Úinvalidr r r! Útest_getitem_dayÊ s$ ûz TestGetItem.test_getitem_dayN)Ú__name__Ú
|
__module__Ú __qualname__r- r= rB rZ rb rf rg rM ÚmarkZarm_slowrz Ú parametrizer r r r r r r! r# / s . "
|
|
þþr# c @ s< e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd S )Ú
|
TestGetLocc
|
C s t dddd}tdd}tjtdd | |¡ W 5 Q R X z| |¡ W n4 tk
|
r~ } z|jd |ksntW 5 d }~X Y nX d S ) Nz2000-1-1ÚAr$ ©r' r Z2012z^Period\('2012', 'A-DEC'\)$rE r )r r rM rN rO Úget_locÚargsr) )r+ r, Z
|
bad_periodÚinstr r r! Útest_get_loc_msgð s
|
zTestGetLoc.test_get_loc_msgc C s t dddg}tdddgdd}||fD ]R}| t¡dks>t| d ¡dksPt| td¡dksft| tj¡dks(tq(d S )Nr. r r3 r? r1 r Únan)r r r r r) ÚfloatrJ r )r+ rt ru r, r r r! Útest_get_loc_natû s zTestGetLoc.test_get_loc_natc C s` t d}t d}t d}t|||g}d}d}| |¡|ks@t| t|¡|ksVt| |¡|ksht| t|¡|ks~td}tjt|d | d¡ W 5 Q R X tjtd d | d
|
¡ W 5 Q R X tjtt
|
t|¡d | |¡ W 5 Q R X t|||g}t d d}d}| |¡|ks(t| t|¡|ks@t| |¡|ksTt| t|¡|ksltd}tjt|d | d¡ W 5 Q R X tjtd d | d
|
¡ W 5 Q R X tjtt
|
t|¡d | |¡ W 5 Q R X t|||g} d}
|
t d d d g¡} | |¡|
|
ks t| t|¡|
|
ks4tt | |¡| ¡ t | t|¡| ¡ d S )Nú
|
2017-09-01ú
|
2017-09-02ú
|
2017-09-03r r z Cannot interpret 'foo' as periodrE Zfooz^1\.1$gñ?r TF)r r r r) ÚstrrM rN rO r ÚreÚescaperV rJ Úarrayr7 Úassert_numpy_array_equal) r+ Úp0Úp1Úp2Úidx0Zexpected_idx1_p1Zexpected_idx1_p2rY r: Úidx2Zexpected_idx2_p1Zexpected_idx2_p2r r r! Ú test_get_loc sL
|
zTestGetLoc.test_get_locc C sl t ddd}| d¡}tjtdd | d¡ W 5 Q R X | d¡}tjtd d | d
|
¡ W 5 Q R X d S ) Nr r r r% Z16801rE i¡A ÚYZ46é. )r Ú to_periodrM rN rO r )r+ ÚdtiÚpiÚpi2r r r! Útest_get_loc_integer@ s
|
|
z TestGetLoc.test_get_loc_integerc C s¦ t dddd}tjtdd | d¡ W 5 Q R X tdddg|d}tjtdd |jd W 5 Q R X tjtdd |d W 5 Q R X d|kstd|ks¢td S ) NÚ2000r r )r r( rE r r r^ )r rM rN rO r r Úlocr) )r+ r¡ rx r r r! Ú+test_get_loc_invalid_string_raises_keyerrorJ s z6TestGetLoc.test_get_loc_invalid_string_raises_keyerrorc C s t ddd}| d¡}| d¡}| |j¡}tjtdd | |d ¡ W 5 Q R X tjtdd | |d ¡ W 5 Q R X d S ) Nr r r r% r zW-SUNrE r )r r Úviewr rM rN rO r )r+ r r¡ r¢ Zpi3r r r! Ú test_get_loc_mismatched_freqZ s
|
|
z'TestGetLoc.test_get_loc_mismatched_freqN) r r r r r r r£ r¦ r¨ r r r r! r
ï s :
|
r
c @ sP e Zd Zdd Zdd Zdd Zej ddd d
|
g¡d d Z d d Z
|
dd Z dS )ÚTestGetIndexerc C s* t d}t d}t d}t d}t d}t d}t d}t|||g}t | |¡tjdd d
|
gtjd ¡ t||||g} t | | d ¡tjd dd d
|
gtjd ¡ t | | d¡tjdd d
|
d gtjd ¡ t | | d¡tjddd d
|
gtjd ¡ |j| dtdd}
|
t |
|
tjddd d gtjd ¡ d S )Nr ú
|
2017-09-04z
|
2017-09-07z
|
2017-08-31r ú
|
2017-09-05z
|
2017-09-09r r r r Úpadr ÚbackfillÚnearestú1 day©Z tolerance) r r r7 r Ú get_indexerrJ r Úintpr
|
) r+ r r Úp3Ztp0Ztp1Ztp2Ztp3r, ÚtargetÚresr r r! Útest_get_indexerk s6 ÿ
|
ÿ
|
ÿ
|
ÿz TestGetIndexer.test_get_indexerc C sÊ t ddd}| d¡}| d¡}tjdddgtjd}| |¡}t ||¡ | |¡}t ||¡ | |¡}t ||¡ | |¡d }t ||¡ | |¡d }t ||¡ | |¡d }t ||¡ d S ) Nr r r r% r r r r ) r r rJ r r² r± r7 r Úget_indexer_non_unique)r+ r r¡ r¢ r< r; r r r! Ú!test_get_indexer_mismatched_dtype s
|
|
|
|
|
z0TestGetIndexer.test_get_indexer_mismatched_dtypec C sP t ddd}| d¡}|}|d d
|¡}tj|jtjd }t ||¡ d S )Nr r r r% r r ) r r r± rJ ZonesÚshaper² r7 r )r+ r" r r¡ Úotherrµ r< r r r! Ú2test_get_indexer_mismatched_dtype_different_length§ s
|
zATestGetIndexer.test_get_indexer_mismatched_dtype_different_lengthÚmethodr¬ r r® c
|
C sØ t ddd}| d¡}|}t d|j d|j ¡}tjt|d |j||d W 5 Q R X d D ]r}| |¡}|d
|
krt
|
|t rq`d d d d|j d|j dfD ¡}tjt|d |j||d W 5 Q R X q`d S )Nr r r r% zCannot compare dtypes z and rE )r¼ )ÚobjectÚcategoryr½ ú|c S s g | ]}t |¡qS r )r r )Ú.0rY r r r! Ú
|
<listcomp>Ä s ÿzPTestGetIndexer.test_get_indexer_mismatched_dtype_with_method.<locals>.<listcomp>z$ not supported between instances of ) r r r r r rM rN Ú TypeErrorr± ZastypeÚ
|
isinstancer Újoin) r+ r" r¼ r r¡ rº rY r Zother2r r r! Ú-test_get_indexer_mismatched_dtype_with_method³ s&
|
|
þþÿ z<TestGetIndexer.test_get_indexer_mismatched_dtype_with_methodc
|
C s t d}t d}t d}t d}t|||g}t||||g}| |¡}tjdddddgtjd }tjdd
|
gtjd } t |d |¡ t |d | ¡ d S ) Nr r rª r« r r r r r r )r r r· rJ r r² r7 r )
|
r+ r r r³ Zp4r: r r; Zexpected_indexerZexpected_missingr r r! Útest_get_indexer_non_uniqueÏ s
|
z*TestGetIndexer.test_get_indexer_non_uniquec C sà t dddjddd}t | |¡tjddd gtjd
|
¡ td d d gdd}t | |d¡tjdddgtjd
|
¡ t | |d¡tjddd gtjd
|
¡ t | |d¡tjdddgtjd
|
¡ t |j|dddtjdddgtjd
|
¡ d}t j
|
t |d |j|ddd W 5 Q R X t |j|dddtjdddgtjd
|
¡ t dt dt dd¡g}t |j|ddd |D dtjdddgtjd
|
¡ t d ¡ t d ¡ t dd ¡g}t j
|
tjd d |j|d|d W 5 Q R X d S ) Nz
|
2000-01-01r r ÚHr] )Úhowr r r r z 1999-12-31T23z 2000-01-01T12z 2000-01-02T01r1 r¬ r r r® z1 hourr° z8Input has different freq=None from PeriodArray\(freq=H\)rE z1 minuter¯ r% c S s g | ]}t |¡qS r )rJ Ú timedelta64)rÀ r@ r r r! rÁ s z4TestGetIndexer.test_get_indexer2.<locals>.<listcomp>z2 hourr? z"Input has different freq=None from)r Zasfreqr7 r r± rJ r r² r rM rN Ú
|
ValueErrorr
|
rÉ Zto_timedelta64Ú libperiodÚIncompatibleFrequency)r+ r, r´ rY Ztol_rawZtol_badr r r! Útest_get_indexer2á sj ÿ ÿ
|
ÿ
|
ÿ
|
ÿþþ
|
ý ÿü
|
|
|
ý ÿz TestGetIndexer.test_get_indexer2N) r r r r¶ r¸ r» rM r r rÅ rÆ rÍ r r r r! r© j s
|
r© c @ s, e Zd Zdd Zdd Zdd Zdd Zd S )
|
Ú TestWherec C s t dddd}dgt| }|}| ||¡}t ||¡ dgdgt|d }ttg|dd
¡ dd}| ||¡}t ||¡ d S ) NÚ20130101r r% rD TFr r1 )r rL Úwherer7 r8 r r Útolist)r+ Z listlike_boxÚiÚcondr< r; r r r! Ú
|
test_where s zTestWhere.test_wherec C sÎ t dddd}tjtfD ]&}|jt||d}|}t ||¡ q| ¡ }t ttg|dd
|
¡ dd}| t||¡}t ||¡ | ¡ }t ttg|dd
|
¡ dd}| t||j ¡}t ||¡ d S )NrÏ r r% rD )rº r r1 ) r rJ r r rÐ r r7 r8 Úcopyr rÑ rv )r+ rÒ Zarrr; r< Úi2r r r! Útest_where_other# s zTestWhere.test_where_otherc
|
C sv t dddd}|dd
¡ }tttg| dd}t|}| ||j¡}tjtj tj g| t
|
d}t |d t stt t ||¡ |j d ¡}tj|d |d
|
g| t
|
d}t |d tjs¾t | ||¡}t ||¡ | d ¡}tj|d |d
|
g| t
|
d}|d tkst | ||¡}t ||¡ td d } tj| | g| t
|
d}|d | ksZt | || ¡}t ||¡ d S )NrÏ r r% rD r r1 r r ztimedelta64[ns]r rh r )Údays)r rÑ r r r rÐ Úasi8rT ÚIndexÚ_valuer½ rà Úintr) r7 r8 r§ rJ rÉ Z to_timestampr
|
)
|
r+ r¡ ÚtailrÖ Úmaskr; r< Ztdir Útdr r r! Útest_where_invalid_dtypes4 s.
|
|
z#TestWhere.test_where_invalid_dtypesc C s~ t dddd}t dddddg¡}t dd¡}tj|d ||d
|
|d |gtd }|d |ksbt| ||¡}t
|
||¡ d S )NrÏ r r% rD TFr Únsr r r r r ) r rJ r rÉ rT rÚ r½ r) rÐ r7 r8 )r+ r¡ rÓ Ztdnatr< r; r r r! Útest_where_mismatched_natR s $ z#TestWhere.test_where_mismatched_natN)r r r rÔ r× rà râ r r r r! rÎ s rÎ c @ s$ e Zd Zdd Zdd Zdd ZdS )ÚTestTakec C sî t ddddd}|fD ]Ð}| dg¡}|tdddks<t| dg¡}|td ddks\t| dd
|
d g¡}t dd ddd}t ||¡ |jdkst|j|jks¦t| dd d g¡}tdd dgddd}t ||¡ |j|jksæt|jdksôt| dd d
|
g¡}tdddgddd}t ||¡ |j|jks6t|jdksFt| dd dg¡}tdd d gddd}t ||¡ |j|jkst|jdkst| dd dg¡}tdd d gddd}t ||¡ |j|jksÚt|jdkstqd S )Nr. r/ r% r, r0 r r1 r z
|
2011-01-06r r r3 r r2 é z
|
2011-01-08r6 r r5 éýÿÿÿz
|
2011-01-29)r Útaker r) r7 r8 r' r r9 r r r! Ú test_take^ s^ ÿ ÿ ÿ ÿ zTestTake.test_takec C s´ t ddddd}ttdddtddd tddd
|
tddd gddd }| d dddg¡}|d dddg }||fD ]>}t ||¡ t|tst|j|jkst|j |j ksptqpd S )Nz1/1/10z12/31/12r% r, )r] Úendr' r( iÚ r r rä é é r0 r é é )
|
r r r ræ r7 r8 rà r) r' r( )r+ rP r< Ztaken1Ztaken2Ztakenr r r! Útest_take_misc s"
|
|
|
|
üø zTestTake.test_take_miscc C sb t dddgddd}| t ddd g¡¡}t dddgddd}t ||¡ |jt ddd g¡d
|
d }t ddd gddd}t ||¡ |jt ddd g¡d d
|
d}t dddgddd}t ||¡ d}tjt|d |jt dddg¡d
|
d W 5 Q R X tjt|d |jt dddg¡d
|
d W 5 Q R X d}tjt |d | t ddg¡¡ W 5 Q R X d S )Nr. z
|
2011-02-01z
|
2011-03-01Zxxxr% )r( r' r r r T)Ú
|
fill_valuer F)Z
|
allow_fillrî zJWhen allow_fill=True and fill_value is not None, all indices must be >= -1rE éþÿÿÿr4 z3index -5 is out of bounds for( axis 0 with)? size 3)
|
r ræ rJ r r7 r8 rM rN rÊ rr )r+ r, r; r< rY r r r! Útest_take_fill_value¥ sF ÿ ÿ ÿ ÿ ÿ$$z TestTake.test_take_fill_valueN)r r r rç rí rð r r r r! rã ] s 1rã c @ s& e Zd Zej dddg¡dd ZdS )Ú TestGetValuer' rÇ r% c C s$ t dddd}| |¡}ttdd|d}|d }| |¡dksDt|| dksTt|j| dksft|tdd }|d
|
krètj t
|
d d | |¡ W 5 Q R X tj t
|
d d || W 5 Q R X tj t
|
d d |j| W 5 Q R X n8| |¡dksút|| dks t|j| dks td S ) Nr r ZMSrD rä r$ r^ r )ÚhoursrÇ z2016-01-01 03:00rE ) r r r r` r r) r¥ r
|
rM rN rO )r+ r' r r¡ rx rX Zts2r r r! Ú test_get_value_datetime_hourlyÍ s$
|
z+TestGetValue.test_get_value_datetime_hourlyN)r r r rM r r ró r r r r! rñ Ì s rñ c @ s$ e Zd Zdd Zdd Zdd ZdS )Ú TestContainsc C s¨ t d}t d}t d}t d}|||g}t|}|D ] }||ksFtt||ks6tq6d}||kshttjt|d | |¡ W 5 Q R X d|kst||ks¤td S )Nr r r rª z2017-09-01 00:00:01rE z2017-09)r r r) r rM rN rO r ) r+ r r r r³ Zps0r ÚpÚkeyr r r! Ú test_containsë s
|
zTestContains.test_containsc C sN t dddd}tddd|ks"ttddd|ks6ttddd|ksJtd S )Nr> r? r$ r r1 r% Z2M)r r r) )r+ rW r r r! Útest_contains_freq_mismatch s z(TestContains.test_contains_freq_mismatchc C s t dddd}t|kstd |ks&ttd|ks6ttj|ksDttdddgdd }t|ksbtd |ksnttd|ks~ttj|kstd S )
|
Nr> r? r$ r r rc r rd r1 )r r r) r rJ r r r* r r r! Útest_contains_nat s z TestContains.test_contains_natN)r r r r÷ rø rù r r r r! rô ê s rô c @ s e Zd Zdd ZdS )Ú TestAsOfLocsc C sú t ddd}| d¡}| d¡}tjdddgtd}d }tjt|d
|
| t j
|
|j tj d|¡ W 5 Q R X tjt|d
|
| t j
|
|j tj d|¡ W 5 Q R X tjt|d
|
| || |¡ W 5 Q R X d }tjtj|d
|
| ||¡ W 5 Q R X d S ) Nr r r r% rÇ r r r z$must be DatetimeIndex or PeriodIndexrE zInput has different freq=H)r r rJ r ÚboolrM rN r Z asof_locsrT rÚ rÙ Úint64Úfloat64rË rÌ )r+ r r¡ r¢ rÞ rY r r r! Ú test_asof_locs_mismatched_type s
|
|
$$z+TestAsOfLocs.test_asof_locs_mismatched_typeN)r r r rþ r r r r! rú s rú )+r r ÚnumpyrJ rM Zpandas._libs.tslibsr rË Z pandas.errorsr ZpandasrT r r r r r r
|
r r r Zpandas._testingZ_testingr7 Zdti4r rÚ r` rW ZfixtureZ tz_localizer Z IntervalIndexZ from_breaksr" r# r
r© rÎ rã rñ rô rú r r r r! Ú<module> sD ,
|
|
÷ÿ
|
A{ -Go 0
|
