fix test

joblib · May 30, 2024 · 1c7736d · 1c7736d
1 parent c13ce83
commit 1c7736d
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/test/test_spark.py b/test/test_spark.py
@@ -36,10 +36,6 @@
 register_spark()
 
 
-def inc(x):
-    return x + 1
-
-
 def slow_raise_value_error(condition, duration=0.05):
     sleep(duration)
     if condition:
@@ -65,6 +61,9 @@ def teardown_class(cls):
         cls.spark.stop()
 
     def test_simple(self):
+        def inc(x):
+            return x + 1
+
         with parallel_backend('spark') as (ba, _):
             seq = Parallel(n_jobs=5)(delayed(inc)(i) for i in range(10))
             assert seq == [inc(i) for i in range(10)]