adapted dataset testcases to new assumption that ignore_attribute is a list

Author: janvanrijn

tests/test_datasets/test_dataset.py

@@ -127,7 +127,7 @@ def test_get_data_with_rowid(self):
         #rval = self.dataset.get_pandas(include_row_id=False)
 
     def test_get_sparse_dataset_with_rowid(self):
-        self.sparse_dataset.row_id_attribute = "a_0"
+        self.sparse_dataset.row_id_attribute = ["a_0"]
         rval, categorical = self.sparse_dataset.get_data(
             include_row_id=True, return_categorical_indicator=True)
         self.assertIsInstance(rval, np.ndarray)
@@ -147,7 +147,7 @@ def test_get_sparse_dataset_with_rowid(self):
         #rval = self.dataset.get_pandas(include_row_id=False)
 
     def test_get_data_with_ignore_attributes(self):
-        self.dataset.ignore_attributes = "condition"
+        self.dataset.ignore_attributes = ["condition"]
         rval = self.dataset.get_data(include_ignore_attributes=True)
         self.assertEqual(rval.dtype, np.float32)
         self.assertEqual(rval.shape, (898, 39))
@@ -163,7 +163,7 @@ def test_get_data_with_ignore_attributes(self):
         # TODO test multiple ignore attributes!
 
     def test_get_sparse_dataset_with_ignore_attributes(self):
-        self.sparse_dataset.ignore_attributes = "a_0"
+        self.sparse_dataset.ignore_attributes = ["a_0"]
         rval = self.sparse_dataset.get_data(include_ignore_attributes=True)
         self.assertIsInstance(rval, np.ndarray)
         self.assertEqual(rval.dtype, np.float32)
@@ -183,8 +183,8 @@ def test_get_sparse_dataset_with_ignore_attributes(self):
         # TODO test multiple ignore attributes!
 
     def test_get_data_rowid_and_ignore_and_target(self):
-        self.dataset.ignore_attributes = "condition"
-        self.dataset.row_id_attribute = "hardness"
+        self.dataset.ignore_attributes = ["condition"]
+        self.dataset.row_id_attribute = ["hardness"]
         X, y = self.dataset.get_data(target="class", include_row_id=False,
                                      include_ignore_attributes=False)
         self.assertEqual(X.dtype, np.float32)
@@ -198,8 +198,8 @@ def test_get_data_rowid_and_ignore_and_target(self):
         self.assertEqual(y.shape, (898, ))
 
     def test_get_sparse_dataset_rowid_and_ignore_and_target(self):
-        self.sparse_dataset.ignore_attributes = "a_0"
-        self.sparse_dataset.row_id_attribute = "a_1"
+        self.sparse_dataset.ignore_attributes = ["a_0"]
+        self.sparse_dataset.row_id_attribute = ["a_1"]
         X, y = self.sparse_dataset.get_data(
             target="class", include_row_id=False,
             include_ignore_attributes=False)