U
|
®ý°d* ã @ s~ d dl mZmZ d dlZd dlZd dlmZ d dlZ d dlm
|
Z
|
m Z m Z m Z d dlmZ d dlmZ G dd dZdd ZdS )
|
é )ÚtimeÚ timedeltaN)ÚOutOfBoundsTimedelta)ÚSeriesÚTimedeltaIndexÚisnaÚ to_timedelta)ÚTimedeltaArrayc @ sü e Zd Zej dddg¡dd Zdd Zdd Zd
|
d Z d d Z
|
ej dddgddgddgddgddgg¡dd Z dd Z ej de d¡ d d ¡e e d¡ d d ¡¡g¡ej d d d d!g¡d"d# Zd$d% Zej dd&d gd&g¡d'd( Zd)d* Zd+d, Zd-d. Zd/d0 Zej d1d2d3d4d5d6d7d8d9d:d;g
|
¡d<d= Zd>d? Zd@dA ZdBdC Z ej dDe j ej g¡dEdF Z dGdH Z dIdJ Z!dKdL Z"ej dMe#d dNd gdOdOgg¡dPdQ Z$ej dRdSdTdUdVdWg¡ej dXej%e&g¡dYdZ Z'd[d\ Z(d]d^ Z)dOS )_ÚTestTimedeltasÚreadonlyTFc C s> t jg td}|r |jdd t|}tg }t ||¡ d S )N©ÚdtypeF)Úwrite)ÚnpÚarrayÚobjectZsetflagsr ÚtmÚassert_index_equal)Úselfr ÚarrÚresultÚexpected© r ú[d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/tools/test_to_timedelta.pyÚtest_to_timedelta_readonly s z)TestTimedeltas.test_to_timedelta_readonlyc C s t ddg}t| ¡ s td S )NÚ )r r ÚallÚAssertionError)r r r r r Útest_to_timedelta_null! s z%TestTimedeltas.test_to_timedelta_nullc C sB t t t dd¡g¡}t t t dd¡g¡¡}t ||¡ d S )Né Ús)r r r Ú timedelta64ÚpdÚIndexr r ©r r r r r r Ú%test_to_timedelta_same_np_timedelta64% s z4TestTimedeltas.test_to_timedelta_same_np_timedelta64c C s: t tddtdddg}tt ddg}t ||¡ d S )Nr ©Údays)r' ÚsecondsZ1dz1days 00:00:01©r r r r Úassert_series_equal©r r r r r r Útest_to_timedelta_series+ s z'TestTimedeltas.test_to_timedelta_seriesc C sB t t dd¡t dd¡ d¡g}tddgdd}t ||¡ d S )Nr Únsé
|
r úm8[ns]©Úunit)r r r! Úastyper r r r$ r r r Útest_to_timedelta_units1 s
|
ÿz&TestTimedeltas.test_to_timedelta_unitsz dtype, unitÚint64r ÚmÚhztimedelta64[s]ztimedelta64[D]ÚDc C sZ t jdgd |d}t||d}|dkr,dnd}tt d|¡gd |d}t ||¡ d S )Nr é r r0 r4 r/ zm8[s])r r r r r! r r )r r r1 r r Z exp_dtyper r r r Ú test_to_timedelta_units_dtypes9 s
|
z-TestTimedeltas.test_to_timedelta_units_dtypesc C s t jtjjd gdd}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 d S )Nr ztimedelta64[m]r zPCannot convert -9223372036854775807 minutes to timedelta64\[s\] without overflow©Úmatch) r r r" ÚNaTÚ_valueÚpytestÚraisesr r r r Z_from_sequence)r r Úmsgr r r Ú test_to_timedelta_oob_non_nanoK s ÿz-TestTimedeltas.test_to_timedelta_oob_non_nanoÚargr. é r8 ÚerrorsÚignoreÚraiseÚcoercec C s* t jtdd t||d W 5 Q R X d S )Nz 1-d arrayr: ©rD )r> r? Ú TypeErrorr )r rB rD r r r Útest_to_timedelta_dataframe[ s z*TestTimedeltas.test_to_timedelta_dataframec C s0 d}t jt|d tdgdd W 5 Q R X d S )Nzerrors must be one ofr: ÚfooÚneverrH ©r> r? Ú
|
ValueErrorr ©r r@ r r r Ú test_to_timedelta_invalid_errorsd s z/TestTimedeltas.test_to_timedelta_invalid_errorsr c C s. d}t jt|d t|dd W 5 Q R X d S )Nz invalid unit abbreviation: foor: rK r0 rM )r rB r@ r r r Ú test_to_timedelta_invalid_unitj s z-TestTimedeltas.test_to_timedelta_invalid_unitc C sL d}t jt|d ttdd W 5 Q R X ttddddtjksHtd S )NzIValue must be Timedelta, string, integer, float, timedelta or convertibler: r )ÚsecondrG rH )r> r? rN r r r" r< r rO r r r Útest_to_timedelta_timeq s
|
ÿz%TestTimedeltas.test_to_timedelta_timec C s. d}t jt|d tddg W 5 Q R X d S )Nz*Could not convert 'foo' to NumPy timedeltar: rK ÚbarrM rO r r r Útest_to_timedelta_bad_valuez s z*TestTimedeltas.test_to_timedelta_bad_valuec C sN t ttjtjgtddgdd¡ t tdtjdgtdddgdd¡ d S )NrK rT rG rH ú1 dayz1 min)r r r r" r< r )r r r r Ú"test_to_timedelta_bad_value_coerce s þþz1TestTimedeltas.test_to_timedelta_bad_value_coercec C s d}|t |ddkstddg}t tj|tdt |dd¡ t ddg¡}t |t |dd¡ t
|
ddg}t |t |dd¡ d S )NZapplerE rH z1 daysr ) r r r Úassert_numpy_array_equalr r r r" r# r r r* )r Z invalid_datar r r Ú'test_to_timedelta_invalid_errors_ignore s
|
þ
|
ÿz6TestTimedeltas.test_to_timedelta_invalid_errors_ignorez val, errors)Z1MT)z1 MT)Z1YT)z1 YT)Z1yT)z1 yT)Z1mF)z1 mF)rV F)Z2dayFc C s8 d}|r,t jt|d t| W 5 Q R X nt| d S )Nz=Units 'M', 'Y' and 'y' do not represent unambiguous timedeltar: rM )r ÚvalrD r@ r r r Ú!test_unambiguous_timedelta_values s
|
z0TestTimedeltas.test_unambiguous_timedelta_valuesc C sL t t dd¡g}t dg t¡}t ||¡ t tdg}t ||¡ d S )Nr r ú00:00:01)r r r! Úapplyr r r* r+ r r r Útest_to_timedelta_via_apply· s
|
z*TestTimedeltas.test_to_timedelta_via_applyc C sN dt jg}t d ¡ t|}W 5 Q R X tt jddt jg}t ||¡ d S )Nr\ r )r( )r" r< r Zassert_produces_warningr r Ú Timedeltar )r Úvalsr r r r r Ú+test_to_timedelta_inference_without_warningÀ s
|
|
z:TestTimedeltas.test_to_timedelta_inference_without_warningc C sr t d¡}ttdt jg}tt dd¡|gtj dd}t ||¡ tdtj gdd}t|}t ||¡ d S )Nr< r\ é Ê;r- r/ r )
|
r r! r r Únanr ZENDIANr* r" r< )r Z timedelta_NaTÚactualr Úserr r r Ú#test_to_timedelta_on_missing_valuesÊ s
|
|
þ z2TestTimedeltas.test_to_timedelta_on_missing_valuesrZ c C s& t |}|jt d¡ d¡ks"td S )Nr< r4 )r r= r r! r2 r )r rZ rd r r r Ú*test_to_timedelta_on_missing_values_scalarÙ s z9TestTimedeltas.test_to_timedelta_on_missing_values_scalarc C sF t ddd¡dd
}t|dd}t jddd d
|
d }t |j|¡ d S ) Nr r gíµ ÷Æ°>iöÿÿÿr r0 ið¢;rb iè r4 r )r Úaranger r rX Zasi8)r r r Z expected_asi8r r r Útest_to_timedelta_floatÞ s z&TestTimedeltas.test_to_timedelta_floatc C sF t jdddgtd}t|ddd}tddtjgdd}t ||¡ d S ) Nr rC Úerrorr r- rG ©r1 rD r0 )r r r r r" r< r r )r r r r r r r Ú%test_to_timedelta_coerce_strings_unitå s z4TestTimedeltas.test_to_timedelta_coerce_strings_unitc C s2 t jdddgtd}t|ddd}t ||¡ d S )Nr rC rj r r- rE rk )r r r r r rX )r r r r r r Ú%test_to_timedelta_ignore_strings_unitë s z4TestTimedeltas.test_to_timedelta_ignore_strings_unitzexpected_val, result_valr& Nc C s: t tdd|g}tt d|gdddd}t ||¡ d S )Nr r& ZInt64r r' r0 r) )r Z expected_valZ
|
result_valr r r r r Ú&test_to_timedelta_nullable_int64_dtypeð s z5TestTimedeltas.test_to_timedelta_nullable_int64_dtype)Úinputr )z8:53:08.71800000001z 8:53:08.718)ú8:53:08.718001rp )ú8:53:08.7180000001rq )z-8:53:08.71800000001z -8:53:08.718)z8:53:08.7180000089z8:53:08.718000008Úfuncc C s" t |¡}||}||ks td S )N)r" r_ r )r ro r rr r r r r Ú&test_to_timedelta_precision_over_nanosú s
|
z5TestTimedeltas.test_to_timedelta_precision_over_nanosc C sr | ¡ }t |¡}d}tjt|d t| W 5 Q R X | d¡}t|}t|t j
|
sZt |j | d¡ksnt d S )NzYValue must be Timedelta, string, integer, float, timedelta or convertible, not datetime64r: r/ Úi8) Z to_datetime64r r r> r? rN r ÚviewÚ
|
isinstancer" r_ r r= )r Z fixed_now_tsZdt64rB r@ Zarg2r r r r Útest_to_timedelta_zerodim s
|
ÿ
|
z(TestTimedeltas.test_to_timedelta_zerodimc C sB t dtjg|d}t|}t tjdddtjg}t ||¡ d S )Nr r r- r0 )r r" ZNAr r_ r< r r* )r Úany_numeric_ea_dtypere r r r r r Ú test_to_timedelta_numeric_ea s z+TestTimedeltas.test_to_timedelta_numeric_ea)*Ú__name__Ú
|
__module__Ú __qualname__r> ÚmarkZ parametrizer r r% r, r3 r9 rA r rh Zreshaper" Z DataFramerJ rP rQ rS rU rW rY r[ r^ ra rf rc r< rg ri rl rm r rn r_ r rs rw ry r r r r r
|
s
|
ûþ
|
|
(ÿ
|
öþ
|
|
|
ÿ
|
ûþ
|
r
|
c C sL t d¡ tddg| ¡ dd}t|}tddgdd}t ||¡ d S )NZpyarrowr rC z [pyarrow]r ztimedelta64[ns])r> Z importorskipr Úlowerr r r* )rx re r r r r r Ú test_from_numeric_arrow_dtype$ s
|
|
r )Údatetimer r Únumpyr r> Z pandas.errorsr Zpandasr" r r r r Zpandas._testingZ_testingr Zpandas.core.arraysr r
|
r r r r r Ú<module> s
|
