fixing a missing copy in the constructor and add some tests for leap net

Author: BDonnot

leap_net/proxy/baseProxy.py

@@ -701,7 +701,7 @@ def _get_adds_mults_from_name(self, obss, attr_nm):
         elif attr_nm in ["a_or", "a_ex"]:
             add_tmp = self.dtype(0.)  # because i multiply by the line status, so i don't want any bias
             if hasattr(obs, "thermal_limit"):
-                mult_tmp = obs.thermal_limit
+                mult_tmp = 1.0 * obs.thermal_limit
             elif hasattr(obs, "rho"):
                 mult_tmp = np.abs(obs.a_or / (obs.rho + 1e-2))  # which is equal to the thermal limit
             else:

leap_net/test/test_LeapNetProxy.py

@@ -215,7 +215,9 @@ def _aux_test_tau_from_list_topo(self, proxy=None):
             proxy = ProxyLeapNet(attr_tau=("line_status", "topo_vect",),
                                  topo_vect_to_tau="given_list",
                                  kwargs_tau=[(0, (2, 1, 1)), (0, (1, 2, 1)), (1, (2, 1, 1, 1, 1, 1)),
-                                             (12, (2, 1, 1, 2)), (13, (2, 1, 2)), (13, (1, 2, 2))]
+                                             (12, (2, 1, 1, 2)), (13, (2, 1, 2)), (13, (1, 2, 2)), 
+                                             (1, (2, 1, 2, 1, 2, 1))
+                                             ]
                                  )
             proxy.init([self.obs])
 
@@ -258,7 +260,7 @@ def _aux_test_tau_from_list_topo(self, proxy=None):
         res = proxy.topo_vect_handler(obs)
         assert np.sum(res) == 0
 
-        # test that everything on bys 2 and everything on bus 1 leads to the same result
+        # test that everything on bus 2 and everything on bus 1 leads to the same result
         obs = env.reset()
         act = env.action_space({"set_bus": {"substations_id": [(12, (2, 2, 2, 2))]}})
         obs, reward, done, info = env.step(act)
@@ -271,7 +273,7 @@ def _aux_test_tau_from_list_topo(self, proxy=None):
         res = proxy.topo_vect_handler(obs)
         assert np.sum(res) == 0
 
-        # check the symmetry 2 <-> 1
+        # check cumulative "actions"
         obs = env.reset()
         act = env.action_space({"set_bus": {"substations_id": [(0, (2, 1, 1))]}})
         obs, reward, done, info = env.step(act)
@@ -282,7 +284,7 @@ def _aux_test_tau_from_list_topo(self, proxy=None):
         assert res[0] == 1.
         assert res[4] == 1.
 
-        # test the symmetry (gain...) why not
+        # test the symmetry (again...) why not
         obs = env.reset()
         act = env.action_space({"set_bus": {"substations_id": [(0, (2, 2, 2))]}})
         obs, reward, done, info = env.step(act)
@@ -295,7 +297,30 @@ def _aux_test_tau_from_list_topo(self, proxy=None):
         res = proxy.topo_vect_handler(obs)
         assert np.sum(res) == 1
         assert res[2] == 1.
+        
+        obs = env.reset()
+        act = env.action_space({"set_bus": {"substations_id": [(12, (1, 2, 2, 1))]}})
+        obs, reward, done, info = env.step(act)
+        res = proxy.topo_vect_handler(obs)
+        assert np.sum(res) == 1
+        assert res[3] == 1.
 
+        # test that if a line is disconnected, we are still able to match the topologies
+        # see https://github.com/BDonnot/leap_net/pull/12/files
+        obs = env.reset()
+        act = env.action_space({"set_bus": {"substations_id": [(1, (2, 1, -1, 1, 2, 1))]}})  # (2, 1, 2, 1, 2, 1)
+        obs, reward, done, info = env.step(act)
+        res = proxy.topo_vect_handler(obs)
+        assert np.sum(res) == 1
+        assert res[6] == 1.
+
+        obs = env.reset()
+        act = env.action_space({"set_bus": {"substations_id": [(1, (2, -1, 2, 1, 2, 1))]}})  # (2, 1, 2, 1, 2, 1)
+        obs, reward, done, info = env.step(act)
+        res = proxy.topo_vect_handler(obs)
+        assert np.sum(res) == 1
+        assert res[6] == 1.
+        
     def test_tau_from_online_topo(self):
         self._aux_test_tau_from_online_topo()