B
    0d_                 @   s@   d Z ddlZddlm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 travis cron job).    N)check_return_X_y)partialc             C   sL   |  }d|j jkstd|jjks&t|jds6tt| }t|| d S )N)iP     )iP  z.. _california_housing_dataset:)datashapeAssertionErrortargetZDESCR
startswithr   r   )fetch_california_housing_fxtr   Z
fetch_func r   `/var/www/html/venv/lib/python3.7/site-packages/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)as_frameframe)iP  	   )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 d Q R X d S )Nz;fetch_california_housing with as_frame=True requires pandas)matchT)r   )r   ZraisesImportError)r
   Zhide_available_pandasZexpected_msgr   r   r   test_pandas_dependency_message   s    r   )	__doc__r   Z"sklearn.datasets.tests.test_commonr   	functoolsr   r   r   r   r   r   r   r   <module>   s   
