U
|
¬ý°d] ã @ s d dl Zd dlZd dlmZ d dlmZ d dlmZ d dl
|
m Z d dl Z d dlmZ d dlmZmZ dd Zd d
|
Zd d Zd d Zdd Zej dd ge dd¡dddgfd gdeddgfd gej eddgfd ddge dd¡dgd fd ddge dd¡e dd¡e dd¡gdddgfg¡d d Z d d Z d d! Z d"d# Z d$d% Z!d&d' Z"ej d(e jdd)d*edddgd)d*g¡d+d, Z#d-d. Z$d/d0 Z%dS )1é N)ÚiNaT)ÚIncompatibleFrequency)Ú _registry)Ú PeriodDtype)Ú PeriodArrayÚ period_arrayc C s0 t tjkstt d¡} t d}| |ks,td S )Nz Period[D]ÚD)r ÚregistryZdtypesÚAssertionErrorÚfind©ÚresultÚexpected© r úVd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/arrays/test_period.pyÚtest_registered s
|
r c C s4 t ddd gddj} t ddtg¡}t | |¡ d S )NÚ2000Ú2001r ©ÚfreqéÍ* é;, )r Zasi8ÚnpÚarrayr ÚtmZassert_numpy_array_equalr r r r Ú test_asi8 s r c C s~ t ddgdd} tjtdd$ | jddgd tjdd
|
dd W 5 Q R X d }tjt|d | jddgd d d W 5 Q R X d S )Nr r r r r ©Úmatchr éÿÿÿÿTÚW)Z
|
allow_fillZ
|
fill_valuez6value should be a 'Period' or 'NaT'. Got 'str' insteadZfoo)r ÚpytestÚraisesr ZtakeÚpdÚPeriodÚ TypeError)ÚarrÚmsgr r r Útest_take_raises& s (r' c C sB t dddgdd} tjtdd | | d d
¡ W 5 Q R X d S ) Nr r Ú2002r r ZLengthr é )r r r! Ú
|
ValueErrorÚfillna©r% r r r Útest_fillna_raises0 s r- c C s4 t dddgdd} | t dd¡¡}|| k s0td S )Nr r r( r r )r r+ r" r# r
|
)r% r r r r Útest_fillna_copies6 s r. zkey, value, expectedr r r é r) é r r( r i¨- c C s6 t t d¡dd}t |dd}||| < t ||¡ d S )Nr0 r r )r r Úaranger Zassert_period_array_equal)ÚkeyÚvaluer r% r r r Ú test_setitem@ s r4 c C sx t t d¡dd} tjtdd tjddd| d< W 5 Q R X tdd gdd}tjtdd || dd
|
g< W 5 Q R X d S ) Nr0 r r r r r ÚAr r r/ ) r r r1 r r! r r" r# r ©r% Úotherr r r Ú%test_setitem_raises_incompatible_freqU s r8 c C sH t t d¡dd} tjtdd tjdddg| ddg< W 5 Q R X d S ) Nr0 r r Úlengthr r r r/ )r r r1 r r! r* r" r# r, r r r Útest_setitem_raises_length_ s r: c C s8 t t d¡dd} tjtdd d| d< W 5 Q R X d S )Nr0 r r Úintr r/ r )r r r1 r r! r$ r, r r r Útest_setitem_raises_typee s r< c C sD t ddgdd} tjddd}tjtdd | | W 5 Q R X d S )Nr r r r ÚMr r )r r" r# r r! r r6 r r r Útest_sub_periodo s r> c C sr t jdddd} | d¡}t j d|j¡}tjtdd || W 5 Q R X tjtdd || W 5 Q R X d S ) Nz
|
1677-09-22r) r )Zperiodsr Únsl @ô Ìk zOverflow in int64 additionr ) r" Z
|
date_rangeZ to_periodr# Z _from_ordinalr r r! Ú OverflowError)ZdtiÚpiZperr r r Útest_sub_period_overflowv s
|
rB r7 ÚHr c C s~ t tdddgdd¡}t dddg¡}tjtdd |j || ¡ W 5 Q R X | || ¡}|
|
t ¡ || ¡}t ||¡ d S )
|
Nr r r( r r TFr r )r" ZSeriesr r r r r! r Z_whereÚwhereZastypeÚobjectr Zassert_series_equal)r7 ZserZcondÚresr r r r Ú test_where_different_freq_raises s rG c C s, t ddgdd} t| }d}||ks(td S )Nr r r r zF<PeriodArray>
|
['2000-01-01', '2001-01-01']
|
Length: 2, dtype: period[D]©r Ústrr
|
©r% r r r r r Útest_repr_small s
|
ÿrK c C s0 t ddgd dd} t| }d}||ks,td S )Nr r iô r r aM <PeriodArray>
|
['2000-01-01', '2001-01-01', '2000-01-01', '2001-01-01', '2000-01-01',
|
'2001-01-01', '2000-01-01', '2001-01-01', '2000-01-01', '2001-01-01',
|
...
|
'2000-01-01', '2001-01-01', '2000-01-01', '2001-01-01', '2000-01-01',
|
'2001-01-01', '2000-01-01', '2001-01-01', '2000-01-01', '2001-01-01']
|
Length: 1000, dtype: period[D]rH rJ r r r Útest_repr_large¦ s
|
ÿ rL )&Únumpyr r Zpandas._libs.tslibsr Zpandas._libs.tslibs.periodr Zpandas.core.dtypes.baser r Zpandas.core.dtypes.dtypesr Zpandasr" Zpandas._testingZ_testingr Zpandas.core.arraysr r r r r' r- r. ÚmarkZ parametrizer# Únanr4 r8 r: r< r> rB rG rK rL r r r r Ú<module> sL
|
|
ýûþ
|
|
|
þ
|
|
