U
|
ý°dZ; ã @ sX d dl Zd dlZd dlm mZ d dlmZm Z m
|
Z
|
m Z d dl m Z G dd dZdS )é N)Ú DataFrameÚNaTÚSeriesÚ
|
date_rangec
|
@ sD e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
e j dd Z e j dd Zdd Zej dejddddejd dd dejd d d dg¡d!d" Zd#d$ Zej d%d&ejd'e j d(g¡d)d* Zd+d, Zd-d. Zd/d0 Zd1d2 Zd3d4 Zd5d6 Z ej dd7d8g¡d9d: Z ej d;d<d=d>g¡d?d@ Z dAS )BÚTestDataFrameInterpolatec C s¬ t tddd}| ¡ }t|d< || ¡ }||}t ||¡ |j d¡}|| ¡ }||j d¡}t ||¡ ||d } || ¡ }
|
|||d } t |
|
| ¡ d S )Nz
|
2012-01-01é )Úperiodsé z
|
US/Pacificr ) r r Úcopyr Ú interpolateÚtmÚ assert_equalÚdtZ tz_localize) ÚselfÚframe_or_seriesÚorigZserÚresÚexpectedZser_tzZres_tzZ expected_tzZser_tdZres_tdZ expected_td© r úbd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/methods/test_interpolate.pyÚ$test_interpolate_datetimelike_values s z=TestDataFrameInterpolate.test_interpolate_datetimelike_valuesc C s |r&|t kr&tjjdd}|j |¡ |dtjdg}|j}|j dd |dddg}t
|
||¡ t ||j¡stt | ¡ d dkst d S )Nz'.values-based in-place check is invalid)Úreasoné r T©Úinplaceg ø?)r ÚpytestÚmarkZxfailÚnodeZ
|
add_markerÚnpÚnanÚvaluesr r r Ú shares_memoryÚAssertionErrorZsqueeze)r r Úusing_array_managerÚrequestr Úobjr r r r r Útest_interpolate_inplace& s z1TestDataFrameInterpolate.test_interpolate_inplacec C s> t ddtjdgdddtjgddddgtdd}t d d
|
d d gd d d d gddddgtdd}| ¡ }t ||¡ |d j}|d j}|rºt ||d j¡s¢t
|
t ||d j¡sæt
|
n,t ||d j¡rÐt
|
t ||d j¡ræt
|
|jdd}|d ksþt
|
t ||¡ t |d j|¡s"t
|
t |d j|¡s:t
|
d S )Nr r é é é r Úabcd©ÚAÚBÚCÚDç ð?ç @ç @ç @ç "@r. r/ Tr ) r r r Úlistr r Úassert_frame_equalZ_valuesr r! r" )r Úusing_copy_on_writeÚdfr ÚresultZcvaluesZdvaluesr r r r Útest_interp_basic7 s8
|
üÿ
|
|
|
üÿ
|
|
z*TestDataFrameInterpolate.test_interp_basicc C sn t ddtjdgdddtjgddddgtdd}| d ¡ ¡ }| d ¡}d|jd
|
< d|jd < t ||¡ d S ) Nr r r' r( r) r r* r+ r. ©r) r, )r r- ) r r r r5 Ú set_indexr Úlocr r6 ©r r8 r9 r r r r Ú&test_interp_basic_with_non_range_index] s
|
üÿ
|
|
|
z?TestDataFrameInterpolate.test_interp_basic_with_non_range_indexc C s. t }| ¡ }||k st|}t ||¡ d S )N)r r r" r r6 r> r r r Útest_interp_emptym s
|
z*TestDataFrameInterpolate.test_interp_emptyc C s` t ddtjdgdddtjgddddgtdd}d }tjt|d
|
|jd d W 5 Q R X d S ) Nr r r' r( r) r r* r+ a method must be one of \['linear', 'time', 'index', 'values', 'nearest', 'zero', 'slinear', 'quadratic', 'cubic', 'barycentric', 'krogh', 'spline', 'polynomial', 'from_derivatives', 'piecewise_polynomial', 'pchip', 'akima', 'cubicspline'\]. Got 'not_a_method' instead.©ÚmatchZ not_a_method©Úmethod)r r r r5 r ÚraisesÚ
|
ValueErrorr ©r r8 Úmsgr r r Útest_interp_bad_methodu s
|
üÿ ÿz/TestDataFrameInterpolate.test_interp_bad_methodc C s t ddtjdgdddtjgdddd gtd
|
d }|d ¡ }tddd dgd d}t ||¡ |d jdd}tddddgd d}t ||¡ d S )Nr0 r1 r3 r r' r( r r) r r* r+ r, r2 )ÚnameÚinfer©Údowncast)r r r r5 r r r Zassert_series_equalr> r r r Útest_interp_combo s
|
üÿ z*TestDataFrameInterpolate.test_interp_comboc C sZ t ddtjdgtjdddgd}| d¡}d}tjt|d |jd d
|
W 5 Q R X d S ) Nr r r' r) ©r, r- r, zkInterpolation with NaNs in the index has not been implemented. Try filling those NaNs before interpolating.rA r rC )r r r r< r rE ÚNotImplementedErrorr rG r r r Útest_interp_nan_idx s "
|
ÿz,TestDataFrameInterpolate.test_interp_nan_idxc C sV t ddtjddtjdgddddddd gd
|
}| d ¡}| ¡ }|jd dd }d|jd< d|jd< t ||¡ |jdd}d|jd< d|jd< t ||¡ |jdd}d|jd< d|jd< tj||dd |jdd}d|jd< d|jd< t ||¡ |jd d}d|jd< d |jd< t ||¡ |jd d}d |jd< d|jd< tj||dd d S ) Nr r r' r é r) é é é ©r, r. r. Ú
|
polynomial©rD Úorderg¥ÝÇUUU@r; gáu:±@)rT r, ZcubicrC gvä:@gb+hZ @ZnearestF)Z check_dtypeZ quadraticgå^¢r@g!ëg¹
@ZslineargXè ;±@Zzeror1 ) r r r r< r
|
r r= r r6 ©r r8 r r9 r r r Útest_interp_various¤ s: (ÿ
|
|
|
|
|
|
|
|
|
|
|
|
|
z,TestDataFrameInterpolate.test_interp_variousc C sà t ddtjddtjdgddddddd gd
|
}|jd d }| ¡ }d|jd < d|jd< t ||¡ |jd dd}t || tj ¡¡ |jdd }| ¡ }|d |d< t ||¡ |jdd }d|jd < d|jd< t ||¡ d S )Nr r r' r rR r) rS rT rU rV Z barycentricrC )r r, é )r r, rK )rD rM Zkroghr, Zpchipç @)
|
r r r r r
|
r= r r6 ÚastypeÚint64)r r8 r9 r Z expectedkr r r Útest_interp_alt_scipyË s$ (ÿ
|
|
|
|
z.TestDataFrameInterpolate.test_interp_alt_scipyc C sÔ t ddtjdgdddtjgtjdddgdtjddgddddgd}|jdd }| ¡ }d|jd
|
< d|jd < d|jd < |d tj¡|d< t ||¡ |jdd d}t ||¡ |jdd }| ¡ }t ||¡ d S )Nr r r' r) r r\ rR )r r r r) r' ©Úaxis)r) r )r r )r r) r ©rb rD r )
|
r r r r r
|
r= r^ Úfloat64r r6 r> r r r Útest_interp_rowwiseä s(
|
ûÿ
|
|
|
z,TestDataFrameInterpolate.test_interp_rowwisezaxis_name, axis_numberÚrowsr Zrows_0)ÚidÚindexZindex_0Úcolumnsr Z columns_1c C s\ dt jdgdt jdgdddgd}t|t jd }|j|d
|
d }|j|d
|
d }t ||¡ d S ) Nr r\ r rR r r rS )r r r ©ZdtypeÚlinearrc )r r r rd r r r6 )r Ú axis_nameÚ axis_numberÚdatar8 r9 r r r r Útest_interp_axis_namesý s
|
|
"z/TestDataFrameInterpolate.test_interp_axis_namesc C sH t dddtjddtjtjdg ddd dd dddd
|
g d }|jdd d S ) Nr ç à?r0 r' rS é@ r r r) éÿÿÿÿ)r r ra )r r r r )r r8 r r r Útest_rowwise_alt s þÿz)TestDataFrameInterpolate.test_rowwise_altÚ check_scipyFT)Zmarksc C sr t tjtjdddgtjddtjdgd}| ¡ }| ¡ }d|jd < t ||¡ |rn|jd
|
d d }t ||¡ d S ) Nrp g Ð?r éýÿÿÿg ÀéüÿÿÿrO g À)r) r- rW r rX )r r r r r
|
r= r r6 )r rt r8 r9 r r r r Útest_interp_leading_nans s $ÿ
|
z1TestDataFrameInterpolate.test_interp_leading_nansc C sx t ddtjdgddddgtjddd gtjtjd
|
d
|
gddd dgd }d }tjt|d | d¡j|d W 5 Q R X d S )Nr r r' ÚaÚbÚcÚdr rR r( r) )r, r- r. r/ ÚEúvCannot interpolate with all object-dtype columns in the DataFrame. Try setting at least one column to a numeric dtype.rA Úobjectra )r r r r rE Ú TypeErrorr^ r )r rb r8 rH r r r Ú test_interp_raise_on_only_mixed( s
|
|
ûÿ
|
ÿz8TestDataFrameInterpolate.test_interp_raise_on_only_mixedc C sH t dddgdddgddd }d
|
}tjt|d | ¡ W 5 Q R X d S ) Nr r r) r' r r\ rO r~ rj r} rA )r r rE r r rG r r r Ú%test_interp_raise_on_all_object_dtype: s
|
ÿz>TestDataFrameInterpolate.test_interp_raise_on_all_object_dtypec C s t dddtjdgi}t dddddgi}| ¡ }| ¡ }|d jdd}|d ksVt|rht ||¡ n t ||¡ | ¡ }|d jddd }|d kst|r¬t ||¡ nt || d
|
¡¡ d S ) Nrx r0 r1 r3 r2 Tr rK )r rM r_ ) r r r r
|
r r" r r6 r^ )r r7 r8 r Z expected_cowr9 Ú return_valuer r r Útest_interp_inplaceE s z,TestDataFrameInterpolate.test_interp_inplacec C sf t ddddgtjdddgddddgd}|jdd d
|
d }|jdd d d }|d ksVtt ||¡ d S ) Nr0 r1 r2 r3 r) r )rx ry rz rk r F)rD rb r T)r r r r r" r r6 )r r9 r r r r r Útest_interp_inplace_rowY s $ÿ z0TestDataFrameInterpolate.test_interp_inplace_rowc C sÔ t ddtjdgddddgddtjdgddddgd }t tjddddgd
|
d tjddddgd d tjddddgd
|
d tjddddgd
|
d d }|jd d }t ||¡ |ddg jd d }t ||ddg ¡ d S )Nr r r' r) r0 r1 r3 r2 r+ rd rj r_ rL r- r/ )r r r Úarrayr r r6 rZ r r r Útest_interp_ignore_all_goodc s$
|
|
üÿüÿ z4TestDataFrameInterpolate.test_interp_ignore_all_goodc C st d}t d|d}tj ||¡}tj||dk < t|||d}|jddd}|jddd d
|
}|d ksdtt ||¡ d S ) Nr z
|
2014-01-01)Ústartr rp )rh ri rn r Útimerc T)rb rD r )
|
r r ÚrandomZrandr r r r" r r6 )r r Úidxrn r r9 r r r r Ú test_interp_time_inplace_axis} s z6TestDataFrameInterpolate.test_interp_time_inplace_axis)rh r )ri r c C sl t ddd¡}t |¡}tt |d¡t d¡|dj|d d}|jd |d
|
}|jd |d
|
}t ||¡ d S ) Nr éd iè )é
|
r r )rn rh ri g®Gázð?)ri rk ©rD rb )
|
r ZlinspaceÚsinr ZtileZarangeZreindexr r r6 )r rl rm ÚxÚyr8 r9 r r r r Útest_interp_string_axis s
|
|
ÿþz0TestDataFrameInterpolate.test_interp_string_axisrD ZffillZbfillÚpadc C s~ |r|dkrt |¡ tddddtjdgdddtjdd gddd
|
tjtjd gd }|j||d }|j||d}t ||¡ d S )N)r ri r0 r1 r2 r3 g @r] g @g $@r4 g >@)r, r- r. rc r ) ÚtdZ&mark_array_manager_not_yet_implementedr r r Zfillnar r r6 )r r$ rb rD r# r8 r r9 r r r Útest_interp_fillna_methods s
|
ýÿz3TestDataFrameInterpolate.test_interp_fillna_methodsN) Ú__name__Ú
|
__module__Ú __qualname__r r& r: r? r@ rI rN rQ r Zskip_if_no_scipyr[ r` re r r Z parametrizeÚparamro rs rw r r r r r r r r r r r r r sJ &
|
|
&
|
ýþ
|
|
ÿ
|
|
|
r )Únumpyr r Z pandas.util._test_decoratorsÚutilZ_test_decoratorsr Zpandasr r r r Zpandas._testingZ_testingr r r r r r Ú<module> s
|
|
