Skip to content

Commit 9535958

Browse files
committed
extend tests
1 parent cab77e3 commit 9535958

1 file changed

Lines changed: 14 additions & 0 deletions

File tree

tests/test_fluxclusterexecutor.py

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -40,6 +40,20 @@ def test_executor(self):
4040
cloudpickle_register(ind=1)
4141
fs1 = exe.submit(mpi_funct, 1)
4242
self.assertFalse(fs1.done())
43+
self.assertEqual(len(os.listdir("executorlib_cache")), 2)
44+
self.assertEqual(fs1.result(), [(1, 2, 0), (1, 2, 1)])
45+
self.assertTrue(fs1.done())
46+
47+
def test_executor_no_cwd(self):
48+
with FluxClusterExecutor(
49+
resource_dict={"cores": 2},
50+
block_allocation=False,
51+
cache_directory="executorlib_cache",
52+
) as exe:
53+
cloudpickle_register(ind=1)
54+
fs1 = exe.submit(mpi_funct, 1)
55+
self.assertFalse(fs1.done())
56+
self.assertEqual(len(os.listdir("executorlib_cache")), 2)
4357
self.assertEqual(fs1.result(), [(1, 2, 0), (1, 2, 1)])
4458
self.assertTrue(fs1.done())
4559

0 commit comments

Comments
 (0)