Skip to content

Commit 589755d

Browse files
committed
add psij option in conftest.py
1 parent 089c360 commit 589755d

File tree

1 file changed

+9
-4
lines changed

1 file changed

+9
-4
lines changed

pydra/conftest.py

Lines changed: 9 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -7,20 +7,23 @@
77

88
def pytest_addoption(parser):
99
parser.addoption("--dask", action="store_true", help="run all combinations")
10+
parser.addoption("--psij", action="store", help="run with psij subtype plugin", choices=["local", "slurm"])
1011

1112

1213
def pytest_generate_tests(metafunc):
1314
if "plugin_dask_opt" in metafunc.fixturenames:
1415
if bool(shutil.which("sbatch")):
15-
Plugins = ["psij-slurm"]
16+
Plugins = ["slurm"]
1617
else:
17-
Plugins = ["psij-local"]
18+
Plugins = ["cf"]
1819
try:
1920
if metafunc.config.getoption("dask"):
2021
Plugins.append("dask")
2122
except ValueError:
2223
# Called as --pyargs, so --dask isn't available
2324
pass
25+
if metafunc.config.getoption("psij"):
26+
Plugins.append("psij-" + metafunc.config.getoption("psij"))
2427
metafunc.parametrize("plugin_dask_opt", Plugins)
2528

2629
if "plugin" in metafunc.fixturenames:
@@ -32,9 +35,11 @@ def pytest_generate_tests(metafunc):
3235
if use_dask:
3336
Plugins = []
3437
elif bool(shutil.which("sbatch")):
35-
Plugins = ["psij-slurm"]
38+
Plugins = ["slurm"]
3639
else:
37-
Plugins = ["psij-local"]
40+
Plugins = ["cf"]
41+
if metafunc.config.getoption("psij"):
42+
Plugins.append("psij-" + metafunc.config.getoption("psij"))
3843
metafunc.parametrize("plugin", Plugins)
3944

4045

0 commit comments

Comments
 (0)