MCPcopy Create free account
hub / github.com/ahalev/python-microgrid / test_init_from_modules

Method test_init_from_modules

tests/envs/test_discrete.py:24–34  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

22 self.assertEqual(env.observation_space.shape, (n_obs,))
23
24 def test_init_from_modules(self):
25 microgrid = get_modular_microgrid()
26 env = DiscreteMicrogridEnv(microgrid.modules.to_tuples(), add_unbalanced_module=False)
27
28 self.assertEqual(env.modules, microgrid.modules)
29 self.assertIsNot(env.modules.to_tuples(), microgrid.modules.to_tuples())
30
31 # Add one for net load
32 n_obs = 1 + sum([x.observation_space['normalized'].shape[0] for x in microgrid.modules.to_list()])
33
34 self.assertEqual(env.observation_space.shape, (n_obs,))
35
36
37class TestDiscreteEnvScenario(TestCase):

Callers

nothing calls this directly

Calls 5

get_modular_microgridFunction · 0.90
to_tuplesMethod · 0.80
assertEqualMethod · 0.80
to_listMethod · 0.45

Tested by

no test coverage detected