From 4e1facf11532228541c69935dd04eeeb3fce0b09 Mon Sep 17 00:00:00 2001
From: Jan Janssen <janssen@mpie.de>
Date: Mon, 28 Oct 2024 15:50:30 +0100
Subject: [PATCH] tests working

---
 tests/test_cache_executor_pysqa_flux.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/tests/test_cache_executor_pysqa_flux.py b/tests/test_cache_executor_pysqa_flux.py
index c599be8e..4f389406 100644
--- a/tests/test_cache_executor_pysqa_flux.py
+++ b/tests/test_cache_executor_pysqa_flux.py
@@ -3,9 +3,11 @@
 import unittest
 import shutil
 
+from executorlib import Executor
+from executorlib.standalone.serialize import cloudpickle_register
+
 try:
     import flux.job
-    from executorlib import Executor
 
     skip_flux_test = "FLUX_URI" not in os.environ
     pmi = os.environ.get("PYMPIPOOL_PMIX", None)
@@ -35,7 +37,7 @@ def test_executor(self):
             backend="pysqa_flux",
             cache_directory="cache",
             max_cores=1,
-            resource_dict={"cores": 2, "cwd": "cache"},
+            resource_dict={"cores": 2, "cwd": "."},
             flux_executor=None,
             flux_executor_pmi_mode=None,
             flux_executor_nesting=False,
@@ -47,6 +49,7 @@ def test_executor(self):
             refresh_rate=0.01,
             plot_dependency_graph=False,
         ) as exe:
+            cloudpickle_register(ind=1)
             fs1 = exe.submit(mpi_funct, 1)
             self.assertFalse(fs1.done())
             self.assertEqual(fs1.result(), [(1, 2, 0), (1, 2, 1)])