U
    }hX                     @   s@   d Z ddlmZ ddlZddlmZ dd Zdd Zd	d
 ZdS )zTest the california_housing loader, if the data is available,
or if specifically requested via environment variable
(e.g. for CI jobs).    )partialN)check_return_X_yc                 C   sL   |  }d|j jkstd|jjks&t|jds6tt| }t|| d S )N)P     )r   z.. _california_housing_dataset:)datashapeAssertionErrortargetZDESCR
startswithr   r   )fetch_california_housing_fxtr   Z
fetch_func r   R/tmp/pip-unpacked-wheel-ig1s1lm8/sklearn/datasets/tests/test_california_housing.py
test_fetch   s    r   c                 C   sb   t d}| dd}|j}t|ddks,t|jdks:tt|j|jsLtt|j	|j
s^td S )NZpandasTZas_frameframe)r   	   )pytestZimportorskipr   hasattrr   r   
isinstancer   Z	DataFramer	   ZSeries)r   pdZbunchr   r   r   r   test_fetch_asframe   s    

r   c              	   C   s,   d}t jt|d | dd W 5 Q R X d S )Nz;fetch_california_housing with as_frame=True requires pandas)matchTr   )r   ZraisesImportError)r   Zhide_available_pandasZexpected_msgr   r   r   test_pandas_dependency_message    s    r   )	__doc__	functoolsr   r   Z"sklearn.datasets.tests.test_commonr   r   r   r   r   r   r   r   <module>   s   
