U
|
¬ý°dÎ ã @ s d dl Zd dlZd dlmZ d dlmZ d dlZd dl m
|
Z d dl m Z mZ ej de dd¡gddgfe dd¡gddgfd gddgfdgddgfe dd¡gejj ¡ dgfe dd¡dgddegfe ejdd
|
d ¡ddd d gfejdd
|
d ddd d gfejdddddddddgfg ¡dd Zdd Zdd Zej de dd¡e dd ¡gdd fe dd¡gd d fg¡d d Zd d! Z d"d# Z d$d% Z d&d' Z dS )(é N)ÚiNaT)ÚIncompatibleFrequency)Ú PeriodArrayÚ period_arrayzdata, freq, expectedÚ2017ÚDiC iá é )ÚperiodsiC iC é ÚQ©r Úfreqé¼ é½ é¾ é¿ c C s. t | |dj}tj|tjd}t ||¡ d S )N©r ©Údtype)r Úasi8ÚnpÚasarrayÚint64ÚtmZassert_numpy_array_equal)Údatar ÚexpectedÚresult© r úcd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/arrays/period/test_constructors.pyÚtest_period_array_ok s r c C s t t d¡g} tj| dd}|jdd t |}t || ¡ t |¡}t |t | ¡¡ t
|
d|i¡}t |t
|
d| i¡¡ d S )Nz
|
2019-01-01Úobjectr F)ÚwriteÚA) r ÚpdÚPeriodr r Zsetflagsr Úassert_period_array_equalÚSeriesZassert_series_equalÚ DataFrameZassert_frame_equal)ÚpaÚarrr r r r Ú!test_period_array_readonly_object# s
|
r* c C s@ t jdddd} tj| dd}tdddgdd}t ||¡ d S )Nr r r r ÚMr z
|
2017-01-01)r# Ú
|
date_ranger Z_from_datetime64r r r% )r) r r r r r Ú!test_from_datetime64_freq_changes3 s r- zdata, freq, msgr" zInput has different freqc C s( t jt|d t| | W 5 Q R X d S )N©Úmatch)ÚpytestÚraisesr r )r r Úmsgr r r Útest_period_array_raises; s r3 c C s: t dddg¡} tjtdd t| dd W 5 Q R X d S )Né é r r r. r r )r# r& r0 r1 Ú TypeErrorr )Zserr r r Ú)test_period_array_non_period_series_raiesK s r7 c C sh t ddgdd} tjtdd t| dd W 5 Q R X tjtdd t| tjj ¡ d W 5 Q R X d S )NÚ2000Ú2001r r r r. r+ ) r r0 r1 r r r# ÚtseriesÚoffsetsZMonthEnd)r) r r r Ú test_period_array_freq_mismatchQ s
|
r< c C s~ t ddgdd} t| d j}tjt|d tj| j| j d W 5 Q R X tjt|d tjt
|
| j| j d W 5 Q R X d S )Nr8 r9 r r r r. r ) r ÚstrZordinalr0 r1 r6 r Ú_from_sequencer r Úlist)r) r2 r r r Ú test_from_sequence_disallows_i8Z s r@ c C s. t j d¡} t jddddj}tjd gtd}| |d< d}tj t
|
|d t j ||d W 5 Q R X tj t
|
|d t j ||d W 5 Q R X tj t
|
|d t j||d W 5 Q R X tj t
|
|d t j||d W 5 Q R X tj t
|
|d t j||d W 5 Q R X tj t
|
|d t j||d W 5 Q R X d S )
|
Nzm8[ns]z
|
2005-01-01r r r r r z2Value must be Period, string, integer, or datetimer. )r# ZNaTZto_numpyÚ period_ranger r Úarrayr r0 r1 Ú
|
ValueErrorr r> Z PeriodIndexZIndexr& r' )Útdr r) r2 r r r Ú!test_from_td64nat_sequence_raisese s" rE ) Únumpyr r0 Zpandas._libs.tslibsr Zpandas._libs.tslibs.periodr Zpandasr# Zpandas._testingZ_testingr Zpandas.core.arraysr r ÚmarkZ parametrizer$ r: r; ZDayr& r, rA r r* r- r3 r7 r< r@ rE r r r r Ú<module> sH ÷þ
|
ýúþ
|
|
