U
|
ý°d¡ ã @ sP d dl Zd dlZd dlZd dlmZ d dlm Z d dl
|
m Z G dd de Z dS )é N)Ú is_sparse)ÚBaseExtensionTestsc @ s e Zd Zdd Zej dddg¡dd Zdd Zd
|
d Z d d Z
|
dd Z dd Z dd Z dd Zdd Zdd Zdd Zd d Zd d Zd S )!ÚBaseMissingTestsc C s~ t ddg¡}t |¡}t ||¡ t |¡ ¡ }t |¡}| ||¡ t |¡ ddg¡ ¡ }tjg t d}| ||¡ d S )NTFr é ©Údtype)
|
ÚnpÚarrayÚpdÚisnaÚtmZassert_numpy_array_equalÚSeriesÚassert_series_equalZdropÚbool)ÚselfÚ data_missingÚexpectedÚresult© r úZd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/extension/base/missing.pyÚ test_isna s
|
|
zBaseMissingTests.test_isnaÚna_funcr Znotnac C sL t |¡}| ¡ }t|| }t|r0t |¡}d|d d
< | ||¡ d S )NT)r
|
r ÚcopyÚgetattrr r r r )r r r r r Úmaskr r r Útest_isna_returns_copy s
|
|
z'BaseMissingTests.test_isna_returns_copyc C s" | ¡ }|dg }| ||¡ d S ©Nr )ÚdropnaÚ assert_extension_array_equal)r r r r r r r Útest_dropna_array% s
|
z"BaseMissingTests.test_dropna_arrayc C s. t |¡}| ¡ }|jdg }| ||¡ d S r )r
|
r r Úilocr )r r Úserr r r r r Útest_dropna_series* s
|
z#BaseMissingTests.test_dropna_seriesc C s t d|i¡}| ¡ }|jdg }| ||¡ |jdd}t jt d¡t g ¡d}| ||¡ t |dtjgd¡}| ¡ }|jd d
}| ||¡ d S ) NÚAr Úcolumns)Zaxisé )Úindexr$ ©r# ÚBr ) r
|
Ú DataFramer r Úassert_frame_equalZ
|
RangeIndexZIndexr Únan)r r Zdfr r r r r Útest_dropna_frame0 s z"BaseMissingTests.test_dropna_framec C s, |d }| |¡}| |¡}| ||¡ d S r )Úfillnar )r r Úvalidr r r r r Útest_fillna_scalarC s
|
|
z#BaseMissingTests.test_fillna_scalarc C sR | dddddg¡}t |¡jddd}t | dddddg¡¡}| ||¡ d S )Nr r Úffillr% ©ÚmethodÚlimit©Ztaker
|
r r- r ©r r Zarrr r r r r Útest_fillna_limit_padI s z&BaseMissingTests.test_fillna_limit_padc C sR | dddddg¡}t |¡jddd}t | dddddg¡¡}| ||¡ d S )Nr r Úbackfillr% r1 r4 r5 r r r Útest_fillna_limit_backfillO s z+BaseMissingTests.test_fillna_limit_backfillc C s` || ¡ }|d }| |¡}||k s,t| ||¡ |jdd}||k sPt| ||¡ d S )Nr r7 ©r2 )r r- ÚAssertionErrorr )r Údatar. r r r r Ú test_fillna_no_op_returns_copyU s
|
z/BaseMissingTests.test_fillna_no_op_returns_copyc C sr |d }t |¡}| |¡}t |j||g|jd¡}| ||¡ | |¡}| ||¡ | |¡}| ||¡ d S )Nr r ©r
|
r r- Ú_from_sequencer r )r r Ú
|
fill_valuer! r r r r r Útest_fillna_seriesa s
|
|
ÿÿ
|
|
z#BaseMissingTests.test_fillna_seriesc C sZ |d }|dkr |d d d
}t |¡j|d}t |j||g|jd¡}| ||¡ d S )Nr r0 éÿÿÿÿr9 r r= )r r Z fillna_methodr? r r r r r Útest_fillna_series_methodu s ÿÿz*BaseMissingTests.test_fillna_series_methodc C sV |d }t |ddgd¡ |¡}t |j||g|jdddgd¡}| ||¡ d S )Nr r% r' r )r
|
r) r- r> r r* )r r r? r r r r r Útest_fillna_frame s ÿüÿ z"BaseMissingTests.test_fillna_framec C sP t |tjgt| d¡ ddi¡}t |dgt| d¡}| ||¡ d S )Nr' r( g )r
|
r) r r+ Úlenr- r* )r r; r r r r r Útest_fillna_fill_other s &z'BaseMissingTests.test_fillna_fill_otherc C sB t |¡}| ¡ }t dd¡ | ¡ }W 5 Q R X | ||¡ d S )Nzmode.use_inf_as_naT)r
|
r r Zoption_contextr )r r r! r r r r r Ú test_use_inf_as_na_no_effect s
|
|
z-BaseMissingTests.test_use_inf_as_na_no_effectN)Ú__name__Ú
|
__module__Ú __qualname__r ÚpytestÚmarkZ parametrizer r r" r, r/ r6 r8 r< r@ rB rC rE rF r r r r r
|
s
|
|
r ) Únumpyr rJ Zpandasr
|
Zpandas._testingZ_testingr Zpandas.api.typesr Z pandas.tests.extension.base.baser r r r r r Ú<module> s
|
