U
|
¬ý°dHf ã @ sÀ d Z ddlmZ ddlZzddlmZ W n ek
|
r@ dZY nX ddlZddl Z ddl
|
m Z m Z ddl mZ ddlZddlmZ ddlmZmZ G dd d ZG d
|
d d ZG d d d ZdS )z
|
Tests for DatetimeArray
|
é )Ú timedeltaN)ÚZoneInfo)Únpy_unit_to_abbrevÚ
|
tz_compare)ÚDatetimeTZDtype)Ú DatetimeArrayÚTimedeltaArrayc @ sV e Zd Zejdddgddd Zejdd Zejd d
|
Zejd d Zd d Z ej
|
de j e j ¡dd Zdd Zdd Zdd Zej
|
d¡dd Zdd Zd d Zd d Zej
|
d!d"d#d$g¡d%d& Zd'd( Zd)d* Zd+d, Zd-d. Z ej
|
d/e d0d1e j d0d1e !d0d2¡e !d3d¡e j" #d4¡e j" $d0¡g¡d5d6 Z%d7d8 Z&d9d: Z'd;S )<Ú TestNonNanoÚsÚmsÚus)Úparamsc C s |j S )z)Fixture returning parametrized time units)Úparam)ÚselfÚrequest© r úYd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/arrays/test_datetimes.pyÚunit s zTestNonNano.unitc C s. |}|d kr t d| d¡S t||dS d S )Nz datetime64[ú])r Útz)ÚnpÚdtyper )r r Útz_naive_fixturer r r r r $ s zTestNonNano.dtypec C sz t |dd }tjddd|d}|d kr@t |¡ d| d¡}n$t | d¡ d ¡¡ d| d¡}tj ||d }||fS )
|
Nr ú
|
2016-01-01é7 ÚD©ÚperiodsÚfreqr úM8[r ÚUTC©r )
|
ÚgetattrÚpdÚ
|
date_ranger ÚasarrayÚastypeÚ
|
tz_convertÚ tz_localizer Ú _simple_new)r r r r ÚdtiÚarrÚdtar r r Údta_dti, s
|
ÿzTestNonNano.dta_dtic C s |\}}|S ©Nr )r r- r, r* r r r r, ; s zTestNonNano.dtac C s t jdt jd d| d¡}tj||d}|j|ks:t|d j|ksLtt |j
|
|d j
|
sbt|d |d d
k ¡ s~td S )Né r! r r r é ) r ÚarangeÚint64Úviewr r) r ÚAssertionErrorr r r Úall)r r r r+ r, r r r Ú test_non_nano@ s zTestNonNano.test_non_nanoÚfieldc C s> |\}}||k ¡ stt||}t|j|}t ||¡ d S r. )r5 r4 r" Ú_dataÚtmÚassert_numpy_array_equal) r r r7 r r- r, r* ÚresÚexpectedr r r Ú test_fieldsI s
|
|
zTestNonNano.test_fieldsc C s t jdddd}t |¡ d| d¡}tj||jd}|jrBt t |
|
¡ ¡ d| d¡}tj||jd}|
|
¡ }t ||¡ d S )Nz2016-01-01 06:00:00r r ©r r r r r! ) r# r$ r r% r& r r) r Ú is_normalizedr4 Ú normalizer9 Ú assert_extension_array_equal)r r r* r+ r, Úexpr< r; r r r Útest_normalizeU s
|
zTestNonNano.test_normalizec C sz t jdt jd d| d¡}t|d}tj||d}|j|ksDttdd}t j
|
tdd tj||d W 5 Q R X d S ) Nr/ r! r r r ÚnsÚ ©Úmatch) r r1 r2 r3 r r r) r r4 ÚpytestÚraises)r r r+ r r, Zwrongr r r Ú test_simple_new_requires_matchd s
|
|
z*TestNonNano.test_simple_new_requires_matchc C sj t jdddd}t |¡ d| d¡}tj||jd}| ¡ }|j |j ksPt
|
|| ¡ |¡ksft
|
d S )Nr r r r> r r r! ) r# r$ r r% r& r r) r ZstdÚ_cresor4 Úfloor)r r r* r+ r, r; r r r Útest_std_non_nanoo s z TestNonNano.test_std_non_nanoz.ignore:Converting to PeriodArray.*:UserWarningc C s. |\}}| d¡}|j d¡}t ||¡ d S )Nr )Ú to_periodr8 r9 rA ©r r- r, r* Úresultr< r r r Útest_to_periodz s
|
zTestNonNano.test_to_periodc C sV t t|}|d }t|tjks&t|j|jks6t|j|jksFt||ksRtd S )Nr )ÚnextÚiterÚtyper# Ú Timestampr4 Ú_valuerK )r r, r; r< r r r Ú test_iter s zTestNonNano.test_iterc sD t¡}t fdd|D s$ttdd t| D s@td S )Nc 3 s | ]}|j j kV qd S r. )rK )Ú.0Úx©r, r r Ú <genexpr> s z1TestNonNano.test_astype_object.<locals>.<genexpr>c s s | ]\}}||kV qd S r. r )rX rY Úyr r r r[ s )r& Úobjectr5 r4 Úzip)r r, rP r rZ r Útest_astype_object s
|
z TestNonNano.test_astype_objectc C s( |\}}| ¡ }| ¡ }t ||¡ d S r. )Ú to_pydatetimer9 r: rO r r r Útest_to_pydatetime s z TestNonNano.test_to_pydatetimeÚmethÚtimeÚtimetzÚdatec C s, |\}}t ||}t ||}t ||¡ d S r. )r" r9 r: )r r- rb r, r* rP r< r r r Útest_time_date s
|
|
zTestNonNano.test_time_datec C s* |\}}| ¡ }|j ¡ }t ||¡ d S r. )Z_format_native_typesr8 r9 r: )r r r r- r, r* r; rB r r r Útest_format_native_types s
|
z$TestNonNano.test_format_native_typesc C s0 |\}}t |t |j dd| ¡ks,td S )Nz[nsú[)Úreprr8 Úreplacer4 )r r- r r, r* r r r Ú test_repr¨ s zTestNonNano.test_reprc C sF |}t t j¡}t j|j|jd |jgt jd}t |¡ d¡}| d¡}tj||j d}tj||j d}|t
|
j krt dddg¡} nN|t
|
j kr¢t dddg¡} n2|t
|
j t
|
jfkrÄt dddg¡} nt dddg¡} |||}
|
t |
|
| ¡ ||d |}
|
t |
|
| ¡ |t
|
j t
|
j fkrB||j|j} t | dd
| dd
¡ d S )Nr0 r! úM8[ns]úM8[s]FT)r Úiinfor2 ÚarrayÚminÚmaxr3 r r) r ÚoperatorÚeqÚneÚltÚler9 r: Ú_ndarray) r Ú comparison_opÚoprn Úvalsr+ Zarr2ÚleftÚrightr< rP Znp_resr r r Ú#test_compare_mismatched_resolutions® s*
|
|
|
|
z/TestNonNano.test_compare_mismatched_resolutionsc C sJ t jdd}t jddd| }|j d¡}|| d¡ }|jdksFtd S )Nr0 )Ú microsecondsr é ©r r )r# Ú Timedeltar$ r8 Úas_unitr r4 )r Útdr* r, r; r r r Ú(test_add_mismatched_reso_doesnt_downcastÑ s
|
z4TestNonNano.test_add_mismatched_reso_doesnt_downcastÚscalaré )ÚhoursÚhi Ým éx c
|
C s |\}}t |¡}t|j|j}t|}|| j |¡}|| } t | |¡ || } t | |¡ || j |¡}|| } t | |¡ d S r. ) r# r rq rK r r8 r r9 rA )
|
r r- r
r, r* r Zexp_resoZexp_unitr< rP r r r Ú-test_add_timedeltalike_scalar_mismatched_resoÜ s
|
z9TestNonNano.test_add_timedeltalike_scalar_mismatched_resoc C sb t jddd}|j d¡}|d d¡}|| }||d j d¡}|jdksRtt ||¡ d S )Nr r r r r r
|
zm8[us])r# r$ r8 r r r4 r9 rA )r r* r, ÚtsrP r< r r r Ú%test_sub_datetimelike_scalar_mismatchù s z1TestNonNano.test_sub_datetimelike_scalar_mismatchc C st t jddd}|j d¡}| d¡}|| }tjdddgdd}tj||jd}t
|
||¡ || }t
|
||¡ d S ) Nr r r r
|
r r zm8[ms]r! ) r# r$ r8 r r ro r r) r r9 rA )r r* r{ r| rP Z
|
exp_valuesr< Zresult2r r r Ú!test_sub_datetime64_reso_mismatch s
|
þ z-TestNonNano.test_sub_datetime64_reso_mismatchN)(Ú__name__Ú
|
__module__Ú __qualname__rH Zfixturer r r- r, r6 ÚmarkÚ parametrizer Z
|
_field_opsZ _bool_opsr= rC rJ rM ÚfilterwarningsrQ rW r_ ra rf rg rk r} r r r# r r Ú timedelta64ÚoffsetsZMinuteZHourr r r r r r r r sR
|
|
|
|
|
ÿ
|
|
|
|
#
|
|
|
|
|
úþ
|
r c @ s e Zd Zdd ZdS )Ú TestDatetimeArrayComparisonsc C sÒ |}t jdddd d}t|}|j|jks.t|j|jks>t|}tjt|t d}|j
|
dkrd| }|||}t ||¡ |t |¡t|t|| t¡fD ]0}|||}t ||¡ |||}t ||¡ qd S )Nz 2016-01-1ZMSé )r r r r! )rt Úgtru )r# r$ r r r4 r r ZonesÚlenÚboolr r9 r: ro ÚlistÚtupler& r] ) r rx ry r* r+ r| r< rP Úotherr r r Ú test_cmp_dt64_arraylike_tznaive s*
|
|
û
|
|
z<TestDatetimeArrayComparisons.test_cmp_dt64_arraylike_tznaiveN)r r r r r r r r r s r c @ sH e Zd Zdd Zdd Zdd Zej ddd g¡ej d
|
dd d g¡d d Z ej de
|
e j e j ddg¡dd Zdd Zdd Zdd Zdd Zdd Zej d e d ¡e d ¡ ¡ e d ¡ ¡ g¡d d Zd d! Zd"d# Zej d$d%d&g¡d'd( Zd)d* Z d+d, Z d-d. Z ej d/d0d1g¡d2d3 Z ej d/d0d1g¡d4d5 Z ej d
|
d6e d6¡d7e !d8¡ej"d9d:d;e j#d<d=d>d? d@ dA e #d<¡ $dB¡d? d@ dA e d ¡ %dC¡g ¡ej d/d0d1g¡dDdE Z&dFdG Z'dHdI Z(dJdK Z)dLdM Z*dNdOgZ+e,dPk r.z e,dNZ-W n e.k
|
r" Y n X e+ /e-¡ ej dQe+¡dRdS Z0dPS )TÚTestDatetimeArrayc C s\ t jddd}| d¡}|jdks&t|j}| d¡}|jdksDtt|t jjj sXtd S )Nr r r rm )
|
r# r$ r& r r4 r8 Ú
|
isinstanceÚcoreZarraysr )r r* r; r, r r r Ú test_astype_non_nano_tznaive9 s
|
|
z.TestDatetimeArray.test_astype_non_nano_tznaivec C s t jdddd}| d¡}|jdks(t|j}| d¡}|jdksFt| d¡}|jdks^tt ||¡rnt|jddd}|jdkstt ||¡std S ) Nr r r ©r r zM8[s, US/Pacific]z
|
M8[s, UTC]F©Úcopy)r# r$ r& r r4 r8 r9 Z shares_memory)r r* r; r, Úres2Úres3r r r Ú test_astype_non_nano_tzawareD s
|
|
|
z.TestDatetimeArray.test_astype_non_nano_tzawarec C s: t jdgtddd}|jtdddd}||ks6td S )NÚ2000ú
|
US/Central©r r! Fr¤ )r Ú_from_sequencer r& r4 )r r+ rP r r r Útest_astype_to_sameW s ÿz%TestDatetimeArray.test_astype_to_samer údatetime64[ns]zdatetime64[ns, UTC]r zdatetime64[ns, CET]c C s t jddg|d}| ¡ }d}|dk|dkA r2d}|rn|dkrDd}nd}tjt|d | |¡ W 5 Q R X n$| |¡}t j|d d
< t ||¡ d S )
|
Nr0 r r! Fr® Tz<Use obj.tz_localize instead or series.dt.tz_localize insteadz1from timezone-aware dtype to timezone-naive dtyperF )
|
r# ÚSeriesr¥ rH rI Ú TypeErrorr& ÚNaTr9 Úassert_series_equal)r r r ZserÚorigÚerrÚmsgÚtr r r Útest_astype_copies^ s
|
z$TestDatetimeArray.test_astype_copiesZuint32Zuint64c C sx t t d¡t d¡g¡}t |¡tjkrRtjt dd |
|
|¡ W 5 Q R X d S |
|
|¡}|j d¡}t ||¡ d S )Nr© Z2001zDo obj.astype\('int64'\)rF Úi8)r r¬ r# rU r r r2 rH rI r° r& rw r3 r9 r: )r r r+ rP r< r r r Útest_astype_intx s
|
z!TestDatetimeArray.test_astype_intc C s@ t jddd}|j}| d¡}|jdks,t||k ¡ s<td S )Nr é r zSparse[datetime64[ns]])r# r$ r8 r& r r4 r5 )r r* r, rP r r r Útest_astype_to_sparse_dt64
s
|
|
z,TestDatetimeArray.test_astype_to_sparse_dt64c C s: t jdgtddd}tjtdd d|_W 5 Q R X d S )Nr© rª r« r! r( rF r )r r¬ r rH rI ÚAttributeErrorr )r r+ r r r Útest_tz_setter_raises s ÿz'TestDatetimeArray.test_tz_setter_raisesc C s¦ |}t jdddgdd}|d kr&|jnt|d}t||d}| ¡ }t d¡ |¡}t | d ¡}||d< ||d< t
|
||¡ ||d< ||g|d d
< t
|
||¡ d S ) Nr0 r r rl r! r« z2020-09-08 16:50r ) r ro r r r r¥ r# rU r( Ústrr9 Ú assert_equal) r r r Údatar r+ r< r Úsetterr r r Útest_setitem_str_impute_tz s z,TestDatetimeArray.test_setitem_str_impute_tzc C s t jdddgdd}t|dtddd }tjtd
|
d t d ¡|d < W 5 Q R X tjd dd}||d < |d | d¡ks|t
|
d S )Nr0 r r rl r! Frª r« )r¥ r z$Cannot compare tz-naive and tz-awarerF r© r ú
|
US/Eastern) r ro r r rH rI r° r# rU r' r4 )r rÀ r+ r r r r Ú test_setitem_different_tz_raises¬ s z2TestDatetimeArray.test_setitem_different_tz_raisesc C s: t tjddddd}tjddd|d< |jd ks6td S )Nr© r r rª r r« r )r r# r$ rU r r4 )r Úar r r Útest_setitem_clears_freq¸ s z*TestDatetimeArray.test_setitem_clears_freqÚobjz
|
2021-01-01c C s2 t jdddd}|j}||d< |d |ks.td S )Nr© r r r> r )r# r$ r8 r4 )r rÇ r* r+ r r r Útest_setitem_objects½ s
|
z&TestDatetimeArray.test_setitem_objectsc C sJ t jddddd}t|}| ddg¡}t|jd |jd}t ||¡ d S )Nr© r r rª r r0 )r r )r# r$ r ÚrepeatZasi8r r9 r¿ )r r* r+ Zrepeatedr< r r r Útest_repeat_preserves_tzÍ s
|
z*TestDatetimeArray.test_repeat_preserves_tzc C s t jddddd}t| ddg¡}| ¡ }|j |¡s<tt j|d< |jd d
|
}t j ddd g|d |d t jgd d}t
|
||¡ d S )Nr© r r rª r rº r éþÿÿÿF)Zdropnar0 r Úcount)ÚindexÚname) r# r$ r rÉ Z value_countsrÍ Úequalsr4 r± r¯ r9 r² )r r* r+ rP r< r r r Ú test_value_counts_preserves_tz× s
|
&z0TestDatetimeArray.test_value_counts_preserves_tzÚmethodÚpadÚbackfillc C s¶ t jddddd}t|dd}t j|d< |d kr8|d
|
n|d }tj|d |d
|
||d |d gtddd}|j|d}t ||¡ |d t jkst |d t j
|
dddks²t d S )Nz
|
2000-01-01r/ r rª r Tr¤ r rÒ r0 r r rº r« r! ©rÑ z
|
2000-01-03) r# r$ r r± r¬ r Úfillnar9 rA r4 rU )r rÑ r* r+ Zfill_valr< rP r r r Útest_fillna_preserves_tzå s
|
þ z*TestDatetimeArray.test_fillna_preserves_tzc C sd t jdddd}|j dd¡ ¡ }t j|d< t j|d< |jd d
|
}| ¡ }|d |d< t ||¡ |jd d
|
}| ¡ }| ¡ }|d |d< |d |d< t ||¡ | |j
|
jdd¡}|j
|
j d sÆt |j
|
j d rÖt t ||¡ |jd d
|
}t ||¡ |jd d
|
} t | |¡ t |¡}
|
|
|
jd d
|
} t |¡} t | | ¡ |
|
jd d
|
} t |¡} t | | ¡ d S )Nr é ú
|
US/Pacificr£ r r )r r0 )r0 r rÒ rÔ )r r rÓ )r r )r0 r0 ÚF)ÚorderZ F_CONTIGUOUSZ C_CONTIGUOUS)r# r$ r8 Úreshaper¥ r± rÕ r9 rA Z_from_backing_datarw Úflagsr4 Z DataFrameZassert_frame_equal) r r* r, Zres1Z expected1r¦ Z expected2Zdta2r§ Zres4Zdfr; r< r r r Útest_fillna_2dø s:
|
|
|
|
|
z TestDatetimeArray.test_fillna_2dc C s d}t tjdd|d}t |¡}tjtjd|dtjd|dgtd}t ||¡ tj|td}t ||¡ tj|d d}tjd
|
d gd d}t ||¡ d S ) Nrª Ú2017r r£ ú2017-01-01T00:00:00r« ú2017-01-02T00:00:00r! rl z2017-01-01T06:00:00z2017-01-02T06:00:00)
|
r r# r$ r r% ro rU r] r9 r: )r r rÀ rP r< r r r Útest_array_interface_tz! s$
|
þû ÿz)TestDatetimeArray.test_array_interface_tzc C sv t tjddd}tjddgdd}t |¡}t ||¡ tj|td}tjt d¡t d¡gtd}t ||¡ d S )NrÞ r r rß rà r® r! )
|
r r# r$ r ro r% r9 r: r] rU )r rÀ r< rP r r r Útest_array_interface9 s ÿ
|
þz&TestDatetimeArray.test_array_interfacerÍ TFc C s¢ t jdddd d d }t|dd d ¡}|r:t |¡}| |d
|
¡}| |d
|
d ¡¡}||ksht| |d
|
d
¡}| |d
|
d
d ¡¡}t
|
||¡ d S ) Né
|
r¸ r! é é é Ê;r ©r ú
|
Asia/Tokyor r r× ) r r1 r r( r# ÚIndexÚ searchsortedr' r4 r9 r¿ )r rÍ rÀ r+ r< rP r r r Ú test_searchsorted_different_tzI s
|
z0TestDatetimeArray.test_searchsorted_different_tzc C sÞ t jdddd d d }t|dd}|r4t |¡}| d ¡}d
|
}tjt|d | |d ¡ W 5 Q R X tjt|d | |¡ W 5 Q R X tjt|d | |d ¡ W 5 Q R X tjt|d | |¡ W 5 Q R X d S ) Nrã r¸ r! rä rå ræ r rç rè z:Cannot compare tz-naive and tz-aware datetime-like objectsrF r )
|
r r1 r r# ré r( rH rI r° rê )r rÍ rÀ r+ Zmismatchrµ r r r Ú$test_searchsorted_tzawareness_compatX s
|
|
z6TestDatetimeArray.test_searchsorted_tzawareness_compatr0 g ð?r± r )ÚdaysÚinvalidrã r¸ r! rä rå ræ ztimedelta64[ns]r c C sj t jdddd d d }t|dd}|r4t |¡}d d
|
d g¡}tjt|d | |¡ W 5 Q R X d S ) Nrã r¸ r! rä rå ræ r rç ú|z0searchsorted requires compatible dtype or scalarz<value should be a 'Timestamp', 'NaT', or array of those. GotrF )
|
r r1 r r# ré ÚjoinrH rI r° rê )r r rÍ rÀ r+ rµ r r r Ú test_searchsorted_invalid_typesl s
|
þÿz1TestDatetimeArray.test_searchsorted_invalid_typesc C s¬ t jddd}|j}tt |jd¡}|d }|| ¡ | ¡ fD ] }|j d|d}t
|
||¡ q@| d¡}| d¡}|d }|| ¡ fD ] }|j d|d}t
|
||¡ qd S )Nr r r r0 éÿÿÿÿ©Ú
|
fill_valuer ) r# r$ r8 r r Zrollrw r` Ú to_datetime64Úshiftr9 Úassert_datetime_array_equalr( )r r* r, r< Úfvrô rP r r r Útest_shift_fill_value s
|
|
z'TestDatetimeArray.test_shift_fill_valuec
|
C s¸ t jddd}|j}|d d¡}|| ¡ fD ],}tjtdd |jd|d W 5 Q R X q.| d¡}|d d ¡}|| ¡ | ¡ fD ],}tjtdd |jd|d W 5 Q R X qd S )
|
Nr r r rò r zCannot comparerF r0 ró )
|
r# r$ r8 r( r` rH rI r° rö rõ )r r* r, rø rî r r r Ú%test_shift_value_tzawareness_mismatch s
|
z7TestDatetimeArray.test_shift_value_tzawareness_mismatchc C sV t jdddd}|j}t jddd}|jd|d }|jd| d¡d }t ||¡ d S )
|
Nr r r r£ z2020-10-18 18:44rØ r« r0 ró )r# r$ r8 rU rö r' r9 r¿ )r r* r, rô rP r< r r r Útest_shift_requires_tzmatch¬ s z-TestDatetimeArray.test_shift_requires_tzmatchc C sd t jdddd}|j dd¡}| d ¡}| ¡ d ¡ |j¡}t ||¡ | d¡}t ||¡ d S )Nz
|
1994-05-12é rØ r£ r rº ) r# r$ r8 rÛ r( ZravelÚshaper9 r÷ )r r* r, rP r< Z roundtripr r r Útest_tz_localize_t2d¸ s
|
|
z&TestDatetimeArray.test_tz_localize_t2drà zdateutil/US/EasternNr c C sª t jddddgt jd}|d9 }t| d¡ |¡}|d }t|d }t|t|ks\t| ¡ | ¡ kspt|
|
t ¡d }t|t|kst| ¡ | ¡ ks¦td S ) Ni@¶NiP¶Ni`"¶Nip0¶Nr! ræ r r ) r ro r2 r r( r' r r¾ r4 Ú utcoffsetr& r] )r r Zutc_valsr, r{ r| Zright2r r r Útest_iter_zoneinfo_foldÍ s
|
ÿ z)TestDatetimeArray.test_iter_zoneinfo_fold)1r r r r¢ r¨ r rH r r r· Úintr Úint32r2 r¹ r» r½ r rÄ rÆ r# rU rõ r` rÈ rÊ rÐ rÖ rÝ rá râ rë rì r r r1 r3 rN rñ rù rú rû rþ Zeastsr r ÚKeyErrorÚappendr r r r r r 8 s~ ÿ
|
ýþ
|
|
|
)
|
|
|
÷þ
|
|
r )Ú__doc__Údatetimer rr Zzoneinfor Ú ImportErrorÚnumpyr rH Zpandas._libs.tslibsr r Zpandas.core.dtypes.dtypesr Zpandasr# Zpandas._testingZ_testingr9 Zpandas.core.arraysr r r r r r r r r Ú<module> s"
|
w$
|
