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
|
"""Test the covtype loader.
Skipped if covtype is not already downloaded to data_home.
"""
from sklearn.datasets import fetch_covtype
from sklearn.utils.testing import assert_equal, SkipTest
from sklearn.datasets.tests.test_common import check_return_X_y
from functools import partial
def fetch(*args, **kwargs):
return fetch_covtype(*args, download_if_missing=False, **kwargs)
def test_fetch():
try:
data1 = fetch(shuffle=True, random_state=42)
except IOError:
raise SkipTest("Covertype dataset can not be loaded.")
data2 = fetch(shuffle=True, random_state=37)
X1, X2 = data1['data'], data2['data']
assert_equal((581012, 54), X1.shape)
assert_equal(X1.shape, X2.shape)
assert_equal(X1.sum(), X2.sum())
y1, y2 = data1['target'], data2['target']
assert_equal((X1.shape[0],), y1.shape)
assert_equal((X1.shape[0],), y2.shape)
# test return_X_y option
fetch_func = partial(fetch)
check_return_X_y(data1, fetch_func)
|
