TST covertype loader

larsmans · amueller · commit 81428f70ce21 · 2013-02-13T21:01:19.000+01:00
diff --git a/sklearn/datasets/tests/test_covtype.py b/sklearn/datasets/tests/test_covtype.py
@@ -0,0 +1,32 @@
+"""Test the covtype loader.
+
+Skipped if covtype is not already downloaded to data_home.
+"""
+
+import errno
+from sklearn.datasets import fetch_covtype
+from sklearn.utils.testing import assert_equal, SkipTest
+
+
+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 as e:
+        if e.errno == errno.ENOENT:
+            raise SkipTest()
+
+    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)
