U
|
ý°d& ã @ sn d Z ddlZddlmZmZmZmZmZm Z m
|
Z
|
ddl m Z G dd dZdd Zdd Zd
|
d Zd d ZdS )zC
|
Tests for DataFrame.mask; tests DataFrame.where as a side-effect.
|
é N)ÚNAÚ DataFrameÚ Float64DtypeÚSeriesÚ StringDtypeÚ TimedeltaÚisnac @ s< e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd S )ÚTestDataFrameMaskc C s¢ t tj dd¡}|dk}| |tj¡}t || |dk¡¡ t || | ¡¡ t tj dd¡}| ||¡}t || |dk|¡¡ t || | |¡¡ d S )Né é r ) r ÚnpÚrandomÚrandnÚwhereÚnanÚtmÚassert_frame_equalÚmask)ÚselfÚdfÚcondÚrsÚother© r ú\d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/indexing/test_mask.pyÚ test_mask s zTestDataFrameMask.test_maskc C s< t ddg}| dgdgg¡}t tjdg}t ||¡ d S ©Né é TF©r r r r r r )r r ÚresÚexpr r r Ú
|
test_mask2! s z TestDataFrameMask.test_mask2c C sº t tj dd¡}|dk}| ¡ }|j|dd}|d ks<tt || |¡¡ t || | ¡¡ | ¡ }|j|| dd}|d kstt || || ¡¡ t || | | ¡¡ d S )Nr
|
r r T©Zinplace)
|
r r r r Úcopyr ÚAssertionErrorr r r )r r r ÚrdfZ return_valuer r r Útest_mask_inplace) s z#TestDataFrameMask.test_mask_inplacec C sB t ddgg}| t ddgg¡}t tjdgg}t ||¡ d S r r )r r r Zexpecr r r Ú test_mask_edge_case_1xN_frame; s z/TestDataFrameMask.test_mask_edge_case_1xN_framec C sH t dddgdddgddd gg}| d
|
d d d ¡}t dddgdddgdd d gg}t ||¡ t || |dk|d ¡¡ | dd dd ¡}t dddgdddgddd gg}t ||¡ t || |d dkd¡¡ |d dd dd ¡}t dddgdddgdddgg}t ||¡ t ||d |d dk|d d ¡¡ d S )Nr r r é r
|
é é é é c S s | dkS )Nr) r ©Úxr r r Ú<lambda>E ó z6TestDataFrameMask.test_mask_callable.<locals>.<lambda>c S s | d S )Nr r r. r r r r0 E r1 é
|
c S s | d dkj S )Nr r )Úvaluesr. r r r r0 K r1 c S s dS )Néc r r. r r r r0 K r1 r4 r c S s | dkS )Nr, r r. r r r r0 Q r1 c S s | d S )Nr2 r r. r r r r0 Q r1 é é é )r r r r )r r Úresultr! r r r Útest_mask_callableB s z$TestDataFrameMask.test_mask_callablec C s\ t tj dd¡d}| |dk¡}|dk}t|}| t¡ |¡}| |¡}t
|
||¡ d S )Néd é2 )Údatar ) r r r r r r ZastypeÚobjectr r r )r r Zboolsr Úexpectedr8 r r r Ú test_mask_dtype_bool_conversionV s
|
z1TestDataFrameMask.test_mask_dtype_bool_conversionN) Ú__name__Ú
|
__module__Ú __qualname__r r" r' r( r9 r? r r r r r s r c C s° t ddddtgiddddgt d }t dd
|
d giddgt d }t dtd
|
d tgiddddgt d }| tkr|d }|d }|d }td d d d g}| ||¡}t ||¡ d S )NÚAZfooÚbarZbazZid1Zid2Zid3Zid4)ÚindexÚdtypeÚthisÚthatFT)r r r r r r Z assert_equal)Zframe_or_seriesÚobjZ filtered_objr> Z
|
filter_serr8 r r r Útest_mask_stringdtypea s,
|
ý
|
ÿ
|
ý rJ c C s t dd tdD } t tjdtjdd}t | | ¡ ¡|¡ t tjtjtjt dt dg}t |
|
| t dd d
|
k¡|¡ d S ) Nc S s g | ]}t |d dqS )Úd©Úunit)r )Ú.0Úir r r Ú
|
<listcomp>} s z3test_mask_where_dtype_timedelta.<locals>.<listcomp>r
|
ztimedelta64[ns]©rF z3 dayz4 dayr rK rL ) r Úranger Úfullr r r r Znotnar r )r r> r r r Ú test_mask_where_dtype_timedelta{ s ÿrT c C sb t ddddgt d} | ¡ }t ddddg}t ddddg| jd}| ||¡}t ||¡ d S )Ng g ð?g @g @rQ TF)r r r rF r r Zassert_series_equal)Zserr r Zexceptedr8 r r r Útest_mask_return_dtype s
|
rU c C sd t ddgddgd} t ddgddgd}| j|dd t tjdgdtjgd}t | |¡ d S ) Nr r r/ Úy)ÚaÚbTFr# r )r r r> r r r Útest_mask_inplace_no_other s
|
rY )Ú__doc__Únumpyr Zpandasr r r r r r r Zpandas._testingZ_testingr r rJ rT rU rY r r r r Ú<module> s $ N
|
