U
|
ý°d ã @ s> d dl Zd dlZd dlmZmZ d dlmZ G dd dZ dS )é N)Ú DataFrameÚSeriesc @ sú e Zd Zdd Zdd Zdd Zdd Zej d d
|
d g¡d d Z
|
ej d d
|
d g¡ej ddddge dddg¡g¡ej dddddgdddgdddggfddddgdddgdddggfg¡dd Z ej d ddd g¡d d Zd!d" Zd#d$ Zd S )%ÚTestDataFrameClipc C sL | ¡ ¡ }| ¡ }|j||d}|j|k ¡ r4t|j|jk ¡ sHtd S )N©ÚupperÚlower)ÚmedianÚcopyÚclipÚvaluesÚanyÚAssertionErrorÚall)ÚselfÚ float_framer ÚoriginalÚdouble© r ú[d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/methods/test_clip.pyÚ test_clip s
|
zTestDataFrameClip.test_clipc C sF | ¡ ¡ }| ¡ }|j||dd}|d ks0t|j|k ¡ rBtd S )NT)r r Úinplace)r r r
|
r r r )r r r Z
|
frame_copyZ return_valuer r r Útest_inplace_clip s
|
z#TestDataFrameClip.test_inplace_clipc C s® t tj dd¡}dD ]\}}| ||¡}t||t|| }}|j|k}|j|k}| | @ }|j| |k ¡ svt |j| |k ¡ st |j| |j| k ¡ st qd S )Néè é ))éÿÿÿÿé )r r )
|
r ÚnpÚrandomÚrandnr
|
ÚminÚmaxr r r )r ÚdfÚlbÚubÚ
|
clipped_dfÚlb_maskÚub_maskÚmaskr r r Útest_dataframe_clip s
|
|
z%TestDataFrameClip.test_dataframe_clipc C s t dddgdtjdgd}| dd¡}t dddgdtjdgd}t ||¡ t dddgdd d
|
ggd d d gd}|j}|jddj}t ||¡ d S )Nr r é g ð?ç @)ÚAÚBç @g333333 @é gffffff@ZfooÚbarZbaz©Úcolumns)r )r r Únanr
|
ÚtmÚassert_frame_equalZdtypesÚassert_series_equal)r r! ÚresultÚexpectedr r r Útest_clip_mixed_numeric. s "z)TestDataFrameClip.test_clip_mixed_numericr TFc C s t tj dd¡}ttj d¡}|d }| ¡ }|j||d|d}|rL|}tdD ]¼}|jd d
|f |k}|jd d
|f |k} | | @ }
|
|j ||f } t
|
j | || dd | j |ksÀt |j | |f } t
|
j | || dd | j |ksðt t
|
|j |
|
|f |j |
|
|f ¡ qTd S )Nr r r r )Úaxisr F)Z check_names)r r r r r r r
|
ÚrangeZilocÚlocr3 r5 Únamer ) r r r! r" r# r r$ Úir% r& r' r6 r r r Útest_clip_against_series; s$ z*TestDataFrameClip.test_clip_against_seriesr r r) r. zaxis,resr r- r* g @g @g @g @r c C sR |j dd}|j|dddg||d}t||j|jd}|r>|}tj||dd d S ) NT)Údeepé é é )r r r9 r )r1 Úindex)Z check_exact)r r
|
r r1 rC r3 r4 ) r Z simple_framer r r9 Úresr r6 r7 r r r Útest_clip_against_list_likeX s z-TestDataFrameClip.test_clip_against_list_liker9 Nc C s t tj dd¡}t tj dd¡}|d }|j|||d}||k}||k}| | @ }t || || ¡ t || || ¡ t || || ¡ d S )Nr r r )r9 )r r r r r
|
r3 r4 ) r r9 r! r" r# r$ r% r& r' r r r Útest_clip_against_framel s z)TestDataFrameClip.test_clip_against_framec
|
C sê t tj dd¡ddddgd}t tj dd¡ddddgd}t |jd ddddgd}|jd |d
|
}|jd ||j d
|
}|j|d d
|
}|j||j d d
|
}|j||d
|
}|j||j ||j d
|
} t ||¡ t ||¡ t || ¡ d S ) Nr r. r+ r, ÚCÚDr0 r r )r r r) ) r r r r r r
|
r1 r3 r4 )
|
r Zdf1Zdf2Zdf3Z result_upperZexpected_upperZ result_lowerZexpected_lowerZresult_lower_upperZexpected_lower_upperr r r Ú#test_clip_against_unordered_columns| s z5TestDataFrameClip.test_clip_against_unordered_columnsc C sR t | tj¡|¡ t |jtjtjd|¡ tdddgdddgdd d
|
gd }|jddtjgd d }tdddgdddgdd d
|
gd }t ||¡ |jddtjgdd }tdddgdddgdd d
|
gd }t ||¡ d
|
dd ddgdddd dgd}t|}tddtjddg}|j|d d }td
|
dd ddgdddd dgd}t ||¡ dS )z&Should process np.nan argument as Noner r r r) r. r@ rA rB é é )Úcol_0Úcol_1Zcol_2r )r r9 éýÿÿÿr éþÿÿÿiùÿÿÿéûÿÿÿ)rL rM éüÿÿÿN)r3 r4 r
|
r r2 r r ÚNaN)r r r! r6 r7 ÚdataÚtr r r Útest_clip_with_na_args s& " ÿ ÿ "z(TestDataFrameClip.test_clip_with_na_args)Ú__name__Ú
|
__module__Ú __qualname__r r r( r8 ÚpytestÚmarkZ parametrizer> r ZasarrayrE rF rI rU r r r r r s&
|
|
" þþ
|
r )
|
Únumpyr rY Zpandasr r Zpandas._testingZ_testingr3 r r r r r Ú<module> s
|
