File tree Expand file tree Collapse file tree 2 files changed +6
-4
lines changed
Expand file tree Collapse file tree 2 files changed +6
-4
lines changed Original file line number Diff line number Diff line change @@ -39,10 +39,11 @@ class ParallelTestSuite(unittest.BaseTestSuite):
3939 Creates worker threads, manages the task queue, and combines the results.
4040 """
4141
42- def __init__ (self ):
42+ def __init__ (self , max_cores ):
4343 super (ParallelTestSuite , self ).__init__ ()
4444 self .processes = None
4545 self .result_queue = None
46+ self .max_cores = max_cores
4647
4748 def run (self , result ):
4849 test_queue = self .create_test_queue ()
@@ -73,10 +74,11 @@ def reversed_tests(self):
7374 return tests [::- 1 ]
7475
7576 def init_processes (self , test_queue ):
77+ use_cores = min (self .max_cores , num_cores ())
78+ print ('Using %s parallel test processes' % use_cores )
7679 self .processes = []
7780 self .result_queue = multiprocessing .Queue ()
78- self .dedicated_temp_dirs = [tempfile .mkdtemp () for x in range (num_cores ())]
79- print ('Using %s parallel test processes' % len (self .dedicated_temp_dirs ))
81+ self .dedicated_temp_dirs = [tempfile .mkdtemp () for x in range (use_cores )]
8082 for temp_dir in self .dedicated_temp_dirs :
8183 p = multiprocessing .Process (target = g_testing_thread ,
8284 args = (test_queue , self .result_queue , temp_dir ))
Original file line number Diff line number Diff line change @@ -1948,7 +1948,7 @@ def suite_for_module(module, tests):
19481948 has_multiple_tests = len (tests ) > 1
19491949 has_multiple_cores = parallel_runner .num_cores () > 1
19501950 if suite_supported and has_multiple_tests and has_multiple_cores :
1951- return parallel_runner .ParallelTestSuite ()
1951+ return parallel_runner .ParallelTestSuite (len ( tests ) )
19521952 return unittest .TestSuite ()
19531953
19541954
You can’t perform that action at this time.
0 commit comments