add test for bug fix for #18

SimonBlanke · Mar 1, 2021 · e6fbd0c · e6fbd0c
1 parent c67af0a
commit e6fbd0c
Showing 1 changed file with 15 additions and 3 deletions.
diff --git a/tests/test_distribution.py b/tests/test_distribution.py
@@ -18,25 +18,39 @@ def test_n_jobs_0():
     hyper.add_search(objective_function, search_space, n_iter=15, n_jobs=2)
     hyper.run()
 
+    assert len(hyper.results_list) == 2
+
 
 def test_n_jobs_1():
     hyper = Hyperactive()
     hyper.add_search(objective_function, search_space, n_iter=15, n_jobs=4)
     hyper.run()
 
+    assert len(hyper.results_list) == 4
+
 
 def test_n_jobs_2():
     hyper = Hyperactive()
     hyper.add_search(objective_function, search_space, n_iter=15, n_jobs=8)
     hyper.run()
 
+    assert len(hyper.results_list) == 8
+
 
 def test_n_jobs_3():
     hyper = Hyperactive()
     hyper.add_search(objective_function, search_space, n_iter=15, n_jobs=-1)
     hyper.run()
 
 
+def test_n_jobs_4():
+    hyper = Hyperactive()
+    hyper.add_search(objective_function, search_space, n_iter=15, n_jobs=100)
+    hyper.run()
+
+    assert len(hyper.results_list) == 100
+
+
 def test_multiprocessing_0():
     hyper = Hyperactive(distribution="multiprocessing")
     hyper.add_search(objective_function, search_space, n_iter=15, n_jobs=2)
@@ -69,8 +83,7 @@ def joblib_wrapper(process_func, search_processes_paras, **kwargs):
         n_jobs = len(search_processes_paras)
 
         jobs = [
-            delayed(process_func)(**info_dict)
-            for info_dict in search_processes_paras
+            delayed(process_func)(**info_dict) for info_dict in search_processes_paras
         ]
         results = Parallel(n_jobs=n_jobs, **kwargs)(jobs)
 
@@ -80,4 +93,3 @@ def joblib_wrapper(process_func, search_processes_paras, **kwargs):
     hyper.add_search(objective_function, search_space, n_iter=15, n_jobs=2)
 
     hyper.run()
-