U
|
ý°dºQ ã @ s d dl Zd dlZd dlmZ d dlmZ d dlmZm Z m
|
Z
|
m Z m Z m Z mZmZ d dlmZ d dlmZ G dd dZG dd d ZG d
|
d d ZdS ) é N)ÚIncompatibleFrequency)Ú PeriodDtype)ÚIndexÚNaTÚPeriodÚ PeriodIndexÚSeriesÚ
|
date_rangeÚoffsetsÚ period_range)Ú PeriodArrayc @ s e Zd Zdd Zdd Zej dee j
|
e e j g¡dd Zdd Zd
|
d Zd d Zdd Zdd Zdd Zdd Zdd Zdd Zej ddd d g¡d d Zd d! Zd"d# Zd$d% Zd&d' Z d(d) Z d*d+ Z d,d- Z ej d.d/d0ge
|
d/d0g¡g¡d1d2 Z d3d4 Z!d5d6 Z"d7d8 Z#ej d9d:d;d<d=d>g¡ej d?d@dAdBdCdDg¡dEdF Z$dGdH Z%dIdJ Z&ej d9d;dKd:d<dLd=d>dMdNdOdPg ¡dQdR Z'dSdT Z(dS )UÚTestPeriodIndexc C sð t dddtt dddg}t t|t|¡ t tt |¡tt |¡¡ tjtt dddg}t t|t|¡ t tt |¡tt |¡¡ t dddtt dddg}t t|t|t d¡ t tt |¡tt |¡t d¡ d S )Nú2011-01ÚM©Úfreqú2011-03ÚD©Údtype)
|
r r ÚtmÚassert_index_equalr r ÚnpÚarrayÚnanÚobject)ÚselfÚarr© r údd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/indexes/period/test_constructors.pyÚ"test_construction_base_constructor s z2TestPeriodIndex.test_construction_base_constructorc C s: t d}dddg}t||d}t||d}t ||¡ d S )Nr ú
|
2011-01-01z
|
2012-03-04z
|
2014-05-01r )r r r r r )r r ÚvaluesÚresultÚexpectedr r r Ú'test_base_constructor_with_period_dtype& s
|
|
z7TestPeriodIndex.test_base_constructor_with_period_dtypeÚvalues_constructorc C sV t dddtt dddg}||}t|td}t|tks>tt |jt
|
|¡¡ d S )Nr r r r r ) r r r r ÚtypeÚAssertionErrorr Úassert_numpy_array_equalr" r r )r r& Úperiodsr" r# r r r Útest_index_object_dtype. s
|
|
ý z'TestPeriodIndex.test_index_object_dtypec C s6 t ddd}tdddd}t|dd}t ||¡ d S )Nz4/2/2012ÚBr é
|
©Ústartr* r ©r/ r* )r r r r )r Úpr$ Úindexr r r Ú test_constructor_use_start_freq> s z/TestPeriodIndex.test_constructor_use_start_freqc
|
C s: t dd¡ d¡dd
}t t dd¡d¡dd
}t||d d
|
}td d d d }t ||¡ t||dd
|
}t |j |j ¡ t||d}t ||¡ dddg}ddg}d}t
|
j t |d t||dd W 5 Q R X t
|
j t |d t||dd W 5 Q R X dddg}dddg}t||dd}tdddd} t || ¡ d S )NiÆ iÚ é é éþÿÿÿé é é úQ-DEC©ÚyearÚquarterr Z1990Q3Z2009Q2r z2Q-DEC©r< r= é× z Mismatched Period array lengths©Úmatchr )r< Úmonthr Ú2Mé ú2007-01©r* r ) r ÚarangeÚrepeatÚtiler r r r r) Úasi8ÚpytestÚraisesÚ
|
ValueError)
|
r ÚyearsÚquartersr2 r$ Úindex2ÚmonthsÚmsgÚidxÚexpr r r Ú test_constructor_field_arraysF s*
|
|
|
z-TestPeriodIndex.test_constructor_field_arraysc C s, t jtdd tdddd W 5 Q R X d S )NzInvalid frequency: Xr@ z2007-1-1iô ÚXrF ©rK rL rM r )r r r r Útest_constructor_Ue s z"TestPeriodIndex.test_constructor_Uc C sb t tdddtddddd}ttdddtdddtdddtdddgdd}t ||¡ d S ) Nr7 ÚN)Zordinalr r4 ©r/ Úendr r5 rD r )r r r r r ©r rS rT r r r Útest_constructor_nanoj s
|
|
ÿ
|
|
|
|
üù z%TestPeriodIndex.test_constructor_nanoc C sl t jddt jd d¡}t t jddddgt jdd¡}t||d }t |j t
|
|¡ t |j t
|
|¡ d S )
|
Ni¨ éÐ r r4 r7 r5 rD é( r> ) r rG Úint64rH rI r r r r r< r r= )r rN rO Zpindexr r r Ú%test_constructor_arrays_negative_yeary s
|
z5TestPeriodIndex.test_constructor_arrays_negative_yearc C s> d}t jt|d" ttddttddd W 5 Q R X d S )NzQuarter must be 1 <= q <= 4r@ r^ iÔ r4 r: r; )rK rL rM r ÚrangeÚlist©r rR r r r Ú!test_constructor_invalid_quarters s z1TestPeriodIndex.test_constructor_invalid_quartersc C s, t dddd}t dddd}t ||¡ d S )NrE g %@r rF r- )r r r )r r# rT r r r Útest_constructor_corner s z'TestPeriodIndex.test_constructor_cornerc C s° t dddd}t t|j|¡ t tt|j|¡ d}tjt|d t|j W 5 Q R X tjt|d tt|j W 5 Q R X d}tjt
|
|d tt dd d
|
d W 5 Q R X tt |}t ||¡ t|}t ||¡ t|dd
|
}t ||¡ t|t ¡ d
|
}t ||¡ |jdks&tt|d d
|
}t || d ¡¡ |jd ksTtt|t d ¡d
|
}t || d ¡¡ |jd kstt|dd
|
}| dd¡}t ||¡ d S )NrE r9 r rF z)freq not specified and cannot be inferredr@ z 'Period' object is not iterableZ2007ÚAr )ÚdatarC r5 r Úe)r r r r r" rc rK rL rM rJ Ú TypeErrorr Úiterr
|
ZMonthEndr r( Zasfreq)r rS rR r# rT r r r Ú test_constructor_fromarraylike s: z.TestPeriodIndex.test_constructor_fromarraylikec C sR t jdddt jd}| t d¡¡}t|dd}t| d¡dd}t ||¡ d S ) Ni i° éd r zM8[us]r r zM8[ns]) r rG r` Úviewr r Zastyper r )r ÚvalsÚpir$ r r r Ú test_constructor_datetime64arr¶ s
|
z.TestPeriodIndex.test_constructor_datetime64arrÚboxNÚseriesr2 c C s^ t dddd}|d kr |j}n|dkr.t|}t|dd}tdd d
|
d gdd}t ||¡ d S ) NZ2017r4 r rF rs r r z
|
2017-01-31z
|
2017-02-28z
|
2017-03-31z
|
2017-04-30)r Z_valuesr r r r )r rr rh r# r$ r r r Ú!test_constructor_datetime64arr_ok¿ s
|
ÿz1TestPeriodIndex.test_constructor_datetime64arr_okc C s t ddgdd}t ddgdd}t ||¡ |jdks:tt ddgd d}t ddgd
|
d}t ||¡ |jd ksttt d d gd d}t |dd}t ddgdd}t ||¡ |jdksºtt |dd}t ||¡ |jdksàtd}tjt|d t dgddd W 5 Q R X d S )Nz2013-01z2013-03z period[M]r r r z
|
2013-01-05z
|
2013-03-05z
|
period[3D]Ú3Dz
|
2013-01-01z
|
2013-01-02r z&specified freq and dtype are differentr@ r z period[D])r r )r r r r r( rK rL r )r rS rT ÚresrR r r r Útest_constructor_dtypeÎ s& z&TestPeriodIndex.test_constructor_dtypec C s^ t g dd}t|t stt|dks*t|jdks8ttjtdd t g W 5 Q R X d S )Nr r r úfreq not specifiedr@ )r Ú
|
isinstancer( Úlenr rK rL rM )r rS r r r Útest_constructor_emptyë s z&TestPeriodIndex.test_constructor_emptyc
|
C s¦ t tdddttdddg}t dddgdd}t ||¡ t t tdddttdddg¡}t ||¡ t tttdddtdddg}t ddddgdd}t ||¡ t t tttdddtdddg¡}t ||¡ t ttddgdd}t ||¡ tjt dd t ttg W 5 Q R X tjt dd t t ttg¡ W 5 Q R X tjt dd t ddg W 5 Q R X tjt dd t t ddg¡ W 5 Q R X d S )Nr r r r rx r@ )
|
r r r r r r r rK rL rM r\ r r r Útest_constructor_pi_natô s< ÿ ÿ ÿ ÿÿ z'TestPeriodIndex.test_constructor_pi_natc
|
C sô d}t jt|d$ ttdddttdddg W 5 Q R X t jt|d* tt tdddttdddg¡ W 5 Q R X t jt|d$ tttdddtdddg W 5 Q R X t jt|d* tt ttdddtdddg¡ W 5 Q R X d S )Nz5Input has different freq=D from PeriodIndex\(freq=M\)r@ r r r r )rK rL r r r r r r rd r r r Ú test_constructor_incompat_freq s" (ÿÿ(ÿÿz.TestPeriodIndex.test_constructor_incompat_freqc C s t dttdddg}t dddgdd}t ||¡ t dttdddg}t dddgdd}t ||¡ t tdddtdg}t dddgdd}t ||¡ d S )Nr r r r r! r z
|
2012-01-01)r r r r r r\ r r r Útest_constructor_mixed4 s z&TestPeriodIndex.test_constructor_mixedc C sÄ t ddddd}tjtdd |j|dd W 5 Q R X |j|jdd}t ||¡ d }tjt|d" t|jjt |j
|
|j d
|
W 5 Q R X t|jj|j
|
|j d
|
}|j|dd}t ||¡ d S ) NrE r1 r5 r )Únamer* r ú<class .*PeriodIndex'>r@ ©r z Should be numpy array of type i8r ) r rK rL r( Ú _simple_newÚ_datar r r' r rJ r )r rS r# rR r r r r Útest_constructor_simple_newA s &z+TestPeriodIndex.test_constructor_simple_newc C sV t g ddd}tjtdd |j|dd W 5 Q R X |j|jdd}t ||¡ d S )Nr r1 )r r r r@ r )r rK rL r( r r r r )r rS r# r r r Ú!test_constructor_simple_new_emptyS s
|
z1TestPeriodIndex.test_constructor_simple_new_emptyÚfloatsgñ?gÍÌÌÌÌÌ @c C sN t jtdd t |¡ W 5 Q R X d}t jt|d t| W 5 Q R X d S )Nz<class r@ z9PeriodIndex does not allow floating point in construction)rK rL r( r r rj )r r rR r r r Útest_constructor_floats\ s
|
z'TestPeriodIndex.test_constructor_floatsc C sX d}t jt|d tdddd W 5 Q R X t jt|d tdddd W 5 Q R X d S )Nz start and end must not be NaTr@ r r! r rZ rW rd r r r Útest_constructor_nate s
|
z$TestPeriodIndex.test_constructor_natc C s\ t dddg}|d }t||d}dd t||D }ttt|}t|}t ||¡ d S )NiÑ iÒ iÓ r^ r> c S s& g | ] }|d dd|d dqS )r ÚdÚQr7 r )Ú.0Útr r r Ú
|
<listcomp>p s zETestPeriodIndex.test_constructor_year_and_quarter.<locals>.<listcomp>)r r Úziprc Úmapr r r )r r< r= rS ÚstrsZlopsr1 r r r Ú!test_constructor_year_and_quarterl s z1TestPeriodIndex.test_constructor_year_and_quarterc C s t dddd}tddddgdd}t ||¡ t d d
|
d d }td d dddgd d}t ||¡ t dddd}tdddgdd}t ||¡ d}tjt|d tdgdd W 5 Q R X d}tjt|d tdgd d W 5 Q R X d}tjt|d t ddd d W 5 Q R X d S ) Nz2014-01rC r4 ©r/ r r* z2014-03z2014-05z2014-07r z
|
2014-01-02z
|
2014-01-15ru rZ z
|
2014-01-05z
|
2014-01-08z
|
2014-01-11z
|
2014-01-14z2014-01-01 17:00Z4HrD )r[ r r* z2014-01-01 09:00z2014-01-01 13:00z;Frequency must be positive, because it represents span: -1Mr@ r z-1Mz:Frequency must be positive, because it represents span: 0MZ0MrF )r r r r rK rL rM )r Úpidxr$ rR r r r Útest_constructor_freq_multu s0 þ ÿ z*TestPeriodIndex.test_constructor_freq_multr rg r r ÚTÚSÚmultr7 r5 rD r4 r8 c C s> t || }td|dd}td|dd |¡}t ||¡ d S )Nz
|
2014-04-01r- r )Ústrr r Z to_periodr r )r r r Zfreqstrr r$ r r r Ú%test_constructor_freq_mult_dti_compat s ÿz5TestPeriodIndex.test_constructor_freq_mult_dti_compatc C sb dD ]$}t ddg|d}t ddgdd}qdD ].}tdd|d }t dd
|
gdd}t ||¡ q.d S ) N)Z1D1HZ1H1Dz
|
2016-01-01z
|
2016-01-02r z2016-01-01 00:00z2016-01-02 00:00Z25Hr5 r. z2016-01-02 01:00)r r r r )r r r r$ r r r Ú test_constructor_freq_combined s z.TestPeriodIndex.test_constructor_freq_combinedc C sè t dddd}t|dks tt dddd}t|dks<tt dddd}t|d ksZtt d
|
dd d}t|d ksxtt d dd d}t|dkstt dddd}t|dks´tt dddd}t|dksÒtt dddd}t|dksðttdd }t |dd}t|dkst|j|jks*t|d |ks<ttd d }t |d d }t|d ksdt|j|jksvt|d |ksttd d!}t |d d }t|t|ks´t||k ¡ sÆt|j|jksØttd"d }t ||d#}t|td$d g}t|d%kst|d |ks$ttt |td$d g¡}t|d%ksNt|d |ks`t|td d&g}d'}t j
|
t |d( t| W 5 Q R X t |¡}t j
|
t |d( t| W 5 Q R X t j
|
t d)d( td d* W 5 Q R X d S )+Nrg z1/1/2001z 12/1/2009)r r/ r[ é r é$ r él r z
|
12/31/2009i× r, i- ÚHz12/31/2001 23:00i8" ZMinz1/1/2001 23:59i r z1/1/2001 23:59:59iQ z 02-Apr-2005r9 r0 r z
|
2006-12-31ÚWr- )r[ r* éÿÿÿÿZ1wz
|
2005-05-01)r/ r[ z
|
2005-05-05r5 Úwz9Input has different freq=W-SUN from PeriodIndex\(freq=B\)r@ zpass as a string instead)r¡ r7 ) r rz r( r r Úallr r r rK rL r rj )r rp r/ Úi1Zend_intvÚi2ro rR r r r Útest_constructor§ s`
|
|
|
|
|
z TestPeriodIndex.test_constructorr r, ÚLÚUrY r c C s, t d|dd}t|j|d}t ||¡ d S )Nz
|
2001/04/01r7 r r )r r r" r r )r r ÚorgrS r r r Útest_recreate_from_dataë s z'TestPeriodIndex.test_recreate_from_datac C sf dddg}t |dd}tdd |D }| t¡}t|ts@ttdd |D sVtt ||¡ d S )
|
NiÕ r? iÙ rg r c S s g | ] }t |qS r )r )r Únumr r r r ÷ s zDTestPeriodIndex.test_map_with_string_constructor.<locals>.<listcomp>c s s | ]}t |tV qd S )N)ry r )r Zresir r r Ú <genexpr>þ s zCTestPeriodIndex.test_map_with_string_constructor.<locals>.<genexpr>) r r r r ry r( r¢ r r )r Úrawr2 r$ rv r r r Ú test_map_with_string_constructoró s
|
|
z0TestPeriodIndex.test_map_with_string_constructor))Ú__name__Ú
|
__module__Ú __qualname__r r% rK ÚmarkZ parametrizerc r r r r Z_from_sequencer+ r3 rU rX r] ra re rf rl rq rt rw r{ r| r} r~ r r
r r r r r r r¥ r© r r r r r r sR ÿ
|
*
|
(
|
D ÿ
|
r c @ s$ e Zd Zdd Zdd Zdd ZdS )ÚTestShallowCopyc C s( t g dd}| ¡ }|}t ||¡ d S )Nr r )r Z_viewr r )r rS r# r$ r r r Útest_shallow_copy_empty s z'TestShallowCopy.test_shallow_copy_emptyc C s8 t dddd}tjtdd | |j¡ W 5 Q R X d S )Nú
|
2018-01-01rD Ú2DrF Zndarrayr@ )r rK rL r( Ú _shallow_copyrJ ©r rp r r r Ú test_shallow_copy_disallow_i8 s z-TestShallowCopy.test_shallow_copy_disallow_i8c C s6 t dddd}tjtdd | |¡ W 5 Q R X d S )Nr´ rD rµ rF r r@ )r rK rL r( r¶ r· r r r Ú0test_shallow_copy_requires_disallow_period_index s z@TestShallowCopy.test_shallow_copy_requires_disallow_period_indexN)r® r¯ r° r³ r¸ r¹ r r r r r² s r² c @ s e Zd Zdd Zdd ZdS )ÚTestSeriesPeriodc C s8 d}t jt|d ttddddtd W 5 Q R X d S )Nz(Cannot cast PeriodIndex to dtype float64r@ z
|
2000-01-01r- r rF r )rK rL rj r r Úfloatrd r r r Ú!test_constructor_cant_cast_period s z2TestSeriesPeriod.test_constructor_cant_cast_periodc C s8 t tdddtdd}t tddd}t ||¡ d S )Nz1/1/2000r- )r* r r )r r r r Zassert_series_equal)r ÚsrT r r r Ú test_constructor_cast_object s z-TestSeriesPeriod.test_constructor_cast_objectN)r® r¯ r° r¼ r¾ r r r r rº s rº )Únumpyr rK Zpandas._libs.tslibs.periodr Zpandas.core.dtypes.dtypesr Zpandasr r r r r r r
|
r Zpandas._testingZ_testingr Zpandas.core.arraysr r r² rº r r r r Ú<module> s (
|
q
|
