#558: fix CI issues

Author: ppebay

tests/unit/Execution/test_lbs_recursive_transfer_strategy.py

@@ -84,7 +84,6 @@ def setUp(self):
 
         # Define the rank and declare known peers
         self.rank = Rank(r_id=0, migratable_objects=self.migratable_objects, sentinel_objects=self.sentinel_objects, logger=self.logger)
-        self.rank.set_shared_blocks(self.block_set)
         self.known_peers = {}
 
         # Instantiate the phase

tests/unit/Model/test_lbs_node.py

@@ -53,18 +53,13 @@ class TestConfig(unittest.TestCase):
     def setUp(self):
         self.logger = logging.getLogger()
         self.node_id = 0
+        self.ranks = set()
         self.rank_ids = set(i for i in range(10))
         self.node = Node(logger=self.logger, n_id=self.node_id)
 
     def test_lbs_node_get_id(self):
         self.assertEqual(self.node.get_id(), self.node_id)
 
-    def test_lbs_node_rank_ids(self):
-        for rank_id in self.rank_ids:
-            self.node.add_rank_id(rank_id)
-        self.assertEqual(self.node.get_number_of_ranks(), len(self.rank_ids))
-        self.assertEqual(self.node.get_rank_ids(), self.rank_ids)
-
     def test_lbs_node_max_memory_usage(self):
         phase = Phase(lgr=self.logger, p_id=0)
         phase_ranks = set()
@@ -82,9 +77,10 @@ def test_lbs_node_max_memory_usage(self):
             rank = Rank(
                 logger=self.logger,
                 r_id=rank_id,
-                migratable_objects={obj},
-                node=self.node
-            )
+                migratable_objects={obj})
+            self.ranks.add(rank)
+            rank.set_node(self.node)
+            
             all_migratable_objs.add(obj)
             phase_ranks.add(rank)