update test

Author: jan-janssen

tests/test_mpiexecspawner.py

@@ -443,13 +443,13 @@ def test_execute_task_failed_no_argument(self):
         q.put({"fn": calc_array, "args": (), "kwargs": {}, "future": f})
         q.put({"shutdown": True, "wait": True})
         cloudpickle_register(ind=1)
+        execute_tasks(
+            future_queue=q,
+            cores=1,
+            openmpi_oversubscribe=False,
+            spawner=MpiExecSpawner,
+        )
         with self.assertRaises(TypeError):
-            execute_tasks(
-                future_queue=q,
-                cores=1,
-                openmpi_oversubscribe=False,
-                spawner=MpiExecSpawner,
-            )
             f.result()
         q.join()
 
@@ -459,13 +459,13 @@ def test_execute_task_failed_wrong_argument(self):
         q.put({"fn": calc_array, "args": (), "kwargs": {"j": 4}, "future": f})
         q.put({"shutdown": True, "wait": True})
         cloudpickle_register(ind=1)
+        execute_tasks(
+            future_queue=q,
+            cores=1,
+            openmpi_oversubscribe=False,
+            spawner=MpiExecSpawner,
+        )
         with self.assertRaises(TypeError):
-            execute_tasks(
-                future_queue=q,
-                cores=1,
-                openmpi_oversubscribe=False,
-                spawner=MpiExecSpawner,
-            )
             f.result()
         q.join()