Fix tests for exceptions

juaml · Jan 23, 2025 · ff89587 · ff89587
1 parent 82bff5e
commit ff89587
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/joblib_htcondor/tests/test_runtime.py b/joblib_htcondor/tests/test_runtime.py
@@ -5,6 +5,7 @@
 
 import logging
 import socket
+from concurrent import futures
 
 import pytest
 from joblib import Parallel, delayed, parallel_config
@@ -32,8 +33,6 @@ def test_normal() -> None:
     assert out == [-1, -2, -3, -4, -5]
 
 
-
-
 def test_exception() -> None:
     """Test running with an exception."""
     def neg_with_exception(a: int) -> int:
@@ -68,6 +67,5 @@ def neg_with_exception(a: int) -> int:
         verbose=1000,
         worker_log_level=logging.INFO,
     ):
-        out = Parallel()(delayed(neg_with_exception)(i + 1) for i in range(5))
-
-    assert out == [-1, -2, -3, -4, -5]
+        with pytest.raises(ValueError, match="This is an exception"):
+            Parallel()(delayed(neg_with_exception)(i + 1) for i in range(5))