U
|
ý°dÎZ ã @ sl d dl Zd dlZd dlm mZ d dlZd dlm Z m
|
Z
|
m Z m Z m Z mZmZ d dlmZ G dd dZdS )é N)ÚCategoricalIndexÚ DataFrameÚIndexÚNaTÚSeriesÚ
|
date_rangeÚoffsetsc @ s e Zd Zej dejddg fejddej dgfg¡dd Z
|
dd Z ej d
|
e e d ¡ged d dde ed edd ddg¡ej ddddg¡dd Zdd Zdd Zej d d d g¡d d! Zej d"ddd#d$g¡d%d& Zd'd( Zej d)d*d+d,d-g¡d.d/ Zd0d1 Zd2d3 Zd4d5 Zd6d7 Zd8d9 Zd:d; Z d<d= Z d>d? Z d@dA Z dBdC Z dDdE Z!dFdG Z"dHdI Z#dJdK Z$dLdM Z%e&j'dNdO Z(dPdQ Z)dRdS Z*dTdU Z+dVdW Z,dXdY Z-ej dZej.d[ejj/d\d]d^d_g¡ejjd`edaddedaddbdce0j1dadddde0j2dedddde0j3dd#ddej.e0j4ddgdfdgejj/d\d]d^ej.e0j4ddgdhdgejj/d\d]d^gdidj dkdldm Z5dndo Z6dpdq Z7drS )sÚTestDataFrameShiftzinput_data, output_data)r )Úshape)é ç ð?c C s\ |j dd || d¡}|tk r@||tt|tddd}n ||dd}t ||¡ d S )NF)Úwriteé Úfloat64)ÚindexÚcolumnsÚdtype©r )ZsetflagsÚshiftr ÚrangeÚlenÚtmÚ assert_equal)ÚselfZ
|
input_dataZ output_dataÚframe_or_seriesÚresultÚexpected© r ú\d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/methods/test_shift.pyÚ test_shift_non_writable_array s
|
ü z0TestDataFrameShift.test_shift_non_writable_arrayc C sz |t j d¡tddddd}|jddd}|jjddd}t |j|¡ |jdd d}|jt d
|
¡ }t |j|¡ d S ) Né ú1/1/2000ÚH©ÚperiodsÚfreq©r r Z5T©r% Z4Hé )
|
ÚnpÚrandomÚrandnr r r r Úassert_index_equalr ZHour)r r Útsr Z exp_indexr r r Útest_shift_mismatched_freq* s
|
ÿz-TestDataFrameShift.test_shift_mismatched_freqÚobjr z1/1/2011é r" r# Z2017©r$ r& Ú
|
shift_sizer r r c C s& |t k r| ¡ }| |¡|k s"td S ©N)r Úto_framer ÚAssertionError)r r/ r2 r r r r Útest_shift_always_copy8 s z)TestDataFrameShift.test_shift_always_copyc C s¾ t td}tjtdd |jdg d W 5 Q R X | ¡ }tjtdd |jdt d¡d W 5 Q R X | t
|
¡}|jdi d}|d i kst | ¡ }|jdi d}|j d i ksºt d S )Né zfill_value must be a scalar©Úmatchr ©Ú
|
fill_valuer )r r ) r r ÚpytestÚraisesÚ
|
ValueErrorr r4 r) ÚarangeÚastypeÚobjectr5 Úiloc)r ÚserÚdfZobj_serr Zobj_dfr r r Ú!test_shift_object_non_scalar_fillG s
|
z4TestDataFrameShift.test_shift_object_non_scalar_fillc C s< t ||¡ t¡}| d¡}| t¡ d¡}t ||¡ d S )Nr )r Úget_objr@ Úintr Úfloatr )r Údatetime_framer r- Úshiftedr r r r Útest_shift_intY s
|
z!TestDataFrameShift.test_shift_intr Úint32Úint64c C sb t ddd}tjd|d}|||d}|d }|j|d}|tjdddd g|d}t ||¡ d S )
|
Nz
|
2000-01-01r r1 r r& r r r r7 )r r) r? r Únanr r ) r r r r ZarrÚs1Úpr r r r r Útest_shift_32bit_take_ s z(TestDataFrameShift.test_shift_32bit_taker$ r7 r( c C sP |t |td|ddd}| dd¡}|t |td|ddd}t ||¡ d S )Nz2016-1-1 00:00:00r" r# r& r Z2Hz2016-1-1 02:00:00)r r r r r )r r$ r r/ r r r r r Útest_shift_preserve_freqstrk s þ þz.TestDataFrameShift.test_shift_preserve_freqstrc C sâ t ddddd}||}| d¡}t ||¡ t |¡dks@t| d¡}tg| t¡j
|
¡ d d
}||}t ||¡ t |¡dkst| d
|
¡}| t¡j
|
¡ d d
ttg }||}t ||¡ t |¡dksÞtd S ) Nú
|
2016-11-06r" é
|
ú
|
US/Eastern©r% r$ Útzr údatetime64[ns, US/Eastern]r é éþÿÿÿr ) r r r r Ú get_dtyper5 r r@ rA ÚvaluesÚtolist)r r Údatesr/ ÚresZexp_valsÚexpr r r Útest_shift_dst{ s
|
|
|
z!TestDataFrameShift.test_shift_dstÚexrT iöÿÿÿé iìÿÿÿc C sV t ddddd}||}| |¡}|tgd dd}t ||¡ t |¡dksRtd S )NrS r" rT rU rV rX r )r r r r r r[ r5 )r r rb r^ r/ r_ r` r r r Útest_shift_dst_beyond s
|
z(TestDataFrameShift.test_shift_dst_beyondc C s& t ||¡}| d¡}t ||¡ d S ©Nr )r rF r r )r rI r r/ Ú unshiftedr r r Útest_shift_by_zero s
|
z%TestDataFrameShift.test_shift_by_zeroc C sÌ |d }| d¡}t |j|j¡ | d¡}t |d |¡ | d¡}t |j|j¡ | d¡}t |d |¡ | d¡ d¡}t | ¡ j|jd d
¡ | d¡ d¡}t | ¡ j|jd d
¡ d S )NÚAr éûÿÿÿ)r r r, r Úassert_series_equalÚassert_numpy_array_equalÚdropnar\ )r rI rC rJ Z shifted_serrf Z unshifted_serr r r Ú
|
test_shift s
|
|
|
|
ÿz TestDataFrameShift.test_shiftc
|
C sÜ t ||¡}t ¡ }|jd|d}t|t|ks6t|jd|d}t ||¡ |jddd}t ||¡ |jd|d}t ||¡ |jd }||d } |t krÀt j
|
| |¡| | ¡dd nt |j | |j | ¡ d S )Nr r' ri ÚBr F)Z check_names)r rF r ÚBDayr r r5 r r r rj ÚxsZassert_almost_equalÚat)
|
r rI r r/ ÚoffsetrJ rf Úshifted2ÚdZ shifted_dr r r Útest_shift_by_offset¸ s
|
z'TestDataFrameShift.test_shift_by_offsetc
|
C s@ t ¡ }t ||¡}| d¡}| d¡}t |j|j¡ t |j|j¡ |tkrt |jd d
df ¡ j
|
|jd d
df j
|
¡ n t | ¡ j
|
|j
|
d d
¡ | dd¡}| dt ¡ ¡}t ||¡ t || dd¡¡ d}tjt|d |jdd W 5 Q R X |jddd}t ||¡ |jdt ¡ d} t | |¡ d S ) Nr éÿÿÿÿr rn z does not match PeriodIndex freqr8 ÚDr' )r ÚmakePeriodFramerF r r, r r rk rB rl r\ r ro r r< r= r> )
|
r r ÚpsrJ rf rs Úshifted3ÚmsgZshifted4Zshifted5r r r Útest_shift_with_periodindexÏ s.
|
|
ÿ z.TestDataFrameShift.test_shift_with_periodindexc C sf t tj dd¡}tjt tj|jdgd|jd d
dd
f gddd}|j ddd }t
|
||¡ d S )
|
NrT r r ©r r rv Tr ©Z ignore_indexÚaxis©r © r r) r* ÚrandÚpdÚconcatrN r rB r r Úassert_frame_equal©r rD r r r r r Útest_shift_other_axisï s (ýz(TestDataFrameShift.test_shift_other_axisc C sf t tj dd¡}tjt tj|jdgd|jd d
dd
f gddd}|j dd d
|
}t
|
||¡ d S ) NrT r r r} rv Tr r~ r r r r r r r Útest_shift_named_axisû s (ýz(TestDataFrameShift.test_shift_named_axisc C sR |j }t ¡ }|jd|dd}t|t|ks2t|jd|dd}t ||¡ d S )Nr r )r% r ri )ÚTr ro r r r5 r r )r rI r/ rr rJ rf r r r Ú test_shift_other_axis_with_freq s z2TestDataFrameShift.test_shift_other_axis_with_freqc C sZ t ddgddgd}| d¡}t tjtjtjgddggtdddgd}t ||¡ d S ) NTF)ÚhighÚlowr r r r ©r )r r r) ÚarrayrN rA r r
)r rD ÚrsÚxpr r r Útest_shift_bool s
|
þz"TestDataFrameShift.test_shift_boolc C sD |ddddgdd}| d¡ d¡}t |jd d
| ¡ ¡ d d
|
}||}| d¡}t |j|j¡ t ||j d d
dk¡st
|
t |j d d
||j dd
k¡s²t
|
| d ¡}t |j|j¡ t ||j d d
dk¡sìt
|
t |j d d
||j d d
k¡st
|
t |j ||j ¡ t |j ||j ¡ d S ) NÚaÚbÚcrt Úcategoryr r rv c S s | j jd S re )Ú_mgrÚarrays)Zndframer r r Úget_cat_values s zBTestDataFrameShift.test_shift_categorical1.<locals>.get_cat_valuesrZ r ) r r r rB rl r, r r) ÚallÚcodesr5 Ú
|
categories)r r r/ Úrtr ÚcatZsp1Zsn2r r r Útest_shift_categorical1 s
|
*
|
,z*TestDataFrameShift.test_shift_categorical1c C sf t dddgdd}t dddgdd}t||d }| d
|
¡}t| d
|
¡| d
|
¡d }t ||¡ d S ) Nr r r r r rh rn ÚC)ZoneZtwor )r r r r r
)r rO Ús2rD r r r r r Útest_shift_categorical4 s
|
z)TestDataFrameShift.test_shift_categoricalc C s |ddddgdd}|j ddd}|tjddddgddddgd d
|
}t ||¡ d }tjt|d |j dd d W 5 Q R X d S )Nr r r rt r r r r: F)r Zorderedz9Cannot setitem on a Categorical with a new category \(f\)r8 Úf)r r Z Categoricalr r r< r= Ú TypeError)r r r- r_ r r{ r r r Ú!test_shift_categorical_fill_value= s
|
|
ÿÿ z4TestDataFrameShift.test_shift_categorical_fill_valuec C s2 t dddd}|ddddd g|d
|
}|d ddddg|d
|
}|jd d d }t ||¡ |d d dddg|d
|
}|jdd d }t ||¡ |d ddg}|jddd }t |¡t |¡ks¸t|d ddddg|d
|
}|dd dddg|d
|
}|jd dd }t ||¡ |ddd ddg|d
|
}|jddd }t ||¡ d S )Nr! r r" r# r g @g @g @g @r& g r r: r r7 r r( )r r r r r[ r5 )r r Zdtir- r` r r_ r/ r r r Útest_shift_fill_valueL s$ z(TestDataFrameShift.test_shift_fill_valuec C s& t dg i}| d¡}t ||¡ d S )NÚfoorv )r r r r
)r rD r r r r Útest_shift_emptyh s
|
z#TestDataFrameShift.test_shift_emptyc C sð t tddgd dddddgg}tj dd¡}g }|D ]`}t| ¡ |d}tdD ].}|jd d
|f |d ¡|jd d
|f< qTtd|_ |
|
|¡ q8|d ¡ ¡ }t |ttdddd ¡ t |d |d ¡ t |d |d ¡ d S )
|
Nr r r rc r r é rM r )Úlistr r) r* r+ r ÚcopyrB r r ÚappendZisnaÚsumr rj r r
)r Z column_listsÚdatarJ r rD ÚsZnullsr r r Ú test_shift_duplicate_columnso s " ,
|
z/TestDataFrameShift.test_shift_duplicate_columnsc C s@ t tjjddd}t tjjddd}tj||gdd}|sRt|jjdksRt |j
|
ddd}|j ddd ddgdd}| d
|
d ¡}tj |jd d
d d
f< |j|_t ||¡ tj||gdd}|sÜt|jjdksÜt |j
|
d dd}|j dd dddgdd}| dd ¡}tj |jd d
d d
f< |j|_t ||¡ d S )Néè ©r r7 ©Úsize©r r r r r rv r c S s0 | j t| jd dd ddd¡j | jddS )Nr r rH )r r ©Zset_axisr r
|
r@ r ©rD r r r Ú<lambda> s
|
ÿ þzETestDataFrameShift.test_shift_axis1_multiple_blocks.<locals>.<lambda>rZ r7 r( c S s0 | j t| jd dd ddd¡j | jddS )Nr r rH )r7 r( rµ r¶ r r r r· § s
|
ÿ þ)r r) r* Úrandintr r r r Úblocksr5 r ÚtakeÚpiperN rB r r r
)r Zusing_array_managerÚdf1Údf2Údf3r r r r r Ú test_shift_axis1_multiple_blocks
s0 ÿ ÿz3TestDataFrameShift.test_shift_axis1_multiple_blocksc C sz t tjjddd}t tjjddd}tj|jd d
dd
f |jd d
d d
f gdd}|jd dt d
|
¡d }t |j
|
j d kst |j d d d
|
dgdd}t d
|
¡|jd d
d d
f< |j|_t ||¡ tj|jd d
dd
f |jd d
d d
f gdd}|jd dt d
|
¡d }t |j
|
j d ks0t |j d dd d gdd}t d
|
¡|jd d
d d
f< |j|_t ||¡ d S )Nr° r± r² r´ r( r r7 r r r ©r r; rv rZ )r r) r* r¸ r r rB r Úint_r r r¹ r5 rº r r r
)r r¼ r½ r¾ r r r r r Ú.test_shift_axis1_multiple_blocks_with_int_fill° s 6 6 zATestDataFrameShift.test_shift_axis1_multiple_blocks_with_int_fillc C st t ¡ }t ||¡}|jddd}|jddd}t ||¡ |jdd}t ||¡ |jt ¡ d}t ||¡ d S )Nr Úinferr' rv rn )r rx rF r r r ro )r r ry rJ rf rs rz r r r Ú'test_period_index_frame_shift_with_freqÊ s z:TestDataFrameShift.test_period_index_frame_shift_with_freqc C sÎ t ||¡}|jddd}|jddd}t ||¡ |j|jjd}t ||¡ t|jtt
|
|j¡|j d}t ||¡}|jddd}|jddd}|j d ¡|_t ||¡ |jddd}t ||¡ d S )Nr rà r' rv r ) r rF r r r r% r r\ r r) Zasarrayr Z
|
_with_freq) r rI r ÚdtobjrJ rf rs Z inferred_tsr r r r Ú#test_datetime_frame_shift_with_freqØ s$ ý z6TestDataFrameShift.test_datetime_frame_shift_with_freqc C sB t ¡ }t ||¡}d}tjt|d |jdd W 5 Q R X d S )Nz.Given freq M does not match PeriodIndex freq Br8 ÚMr' )r rx rF r< r= r> r )r r ry r{ r r r Ú-test_period_index_frame_shift_with_freq_errorï s
|
z@TestDataFrameShift.test_period_index_frame_shift_with_freq_errorc C sJ t ||¡}|jdddg }d}tjt|d |jdd W 5 Q R X d S )Nr r é z6Freq was not set in the index hence cannot be inferredr8 rà r' )r rF rB r< r= r> r )r rI r rÅ Zno_freqr{ r r r Ú)test_datetime_frame_shift_with_freq_errorö s
|
z<TestDataFrameShift.test_datetime_frame_shift_with_freq_errorc C s t t d¡t d¡g}tjtdd |jddd W 5 Q R X | ¡ }tjtdd |jddd W 5 Q R X t||d}| ¡ |jdddd }tddg|d
|
d}t
|
||¡ td
|
|i}||d < t |j jd ksàt|jdddd }t
|
||¡ d S ) Nú
|
2020-01-01z
|
2020-01-02zvalue should be ar8 r r r: ©rh rn rÀ rh rn r )r r Ú Timestampr< r= r£ r r4 r Ú_consolidate_inplacer r
r r r r5 )r rC rD r½ r r r¾ r r r Ú)test_shift_dt64values_int_fill_deprecatedÿ s z<TestDataFrameShift.test_shift_dt64values_int_fill_deprecatedÚas_catTz1_can_hold_element incorrectly always returns True)Úreason)ZmarksFÚvalsrË z
|
US/Pacific)r$ rW rw z 2020 DaysZInt64r ZFloat32c C s
|
t | jS r3 )Ústrr )Úxr r r r· : ó zTestDataFrameShift.<lambda>)Úidsc C sÞ t |}|r| d¡}td|i}|jdddd}tdddgi}t ||¡ t||d}| ¡ |jdddd}t|d ddgd}t ||¡ td|i}||d< t|jj d ks¾t
|
|jdddd}t ||¡ d S )
|
Nr rh rv r r¦ rÀ rÌ rn r ) r r@ r r r r
rÎ r r r r5 ) r rÒ rÐ rC rD r r r½ r¾ r r r Ú(test_shift_dt64values_axis1_invalid_fill s" %
|
z;TestDataFrameShift.test_shift_dt64values_axis1_invalid_fillc C sÎ t dddg}tddgddgdd gd
|
|d d
|d }|jdd }ttjtjgddgddgd
|
|d d
|d }t ||¡ |jddd }ttjtjgtjtjgddgd
|
|d d
|d }t ||¡ d S )Nr r r r r7 r r( r r¨ )r r r rv r} r )r r r r) rN r r
)r ÚcirD r r r r r Ú$test_shift_axis1_categorical_columnsV s(
|
ÿ
|
ÿ
|
ýz7TestDataFrameShift.test_shift_axis1_categorical_columnsc C sX t tj dd¡}|jddd d}|tj }t ||¡ |jddd d}t ||¡ d S )Nr r7 r¨ r rÀ iúÿÿÿ)r r) r* r r rN r r
)r rD rJ r rs r r r Ú test_shift_axis1_many_periodsl s
|
z0TestDataFrameShift.test_shift_axis1_many_periodsN)8Ú__name__Ú
|
__module__Ú __qualname__r< ÚmarkZ parametrizer) ÚemptyZonesrN r r. r r? r r r6 rE rK rQ rR ra rd rg rm ru r| r r r r r r¡ r¤ r¥ r§ r¯ r¿ ÚtdZ&skip_array_manager_not_yet_implementedr rÄ rÆ rÈ rÊ rÏ ÚparamZxfailr Z period_rangeZtimedelta_rangeZinterval_ranger r× rÙ rÚ r r r r r s¤ $þ
|
ýþ
|
|
|
|
+
|
ÿþùþ
|
ÿþÿþôë r )Únumpyr) r< Z pandas.util._test_decoratorsÚutilZ_test_decoratorsrà Zpandasr r r r r r r r Zpandas._testingZ_testingr r r r r r Ú<module> s $
|
