U
|
ý°d"
|
ã @ sB d Z ddlZddlZddlmZmZ ddlmZ G dd dZ
|
dS )z%
|
Includes test for last_valid_index.
|
é N)Ú DataFrameÚSeriesc @ s< e Zd Zdd Zej de ee deg g e dg¡dd Z
|
ej dd d
|
d d gid
|
d
|
d gd
|
d fd d
|
d d gid
|
d d gd
|
d fd d
|
d d d giddddgddfd d
|
e j d gid
|
d
|
d gd
|
d fd e j e j d gid
|
d
|
d gd d fd d
|
e j d gid
|
d d gd
|
d fg¡dd Z ej dejejg¡dd Zej dejejg¡dd ZdS )ÚTestFirstValidIndexc C s: |t jg}| ¡ d ks t|jd d
¡ d ks6td S )Nr )ÚnpÚnanÚfirst_valid_indexÚAssertionErrorZiloc)ÚselfZframe_or_seriesÚobj© r úhd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/methods/test_first_valid_index.pyÚ!test_first_valid_index_single_nan s z5TestFirstValidIndex.test_first_valid_index_single_nanÚempty)Údtype)Úindexr c C s$ | ¡ d kst| ¡ d ks td S )N)Úlast_valid_indexr r )r r r r r Ú test_first_valid_index_empty s z0TestFirstValidIndex.test_first_valid_index_emptyz%data,idx,expected_first,expected_lastÚAé é é é Údc C s0 t ||d}|| ¡ ks t|| ¡ ks,td S )N©r )r r r r )r ÚdataÚidxZexpected_firstZ expected_lastZdfr r r Útest_first_last_valid_frame s z/TestFirstValidIndex.test_first_last_valid_frameÚ
|
index_funcc C s¨ d}||}t j |¡}t j|d d
< t j|dd
< td|i|d}| ¡ |jd ksZt| ¡ |jd kspt|d }| ¡ |jd kst| ¡ |jd ks¤td S )Né é éûÿÿÿÚfoor iúÿÿÿ) r ÚrandomZrandnr r r r r r )r r ÚNr ÚmatÚframeÚserr r r Útest_first_last_valid/ s z)TestFirstValidIndex.test_first_last_validc C sf |d}t tjdg|d}| ¡ d ks*t| ¡ d ks:t|d }| ¡ d ksRt| ¡ d ksbtd S )Nr r! )Úcolumnsr )r r r r r r )r r r r% r& r r r Ú test_first_last_valid_all_nan? s z1TestFirstValidIndex.test_first_last_valid_all_nanN)Ú__name__Ú
|
__module__Ú __qualname__r ÚpytestÚmarkZ parametrizer r Úobjectr r r r ÚtmZmakeStringIndexZ makeDateIndexr' r) r r r r r s* ÿ
|
úþ
|
|
r ) Ú__doc__Únumpyr r- Zpandasr r Zpandas._testingZ_testingr0 r r r r r Ú<module> s
|
|
