From cbf3a6f599f2dd29c8a05a26b8025a884a6865ef Mon Sep 17 00:00:00 2001 From: ahalev Date: Fri, 16 Dec 2022 20:57:14 -0800 Subject: [PATCH] delete old tests file --- tests/envs/discrete.py | 166 ----------------------------------------- 1 file changed, 166 deletions(-) delete mode 100644 tests/envs/discrete.py diff --git a/tests/envs/discrete.py b/tests/envs/discrete.py deleted file mode 100644 index 27b9ea5d..00000000 --- a/tests/envs/discrete.py +++ /dev/null @@ -1,166 +0,0 @@ -from copy import deepcopy - -from tests.helpers.test_case import TestCase -from tests.helpers.modular_microgrid import get_modular_microgrid - -from pymgrid.envs import DiscreteMicrogridEnv - - -class TestDiscreteEnv(TestCase): - - def test_init_from_microgrid(self): - microgrid = get_modular_microgrid() - env = DiscreteMicrogridEnv(microgrid) - - self.assertEqual(env.modules, microgrid.modules) - self.assertIsNot(env.modules.module_tuples(), microgrid.modules.module_tuples()) - - n_obs = sum([x.observation_spaces['normalized'].shape[0] for x in microgrid.module_list]) - - self.assertEqual(env.observation_space.shape, (n_obs,)) - - def test_init_from_modules(self): - microgrid = get_modular_microgrid() - env = DiscreteMicrogridEnv(microgrid.modules.module_tuples(), add_unbalanced_module=False) - - self.assertEqual(env.modules, microgrid.modules) - self.assertIsNot(env.modules.module_tuples(), microgrid.modules.module_tuples()) - - n_obs = sum([x.observation_spaces['normalized'].shape[0] for x in microgrid.module_list]) - - self.assertEqual(env.observation_space.shape, (n_obs,)) - - -class TestDiscreteEnvScenario(TestCase): - microgrid_number = 0 - - def setUp(self) -> None: - self.env = DiscreteMicrogridEnv.from_scenario(microgrid_number=self.microgrid_number) - - def test_run_once(self): - env = deepcopy(self.env) - # sample environment then get log - self.assertEqual(len(env.log), 0) - for j in range(10): - with self.subTest(step=j): - action = env.sample_action(strict_bound=True) - env.step(action) - self.assertEqual(len(env.log), j+1) - - def test_reset_after_run(self): - env = deepcopy(self.env) - env.step(env.sample_action(strict_bound=True)) - env.reset() - self.assertEqual(len(env.log), 0) - - def test_run_again_after_reset(self): - env = deepcopy(self.env) - env.step(env.sample_action(strict_bound=True)) - - self.assertEqual(len(env.log), 1) - - env.reset() - - self.assertEqual(len(env.log), 0) - - for j in range(10): - with self.subTest(step=j): - action = env.sample_action(strict_bound=True) - env.step(action) - self.assertEqual(len(env.log), j+1) - - -class TestDiscreteEnvScenario1(TestDiscreteEnvScenario): - microgrid_number = 1 - - -class TestDiscreteEnvScenario2(TestDiscreteEnvScenario): - microgrid_number = 2 - - -class TestDiscreteEnvScenario3(TestDiscreteEnvScenario): - microgrid_number = 3 - - -class TestDiscreteEnvScenario4(TestDiscreteEnvScenario): - microgrid_number = 4 - - -class TestDiscreteEnvScenario5(TestDiscreteEnvScenario): - microgrid_number = 5 - - -class TestDiscreteEnvScenario6(TestDiscreteEnvScenario): - microgrid_number = 6 - - -class TestDiscreteEnvScenario47(TestDiscreteEnvScenario): - microgrid_number = 7 - - -class TestDiscreteEnvScenario8(TestDiscreteEnvScenario): - microgrid_number = 8 - - -class TestDiscreteEnvScenario9(TestDiscreteEnvScenario): - microgrid_number = 9 - - -class TestDiscreteEnvScenario10(TestDiscreteEnvScenario): - microgrid_number = 10 - - -class TestDiscreteEnvScenario11(TestDiscreteEnvScenario): - microgrid_number = 11 - - -class TestDiscreteEnvScenario12(TestDiscreteEnvScenario): - microgrid_number = 12 - - -class TestDiscreteEnvScenario13(TestDiscreteEnvScenario): - microgrid_number = 13 - - -class TestDiscreteEnvScenario14(TestDiscreteEnvScenario): - microgrid_number = 14 - - -class TestDiscreteEnvScenario15(TestDiscreteEnvScenario): - microgrid_number = 15 - - -class TestDiscreteEnvScenario16(TestDiscreteEnvScenario): - microgrid_number = 16 - - -class TestDiscreteEnvScenario17(TestDiscreteEnvScenario): - microgrid_number = 17 - - -class TestDiscreteEnvScenario18(TestDiscreteEnvScenario): - microgrid_number = 18 - - -class TestDiscreteEnvScenario19(TestDiscreteEnvScenario): - microgrid_number = 19 - - -class TestDiscreteEnvScenario20(TestDiscreteEnvScenario): - microgrid_number = 20 - - -class TestDiscreteEnvScenario21(TestDiscreteEnvScenario): - microgrid_number = 21 - - -class TestDiscreteEnvScenario22(TestDiscreteEnvScenario): - microgrid_number = 22 - - -class TestDiscreteEnvScenario23(TestDiscreteEnvScenario): - microgrid_number = 23 - - -class TestDiscreteEnvScenario24(TestDiscreteEnvScenario): - microgrid_number = 24