
    sgY                     :    d Z ddlmZ ddlZddlmZ d Zd Zd Zy)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                      |        }d|j                   j                  k(  sJ d|j                  j                  k(  sJ |j                  j	                  d      sJ t        |       }t        ||       y )N)P     )r   z.. _california_housing_dataset:)datashapetargetDESCR
startswithr   r   )fetch_california_housing_fxtr   
fetch_funcs      a/var/www/html/venv/lib/python3.12/site-packages/sklearn/datasets/tests/test_california_housing.py
test_fetchr      sh    ')D(((t{{(((((::  !BCCC 56JT:&    c                 "   t        j                  d      } | d      }|j                  }t        |d      du sJ |j                  dk(  sJ t        |j                  |j                        sJ t        |j                  |j                        sJ y )NpandasTas_frameframe)r   	   )
pytestimportorskipr   hasattrr	   
isinstancer   	DataFramer
   Series)r   pdbunchr   s       r   test_fetch_asframer       s|    			X	&B($7EKKE5'"d***;;*$$$ejj",,///ellBII...r   c                 z    d}t        j                  t        |      5   | d       d d d        y # 1 sw Y   y xY w)Nz;fetch_california_housing with as_frame=True requires pandas)matchTr   )r   raisesImportError)r   hide_available_pandasexpected_msgs      r   test_pandas_dependency_messager'   !   s5     QL	{,	7 4$d34 4 4s   
1:)	__doc__	functoolsr   r   "sklearn.datasets.tests.test_commonr   r   r    r'    r   r   <module>r,      s$      ?'/4r   