U
|
ý°dp ã @ s: d dl Zd dlZd dlmZ d dlmZ G dd dZdS )é N)ÚCategoricalIndexc @ s$ e Zd Zdd Zdd Zdd ZdS )Ú
|
TestFillNAc C s t dtjddgdd}t ddddgdd}t | d¡|¡ |j}d}tjt |d | d¡ W 5 Q R X | d¡}|
|
t ¡ d¡}t ||¡ d S )Ng ð?g @Úx)Únameú3Cannot setitem on a Categorical with a new category©Úmatchg @) r ÚnpÚnanÚtmÚassert_index_equalÚfillnaÚ_dataÚpytestÚraisesÚ TypeErrorZastypeÚobject)ÚselfÚidxÚexpÚcatÚmsgÚresultÚexpected© r úcd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/indexes/categorical/test_fillna.pyÚtest_fillna_categorical s
|
z"TestFillNA.test_fillna_categoricalc C sx t dddg}| d¡}||k s$tt ||¡s4t|j}| d¡}|j|jk sTt|jjd ksdtt ||¡rttd S )Nr é )r r ÚAssertionErrorr Z shares_memoryr Z_ndarrayÚbase)r Úcir r r r r Ú test_fillna_copies_with_no_nas s
|
|
z)TestFillNA.test_fillna_copies_with_no_nasc C sV t dddg}|j}d}| d¡}t ||¡ tjt|d | d¡ W 5 Q R X d S )Né é r Fr )r r r r r r r r )r r r r Úresr r r Ú!test_fillna_validates_with_no_nas* s
|
z,TestFillNA.test_fillna_validates_with_no_nasN)Ú__name__Ú
|
__module__Ú __qualname__r r! r% r r r r r s r ) Únumpyr r Zpandasr Zpandas._testingZ_testingr r r r r r Ú<module> s
|
