Merge pull request #215 from openml/FIX/#199

FIX issue #189, delete cached dataset after test

Author: janvanrijn

tests/test_datasets/test_dataset_functions.py

@@ -1,5 +1,6 @@
 import unittest
 import os
+import shutil
 import sys
 
 if sys.version_info[0] >= 3:
@@ -26,6 +27,22 @@
 
 class TestOpenMLDataset(TestBase):
 
+    def setUp(self):
+        super(TestOpenMLDataset, self).setUp()
+        self._remove_did1()
+
+    def tearDown(self):
+        super(TestOpenMLDataset, self).tearDown()
+        self._remove_did1()
+
+    def _remove_did1(self):
+        cache_dir = self.static_cache_dir
+        did_1_dir = os.path.join(cache_dir, 'datasets', '1')
+        try:
+            shutil.rmtree(did_1_dir)
+        except:
+            pass
+
     def test__list_cached_datasets(self):
         openml.config.set_cache_directory(self.static_cache_dir)
         cached_datasets = openml.datasets.functions._list_cached_datasets()