U
|
¬ý°d» ã @ sÜ d dl mZ d dlZd dlZd dlZd dlZd dlmZm Z m
|
Z
|
d dl m m Z d dlZd dlmZmZmZmZ d dlmZ d dlmZmZmZm Z d dl m Z d dl m Z ej!dd d
|
d d d gddd Z"ej!dd Z#ej!dd Z$ej!dd Z%G dd dZ&G dd de&Z'G dd d e&Z(G d d d e&Z)ej*j+d ed d!d"gj,ee -d"d#¡ffej.d$d%dd&j,ee /d"d#¡ffej0d$d%dd&j,effgd'd( d)d*d+ Z1ej*j+d,ed d!d"gj,e /d"d#¡ej2ffej.d$d%dd&j,e -d"d#¡ej2ffej0d$d%dd&j,e /d"d#¡e -d"d#¡ej2ffgd-d( d)d.d/ Z3ej* +d0ej.d1d2d3jej4d1d2d3jg¡d4d5 Z5ej* +d6d7d8g¡ej* +d9e 6d:d;g¡ed<d=gdd>ed:d;gdd?g¡ej* +d@e7ejejej8ej9ej:ej;g¡dAdB Z<ej* +d9e 6d:d;g¡ed<d=gdd>ed:d;gdd?g¡ej* +dCd<d=gdDdEge
|
d:dFdGgd= g¡dHdI Z=ej* +d@e7e>ejej8g¡dJdK Z?ej* +dLdMdNg¡dOdP Z@ej!dQdRejAdSe BdT¡dUejAdVe BdV¡dUgddWdX ZCej* +dLdMdNg¡dYdZ ZDdS )[é )Ú annotationsN)ÚNaTÚOutOfBoundsDatetimeÚ Timestamp)Ú DatetimeIndexÚPeriodÚ PeriodIndexÚTimedeltaIndex)Ú DatetimeArrayÚ PandasArrayÚ PeriodArrayÚTimedeltaArray)Ú_sequence_to_dt64ns)Úsequence_to_td64nsÚDÚBÚWÚMÚQÚY)Úparamsc C s | j S )z:Fixture returning parametrized frequency in string format.)Úparam)Úrequest© r ú\d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/arrays/test_datetimelike.pyÚfreqstr# s r c C s t jtdd| d}|S )zò
|
A fixture to provide PeriodIndex objects with different frequencies.
|
|
Most PeriodArray behavior is already tested in PeriodIndex tests,
|
so here we just test that the PeriodArray behavior matches
|
the PeriodIndex behavior.
|
ú
|
2000-01-01éd ©ÚstartÚperiodsÚfreq)ÚpdÚ period_ranger )r Úpir r r Ú period_index) s
|
r% c C s t jtdd| d}|S )zü
|
A fixture to provide DatetimeIndex objects with different frequencies.
|
|
Most DatetimeArray behavior is already tested in DatetimeIndex tests,
|
so here we just test that the DatetimeArray behavior matches
|
the DatetimeIndex behavior.
|
r r r )r" Ú
|
date_ranger )r Údtir r r Údatetime_index7 s
|
r( c C s t dddgS )a
|
A fixture to provide TimedeltaIndex objects with different frequencies.
|
Most TimedeltaArray behavior is already tested in TimedeltaIndex tests,
|
so here we just test that the TimedeltaArray behavior matches
|
the TimedeltaIndex behavior.
|
ú1 Dayú3 Hoursr )r r r r r Útimedelta_indexE s r+ c @ s e Zd ZU ded< ejdd Zdd Zej de
|
j dd d
|
e
|
j dd d d e
|
j d d d
|
e
|
j dd ddg¡dd Zej dddg¡ej dddg¡dd Zdd Zej dd d ed d d d jg¡d!d" Zd#d$ Zd%d& Zd'd( Zd)d* Zd+d, Zd-d. Zd/d0 Zej d1d2d3g¡d4d5 Zd6d7 Z ej d8d9d:d;g¡d<d= Z d>d? Z d@dA Z dBdC Z dDdE Z!dFdG Z"ej d8e
|
j#e
|
j$e%j&e'e(g¡dHdI Z)dJdK Z*ej dddg¡dLdM Z+dNdO Z,ej d8e'e%j&e
|
j#e
|
j$g¡dPdQ Z-dRdS Z.dTdU Z/dVdW Z0dXdY Z1d9S )ZÚ SharedTestsz2type[DatetimeIndex | PeriodIndex | TimedeltaIndex]Ú index_clsc C s, t jdddd d d }| j|dd}|S ) z5Fixture returning DatetimeArray with daily frequency.é
|
Úi8©Údtypeé é é Ê;r ©r! )ÚnpÚarangeÚ array_cls©ÚselfÚdataÚarrr r r Úarr1dT s zSharedTests.arr1dc C sd |}| |¡}tjtdd ||d d
k W 5 Q R X tjtdd ||dg k W 5 Q R X d S )NzLengths must match©Úmatché r )r- ÚpytestÚraisesÚ
|
ValueError)r: r= r< Úidxr r r Útest_compare_len1_raises[ s
|
z$SharedTests.test_compare_len1_raisesÚresultZ2020é ©r ÚUTC)r Útzz0 daysÚ2020Q1r ©r r! c C s. t |¡}t||kstt||kr*td S ©N)r" Ú CategoricalÚallÚAssertionErrorÚany)r: rF Úexpectedr r r Ú test_compare_with_Categoricalh s
|
|
z)SharedTests.test_compare_with_CategoricalÚreverseTFÚas_indexc C sÎ t j||d}|r t |¡}|| }}|r4|| }}tj|jtd}| } ||k}
|
t |
|
|¡ ||k}
|
t |
|
| ¡ |sÊ|sÊ||k }
|
t |
|
| ¡ ||k}
|
t |
|
|¡ ||k}
|
t |
|
| ¡ ||k}
|
t |
|
|¡ d S )N)Úorderedr0 ) r" rN ÚCategoricalIndexr6 ÚonesÚshapeÚboolÚtmÚassert_numpy_array_equal) r: r= rU rT rV ÚotherÚleftÚrightrX ÚzerosrF r r r Ú test_compare_categorical_dtypev s*
|
|
|
z*SharedTests.test_compare_categorical_dtypec C s¼ t jdddd d d }t j |¡ | jtk r4d nd}| j||d}| j |¡}d d
|
d g}| |¡}| |¡}t
|
| |¡|¡ t d d
|
d g¡}| |¡}| |¡}t
|
| |¡|¡ d S ) Nr r/ r0 r2 r3 r4 r r5 r@ é é^ ) r6 r7 ÚrandomÚshuffler8 r r- Ú _simple_newÚtaker[ Úassert_index_equalÚarray)r: r; r! r< rD ZtakersrF rR r r r Ú test_take s
|
|
|
|
|
zSharedTests.test_takeÚ
|
fill_valueé g @iå r@ é c C sj t jdddd d d }| j|dd}d |jj d
|
}tjt|d |jd d gd|d W 5 Q R X d S )Nr. r/ r0 r2 r3 r4 r r5 úvalue should be a 'ú' or 'NaT'. Gotr> r r@ T©Z
|
allow_fillrk ) r6 r7 r8 Ú _scalar_typeÚ__name__rA rB Ú TypeErrorrg )r: rk r; r< Úmsgr r r Útest_take_fill_raises® s
|
z!SharedTests.test_take_fill_raisesc C s t jdddd d d }| j|dd}|jd d
|
gd d d }|d tksLt|jd d
|
gd t jd }|d tksrt|jd d
|
gd td }|d tkstd S )Nr. r/ r0 r2 r3 r4 r r5 éÿÿÿÿr@ Trp r )r6 r7 r8 rg r rP Únan©r: r; r< rF r r r Útest_take_fill¸ s zSharedTests.test_take_fillc C sv |j ddgdt|d d}|ddg }t ||¡ d|jj d}tjt|d |j ddgddd W 5 Q R X d S ) Nrv r@ Trp rn ro r> Úfoo) rg Ústrr[ Ú assert_equalrq rr rA rB rs )r: r= rF rR rt r r r Útest_take_fill_strÆ s z SharedTests.test_take_fill_strc C s |}| |¡}| dt¡}| |¡}| |d d
|dd
|g¡}| t¡}| t |d d
|dd
|g¡d ¡}t
|
| |¡|¡ d S )Nr rv r@ ) r- Úinsertr r8 Ú_concat_same_typeÚastypeÚobjectr6 Z concatenater[ rh )r: r= r< rD rF Úarr2rR r r r Útest_concat_same_typeÐ s
|
|
|
(z!SharedTests.test_concat_same_typec C s t jdddd d d }| j|dd}| |d ¡}|jjj}t||sNt| t ¡}t||sftd
|
| j
|
j d }t j t|d | d ¡ W 5 Q R X d S )Nr. r/ r0 r2 r3 r4 r r5 r z'value' should be a Ú.r> rz )r6 r7 r8 Z _unbox_scalarÚ_ndarrayr1 ÚtypeÚ
|
isinstancerP r Ú scalar_typerr rA rB rC )r: r; r< rF rR rt r r r Útest_unbox_scalarÜ s
|
|
z SharedTests.test_unbox_scalarc C sV t jdddd d d }| j|dd}| |d ¡ | |d d
|
¡ | t¡ d S ) Nr. r/ r0 r2 r3 r4 r r5 r r@ )r6 r7 r8 Z_check_compatible_withr r9 r r r Útest_check_compatible_withê s
|
z&SharedTests.test_check_compatible_withc C sN t jdddd d d }| j|dd}| t|d ¡}||d ksJtd S )
|
Nr. r/ r0 r2 r3 r4 r r5 r )r6 r7 r8 Z_scalar_from_stringr{ rP rx r r r Útest_scalar_from_stringò s z#SharedTests.test_scalar_from_stringc C sT t jdddd d d }| j|dd}d }tjt|d
|
| d ¡ W 5 Q R X d S ) Nr. r/ r0 r2 r3 r4 r r5 z)does not support reduction 'not a method'r> z not a method)r6 r7 r8 rA rB rs Z_reduce)r: r; r< rt r r r Útest_reduce_invalidø s
|
z SharedTests.test_reduce_invalidÚmethodÚpadZbackfillc C sx t jdddd d d }| j|dd}t|d < |d
|
kr@|d n|d }|j|d }|d |ksdt|d tksttd S )Nr. r/ r0 r2 r3 r4 r r5 rb r rG é )r )r6 r7 r8 r ZfillnarP )r: r r; r< rk rF r r r Ú%test_fillna_method_doesnt_change_orig s z1SharedTests.test_fillna_method_doesnt_change_origc C sâ t jdddd d d }| j|dd}| |d ¡}|d ksBt|j|d
|
d d }|d ks`t| |d d
¡}t jd d
|
gt jd}t ||¡ |j|d d
d d }t jd
|
d gt jd}t ||¡ | t ¡}|dksÞtd S )Nr. r/ r0 r2 r3 r4 r r5 r@ rl r_ ©ZsiderG )
|
r6 r7 r8 Ú searchsortedrP ri Úintpr[ r\ r )r: r; r< rF rR r r r Útest_searchsorted s
|
z SharedTests.test_searchsortedÚboxNÚindexZseriesc
|
C sR |}|d krn |dkr"| |¡}n
|
t |¡}| t|d ¡}|dksJt|jt|d dd}|dkslt| dd |dd
D ¡}tjddgtjd }t
|
||¡ t j t t d
|
|jj d ¡d | d ¡ W 5 Q R X |dkrêdnd}t d|¡N t j t t d
|
|jj d| d¡d | t|d dg¡ W 5 Q R X W 5 Q R X d S )Nr r@ rl r_ r rG c S s g | ] }t |qS r ©r{ ©Ú.0Úxr r r Ú
|
<listcomp>: s zBSharedTests.test_searchsorted_castable_strings.<locals>.<listcomp>r0 rn z/', 'NaT', or array of those. Got 'str' instead.r> rz ÚpythonZ StringArrayZArrowStringArrayÚstring_storagez"', 'NaT', or array of those. Got 'z
|
' instead.Zbaz)r- r" ÚSeriesr r{ rP r6 ri r r[ r\ rA rB rs ÚreÚescaperq rr Zoption_context)r: r= r r r< rF rR Zarr_typer r r Ú"test_searchsorted_castable_strings' s:
|
ÿþÿþz.SharedTests.test_searchsorted_castable_stringsc C s\ t jdd tddD dd}| j|dd}|d t |¡}|d t |¡}|d d S )
|
Nc S s g | ]}t j| qS r )r Ú_value)r Únr r r r V s zGSharedTests.test_getitem_near_implementation_bounds.<locals>.<listcomp>r@ r r/ r0 Únsr5 r )r6 ÚasarrayÚranger8 r" ÚIndexr )r: Zi8valsr< r Zserr r r Ú'test_getitem_near_implementation_boundsS s
|
|
z3SharedTests.test_getitem_near_implementation_boundsc C s t ||jd d
tjf |jd}|d d
tjf }t ||¡ |}t ||jd d
df |jd}|d d
df }t ||¡ |d }|d }||kstd S )Nr0 rG r )rv r rv )r r
r6 Únewaxisr1 r[ r| rP )r: r= rR rF Úarr2dr r r Útest_getitem_2d` s " zSharedTests.test_getitem_2dc C s~ |j d d
tjf }t|j||jd}t|}t|dks@t|D ]4}t |t|sZt|j
|
dksht|j|jksDtqDd S )NrG r0 r@ ) r
r6 r© r rf r1 ÚlistÚlenrP r Úndim)r: r= Údata2drª rF r r r r Ú test_iter_2dq s zSharedTests.test_iter_2dc
|
C sÄ |j d d
tjf }t|j||jd}t|}t|tr~dt|j d|d
|
¡ d|d
|
¡ d|d
|
¡ d |j
|
}n6dt|j d|d d|d d|d d |j
|
}||ksÀt d S )
|
NrG r0 ú<z>
|
[
|
['r z'],
|
['r@ rl z']
|
]
|
Shape: (3, 1), dtype: ) r
r6 r© r rf r1 Úreprr r rr Z
|
_repr_baserP )r: r= r¯ rª rF rR r r r Ú test_repr_2d{ s
|
@ÿ 4ÿ
|
zSharedTests.test_repr_2dc C s¢ t jdddd d d }| j|dd}|d |d
|
< t jdddd d d }|d |d
|
< t |j|¡ |d d
|d d
< |d d
|d d
< t |j|¡ d S ) Nr. r/ r0 r2 r3 r4 r r5 r@ r éþÿÿÿrl )r6 r7 r8 r[ r\ Úasi8©r: r; r< rR r r r Ú test_setitem s zSharedTests.test_setitemc C s | ¡ d d d
}|jjdkr(| d ¡}|}|tkr>t|}nF|tjkrZt | t¡¡}n*|t krv|tj
|
|td}n|| t¡}||d d
< t ||¡ d S )Nrv )Úmr r0 ) Úcopyr1 ÚkindÚ
|
_with_freqr¬ r6 ri r r r r¥ r[ r| )r: r r= rR Úvalsr r r Útest_setitem_object_dtype¦ s
|
|
|
z%SharedTests.test_setitem_object_dtypec C s | ¡ }|dd
|ddg< | ¡ }dd |dd
D |d d
< t ||¡ | ¡ }|d |d< | ¡ }t|d |d< t ||¡ d S )Nr´ r r@ c S s g | ] }t |qS r r r r r r r Ì s z1SharedTests.test_setitem_strs.<locals>.<listcomp>rl rv )r¹ r[ r| r{ )r: r= rR rF r r r Útest_setitem_strsÄ s z SharedTests.test_setitem_strsc C sd | ¡ d d d
}t|ts&| d ¡}t |¡}|r>t |¡}|d d d
|d d
< t ||¡ d S )Nrv ) r¹ r r r» r" rN rW r[ r| )r: r= rU rR Úcatr r r Útest_setitem_categorical× s
|
|
|
|
z$SharedTests.test_setitem_categoricalc C sØ t jdddd d d }| j|dd}|d }tjtd
|
d ||d < W 5 Q R X tjtd d t |d < W 5 Q R X d}tjt|d |d g|g < W 5 Q R X d}tjt|d |d d
|dd
< W 5 Q R X d S )Nr. r/ r0 r2 r3 r4 r r5 r zindex 12 is out of boundsr> rm z value should be a.* 'object'z<cannot set using a list-like indexer with a different lengthr@ z=cannot set using a slice indexer with a different length thanrG ) r6 r7 r8 rA rB Ú
|
IndexErrorrs r rC )r: r; r< Úvalrt r r r Útest_setitem_raiseså s z SharedTests.test_setitem_raisesc C sp d|j j d}tjt|d |ddg|d d
< W 5 Q R X tjt|d |ddg|d d
< W 5 Q R X d S ) Nrn z ', 'NaT', or array of those. Gotr> r r@ rl g g ð?)rq rr rA rB rs )r: r= r rt r r r Útest_setitem_numeric_raisesú s ÿ z'SharedTests.test_setitem_numeric_raisesc C s t jdddd d d }| j|dd}|tjd d
|
}|tjd d
|
7 }t ||¡ |tjd d
|
}|tjd d
|
8 }t ||¡ d S ) Nr. r/ r0 r2 r3 r4 r r5 r@ ©Údays)r6 r7 r8 r" Ú Timedeltar[ r| r¶ r r r Útest_inplace_arithmetic s z#SharedTests.test_inplace_arithmeticc C sT t jdddd d d }| j|dd}tjtd d
|
|jd d d W 5 Q R X d S ) Nr. r/ r0 r2 r3 r4 r r5 zvalue should be ar> r@ )rk )r6 r7 r8 rA rB rs Úshiftr9 r r r Ú test_shift_fill_int_deprecated s z*SharedTests.test_shift_fill_int_deprecatedc C sô |}t |d dkr |d d
}|t |d }| ¡ }t|t|ksLt||ksXtt|t |d < t|ts|t |d d t |d d
¡ }|jddtks®t| ¡ }t|t|ksÊt||ksÖt|d d
¡ tksît|d d
jddtks t| dd¡}|jd d}t|t|ks:t||ksHt|jd ddtks`t|jdd}t|j |g|j
|
d }t ||¡ |jddd}t|j tg|j
|
d }t ||¡ |jdd}t ||¡ |jddd}t ||¡ d S )
|
Nrl r rv r@ F)Úskipna)Úaxis)rÌ rË r0 ) r Zmedianr rP r r r ZmeanÚreshapeÚ_from_sequencer1 r[ r| )r: r= r< rR rF r Z expected2r r r Ú test_median s>
|
( zSharedTests.test_medianc C sT t jdddgt jd}| j|| jd}tj|dd}| j|| jd}t ||¡ d S )Nr@ rl rG r0 ZInt64)r6 ri Úint64r8 Ú example_dtyper" r[ Ú assert_extension_array_equal)r: r< rR r; rF r r r Útest_from_integer_arrayM s
|
z#SharedTests.test_from_integer_array)2rr Ú
|
__module__Ú __qualname__Ú__annotations__rA Úfixturer= rE ÚmarkÚ parametrizer" r& Útimedelta_ranger# rS ra rj r Útimeru ry r} r r r r r r r r¡ r¨ r« r° r³ r· r§ r r6 ri r¬ r r½ r¾ rÀ rà rÄ rÈ rÊ rÏ rÓ r r r r r, Q sp
|
|
üþ
|
!
|
|
|
|
+
|
ûþ
|
|
|
|
0r, c @ sÒ e Zd ZeZeZeZdZ e
|
j dd Z dd Z dd Zdd Zd
|
d Zd d Zdd Zdd Zdd Zdd Zdd Ze
|
j dej¡dd Ze
|
j dej¡dd Z d d Z d d Z d!d" Z d#d$ Z d%d& Z!d'S )(ÚTestDatetimeArrayúM8[ns]c C s |}t jdd||d}|j}|S )zc
|
Fixture returning DatetimeArray with parametrized frequency and
|
timezones
|
z2016-01-01 01:01:00r ©r r! rJ )r" r& Ú_data)r: Ztz_naive_fixturer rJ r' Údtar r r r= ] s zTestDatetimeArray.arr1dc C sj | |¡}|jdd}|tjdd }| d ¡}t ||¡ |j}|jdd}|j d ¡}t ||¡ d S )NZ2Tr5 r@ )Úminutes) r- Úroundr" rÇ r» r[ rh rß Úassert_datetime_array_equal)r: r= r' rF rR rà r r r Ú
|
test_roundh s
|
|
z TestDatetimeArray.test_roundc C s t |}t |¡}|j}||ks$tt ||¡ tj|dd}||ksJtt ||¡ tj|dd}|j}||ksvtt ||¡ tj|ddd}||kstt ||¡ tj|dd}||k sÄtt ||¡ tj|td}tjt |td}t ||¡ tj|dd}||j
|
k stt ||¡r,t|j
|
¡ }t ||¡ dt fD ]0}tj||d}t |¡ |¡}t ||¡ qJd S )NF©r¹ údatetime64[ns]r0 ©r1 r¹ rÐ Úfloat64)r
|
r6 r¥ r
rP r[ r\ ri r r¬ rµ Úmay_share_memoryr¹ r{ r )r: r( r< rF rR r1 r r r Útest_array_interfacev s<
|
|
z&TestDatetimeArray.test_array_interfacec C sT |}| |¡}t t|¡}tj|td}t ||¡ tj|td}t ||¡ d S ©Nr0 )r- r6 ri r¬ r r[ r\ ©r: r= r< r' rR rF r r r Útest_array_object_dtype s
|
z)TestDatetimeArray.test_array_object_dtypec C s® |}| |¡}|j d¡}tj|dd}t ||¡ tj|dd}t ||¡ tj|ddd}|j|jksnt|jd k s|ttj|ddd}|j|jkst|jd k sªtd S )NrÝ r0 ræ Frç ) r- rµ Úviewr6 ri r[ r\ ÚbaserP rì r r r Ú test_array_tz® s
|
z TestDatetimeArray.test_array_tzc C s| |}| |¡}|j}tj|dd}t ||¡ tj|tjd}t ||¡ tj|ddd}|j|jk sjt|jd ksxtd S )Nr/ r0 Frç ) r- rµ r6 ri r[ r\ rÐ rï rP rì r r r Útest_array_i8_dtype s
|
z%TestDatetimeArray.test_array_i8_dtypec C sL t jddgdd}t|}|j|ks(tt|d d
}|jj|ksHtd S )Nr z
|
2000-01-02rÝ r0 r )r6 ri r
|
r
rP rï )r: r< rà r r r Útest_from_array_keeps_baseÒ s
|
z,TestDatetimeArray.test_from_array_keeps_basec C sR |}| |¡}t|t|ks"tt |¡}t|ts:tt|t|ksNtd S rM )r- r¬ rP r" r§ r r )r: r= r< r' Zdti2r r r Ú test_from_dtiÛ s
|
|
z TestDatetimeArray.test_from_dtic C sN |}| |¡}| d¡}t|tjs(t|jdks6tt|t|ksJtd S ©NÚO©r- r r r6 ÚndarrayrP r1 r¬ )r: r= r< r' Úasobjr r r Útest_astype_objectå s
|
|
z$TestDatetimeArray.test_astype_objectc C sN |}t |}|j|d}|j|d}t|ts2tt t |¡t |¡¡ d S )Nr5 ) r
|
Ú to_periodr r rP r[ rh r" r§ )r: r( r r' r< rR rF r r r Útest_to_periodî s z TestDatetimeArray.test_to_periodc C s` | dd¡}|jd krd nt}t |¡" | d¡}| d¡ dd¡}W 5 Q R X t ||¡ d S )Nr@ rv r )rÍ rJ Ú UserWarningr[ Zassert_produces_warningrú Zassert_period_array_equal)r: r= rª ÚwarnrF rR r r r Útest_to_period_2dú s
|
z#TestDatetimeArray.test_to_period_2dÚpropnamec C sN | |¡}|}|j|jks tt||}tjt|||jd}t ||¡ d S rë ) r- r! rP Úgetattrr6 ri r1 r[ r\ ©r: r= rÿ r' r< rF rR r r r Útest_bool_properties s
|
|
z&TestDatetimeArray.test_bool_propertiesc C s> | |¡}|}t||}tjt|||jd}t ||¡ d S rë )r- r r6 ri r1 r[ r\ r r r r Útest_int_properties s
|
|
|
z%TestDatetimeArray.test_int_propertiesc C sÖ |}| |¡}| |j¡}|jddgd|d}|d |ks>td|jj d}tjt |d |jddgd|| d W 5 Q R X tjt |d |jddgdt
|
d d W 5 Q R X |jd k rÀd nd
|
}| |¡}d }tjt |d |jddgd|d W 5 Q R X t j } d|jj d}tjt |d |jddgd| d W 5 Q R X t d d ¡} tjt |d |jddgd| d W 5 Q R X |jd k rÒ| d¡} |jddgd| d}|jddgd| |jj¡d}
|
t ||
|
¡ d S )Nrv r@ Trp r rn ro r> Z2014Q1z
|
US/Easternz:Cannot compare tz-naive and tz-aware datetime-like objectsr r¤ zAustralia/Melbourne)r- Ú tz_localizerJ rg rP rq rr rA rB rs r r r¢ r6 Ú timedelta64Z
|
tz_convertr1 r[ r| ) r: r= Ú fixed_now_tsr< r' ÚnowrF rt rJ ÚvaluerR r r r Útest_take_fill_valid s>
|
""
|
|
ýz&TestDatetimeArray.test_take_fill_validc C sP |}|j d kr| d¡}n
|
| d ¡}tjtdd | ||g¡ W 5 Q R X d S )NrI z to_concat must have the samer> )rJ r rA rB rC r )r: r= r< r] r r r Ú test_concat_same_type_invalidI s
|
|
z/TestDatetimeArray.test_concat_same_type_invalidc C sf t tjddddd}t tjddddd}t ||g¡}t t dddd g¡ d¡}t ||¡ d S )
|
NÚ2000rl r z
|
US/CentralrÞ ÚHz2000-01-01 00:00:00z2000-01-02 00:00:00z2000-01-01 01:00:00)r
|
r" r& r Ú to_datetimer r[ rã )r: ÚaÚbrF rR r r r Ú$test_concat_same_type_different_freqU s üÿùÿ z6TestDatetimeArray.test_concat_same_type_different_freqc C s6 |}| d¡}tjdd |D td}t ||¡ d S )Nú%Y %bc S s g | ]}| d ¡qS )r ©Ústrftime)r Útsr r r r k s z3TestDatetimeArray.test_strftime.<locals>.<listcomp>r0 ©r r6 ri r r[ r\ ©r: r= r< rF rR r r r Ú test_strftimeg s
|
z TestDatetimeArray.test_strftimec C s> t tdtg}| d¡}tjdtjgtd}t ||¡ d S )Nú
|
2019-01-01ú%Y-%m-%dr0 )
|
r
|
r r r r6 ri rw r r[ r\ ©r: r< rF rR r r r Útest_strftime_natn s
|
z#TestDatetimeArray.test_strftime_natN)"rr rÔ rÕ r r- r
|
r8 r r rÑ rA r× r= rä rê rí rð rñ rò ró rù rû rþ rØ rÙ Ú _bool_opsr Ú
|
_field_opsr r r
|
r r r r r r r rÜ W s2
|
|
*
|
|
|
0 rÜ c @ sf e Zd ZeZeZejZ dZ
|
dd Z dd Z dd Z dd Zej d
|
ej¡d d Zd d Zdd ZdS )ÚTestTimedeltaArrayúm8[ns]c C sX t ddg}t|}t|t|ks(tt |¡}t|t s@tt|t|ksTtd S )Nr) r* )r r r¬ rP r" r§ r )r: Útdir< Ztdi2r r r Ú test_from_tdi} s
|
z TestTimedeltaArray.test_from_tdic C sT t ddg}t|}| d¡}t|tjs.t|jdks<tt|t|ksPtd S )Nr) r* rõ ) r r r r r6 r÷ rP r1 r¬ )r: r r< rø r r r rù s
|
z%TestTimedeltaArray.test_astype_objectc C s, |}t |}| ¡ }| ¡ }t ||¡ d S rM )r Zto_pytimedeltar[ r\ ©r: r+ r r< rR rF r r r Útest_to_pytimedelta s
|
z&TestTimedeltaArray.test_to_pytimedeltac C s. |}t |}| ¡ }| ¡ }t ||j¡ d S rM )r Ú total_secondsr[ r\ Úvaluesr" r r r Útest_total_seconds s
|
z%TestTimedeltaArray.test_total_secondsrÿ c C s< |}t |}t||}tjt|||jd}t ||¡ d S rë )r r r6 ri r1 r[ r\ )r: r+ rÿ r r< rF rR r r r r ¡ s
|
|
z&TestTimedeltaArray.test_int_propertiesc C s t |}t |¡}|j}||ks$tt ||¡ tj|dd}||ksJtt ||¡ tj|dd}|j}||ksvtt ||¡ tj|ddd}||kstt ||¡ tj|dd}||k sÄtt ||¡ tj|td}tjt |td}t ||¡ tj|dd}||j
|
k stt ||¡r,t|j
|
¡ }t ||¡ dt fD ]0}tj||d}t |¡ |¡}t ||¡ qJd S )NFrå ztimedelta64[ns]r0 rç rÐ rè )r r6 r¥ r
rP r[ r\ ri r r¬ rµ ré r¹ r{ r )r: r+ r< rF rR r1 r r r rê « s<
|
|
z'TestTimedeltaArray.test_array_interfacec C sô |}t |}tjdd}|jddgd|d}|d |ks<t|}d|jj d}tjt |d |jddgd|d W 5 Q R X |
|
d
|
¡}tjt |d |jddgd|d W 5 Q R X t d d ¡}tjt |d |jddgd|d W 5 Q R X d S ) Nr@ rÅ rv Trp r rn ro r> r r r¤ ) r r" rÇ rg rP rq rr rA rB rs rú r6 Ú
|
datetime64) r: r+ r r r< Ztd1rF r rt r r r r Õ s
|
z'TestTimedeltaArray.test_take_fill_validN)rr rÔ rÕ r r- r r8 r" rÇ r rÑ r! rù r# r& rA rØ rÙ r r rê r r r r r r w s
|
|
*r c @ s¾ e Zd ZeZeZeZeg ddj Z
|
e j dd Z dd Zdd Zd d
|
Ze j d d d g¡dd Zdd Zdd Ze j dej¡dd Ze j dej¡dd Zdd Zdd Zd d Z d S ) ÚTestPeriodArrayr r5 c C s |j S )zW
|
Fixture returning DatetimeArray from parametrized PeriodIndex objects
|
)rß )r: r% r r r r= ô s zTestPeriodArray.arr1dc C sR | |¡}|}t|t|ks"tt |¡}t|ts:tt|t|ksNtd S rM )r- r¬ rP r" r§ r r )r: r= r$ r< Zpi2r r r Ú test_from_piû s
|
|
z TestPeriodArray.test_from_pic C sN | |¡}|}| d¡}t|tjs(t|jdks6tt|t|ksJtd S rô rö )r: r= r$ r< rø r r r rù s
|
|
z"TestPeriodArray.test_astype_objectc C s |}t j}d|jj d}tjt|d |jddgd|d W 5 Q R X t dd ¡}tjt|d |jddgd|d W 5 Q R X d S )
|
Nrn ro r> rv r@ Trp r r¤ )
|
r r¢ rq rr rA rB rs rg r6 r )r: r= r< r rt r r r r s z$TestPeriodArray.test_take_fill_validÚhowÚSÚEc C sT | |¡}|}t|j|d}|j|d}t|ts8tt t |¡t |¡¡ d S )N)r* ) r- r
|
Ú to_timestampr rP r[ rh r" r§ )r: r* r= r$ r< rR rF r r r Útest_to_timestamp s
|
z!TestPeriodArray.test_to_timestampc C s¤ t jddddj}| ¡ }| ¡ }|jdks0tt ||¡ |d d d
}| ¡ }| ¡ }|jdkshtt ||¡ | d¡}| ¡ }|jdkstt ||¡ d S )Nz
|
2021-10-18rG r rL rl Z2B) r" r& rß rú r- r! rP r[ rÒ ) r: rà ZparrrF Zdta2Zparr2Zresult2Zparr3Zresult3r r r Ú test_to_timestamp_roundtrip_bday( s
|
z0TestPeriodArray.test_to_timestamp_roundtrip_bdayc C s^ t jdddd}d}tjt|d | ¡ W 5 Q R X tjt|d |j ¡ W 5 Q R X d S )NZ1500r rG )r! r z7Out of bounds nanosecond timestamp: 1500-01-01 00:00:00r> )r" r# rA rB r r- rß )r: r$ rt r r r Ú test_to_timestamp_out_of_bounds; s z/TestPeriodArray.test_to_timestamp_out_of_boundsrÿ c C s8 | |¡}|}t||}t t||¡}t ||¡ d S rM ©r- r r6 ri r[ r\ ©r: r= rÿ r$ r< rF rR r r r r E s
|
|
|
z$TestPeriodArray.test_bool_propertiesc C s8 | |¡}|}t||}t t||¡}t ||¡ d S rM r1 r2 r r r r P s
|
|
|
z#TestPeriodArray.test_int_propertiesc C s¼ |}t |¡}t jt|td}t ||¡ t j|td}t ||¡ t j|dd}t ||j¡ d}tj t
|
|d t j|dd W 5 Q R X t j|dd}t |¡ d¡}t ||¡ d S )Nr0 rÐ zEfloat\(\) argument must be a string or a( real)? number, not 'Period'r> rè ZS20) r6 r¥ ri r¬ r r[ r\ rµ rA rB rs r )r: r= r< rF rR rt r r r rê Z s
|
z$TestPeriodArray.test_array_interfacec C s6 |}| d¡}tjdd |D td}t ||¡ d S )Nú%Yc S s g | ]}| d ¡qS )r3 r )r Zperr r r r v s z1TestPeriodArray.test_strftime.<locals>.<listcomp>r0 r r r r r r r s
|
z TestPeriodArray.test_strftimec C sB t tdtgdd}| d¡}tjdtjgtd}t ||¡ d S )Nr z period[D]r0 r )
|
r r r r r6 ri rw r r[ r\ r r r r r y s
|
z!TestPeriodArray.test_strftime_natN) rr rÔ rÕ r r- r r8 r r r1 rÑ rA r× r= r) rù r rØ rÙ r. r/ r0 r r r r rê r r r r r r r( î s(
|
|
|
|
|
|
|
r( zarr,casting_natsr) r* r r¤ r rG rL c C s
|
t | jS rM ©r rr ©r r r r Ú<lambda> ó r6 )Úidsc C sF t | t| d | d g¡}|D ] }| ¡ } || d< t | |¡ q d S )Nr@ rl r )r rÎ r r¹ r[ r| )r< Z casting_natsrR Únatr r r Ú test_casting_nat_setitem_array s
|
r: zarr,non_casting_natsc C s
|
t | jS rM r4 r5 r r r r6 ª r7 c
|
C s4 d}|D ]&}t jt|d || d< W 5 Q R X qd S )Nzwvalue should be a '(Timestamp|Timedelta|Period)', 'NaT', or array of those. Got '(timedelta64|datetime64|int)' instead.r> r )rA rB rs )r< Znon_casting_natsrt r9 r r r Ú test_invalid_nat_setitem_array s
|
ÿr; r< r rb rH c C sÈ t | d< | ¡ }| ¡ }t |d ¡s*t| jdd}|d dksFt| jddd}|d dksdt| j| d ¡ d}|d |d kst| j| d jdd d}|d |d ks¸tt | |¡ d S )
|
Nr rÐ r0 lûÿÿÿ )r1 Úna_valuer@ )r< Frå )r r¹ Zto_numpyr6 ÚisnanrP r[ r| )r< ÚoriginalrF r r r Útest_to_numpy_extra· s r? rU TFr% z
|
2020-01-01z
|
2020-02-01r@ rl )Úunitr5 Úklassc C s< |s
|
| j } | || ¡}tjddg|jd}t ||¡ d S )Nr r@ r0 )rß r r6 ri r1 r[ r\ )r% rA rU rF rR r r r Ú,test_searchsorted_datetimelike_with_listlikeÔ s
|
rB Úargr r z Europe/London)rJ c C s, d}t jt|d | |¡ W 5 Q R X d S )Nz [Unexpected type|Cannot compare]r> )rA rB rs r )r% rC rt r r r Ú:test_searchsorted_datetimelike_with_listlike_invalid_dtypeô s rD c C sB ddgd }| |}t |dd}t dd |D }t ||¡ d S )NrK Z2020Q2rl r r5 c S s g | ] }t |qS r )r )r Úsr r r r s z?test_period_index_construction_from_strings.<locals>.<listcomp>)r r[ rh )rA Ústringsr; rF rR r r r Ú+test_period_index_construction_from_strings s
|
rG r1 rÝ r c C sø t jdddg| d}t|}ttd| }||}||}t ||¡ | |¡}| |¡}t ||¡ tt d| }||d }||d }t
|
||¡ t j t j d| }||j}||j}t
|
||¡ ttd| }||}||}t ||¡ d S )Nr@ rl rG r0 ©rÝ r r )r6 ri r r
|
r r[ rÒ rÎ r r r| r" r Ú to_timedeltar r rh )r1 r; r< ÚclsrF rR ÚfuncÚidx_clsr r r Útest_from_pandas_array s*
|
|
|
|
rM Ú
|
memoryviewri Údaskz
|
dask.array)ZmarksÚxarrayc C s t jdddgt jd}| j}|dkr.t|}nN|dkrDt d|¡}n8|dkrbd d
|
l}|j |¡}n|d kr|d d
|
l}| |¡}||fS ) z¢
|
Fixture giving a numpy array and a parametrized 'data' object, which can
|
be a memoryview, array, dask or xarray object created from the numpy array.
|
r@ rl rG r0 rN ri ÚirO r NrP )r6 ri rÐ r rN Z
|
dask.arrayrP Z DataArray)r r< Únamer; rO Zxrr r r Ú array_likes1 s
|
|
rS c C sÐ |\}}t td| }||}| |¡}t ||¡ ttd| }||d }||d }t ||¡ t|t s¢t
|
j t
|
j d| }||j }||j }t ||¡ ttd| }||}||}t ||¡ d S )NrH r )r
|
r rÎ r[ rÒ r r r| r rN r" r rI ri r r rh ) r1 rS r< r; rJ rR rF rK rL r r r Útest_from_obscure_arrayR s$
|
|
|
|
rT )EÚ
|
__future__r ri r Únumpyr6 rA Z pandas._libsr r r Z pandas.util._test_decoratorsÚutilZ_test_decoratorsÚtdZpandasr" r r r r Zpandas._testingZ_testingr[ Zpandas.core.arraysr
|
r r r Z pandas.core.arrays.datetimesr Z pandas.core.arrays.timedeltasr r× r r% r( r+ r, rÜ r r( rØ rÙ rß r r& r' r# r: r¢ r; rÚ r? r r¬ r r§ rN rW rB rD ÚtuplerG rM r Z
|
skip_if_norS rT r r r r Ú<module> sà
|
|
|
|
|
"w þþ÷ ó
|
þþþ÷ð
|
þþ
|
ýþùþ ýþ ÿ
|
|
üÿ
|
|
