Fix tests

automl · Jul 1, 2022 · bd84c25 · bd84c25
1 parent 12afcaf
commit bd84c25
Showing 1 changed file with 13 additions and 13 deletions.
diff --git a/test/test_CARLEnv.py b/test/test_CARLEnv.py
@@ -152,6 +152,19 @@ def test_dict_observation_space(self):
         next_obs, reward, done, info = env.step(action=action)
         env.close()
 
+    def test_state_context_feature_population(self):
+        env = (  # noqa: F841 local variable is assigned to but never used
+            CARLPendulumEnv(
+                contexts={},
+                hide_context=False,
+                add_gaussian_noise_to_context=False,
+                gaussian_noise_std_percentage=0.01,
+                state_context_features=None,
+                scale_context_features="no",
+            )
+        )
+        self.assertIsNotNone(env.state_context_features)
+
 
 class TestEpisodeTermination(unittest.TestCase):
     def test_episode_termination(self):
@@ -309,19 +322,6 @@ def test_context_mask(self):
         self.assertTrue(len(s_c) == len(list(env.default_context.keys())) - 2)
         self.assertTrue(len(forbidden_in_context) == 0)
 
-    def test_state_context_feature_population(self):
-        env = (  # noqa: F841 local variable is assigned to but never used
-            CARLPendulumEnv(
-                contexts={},
-                hide_context=False,
-                add_gaussian_noise_to_context=False,
-                gaussian_noise_std_percentage=0.01,
-                state_context_features=None,
-                scale_context_features="no",
-            )
-        )
-        self.assertIsNotNone(env.state_context_features)
-
 
 class TestContextSelection(unittest.TestCase):
     @staticmethod