From 12afcaf8ea794b9fed8225b7b149036eedebe39a Mon Sep 17 00:00:00 2001 From: Carolin Benjamins Date: Thu, 30 Jun 2022 16:28:16 +0200 Subject: [PATCH] Add first tests but the fail :') --- test/test_CARLEnv.py | 33 +++++++++++++++++++++++++++++++++ 1 file changed, 33 insertions(+) diff --git a/test/test_CARLEnv.py b/test/test_CARLEnv.py index 155cbcd5..9f318bf6 100644 --- a/test/test_CARLEnv.py +++ b/test/test_CARLEnv.py @@ -289,6 +289,39 @@ def test_context_feature_scaling_unknown_step(self): with self.assertRaises(ValueError): next_obs, reward, done, info = env.step(action=action) + def test_context_mask(self): + context_mask = ["dt", "g"] + env = ( # noqa: F841 local variable is assigned to but never used + CARLPendulumEnv( + contexts={}, + hide_context=False, + context_mask=context_mask, + dict_observation_space=True, + add_gaussian_noise_to_context=False, + gaussian_noise_std_percentage=0.01, + state_context_features=None, + scale_context_features="no", + ) + ) + s = env.reset() + s_c = s["context"] + forbidden_in_context = [f for f in env.state_context_features if f in context_mask] + 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