feat: improve schedular default behaviour

Author: darwintree

tests/test_all.py

@@ -67,7 +67,7 @@ def run():
     parser.add_argument(
         "--max-nodes",
         dest="max_nodes",
-        default=24,
+        default=0,
         type=int,
     )
     parser.add_argument(
@@ -96,6 +96,10 @@ def run():
             "../target/release/conflux"),
         type=str)
     options = parser.parse_args()
+    
+    if options.max_nodes == 0:
+        options.max_nodes = os.cpu_count()
+        print(f"Max nodes not specified, using {options.max_nodes} nodes")
 
     all_failed = set()
 

tests/test_utils/test_scheduler.py

@@ -68,6 +68,11 @@ def _prepare_task_queue(self, test_scripts):
             # Collect completed task results and add them to the queue
             for future, i, script in tasks:
                 result = future.result()
+                if result > self.available_nodes:
+                    raise RuntimeError(f"Cannot run {script} because it requires {result} nodes, "
+                                       f"but only max to {self.available_nodes} nodes are available"
+                                       f"Please specify --max-nodes to run the test")
+
                 task_queue.put((script, result, i))
         print(" Done")
         return task_queue