auto merge of #5053 : deepak1556/servo/script_patch, r=jdm

fixes #5046 

1 files changed, 11 insertions, 4 deletions

diff --git a/python/servo/testing_commands.py b/python/servo/testing_commands.py
index 10959cd021d..95dd71af1e9 100644
--- a/python/servo/testing_commands.py
+++ b/python/servo/testing_commands.py
@@ -16,6 +16,7 @@ from mach.decorators import (
 
 from servo.command_base import CommandBase
 import tidy
+import multiprocessing
 
 
 @CommandProvider
@@ -214,9 +215,12 @@ class MachCommands(CommandBase):
              description='Run the web platform tests',
              category='testing')
     @CommandArgument(
-        'params', default=None, nargs='...',
-        help="Command-line arguments to be passed through to wpt/run.sh")
-    def test_wpt(self, params=None):
+        '--processes', default=None,
+        help="Number of servo processes to spawn")
+    @CommandArgument(
+        "params", default=None, nargs='...',
+        help="command-line arguments to be passed through to wpt/run.sh")
+    def test_wpt(self, processes=None, params=None):
         if params is None:
             params = []
         else:
@@ -227,7 +231,10 @@ class MachCommands(CommandBase):
 
             if path.exists(maybe_path) and wpt_path in maybe_path:
                 params = ["--include",
-                          path.relpath(maybe_path, wpt_path)] + params[1:]
+                          path.relpath(maybe_path, wpt_path)]
+
+        processes = str(multiprocessing.cpu_count()) if processes is None else processes
+        params = params + ["--processes", processes]
 
         return subprocess.call(
             ["bash", path.join("tests", "wpt", "run.sh")] + params,