# Copyright 2022-2026 The Ramble Authors
#
# Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
# https://www.apache.org/licenses/LICENSE-2.0> or the MIT license
# <LICENSE-MIT or https://opensource.org/licenses/MIT>, at your
# option. This file may not be copied, modified, or distributed
# except according to those terms.
import os
import pytest
import ramble.test.modifier_functionality.modifier_helpers as modifier_helpers
import ramble.workspace
from ramble.main import RambleCommand
from ramble.test.dry_run_helpers import SCOPES, dry_run_config
workspace = RambleCommand("workspace")
[docs]
@pytest.mark.long
@pytest.mark.parametrize(
"scopes",
[
[SCOPES.workspace, SCOPES.workspace],
[SCOPES.workspace, SCOPES.application],
[SCOPES.workspace, SCOPES.workload],
[SCOPES.workspace, SCOPES.experiment],
[SCOPES.application, SCOPES.workspace],
[SCOPES.application, SCOPES.application],
[SCOPES.application, SCOPES.workload],
[SCOPES.application, SCOPES.experiment],
[SCOPES.workload, SCOPES.workspace],
[SCOPES.workload, SCOPES.application],
[SCOPES.workload, SCOPES.workload],
[SCOPES.workload, SCOPES.experiment],
[SCOPES.experiment, SCOPES.workspace],
[SCOPES.experiment, SCOPES.application],
[SCOPES.experiment, SCOPES.workload],
[SCOPES.experiment, SCOPES.experiment],
],
)
@pytest.mark.parametrize(
"factories,answers",
[
(
[modifier_helpers.intel_aps_modifier, modifier_helpers.lscpu_modifier],
[modifier_helpers.intel_aps_answer, modifier_helpers.lscpu_answer],
),
],
)
def test_gromacs_multi_modifier_dry_run(
mutable_mock_workspace_path, mutable_applications, scopes, factories, answers, workspace_name
):
test_modifiers = []
for scope, factory in zip(scopes, factories):
test_modifiers.append((scope, factory()))
software_tests = []
script_tests = []
for answer in answers:
sw_test, script_test = answer()
for sw in sw_test:
software_tests.append(sw)
for script in script_test:
script_tests.append(script)
with ramble.workspace.create(workspace_name) as ws1:
ws1.write()
config_path = os.path.join(ws1.config_dir, ramble.workspace.CONFIG_FILE_NAME)
dry_run_config("modifiers", test_modifiers, config_path, "gromacs", "water_bare")
ws1._re_read()
workspace("concretize", global_args=["-D", ws1.root])
workspace("setup", "--dry-run", global_args=["-D", ws1.root])
# Test software directories
software_base_dir = os.path.join(ws1.software_dir, "spack")
modifier_helpers.check_software_env(software_base_dir, software_tests)
exp_script = os.path.join(
ws1.experiment_dir, "gromacs", "water_bare", "test_exp", "execute_experiment"
)
modifier_helpers.check_execute_script(exp_script, script_tests)