add: change tests to actually check the ID

Author: LennartPurucker

tests/test_tasks/test_classification_task.py

@@ -15,7 +15,7 @@ def setUp(self, n_levels: int = 1):
         super().setUp()
         self.task_id = 119  # diabetes
         self.task_type = TaskType.SUPERVISED_CLASSIFICATION
-        self.estimation_procedure = 1
+        self.estimation_procedure = 5
 
     def test_get_X_and_Y(self):
         X, Y = super().test_get_X_and_Y()
@@ -30,7 +30,8 @@ def test_download_task(self):
         assert task.task_id == self.task_id
         assert task.task_type_id == TaskType.SUPERVISED_CLASSIFICATION
         assert task.dataset_id == 20
+        assert task.estimation_procedure_id == self.estimation_procedure
 
     def test_class_labels(self):
         task = get_task(self.task_id)
-        assert task.class_labels == ["tested_negative", "tested_positive"]
+        assert task.class_labels == ["tested_negative", "tested_positive"]

tests/test_tasks/test_regression_task.py

@@ -18,11 +18,11 @@ class OpenMLRegressionTaskTest(OpenMLSupervisedTaskTest):
 
     def setUp(self, n_levels: int = 1):
         super().setUp()
-
+        self.estimation_procedure = 9
         task_meta_data = {
             "task_type": TaskType.SUPERVISED_REGRESSION,
             "dataset_id": 105,  # wisconsin
-            "estimation_procedure_id": 7,
+            "estimation_procedure_id": self.estimation_procedure, # non default value to test estimation procedure id
             "target_name": "time",
         }
         _task_id = check_task_existence(**task_meta_data)
@@ -46,7 +46,7 @@ def setUp(self, n_levels: int = 1):
                     raise Exception(repr(e))
         self.task_id = task_id
         self.task_type = TaskType.SUPERVISED_REGRESSION
-        self.estimation_procedure = 7
+
 
     def test_get_X_and_Y(self):
         X, Y = super().test_get_X_and_Y()
@@ -61,3 +61,4 @@ def test_download_task(self):
         assert task.task_id == self.task_id
         assert task.task_type_id == TaskType.SUPERVISED_REGRESSION
         assert task.dataset_id == 105
+        assert task.estimation_procedure_id == self.estimation_procedure