U
|
¬ý°d ã @ sh d dl Zd dlZd dlmZ d dlZd dlmZ d dl
|
m Z d dl m Z G dd dZG dd dZdS ) é N)ÚDatetimeTZDtype)Ú DatetimeArray)Ú_sequence_to_dt64nsc @ s e Zd Zdd Zdd Zdd Zej de j
|
e e j e jg¡dd Zd
|
d Zd d Zdd Zdd Zdd Zdd Zdd Zej ddddd g¡d d Zd S ) Ú TestDatetimeArrayConstructorc C sD t j t d¡t d¡g¡}tjtdd t |¡ W 5 Q R X d S )Né z Cannot create a DatetimeArray©Úmatch)
|
ÚpdZ
|
MultiIndexZ from_productÚnpÚarangeÚpytestÚraisesÚ TypeErrorr Ú_from_sequence)ÚselfÚmi© r úfd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/arrays/datetimes/test_constructors.pyÚ test_from_sequence_invalid_type s z<TestDatetimeArrayConstructor.test_from_sequence_invalid_typec C sx t jddddgdd d¡}tjtdd t| ddd¡ W 5 Q R X tjtdd t|dg ¡ W 5 Q R X d S )
|
Nr é é é zM8[h]©ÚdtypeúM8[ns]zOnly 1-dimensionalr ) r
|
ÚarrayZastyper r Ú
|
ValueErrorr ÚreshapeZsqueeze©r Úarrr r r Útest_only_1dim_accepted s
|
z4TestDatetimeArrayConstructor.test_only_1dim_acceptedc C sF t jdt jdd d }d}tjt|d t|dd W 5 Q R X d S ) Nr r é é Ê;zRInferred frequency H from passed values does not conform to passed frequency W-SUNr ÚW©Úfreq)r
|
r Úint64r r r r ©r r Úmsgr r r Útest_freq_validation s
|
ÿz1TestDatetimeArrayConstructor.test_freq_validationÚmethc
|
C s` t t d¡tjdddg¡}d}||d d d
fD ]&}tjt|d || W 5 Q R X q4d S )NÚ2000ZCET©ÚtzzuCannot mix tz-aware with tz-naive values|Tz-aware datetime.datetime cannot be converted to datetime64 unless utc=Trueéÿÿÿÿr )r
|
r r Ú Timestampr r r )r r* r r( Úobjr r r Ú test_mixing_naive_tzaware_raises) s ÿz=TestDatetimeArrayConstructor.test_mixing_naive_tzaware_raisesc C sP t tjdtjd¡d d }t |¡ d¡}t jddddj }t
|
||¡ d S ) Nr r r! r" Úinferz
|
1970-01-01ÚH)Úperiodsr% ) r r r
|
r r& r r Z
|
_with_freqÚ
|
date_rangeÚ_dataÚtmÚassert_datetime_array_equal)r r ÚresultÚexpectedr r r Útest_from_pandas_arrayB s z3TestDatetimeArrayConstructor.test_from_pandas_arrayc C s° t tjdgddtddd}tdd}d}tjt|d t ||d W 5 Q R X tjt|d t |t d¡d W 5 Q R X tjt|d t | d ¡|jd W 5 Q R X d S ) Nz2000-01-01T06:00:00r r ú
|
US/Centralr, z
|
US/EasternzEdtype=datetime64\[ns.*\] does not match data dtype datetime64\[ns.*\]r ) r r
|
r r r r r r Z tz_localize)r r r r( r r r Ú test_mismatched_timezone_raisesJ s þ
|
z<TestDatetimeArrayConstructor.test_mismatched_timezone_raisesc C s, t jtdd tdddg W 5 Q R X d S )NÚlistr r r r )r r r r ©r r r r Útest_non_array_raisesZ s z2TestDatetimeArrayConstructor.test_non_array_raisesc C sÐ t jdddgdd}d}tjt|d t| W 5 Q R X d}tjt|d t |¡ W 5 Q R X tjt|d t| W 5 Q R X tjt|d t
|
|¡ W 5 Q R X tjt|d t |¡ W 5 Q R X d S ) Nr r r Úboolr z-Unexpected value for 'dtype': 'bool'. Must ber z2dtype bool cannot be converted to datetime64\[ns\]) r
|
r r r r r r r r r Ú DatetimeIndexÚ to_datetimer' r r r Útest_bool_dtype_raises^ s z3TestDatetimeArrayConstructor.test_bool_dtype_raisesc C s: t jtdd" ttjdddgdddd W 5 Q R X d S ) Nz Unexpected value for 'dtype'.r r r r Úi8r Úcategory©r r r r r
|
r r? r r r Útest_incorrect_dtype_raisesr s z8TestDatetimeArrayConstructor.test_incorrect_dtype_raisesc C s: t jtdd" ttjdddgdddd W 5 Q R X d S )
|
NzFrequency inferencer r r r rE r r2 r$ rG r? r r r Útest_freq_infer_raisesv s z3TestDatetimeArrayConstructor.test_freq_infer_raisesc C sL t jdddgdd}t|dd}|j|ks.tt|dd}|j|k sHtd S ) Nr r r r r F)ÚcopyT)r
|
r r Z_ndarrayÚAssertionError)r Údatar r r r Ú test_copyz s
|
z&TestDatetimeArrayConstructor.test_copyÚunitÚsÚmsÚusÚnsc C sH t jdddgd| dd}t|}|j|ks2t|d j|ksDtd S )Nr r r zM8[ú]r r )r
|
r r rN rK )r rN rL r r r r Útest_numpy_datetime_unit s z5TestDatetimeArrayConstructor.test_numpy_datetime_unitN)Ú__name__Ú
|
__module__Ú __qualname__r r r) r ÚmarkÚ parametrizer r r r rC rB r1 r; r= r@ rD rH rI rM rT r r r r r s* üþ
|
r c @ s6 e Zd Zdd Zdd Zej dddg¡dd Zd
|
S ) ÚTestSequenceToDT64NSc C sH t jdgtddd}tjtdd t j|tddd W 5 Q R X d S )Nr+ r< r, r zdata is already tz-awarer ÚUTC)r r r r r r Ú_from_sequence_not_strictr r r r Ú test_tz_dtype_mismatch_raises s ÿ ÿz2TestSequenceToDT64NS.test_tz_dtype_mismatch_raisesc C s8 t dd}tjdg|d}tj||d}t ||¡ d S )Nr< r, r+ r )r r r r\ r7 Z assert_equal)r r r r9 r r r Útest_tz_dtype_matches s
|
z*TestSequenceToDT64NS.test_tz_dtype_matchesÚorderÚFÚCc C s¸ t jdddd}tj|td dd¡}|dkr4|j}t|}t| ¡ }t
|
|d ¡ |d ¡ |d
|
|d
|
kstt |d |d kst t |¡}t | ¡ ¡ |j¡}t ||¡ d S ) Nz
|
2016-01-01é z
|
US/Pacific)r4 r- r r r r` r r )r r5 r
|
r Úobjectr ÚTr Zravelr7 Zassert_numpy_array_equalrK r r Úshaper8 )r r_ Zdtir Úresr: r r r Útest_2d s
|
z TestSequenceToDT64NS.test_2dN) rU rV rW r] r^ r rX rY rg r r r r rZ s rZ )Únumpyr
|
r Zpandas.core.dtypes.dtypesr Zpandasr Zpandas._testingZ_testingr7 Zpandas.core.arraysr Z pandas.core.arrays.datetimesr r rZ r r r r Ú<module> s ~
|
