U
|
ý°d$ ã @ sR d dl Zd dlZd dlZd dlmZmZmZmZm Z d dl
|
m Z G dd dZ dS )é N)Ú DataFrameÚ DatetimeIndexÚIndexÚSeriesÚ
|
date_rangec
|
@ s e Zd Zdd Zdd Zdd Zdd Zej d d
|
d d d
|
gfd d d d
|
d gfd
|
d dd d
|
gfg¡ej de
|
j d¡dd Z dd Z dd Zd S )ÚTestDataFrameTruncatec C s° |d d d
}t ||¡}|jd |jd }}|jd }|jd }| ¡ }t ||¡ |dd
} | ||¡}t || ¡ | ||¡}t || ¡ |dd
} |j|d}t || ¡ |j|d}t || ¡ |d d
} |j|d}t || ¡ |j|d}t || ¡ |j|jd |jj d}t|dks8t|j|jd |jj d}t|dksdtd
|
}
|
tj t
|
|
|
d 0 |j|jd |jj |jd |jj d W 5 Q R X d S ) Né é é é é )Úbefore)Úafterr éÿÿÿÿz?Truncate: 2000-01-06 00:00:00 must be after 2000-02-04 00:00:00©Úmatch©r r ) ÚtmÚget_objÚindexÚtruncateÚ assert_equalÚfreqÚlenÚAssertionErrorÚpytestÚraisesÚ
|
ValueError) ÚselfZdatetime_frameÚframe_or_seriesÚtsÚstartÚendZ start_missingZ end_missingÚ truncatedÚexpectedÚmsg© r& ú_d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/methods/test_truncate.pyÚ test_truncate s@
|
|
ÿz#TestDataFrameTruncate.test_truncatec C s` t ddddddgiddd d
|
d gd }t ||¡}d }tjt|d |jdd
|
d W 5 Q R X d S )NÚAÚaÚbÚcÚdÚeé r r
|
é r ©r ú truncate requires a sorted indexr r )r r r r r r r )r r Úobjr% r& r& r' Ú test_truncate_nonsortedindexE s
|
$ z2TestDataFrameTruncate.test_truncate_nonsortedindexc C sv t dddd}ttj t|¡tj t|¡d|d}|jddd }d
|
}tjt |d |j
|
d d d W 5 Q R X d S )Nz
|
2011-01-01z
|
2012-01-01ÚW)r )r) ÚBr1 r) F)Z ascendingr2 r z2011-11z2011-12r ) r r ÚnpÚrandomÚrandnr Z sort_valuesr r r r )r Úrngr Z
|
decreasingr% r& r& r' Ú test_sort_values_nonsortedindexO s ÿz5TestDataFrameTruncate.test_sort_values_nonsortedindexc C sp t tj d¡tj d¡tj d¡tj d¡dddddgd}d}tjt|d |jddd
|
d W 5 Q R X d S ) Nr/ )r é r
|
r r r< r
|
r )Úcolumnsr2 r r )r r Zaxis)r r7 r8 r9 r r r r )r Údfr% r& r& r' Ú"test_truncate_nonsortedindex_axis1[ s
|
|
|
|
ü
|
ù z8TestDataFrameTruncate.test_truncate_nonsortedindex_axis1zbefore, after, indicesr r
|
Nr r Údtypúdatetime64[ns]c
|
C s t ddddg|d}t|trX|d k r0t |¡nd }|d k rFt |¡nd }dd |D }|tt||d}|j||d }|j| } t
|
|| ¡ d S )
|
Nr r
|
r r )Zdtypec S s g | ]}t |¡qS r& )ÚpdÚ Timestamp)Ú.0Úir& r& r' Ú
|
<listcomp>x s zHTestDataFrameTruncate.test_truncate_decreasing_index.<locals>.<listcomp>r1 r ) r Ú
|
isinstancer rB rC Úranger r Úlocr r )
|
r r r Úindicesr@ r ÚidxÚvaluesÚresultr$ r& r& r' Ú test_truncate_decreasing_indexk s
|
|
z4TestDataFrameTruncate.test_truncate_decreasing_indexc C s¬ t jjddddgddggddgd }tt|jd
|
|d gd }t ||¡}|jddd }t ddddgddddgddddgd¡}|
|
ddg¡}t ||¡}t ||¡ d S )Nr r
|
r é r) r6 ÚL1ÚL2)Únamesr Úcol)r r= r r/ )rP rQ rS ) rB Z
|
MultiIndexZ from_productr rH Úshaper r r Ú from_dictZ set_indexr )r r ÚmiÚs1rM r> r$ r& r& r' Útest_truncate_multiindex~ s $ "ÿ z.TestDataFrameTruncate.test_truncate_multiindexc C sH t dtddd d¡}|tkr,|jdd}| dd¡}t ||¡ d S ) Nr z
|
2021-06-30r1 r/ r* )Únamez
|
2021-06-28z
|
2021-07-01)r r Úrepeatr Zto_framer r r )r r r3 r# r& r& r' Ú)test_truncate_index_only_one_unique_value s
|
z?TestDataFrameTruncate.test_truncate_index_only_one_unique_value)rA )Ú__name__Ú
|
__module__Ú __qualname__r( r4 r; r? r ÚmarkZ parametrizer ZALL_REAL_NUMPY_DTYPESrN rX r[ r& r& r& r' r s 5
|
*þ r )Únumpyr7 r ZpandasrB r r r r r Zpandas._testingZ_testingr r r& r& r& r' Ú<module> s
|
|
