U
|
ý°d©' ã @ sN d dl Zd dlZd dlZd dlmZmZmZmZ d dl m
|
Z G dd dZ dS )é N)Ú DataFrameÚSeriesÚ TimestampÚ
|
date_rangec @ sæ e Zd Zdd Zej dde d¡g¡dd Z dd Z
|
d d
|
Z d d Z d d Z ej dddg¡dd Zej dddg¡dd Zej dddg¡dd Zej dddg¡dd Zdd Zd d Zd d Zd d! Zd"d# Zd$d% Zd&d' Zd(d) Zd*d+ Zd,d- Zej d.d/e ej d/dd/ej ej ej d/gej dej ej d0dej ej ge ej d1¡ej d2d3d4d5d6d7d8gd9d:d;fde e ej d1¡d/dej dej ej ej d/ge ej d1¡e ej d1¡d9d:d;fg¡d<d= Z d>d? Z d@dA Z!dS )BÚTestDataFrameDiffc C s: t tj dd¡}tjtdd | d¡ W 5 Q R X d S )Né zperiods must be an integer)Úmatchg ø?)r ÚnpÚrandomÚrandnÚpytestZraisesÚ
|
ValueErrorÚdiff©ÚselfÚdf© r ú[d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/methods/test_diff.pyÚtest_diff_requires_integer s z,TestDataFrameDiff.test_diff_requires_integerÚnumé c C s8 |}| |¡}|d |d |¡ }t |d |¡ d S )NÚA)r ÚshiftÚtmÚassert_series_equal)r Údatetime_framer r Úthe_diffÚexpectedr r r Ú test_diff s
|
zTestDataFrameDiff.test_diffc C s> d}|d }t ||g}td|i ¡ }|jd dks:td S )Nl _É r Ús)r r r r ÚAssertionError)r ÚaÚbÚserÚrsr r r Útest_diff_int_dtype s
|
z%TestDataFrameDiff.test_diff_int_dtypec C s: | d¡}| d¡}t |d |d |d d¡ ¡ d S )NÚfloat32r r )Úastyper r r r )r r Útfr r r r Útest_diff_mixed_numeric& s
|
|
z)TestDataFrameDiff.test_diff_mixed_numericc C s^ t tdgtdgd}| ddd¡ |jdd}t tjtdtdd}t ||¡ d S ) Nr é )ÚyÚzr Úxr ©Úaxis)r- r+ r, )r r Úinsertr r Únanr Úassert_frame_equal)r r Úresultr r r r Ú test_diff_axis1_nonconsolidated, s
|
z1TestDataFrameDiff.test_diff_axis1_nonconsolidatedc C sÒ t d¡ dd¡ d¡}t dd¡|d d
df< t|}|jddd }t|d tjt d¡t d¡gd
|
}t
|
||¡ | d¡}|| }|d ¡ ¡ stt
|
||¡ |jd dd }|t j }t
|
||¡ d S ) Né r* r ztimedelta64[ns]ÚNaTÚnsr r r. ©r r éÿÿÿÿ)r ÚarangeZreshaper' Z timedelta64r r Úpdr6 Ú Timedeltar Z assert_equalÚisnaÚallr r1 ©r Zarrr r3 r r r r Ú test_diff_timedelta64_with_nat4 s &
|
|
z0TestDataFrameDiff.test_diff_timedelta64_with_natÚtzNÚUTCc C sd t jdddg|d}t|}| ¡ }| ¡ }t t jt jt jddg¡}t| ¡ }t ||¡ d S )Nr6 z
|
2019-01-01z
|
2019-01-02)rA r )Údays)
|
r; Z DatetimeIndexr Úto_framer ÚTimedeltaIndexr6 r< r r2 )r rA Údtir# r r3 Zex_indexr r r r Ú!test_diff_datetime_axis0_with_natH s z3TestDataFrameDiff.test_diff_datetime_axis0_with_natc C s t dd|d}t|}| ¡ }| ¡ |d< tj|jd d
df< || }|d ¡ ¡ sZt |j
|
ddd}t ||¡ |j
|
ddd}t ||¡ d S )Nú
|
2016-01-01é )ÚperiodsrA r r r. ) r r rD Úcopyr; r6 Zilocr= r> r r r r2 )r rA rF r# r r r3 r r r Ú(test_diff_datetime_with_nat_zero_periodsU s z:TestDataFrameDiff.test_diff_datetime_with_nat_zero_periodsc C sd t tddd|dtddd|dd}|jdd}t t dd g¡t dd g¡d}t ||¡ d S )
|
NÚ2010ÚDr ©ÚfreqrJ rA r8 r r. r6 z1 days©r r r r; rE r r2 ©r rA r r3 r r r r Útest_diff_datetime_axis0i s þÿ þÿz*TestDataFrameDiff.test_diff_datetime_axis0c C sd t tddd|dtddd|dd}|jdd}t t ddg¡t d d g¡d}t ||¡ d S )
|
NrM rN r rO r8 r r. r6 z0 daysrQ rR r r r Útest_diff_datetime_axis1| s þÿ þÿz*TestDataFrameDiff.test_diff_datetime_axis1c C s\ t tdtdgddgd}| ¡ }t tjtjgt d¡dggdd gd
|
}t ||¡ d S ) Nz 20130101 9:01z 20130101 9:02ç ð?ç @)ÚtimeÚvaluez00:01:00r rW rX )Úcolumns)
|
r r r r; r6 r r1 r< r r2 )r r ÚresÚexpr r r Útest_diff_timedelta s þÿ ÿz%TestDataFrameDiff.test_diff_timedeltac C sN t tj dd¡}tjdddddgtd|d< | ¡ }|d jtjksJt d S ) Né r* r r rI ©Údtyper r )
|
r r r
|
r ÚarrayÚobjectr r_ Úfloat64r )r r r3 r r r Útest_diff_mixed_dtype s z'TestDataFrameDiff.test_diff_mixed_dtypec C s( | d¡}|| d¡ }t ||¡ d S )Nr9 )r r r r2 ©r r r$ Zxpr r r Útest_diff_neg_n¥ s
|
z!TestDataFrameDiff.test_diff_neg_nc C s$ | d¡}| d¡}t ||¡ d S )NrU r )r r r2 rd r r r Útest_diff_float_nª s
|
|
z#TestDataFrameDiff.test_diff_float_nc C sh t ddgddgg}t |jddt tjdgtjdgg¡ t |jddt tjtjgddgg¡ d S )NrU rV g @g @r r. r )r r r2 r r r1 r r r r Útest_diff_axis¯ s
|
ÿ
|
ÿz TestDataFrameDiff.test_diff_axisc C sL t ddd d¡}td|i}|jddd}|tj t¡}t ||¡ d S )NrH r* )rJ rN r r r. )
|
r Z to_periodr r r; r6 r' ra r r2 )r Úpir r3 r r r r Útest_diff_period¹ s
|
z"TestDataFrameDiff.test_diff_periodc C sº t tddtjdtjd d}t tjtjtjg|d d d}|jdd}t ||¡ t tjdddtjdd dd
|
}|jdd}t |d tj |d d d
|
}t ||¡ d S )Nr* r r^ ©r ÚBrk r r. r& rb )r! r" r! r" r © r Úranger r: rb r1 r r r2 ©r r r r3 r r r Ú test_diff_axis1_mixed_dtypesà s "" ÿ z.TestDataFrameDiff.test_diff_axis1_mixed_dtypesc C sJ t tddtjdtjd d}|tj }|jddd}t ||¡ d S )Nr* r r^ rj r ©r/ rJ rl rn r r r Ú*test_diff_axis1_mixed_dtypes_large_periodsÔ s "
|
z<TestDataFrameDiff.test_diff_axis1_mixed_dtypes_large_periodsc C s` t tddtjdtjd d}t d|d |d tj d}|jdd d
|
}t ||¡ d S ) Nr* r r^ rj ç ð¿r rk r r9 rp rl rn r r r Ú-test_diff_axis1_mixed_dtypes_negative_periodsÝ s " z?TestDataFrameDiff.test_diff_axis1_mixed_dtypes_negative_periodsc C sT t ddgddggdd}| ¡ }t tjtjgddggt dd¡d}t ||¡ d S ) Nr r z Sparse[int]r^ rU rr Úfloatg )r r r r1 r; Z SparseDtyper r2 )r Z sparse_dfr3 r r r r Útest_diff_sparseæ s
|
ÿz"TestDataFrameDiff.test_diff_sparsez axis,expectedr éþÿÿÿé r* r] é é é é é ©r! r" ÚcÚdÚInt64r^ c C sl t t ddtjdgd¡t ddtjdgd¡t tjd¡t dd¡d ddd}|j|d }t ||¡ d S )
|
Nr r r rw ry r} r r^ r. ) r r Úrepeatr1 Ztiler: r r r2 )r r/ r r r3 r r r Útest_diff_integer_nañ s üù z&TestDataFrameDiff.test_diff_integer_nac C sH t j dd¡}d|j_t|}| ¡ }tt |¡ ¡ }t ||¡ d S )Nr] r F)
|
r r
|
r ÚflagsZ writeabler r r` r r2 r? r r r Útest_diff_readonly s z$TestDataFrameDiff.test_diff_readonlyc C sV t td}| |¡}| ¡ }|dkr*dnd}t tjddddg|d}t ||¡ d S )Nr] )Zint8Úint16r& rb rU r^ )r rm r' r r r1 r r2 )r Zany_int_numpy_dtyper r3 Zexpected_dtyper r r r Útest_diff_all_int_dtype' s
|
ÿz)TestDataFrameDiff.test_diff_all_int_dtype)"Ú__name__Ú
|
__module__Ú __qualname__r r ÚmarkZ parametrizer Úint64r r% r) r4 r@ rG rL rS rT r\ rc re rf rg ri ro rq rs ru r r1 r r r r r r r r r sh
|
|
|
|
|
|
|
üùþ üùþóþ
|
r ) Únumpyr r Zpandasr; r r r r Zpandas._testingZ_testingr r r r r r Ú<module> s
|
|
