added testcases

Author: janvanrijn

tests/test_runs/test_run_functions.py

@@ -2,12 +2,15 @@
 import arff
 import time
 
+import numpy as np
+
 import openml
 import openml.exceptions
 import openml._api_calls
+import sklearn
 
 from openml.testing import TestBase
-from openml.runs.functions import _run_task_get_arffcontent
+from openml.runs.functions import _run_task_get_arffcontent, _get_seeded_model
 
 from sklearn.model_selection._search import BaseSearchCV
 from sklearn.tree import DecisionTreeClassifier, ExtraTreeClassifier
@@ -175,6 +178,39 @@ def test_run_and_upload(self):
                 check_res = self._check_serialized_optimized_run(run.run_id)
                 self.assertTrue(check_res)
 
+    def test_get_seeded_model(self):
+        randomized_clfs = [
+            BaggingClassifier(),
+            RandomizedSearchCV(RandomForestClassifier(),
+                               {"max_depth": [3, None],
+                                "max_features": [1, 2, 3, 4],
+                                "bootstrap": [True, False],
+                                "criterion": ["gini", "entropy"],
+                                "random_state" : [-1, 0, 1, 2]},
+                               ),
+            DummyClassifier()
+        ]
+
+        for clf in randomized_clfs:
+            const_probe = 42
+            clf_seeded = _get_seeded_model(clf, const_probe)
+            all_params = clf_seeded.get_params()
+            params = [key for key in all_params if key.endswith('random_state')]
+            self.assertGreater(len(params), 0)
+
+            for param in params:
+                self.assertTrue(isinstance(all_params[param], int))
+                self.assertIsNotNone(all_params[param])
+
+    def test_get_seeded_model_raises(self):
+        randomized_clfs = [
+            BaggingClassifier(random_state=np.random.RandomState(42)),
+            DummyClassifier(random_state="OpenMLIsGreat")
+        ]
+
+        for clf in randomized_clfs:
+            self.assertRaises(ValueError, _get_seeded_model, model=clf, seed=42)
+
     def test_run_with_classifiers_in_param_grid(self):
         task = openml.tasks.get_task(115)
 

tests/test_setups/test_setup_functions.py

@@ -75,7 +75,7 @@ def test_existing_setup_exists(self):
         setup_id = openml.setups.setup_exists(flow, classif)
         self.assertEquals(setup_id, run.setup_id)
 
-    def test_setup_get(self):
+    def test_get_setup(self):
         # no setups in default test server
         openml.config.server = 'https://www.openml.org/api/v1/xml/'