U
|
ý°d©1 ã @ sJ d dl Zd dlZd dlmZmZmZ d dlmZ dd Z
|
G dd dZ dS )é N)Ú PeriodIndexÚ
|
date_rangeÚ period_rangec C s | tj t| ¡¡S )N)ZtakeÚnpÚrandomZ permutationÚlen)Úobj© r ú^d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/indexes/period/test_setops.pyÚ_permute s r c @ sL e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dS )ÚTestPeriodIndexc C s. t dddd}t dddd}tdddd d
|
d d d ddg
|
dd}t dddd}t dddd}t dddd}t dddd}tg dd} t dddd}
|
t dddd} t dddd} tddddddddd d g
|
dd} td d d gd!d}td d d"gd!d}td d d d"gd!d}t d d#d$d}t d%d#d$d}t d d#d&d}t d'd(dd}t d)d(dd}td*d+d,d-d.d/d0d1d2d3g
|
d(d}td4d5dd6dgdd}t dddd}td4d5dd6ddd7d8d9d:g
|
dd}|||f|||f|| |
|
f| | | f|||f|||f|||f|||ffD ]8\}}} |j||d;} |d kr| ¡ } t | | ¡ qðd S )<Nú1/1/2000ÚDé ©ÚfreqÚperiodsú1/6/2000z
|
2000-01-06z
|
2000-01-07z
|
2000-01-08z
|
2000-01-09z
|
2000-01-10ú
|
2000-01-01z
|
2000-01-02z
|
2000-01-03z
|
2000-01-04z
|
2000-01-05©r ú1/4/2000é ú2000-01-01 09:00ÚHú2000-01-02 09:00ú2000-01-01 10:00ú2000-01-01 11:00ú2000-01-01 12:00ú2000-01-01 13:00z2000-01-02 10:00z2000-01-02 11:00z2000-01-02 12:00z2000-01-02 13:00ú2000-01-01 09:01ú2000-01-01 09:03ú2000-01-01 09:05ÚTz2000-01-01 09:08ÚMé ú
|
2000-04-01é
|
z
|
2003-01-01ÚAú
|
1998-01-01Ú2003Ú2004Ú2005Ú2006Ú2007Z1998Z1999Ú2000Z2001Z2002ú1/3/2000ú1/2/2000ú1/5/2000z1/7/2000z1/8/2000z1/9/2000z 1/10/2000©Úsort)r r ÚunionÚ sort_valuesÚtmÚassert_index_equal) Úselfr3 Úother1Úrng1Ú expected1Úrng2Úother2Ú expected2Úrng3Úother3Ú expected3Úrng4Úother4Ú expected4Úrng5Úother5Ú expected5Úrng6Úother6Ú expected6Úrng7Úother7Ú expected7Zrng8Zother8Z expected8ÚrngÚotherÚexpectedZ result_unionr r r
|
Ú
|
test_union sÒ ö ó ö ó ÿ ÿüù
|
ö ó ÿö óø
|
|
zTestPeriodIndex.test_unionc C sÖ t dddd}|d d
j|dd
|d}t ||¡ t|d d
jt|dd
|d}|d krrt ||¡ t ||¡stt dddd}t dddd}|j||d}| t¡j| t¡|d}t ||¡ d S ) Nr ú 1/20/2000r r éûÿÿÿr& r2 úW-WED) r r4 r6 r7 r Ú equalContentsÚAssertionErrorZastypeÚobject)r8 r3 ÚindexÚresultÚindex2rP r r r
|
Útest_union_misc s & z TestPeriodIndex.test_union_miscc C s t dddd}|d d
j|dd
|d}t ||dd
¡ t|d d
}t|dd
}|j||d}|d krt ||dd
¡ t ||dd
¡s¢tt dddd}t dddd}|j||d}tjg t d }t ||¡ t ddd
|
d}|j||d}t ||¡ d S ) Nr rR r r rS r& r2 rT )ZdtypeZ2D)
|
r Ú intersectionr6 r7 r rU rV ÚpdZIndexrW ) r8 r3 rX rY ÚleftÚrightrZ rP Zindex3r r r
|
Útest_intersection s" z!TestPeriodIndex.test_intersectionc C s t ddddd}t ddddd}t ddddd}t ddddd}t dddd d}t d d
|
ddd}tg ddd }||f||f||ffD ]B\} }
|
|j| |d } t | |
|
¡ | j|
|
jks¶t| j|
|
jkstqtd dddgddd}tddddgddd}tddgddd}tddddgddd}tddgdd d}t d d
|
ddd}tg ddd}||f||f||ffD ]X\} }
|
|j| |d } |d kr|
|
¡ }
|
t | |
|
¡ | j|
|
jks¨t| jdksbtqbt dddd} | dd
| ¡} t
|
| dksît| | dd
¡} t
|
| dkstd S )Nz6/1/2000z 6/30/2000r Úidx)r Únamez 5/15/2000z 6/20/2000rO z7/1/2000z 7/31/2000)rb r r2 z
|
2011-01-05z
|
2011-01-04ú
|
2011-01-02ú
|
2011-01-03z
|
2011-02-02z
|
2011-02-03z 6/15/2000r" r r ) r r r\ r6 r7 rb rV r r5 r r ) r8 r3 Úbaser< r> r? rA rB rD rN rP rY r r r
|
Útest_intersection_casesµ sb ý
|
ý
|
ý
|
ýý
|
z'TestPeriodIndex.test_intersection_casesc C sÊ dddddg}t |dd}tddd d
|
}|}t |dd}tddd d
|
}t dddgdd}t |dd} t g dd}
|
| } d d d ddg}t |dd} tddd d
|
} | }t dddgdd}t ddgdd}t dgdd}ddddddd g}t |d d}td d d d
|
}t dddgd d}d d d!d"d#g}t |d$d}td%d$d&d
|
}t d d!gd$d}|||f|||f| |
|
| f| | |f|||f|||f|||ffD ]B\}}}|j||d'}|d kr¶t|r¶| ¡ }t ||¡ qd S )(Nr/ r0 r r1 r r r r r r r r r r r r r r r r! r" z
|
2000-02-01r z
|
2000-06-01z
|
2000-07-01z
|
2000-05-01z
|
2000-03-01r% r# r$ r) r- r, r+ r* r' r( r r2 )r r Ú
|
differencer r5 r6 r7 ) r8 r3 Z
|
period_rngr: r9 r; r< r= r> r? r@ rA rB rC rD rE rF rG rH rI rJ rK rL rM rN rO rP Zresult_differencer r r
|
Útest_differenceù sh û ÿù ù z TestPeriodIndex.test_differencec C s t dddd}t dddd}tddgdd}| ||¡}t ||¡ t d||¡ t dddd}| ||¡}tddgdd}t ||¡ t d||¡ d S ) NZ20160920Z20160925r r Z20160921Z20160924r Z20160922)r r rg r6 r7 Zassert_attr_equal)r8 r3 rX rO rP Zidx_diffr r r
|
Útest_difference_freq: s z$TestPeriodIndex.test_difference_freqc C s0 t ddd}| |¡}| |¡}t ||¡ d S )Nú
|
2011-01-01é ©r )r Úappendr\ r6 r7 )r8 ra Úidx_duprY r r r
|
Ú"test_intersection_equal_duplicatesL s
|
|
z2TestPeriodIndex.test_intersection_equal_duplicatesc C s^ t ddd}| |¡}t ddd}| |¡}| |¡}tddddddgdd}t ||¡ d S )Nrj rk rl rc rd r r )r rm r4 r r6 r7 )r8 ra rn Zidx2Zidx2_duprY rP r r r
|
Útest_union_duplicatesS s
|
|
|
ú÷ z%TestPeriodIndex.test_union_duplicatesN) Ú__name__Ú
|
__module__Ú __qualname__rQ r[ r` rf rh ri ro rp r r r r
|
r s wDAr ) Únumpyr Zpandasr] r r r Zpandas._testingZ_testingr6 r r r r r r
|
Ú<module> s
|
|
