3
���hX � @ s� d Z ddlZddlZddlmZ ddlmZ ddlm Z ddl
m Z ddl j
ZddlmZ dd � Zejjd
ejdg�d d � �ZdS )
zg
Though Index.fillna and Series.fillna has separate impl,
test here to confirm these works as the same
� N)�iNaT)�needs_i8_conversion)�
ABCMultiIndex)�Index)� allow_na_opsc C sp | }t |t�rtjd� t|�dkr.|jd nd}|j|�}t |t�rTtj ||� n tj
||� ||k slt �d S )Nz MultiIndex doesn't support isnar ) �
isinstancer �pytest�skip�len�values�fillnar �tm�assert_index_equal�assert_series_equal�AssertionError)�index_or_series_obj�obj�
fill_value�result� r �?/tmp/pip-build-5_djhm0z/pandas/pandas/tests/base/test_fillna.py� test_fillna s
r �null_objc C s� |}t |�}t|�s&tj|� d�� n4t|�dk r>tjd� n t|t�rZtjd| � d�� |j}|d }|j� }t |j
�r�t |dd�< ||dd�< n| |dd�< ||dd�< ||�}||�}|j |�}t|t
�r�tj||� n tj||� ||k s�t�d S )Nz doesn't allow for NA operations� z%Test doesn't make sense on empty datazMultiIndex can't hold '�'r � )�typer r r r
r r r �copyr Zdtyper r r r
r r r )r r r �klassr r �expectedr r r r �test_fillna_null% s.
r )�__doc__Znumpy�npr Z pandas._libsr Zpandas.core.dtypes.commonr Zpandas.core.dtypes.genericr Zpandasr Zpandas._testingZ_testingr
Zpandas.tests.base.commonr r �markZ parametrize�nanr r r r r �<module> s
|