U
|
®ý°d¯ ã @ sr d dl Zd dlZd dlmZ d dlmZmZmZm Z m
|
Z
|
m Z d dl m Z dd Zdd Zdd ZG d
|
d d ZdS ) é N)ÚNpyDatetimeUnit)ÚOutOfBoundsDatetimeÚOutOfBoundsTimedeltaÚastype_overflowsafeÚ is_unitlessÚpy_get_unit_from_dtypeÚpy_td64_to_tdstructc C s° t d¡} t| rtt d¡} t| s,tt d¡} t| rBtt d¡} t| sXtd}tjt|d tt t j¡ W 5 Q R X d}tjt|d td W 5 Q R X d S ) NúM8[ns]Ú
|
datetime64úm8[ns]Z timedelta64z'dtype must be datetime64 or timedelta64©Úmatchz#Argument 'dtype' has incorrect typeZfoo) ÚnpÚdtyper ÚAssertionErrorÚpytestÚraisesÚ
|
ValueErrorÚint64Ú TypeError)r Úmsg© r ú[d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/tslibs/test_np_datetime.pyÚtest_is_unitless s
|
|
|
|
r c C sÊ t t d¡tjjkstt t d¡tjjks4tt t d¡tjjksNtt t d¡tj jkshtt t d¡tj
|
jkstt t d¡tj jkstt t d¡tj jks¶tt t d¡tj jksÐtt t d ¡tjjksêtt t d
|
¡tjjkstt t d ¡tjjks"tt t d ¡tjjks>tt t d ¡tjjksZtt t d¡tjjksvtt t d¡tjjkstt t d¡tjjks®tt t d¡tj jksÊtt t d¡tj
|
jksætt t d¡tj jkstt t d¡tj jks tt t d¡tj jks:tt t d¡tjjksVtt t d¡tjjksrtt t d¡tjjkstt t d¡tjjksªtt t d¡tjjksÆtd S )NzM8[Y]zM8[M]zM8[W]úM8[D]zM8[h]zM8[m]zM8[s]zM8[ms]úM8[us]r zM8[ps]zM8[fs]zM8[as]zm8[Y]zm8[M]zm8[W]úm8[D]zm8[h]zm8[m]zm8[s]zm8[ms]úm8[us]r zm8[ps]zm8[fs]zm8[as])r r r r ZNPY_FR_YÚvaluer ZNPY_FR_MZNPY_FR_WZNPY_FR_DZNPY_FR_hZNPY_FR_mÚNPY_FR_sÚ NPY_FR_msÚ NPY_FR_usÚ NPY_FR_nsZ NPY_FR_psZ NPY_FR_fsZ NPY_FR_asr r r r Útest_get_unit_from_dtype' s4 r# c C sØ d} t | tjj}ddddddddddd
|
}||ks8tt | tjj}dd d
|
d dddd d dd
|
}||ksltt | tjj}dd ddddddddd
|
}||ks tt | tjj}dddd ddddddd
|
}||ksÔtd S )Nl Ê*µL r é iÆ i| éê iìï )
|
ÚdaysZhrsÚminÚsecÚmsÚusÚnsÚsecondsÚ microsecondsÚ nanosecondsé é é" i¦0 iJµ é é% é iì2 i i3 é é
|
iÊ) )r r r" r r r! r r ) ÚvalZres1Zexp1Zres2Zexp2Zres3Zexp3Zres4Zexp4r r r Útest_td64_to_tdstructI sj ö ö ö ö r8 c @ s4 e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd S ) ÚTestAstypeOverflowSafec C sl t d¡}t d¡}d}tjt|d t||dd W 5 Q R X tjt|d t||dd W 5 Q R X d S )Né r ú]astype_overflowsafe values.dtype and dtype must be either both-datetime64 or both-timedelta64r T©ÚcopyF)r Úaranger r r r r ©ÚselfÚarrr r r r r Útest_pass_non_dt64_array s
|
|
ÿz/TestAstypeOverflowSafe.test_pass_non_dt64_arrayc C sv t jddd d¡}t d¡}d}tjt|d t||dd W 5 Q R X tjt|d t||d
|
d W 5 Q R X d S ) Nr: Úi8©r r r r; r Tr< F)r r> Úviewr r r r r r? r r r Útest_pass_non_dt64_dtype s
|
ÿz/TestAstypeOverflowSafe.test_pass_non_dt64_dtypec
|
C s¤ t d¡}t dd¡}|t jddd }| |¡}| |j¡}||k ¡ rNtd}tjt |d t
|
|| W 5 Q R X t d ¡}t
|
||}| |¡} t || ¡ d S )
|
Nr ú
|
2262-04-05ÚDr6 r rD z"Out of bounds nanosecond timestampr r ) r r r
|
r> ÚastypeÚallr r r r r ÚtmÚassert_numpy_array_equal©
|
r@ r ÚdtrA ZwrongZ roundtripr Zdtype2ÚresultÚexpectedr r r Ú test_astype_overflowsafe_dt64© s
|
|
|
|
|
z4TestAstypeOverflowSafe.test_astype_overflowsafe_dt64c
|
C s® t d¡}t dd¡}|t jddd }| d¡}| |¡}| |j¡}||k ¡ rXtd}tj t
|
|d t || W 5 Q R X t d ¡}t ||}| |¡} t || ¡ d S )
|
Nr rG rH r6 r rD z@Cannot convert 106752 days to timedelta64\[ns\] without overflowr r )r r r
|
r> rE rI rJ r r r r r rK rL rM r r r Ú test_astype_overflowsafe_td64¾ s
|
|
|
|
|
|
z4TestAstypeOverflowSafe.test_astype_overflowsafe_td64c C sp t jddgdd}t d¡}d}tjt|d t||dd W 5 Q R X t||d
|
d }| |¡}t ||¡ d S ) Ni$úÿÿiÜ r rD r z'Cannot losslessly cast '-1500 ns' to usr F)Zround_okT)
|
r Úarrayr r r r r rI rK rL )r@ rA r r rO rP r r r Ú*test_astype_overflowsafe_disallow_roundingÔ s
|
|
zATestAstypeOverflowSafe.test_astype_overflowsafe_disallow_roundingN)Ú__name__Ú
|
__module__Ú __qualname__rB rF rQ rR rT r r r r r9 s
|
r9 )Únumpyr r Zpandas._libs.tslibs.dtypesr Z pandas._libs.tslibs.np_datetimer r r r r r Zpandas._testingZ_testingrK r r# r8 r9 r r r r Ú<module> s "A
|
