Skip to content

Commit

Permalink
tests working
Browse files Browse the repository at this point in the history
  • Loading branch information
jan-janssen committed Oct 28, 2024
1 parent ba2c2bf commit 4e1facf
Showing 1 changed file with 5 additions and 2 deletions.
7 changes: 5 additions & 2 deletions tests/test_cache_executor_pysqa_flux.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down Expand Up @@ -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,
Expand All @@ -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)])
Expand Down

0 comments on commit 4e1facf

Please sign in to comment.