3 Uщdb1у@sLddlZddlZddlZddlmZmZmZmZddl j ZGddДdГZdS)щN)┌Index┌NaT┌PeriodIndex┌Seriesc@sJeZdZejjdd5d6d7d8d9d:d;d┌TestPeriodIndexOpsz freq,expected┌A┌year┌Q┌quarter┌M┌month┌D┌day┌H┌hour┌T┌minute┌S┌second┌L┌millisecond┌U┌microsecondcCs"tjdd|dН}|j|kstВdS)Nz 2013-04-01щ)┌start┌periods┌freq)┌pd┌period_range┌ resolution┌AssertionError)┌selfr┌expected┌idxйr$·X/var/www/html/virt/lib64/python3.6/site-packages/pandas/tests/indexes/period/test_ops.py┌test_resolution sz"TestPeriodIndexOps.test_resolutioncCsrtjddddН}ttj|jtdt|ГdГГddН}tddd d ddd dddg ddН}ttdddГ|ddН}x$|t|ГgD]}t j |jГ|ГqxWtjddddН}t j|j Г|ГtdddddtgddН}tddgddН}tddg|dН}x$|t|ГgD]}t j |jГ|ГqЎWtddtgddН}tdddg|dН}x*|t|ГgD]}t j |jddН|ГРq@Wt j|j Г|ГdS)Nz2011-01-01 09:00rщ )rrщ)rz2011-01-01 18:00z2011-01-01 17:00z2011-01-01 16:00z2011-01-01 15:00z2011-01-01 14:00z2011-01-01 13:00z2011-01-01 12:00z2011-01-01 11:00z2011-01-01 10:00rZint64)┌index┌dtypez2013-01-01 09:00z2013-01-01 08:00щщ)r)F)Zdropnaщ )rrr┌np┌repeat┌_values┌range┌lenr┌tm┌assert_series_equalZvalue_counts┌assert_index_equal┌uniquer)r!r#Zexp_idxr"┌objr$r$r%┌test_value_counts_uniquesH$z+TestPeriodIndexOps.test_value_counts_uniquerZ3DZ2HZ2TZ3ScCshtjdd|ddН}|jГ}tj||Г|j|jks6tВ|j|Г}|jГ}tj||Г|j|jksdtВdS)Nz 2011-01-01r'r#)rr┌name)rr┌drop_duplicatesr3r5rr ┌append)r!rr#┌resultZidx_dupr$r$r%┌test_drop_duplicates_metadataSs z0TestPeriodIndexOps.test_drop_duplicates_metadatazkeep, expected, index┌firstFr'Tщr┌lastщcCsВtjdd|ddН}|j|ddЕГ}tj|j|dН|Г||}|j|dН}tj||Гt|Гj|dН}tj |t||dНГdS)Nz 2011-01-01r'r#)rrr9r?)┌keep)r)) rrr;r3┌assert_numpy_array_equalZ duplicatedr:r5rr4)r!rrBr"r)r#r<r$r$r%┌test_drop_duplicates`s z'TestPeriodIndexOps.test_drop_duplicatescCs*ddД}tdddgdddН}td d dgdd Н}Рx■||gD]Є}|jГ}tj||Г|||Г|jddН}tj||dddЕГ|||dddЕГ|jddН\}}tj||Гtj|tjdddgГddН|||Г|jdddН\}}tj||dddЕГtj|tjdddgГddН|||ddd ЕГq:WtdddddgdddН}tdddddgdddН}td ddd d gdd Н}td d d ddgdd Н}Рx||f||fgD]Ї\}} |jГ}tj|| Г|||Г|jddН}tj|| ddd!ЕГ|||Г|jddН\}}tj|| ГtjdddddgГ} tj|| ddН|||Г|jdddН\}}tj|| ddd"ЕГtjdddddgГ} tj|| ddН|||ГРqаWtddddgdddН}|jГ}tddddgdddН} tj|| Г|jdkРsшt В|jddН}tddddgdddН} tj|| Г|jdkРs&t ВdS)#NcSst|tГr|j|jkstВdS)N)┌ isinstancerrr )r)Zexpected_indexr$r$r%┌_check_freq|s z9TestPeriodIndexOps.test_order_compat.._check_freqZ2011Z2012Z2013┌pidxr)r9ri█i▄i▌r#)r9F)┌ ascendingr(T)┌return_indexerrr,)┌check_dtype)rIrHZ2015i▀щr+rr r-r-r-r-r-r-) rr┌sort_valuesr3r5rCr.┌arrayrr )r!rFrGZiidxr#┌ordered┌indexerZ pexpectedZ iexpectedr"┌expr<r$r$r%┌test_order_compat{sd z$TestPeriodIndexOps.test_order_compatc CsРxHdDР]>}tdddg|ddН}|jГ}tj||Г|j|jksFtВ|jd d Н}|dddЕ}tj||Г|j|jks|tВ|j|ksКtВ|jdd Н\}}tj||Гtj|tjdddgГd dН|j|jks╥tВ|j|ksрtВ|jdd dН\}}|dddЕ}tj||Гtj|tjdddgГd dН|j|jkРs:tВ|j|kstВqWtdddddgdddН}tdddddgdddН}tdddddgdddН}tdddddgdddН} tt dddt gdddН} tt t dddgdddН}Рx(||f|| f| |fgDР]\}}|jГ}tj||Г|jdkРs$tВ|jd d Н}tj||dddЕГ|jdkРsVtВ|jdd Н\}}tj||ГtjdddddgГ}tj||d dН|jdkРsжtВ|jdd dН\}}tj||dddЕГtjdddddgГ}tj||d dН|jdkРsЎtВРqЎWdS)Nr ┌2D┌4Dz 2011-01-01z 2011-01-02z 2011-01-03r#)rr9F)rHr(T)rIrr,)rJ)rIrHz 2011-01-05┌idx1┌idx2┌idx3rKr+)r rRrSr-r-r-r-) rrLr3r5rr rCr.rMr) r!rr#rNr"rOrTZexp1rUZexp2rVZexp3rPr$r$r%┌ test_order┬sv$zTestPeriodIndexOps.test_ordercCsцtjjtkstВtjgddНjtks(tВtjddgddН}|jsDtВtj|jt j ddgГГ|jdksjtВtj|jt j gt j dНГtjddgddН}|jsаtВtj|jt j dd gГГ|jd ks╞tВtj|jt j d gt j dНГdS)Nr)rz 2011-01-01z 2011-01-02r F)r*rTr()rrZ _na_valuerr Z_can_hold_nar3rCZ_isnanr.rMZhasnansZ _nan_idxsZintp)r!r#r$r$r%┌test_nats zTestPeriodIndexOps.test_natcCsъtjdddg|dН}|j|Гs"tВ|j|jГГs4tВ|j|jtГГsHtВ|jtГj|Гs\tВ|jtГj|jtГГsvtВ|jt|ГГsКtВ|jtj|ГГsаtВtjdddgddН}|j|Гs─tВ|j|jГГs╪tВ|j|jtГГsюtВ|jtГj|ГРstВ|jt|ГГРstВ|jtj|ГГРs4tВtjj |j j |j jddНГ}tj |j|jГ|j|ГРsrtВ|j|jГГРsИtВ|j|jtГГРsаtВ|jtГj|ГРs╕tВ|jt|ГГРs╬tВ|jtj|ГГРsцtВdS)Nz 2011-01-01z 2011-01-02r)rr)rr┌equalsr ┌copyZastype┌object┌listrZ_simple_newr0Zasi8r3rC)r!rr#rUrVr$r$r%┌test_equals%s0zTestPeriodIndexOps.test_equalscCsVtjddddН}tjdГП|jWdQRXtjtddНПtjj Г|_WdQRXdS)NZ2018Q1rKr )rrzcan't set attribute)┌match) rrr3Zassert_produces_warningr┌pytestZraises┌AttributeError┌offsetsZDay)r!r#r$r$r%┌test_freq_setter_deprecatedEs z.TestPeriodIndexOps.test_freq_setter_deprecatedN)rr)r r )rr)r r)rr)rr)rr)rr)rr)┌__name__┌ __module__┌__qualname__r_┌markZparametrizer&r8r=r.ZconcatenateZarangerDrQrWrXr]rbr$r$r$r%r s47& && GQ r) Znumpyr.r_ZpandasrrrrrZpandas._testingZ_testingr3rr$r$r$r%┌s