1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
|
"""Test the california_housing loader, if the data is available,
or if specifically requested via environment variable
(e.g. for CI jobs)."""
from functools import partial
import pytest
from sklearn.datasets.tests.test_common import check_return_X_y
def test_fetch(fetch_california_housing_fxt):
data = fetch_california_housing_fxt()
assert (20640, 8) == data.data.shape
assert (20640,) == data.target.shape
assert data.DESCR.startswith(".. _california_housing_dataset:")
# test return_X_y option
fetch_func = partial(fetch_california_housing_fxt)
check_return_X_y(data, fetch_func)
def test_fetch_asframe(fetch_california_housing_fxt):
pd = pytest.importorskip("pandas")
bunch = fetch_california_housing_fxt(as_frame=True)
frame = bunch.frame
assert hasattr(bunch, "frame") is True
assert frame.shape == (20640, 9)
assert isinstance(bunch.data, pd.DataFrame)
assert isinstance(bunch.target, pd.Series)
def test_pandas_dependency_message(fetch_california_housing_fxt, hide_available_pandas):
# Check that pandas is imported lazily and that an informative error
# message is raised when pandas is missing:
expected_msg = "fetch_california_housing with as_frame=True requires pandas"
with pytest.raises(ImportError, match=expected_msg):
fetch_california_housing_fxt(as_frame=True)
|
