TST update tests to check default n_jobs

Author: glemaitre

imblearn/combine/tests/test_smote_enn.py

@@ -147,12 +147,12 @@ def test_validate_estimator_default():
 
 
 def test_parallelisation():
-    # Check if default job count is 1
+    # Check if default job count is none
     smt = SMOTEENN(random_state=RND_SEED)
     smt._validate_estimator()
-    assert smt.n_jobs == 1
-    assert smt.smote_.n_jobs == 1
-    assert smt.enn_.n_jobs == 1
+    assert smt.n_jobs is None
+    assert smt.smote_.n_jobs is None
+    assert smt.enn_.n_jobs is None
 
     # Check if job count is set
     smt = SMOTEENN(random_state=RND_SEED, n_jobs=8)

imblearn/combine/tests/test_smote_tomek.py

@@ -157,12 +157,12 @@ def test_validate_estimator_default():
 
 
 def test_parallelisation():
-    # Check if default job count is 1
+    # Check if default job count is None
     smt = SMOTETomek(random_state=RND_SEED)
     smt._validate_estimator()
-    assert smt.n_jobs == 1
-    assert smt.smote_.n_jobs == 1
-    assert smt.tomek_.n_jobs == 1
+    assert smt.n_jobs is None
+    assert smt.smote_.n_jobs is None
+    assert smt.tomek_.n_jobs is None
 
     # Check if job count is set
     smt = SMOTETomek(random_state=RND_SEED, n_jobs=8)

imblearn/under_sampling/_prototype_selection/tests/test_condensed_nearest_neighbour.py

@@ -43,7 +43,7 @@ def test_cnn_init():
     cnn = CondensedNearestNeighbour(random_state=RND_SEED)
 
     assert cnn.n_seeds_S == 1
-    assert cnn.n_jobs == 1
+    assert cnn.n_jobs is None
 
 
 def test_cnn_fit_resample():

imblearn/under_sampling/_prototype_selection/tests/test_edited_nearest_neighbours.py

@@ -44,7 +44,7 @@ def test_enn_init():
 
     assert enn.n_neighbors == 3
     assert enn.kind_sel == "all"
-    assert enn.n_jobs == 1
+    assert enn.n_jobs is None
 
 
 def test_enn_fit_resample():

imblearn/under_sampling/_prototype_selection/tests/test_one_sided_selection.py

@@ -38,7 +38,7 @@ def test_oss_init():
     oss = OneSidedSelection(random_state=RND_SEED)
 
     assert oss.n_seeds_S == 1
-    assert oss.n_jobs == 1
+    assert oss.n_jobs is None
     assert oss.random_state == RND_SEED
 
 

imblearn/under_sampling/_prototype_selection/tests/test_repeated_edited_nearest_neighbours.py

@@ -106,7 +106,7 @@ def test_renn_init():
 
     assert renn.n_neighbors == 3
     assert renn.kind_sel == "all"
-    assert renn.n_jobs == 1
+    assert renn.n_jobs is None
 
 
 def test_renn_iter_wrong():

imblearn/under_sampling/_prototype_selection/tests/test_tomek_links.py

@@ -37,7 +37,7 @@
 
 def test_tl_init():
     tl = TomekLinks()
-    assert tl.n_jobs == 1
+    assert tl.n_jobs is None
 
 
 def test_tl_fit_resample():