Skip to content

Commit 468b0ea

Browse files
committed
extend test
1 parent 33063f4 commit 468b0ea

File tree

1 file changed

+8
-3
lines changed

1 file changed

+8
-3
lines changed

tests/test_fluxclusterexecutor.py

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@
1111
import flux.job
1212
from executorlib.task_scheduler.file.hdf import dump
1313
from executorlib.task_scheduler.file.queue_spawner import terminate_with_pysqa, terminate_tasks_in_cache, execute_with_pysqa
14+
from executorlib.task_scheduler.file.shared import _get_execute_command
1415

1516
skip_flux_test = "FLUX_URI" not in os.environ
1617
pmi = os.environ.get("EXECUTORLIB_PMIX", None)
@@ -62,9 +63,13 @@ def test_executor_no_cwd(self):
6263

6364
def test_pysqa_interface(self):
6465
queue_id = execute_with_pysqa(
65-
command=["flux", "run", "sleep", "10"],
66-
file_name="test.h5",
67-
data_dict={"fn": sleep},
66+
command=_get_execute_command(
67+
file_name="test_i.h5",
68+
cores=1,
69+
),
70+
file_name="test_i.h5",
71+
data_dict={"fn": sleep, "args": (10,)},
72+
resource_dict={"cores": 1},
6873
cache_directory="executorlib_cache",
6974
backend="flux"
7075
)

0 commit comments

Comments
 (0)