some more tests

automl · Dec 11, 2023 · 4eca801 · 4eca801
1 parent 36f2c3b
commit 4eca801
Show file tree

Hide file tree

Showing 6 changed files with 83 additions and 10 deletions.
diff --git a/carl/context/context_space.py b/carl/context/context_space.py
@@ -219,7 +219,7 @@ def sample_contexts(
 
         contexts = []
         for _ in range(size):
-            context = {cf.name: cf.sample() for cf in self.context_space.values()}
+            context = {cf.name: cf.rvs() for cf in self.context_space.values()}
             context = self.insert_defaults(context, context_keys)
             contexts += [context]
 

diff --git a/carl/envs/gymnasium/carl_gymnasium_env.py b/carl/envs/gymnasium/carl_gymnasium_env.py
@@ -10,10 +10,10 @@
 
 try:
     pygame.display.init()
-except:
-    import os
+except: # pragma: no cover
+    import os # pragma: no cover
 
-    os.environ["SDL_VIDEODRIVER"] = "dummy"
+    os.environ["SDL_VIDEODRIVER"] = "dummy" # pragma: no cover
 
 
 class CARLGymnasiumEnv(CARLEnv):

diff --git a/test/test_all_envs.py b/test/test_all_envs.py
@@ -16,6 +16,7 @@ def test_init_all_envs(self):
                     env = (  # noqa: F841 local variable is assigned to but never used
                         var()
                     )
+                    env.reset()
                 except Exception as e:
                     print(f"Cannot instantiate {var} environment.")
                     raise e
diff --git a/test/test_context_sampler.py b/test/test_context_sampler.py
@@ -1,4 +1,5 @@
 import unittest
+from omegaconf import DictConfig 
 
 from carl.context.context_space import (
     ContextSpace,
@@ -44,11 +45,25 @@ def test_init(self):
             name="TestSampler",
         )
 
+        with self.assertRaises(ValueError):
+            ContextSampler(
+                context_distributions=0, 
+                context_space=self.cspace,
+                seed=0,
+                name="TestSampler",
+            )
+
     def test_sample_contexts(self):
         contexts = self.sampler.sample_contexts(n_contexts=3)
         self.assertEqual(len(contexts), 3)
         self.assertEqual(contexts[0]["gravity"], 9.8)
 
+        contexts = self.sampler.sample_contexts(n_contexts=1)
+        self.assertEqual(len(contexts), 1)
+        self.assertEqual(contexts[0]["gravity"], 9.8)
+
+
+
 
 if __name__ == "__main__":
     unittest.main()
diff --git a/test/test_context_space.py b/test/test_context_space.py
@@ -88,6 +88,25 @@ def test_verify_context(self):
         is_valid = self.context_space.verify_context(context)
         self.assertEqual(is_valid, False)
 
+    def test_sample(self):
+        context = self.context_space.sample_contexts(["gravity"], size=1)
+        is_valid = self.context_space.verify_context(context)
+        self.assertEqual(is_valid, True)
+
+        contexts = self.context_space.sample_contexts(["gravity"], size=10)
+        self.assertTrue(len(contexts) == 10)
+        for context in contexts:
+            is_valid = self.context_space.verify_context(context)
+            self.assertEqual(is_valid, True)
+
+        contexts = self.context_space.sample_contexts(None, size=10)
+        self.assertTrue(len(contexts) == 10)
+        for context in contexts:
+            is_valid = self.context_space.verify_context(context)
+            self.assertEqual(is_valid, True)
+
+        with self.assertRaises(ValueError):
+            self.context_space.sample_contexts(["false_feature"], size=0)
 
 if __name__ == "__main__":
     unittest.main()
diff --git a/test/test_search_space_encoding.py b/test/test_search_space_encoding.py
@@ -6,20 +6,58 @@
 from carl.context.search_space_encoding import search_space_to_config_space
 
 
+dict_space = {
+                "uniform_integer": (1, 10),
+                "uniform_float": (1.0, 10.0),
+                "categorical": ["a", "b", "c"],
+                "constant": 1337,
+            }
+
+dict_space_2 = { "hyperparameters": [
+    {"name": "x0", 
+    "type": "uniform_float", 
+    "log": False,
+    "lower": -512.0,
+    "upper": 512.0, 
+    "default": -3.0},
+    {"name": "x1", 
+    "type": "uniform_float",
+    "log": False,
+    "lower": -512.0,
+    "upper": 512.0,
+    "default": -4.0}],
+    "conditions": [],
+    "forbiddens": [],
+    "python_module_version": "0.4.17",
+    "json_format_version": 0.2}
+
+str_space = """{
+                "uniform_integer": (1, 10),
+                "uniform_float": (1.0, 10.0),
+                "categorical": ["a", "b", "c"],
+                "constant": 1337,
+            }"""
+
 class TestSearchSpacEncoding(unittest.TestCase):
     def setUp(self):
         self.test_space = None
         self.test_space = ConfigurationSpace(
             name="myspace",
-            space={
-                "uniform_integer": (1, 10),
-                "uniform_float": (1.0, 10.0),
-                "categorical": ["a", "b", "c"],
-                "constant": 1337,
-            },
+            space=dict_space
         )
         return super().setUp()
 
+    def test_init(self):
+        self.test_space = ConfigurationSpace(
+            name="myspace",
+            space=dict_space_2
+        )
+
+        self.test_space = ConfigurationSpace(
+            name="myspace",
+            space=str_space
+        )
+
     def test_config_spaces(self):
         try:
             search_space_to_config_space(self.test_space)