index-list shouldn't consider worker0 - revert unit test changes

Author: jlnav

.pre-commit-config.yaml

@@ -37,4 +37,4 @@ repos:
   rev: v1.19.1
   hooks:
     - id: mypy
-      exclude: ^libensemble/utils/(launcher|loc_stack|runners|pydantic|output_directory)\.py$|^libensemble/tests/regression_tests/support\.py$|^libensemble/tests/functionality_tests/
+      exclude: ^libensemble/utils/(launcher|loc_stack|runners|pydantic|output_directory)\.py$|^libensemble/tests/regression_tests/support\.py$|^libensemble/tests/functionality_tests/|^libensemble/tests/unit_tests/

libensemble/resources/worker_resources.py

@@ -106,7 +106,7 @@ def get_index_list(num_workers: int, num_rsets: int, zero_resource_list: list[in
         """Map WorkerID to index into a nodelist"""
         index = 0
         index_list: list[int | None] = []
-        for i in range(0, num_workers):
+        for i in range(1, num_workers + 1):
             if i in zero_resource_list:
                 index_list.append(None)
             else:

libensemble/tests/unit_tests/test_resources.py

@@ -694,23 +694,23 @@ def test_map_workerid_to_index():
 
     zero_resource_list = []
     index_list = ResourceManager.get_index_list(num_workers, num_rsets, zero_resource_list)
-    for workerID in range(0, num_workers):
-        index = index_list[workerID]
-        assert index == workerID, "index incorrect. Received: " + str(index)
-
-    zero_resource_list = [0]
-    index_list = ResourceManager.get_index_list(num_workers, num_rsets, zero_resource_list)
-    for workerID in range(1, num_workers):
-        index = index_list[workerID]
+    for workerID in range(1, num_workers + 1):
+        index = index_list[workerID - 1]
         assert index == workerID - 1, "index incorrect. Received: " + str(index)
 
-    zero_resource_list = [0, 1]
+    zero_resource_list = [1]
     index_list = ResourceManager.get_index_list(num_workers, num_rsets, zero_resource_list)
-    for workerID in range(2, num_workers):
-        index = index_list[workerID]
+    for workerID in range(2, num_workers + 1):
+        index = index_list[workerID - 1]
         assert index == workerID - 2, "index incorrect. Received: " + str(index)
 
-    zero_resource_list = [0, 2]
+    zero_resource_list = [1, 2]
+    index_list = ResourceManager.get_index_list(num_workers, num_rsets, zero_resource_list)
+    for workerID in range(3, num_workers + 1):
+        index = index_list[workerID - 1]
+        assert index == workerID - 3, "index incorrect. Received: " + str(index)
+
+    zero_resource_list = [1, 3]
     index_list = ResourceManager.get_index_list(num_workers, num_rsets, zero_resource_list)
 
     workerID = 2