Skip to content

Commit 6423c20

Browse files
Fix tests after changing order of variable definition
1 parent f93cb69 commit 6423c20

File tree

2 files changed

+13
-16
lines changed

2 files changed

+13
-16
lines changed

lib/ramble/ramble/test/application.py

Lines changed: 6 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -382,15 +382,15 @@ def test_set_input_path_multi_input(mutable_mock_apps_repo):
382382
assert executable_application_instance.variables["test-input3"] == input3_path
383383

384384

385-
def test_set_default_experiment_variables(mutable_mock_apps_repo):
386-
"""_set_default_experiment_variables"""
385+
def test_set_variables(mutable_mock_apps_repo):
386+
"""Test that set_variables defines workload variables"""
387387

388388
executable_application_instance = mutable_mock_apps_repo.get("basic")
389389

390390
expansion_vars = basic_exp_dict()
391+
del expansion_vars["n_ranks"]
391392

392393
# Set up the instance to pass the initial part of the function
393-
executable_application_instance.expander = ramble.expander.Expander(expansion_vars, None)
394394

395395
test_wl = ramble.workload.Workload("test_wl", executables=["foo"], inputs=["input"])
396396
test_wl2 = ramble.workload.Workload("test_wl2", executables=["bar"], inputs=["input"])
@@ -400,9 +400,8 @@ def test_set_default_experiment_variables(mutable_mock_apps_repo):
400400
executable_application_instance.internals = {}
401401

402402
executable_application_instance.inputs[_FS] = {"input": {"target_dir": "."}}
403-
executable_application_instance.variables = {}
404403

405-
executable_application_instance._set_default_experiment_variables()
404+
executable_application_instance.set_variables(expansion_vars, None)
406405

407406
assert executable_application_instance.variables["n_ranks"] == "1"
408407

@@ -414,9 +413,6 @@ def test_define_commands(mutable_mock_apps_repo):
414413

415414
expansion_vars = basic_exp_dict()
416415

417-
# Set up the instance to pass the initial part of the function
418-
executable_application_instance.expander = ramble.expander.Expander(expansion_vars, None)
419-
420416
test_wl = ramble.workload.Workload("test_wl", executables=["foo"], inputs=["input"])
421417
test_wl2 = ramble.workload.Workload("test_wl2", executables=["bar"], inputs=["input"])
422418
test_wl2.add_variable(ramble.workload.WorkloadVariable("n_ranks", default="1"))
@@ -425,14 +421,13 @@ def test_define_commands(mutable_mock_apps_repo):
425421
executable_application_instance.internals = {}
426422

427423
executable_application_instance.inputs[_FS] = {"input": {"target_dir": "."}}
428-
executable_application_instance.variables = {}
424+
executable_application_instance.set_variables(expansion_vars, None)
429425

430426
exec_graph = executable_application_instance._get_executable_graph("test_wl2")
431427

432428
executable_application_instance.set_formatted_executables(
433429
{"command": {"join_separator": "\n"}}
434430
)
435-
executable_application_instance._set_default_experiment_variables()
436431

437432
executable_application_instance.chain_prepend = []
438433
executable_application_instance._define_commands(exec_graph)
@@ -484,9 +479,6 @@ def test_derive_variables_for_template_path(mutable_mock_apps_repo, workspace_na
484479

485480
expansion_vars = basic_exp_dict()
486481

487-
# Set up the instance to pass the initial part of the function
488-
executable_application_instance.expander = ramble.expander.Expander(expansion_vars, None)
489-
490482
test_wl = ramble.workload.Workload("test_wl", executables=["foo"], inputs=["input"])
491483
test_wl2 = ramble.workload.Workload("test_wl2", executables=["bar"], inputs=["input"])
492484
test_wl2.add_variable(ramble.workload.WorkloadVariable("n_ranks", default="1"))
@@ -495,12 +487,10 @@ def test_derive_variables_for_template_path(mutable_mock_apps_repo, workspace_na
495487
executable_application_instance.internals = {}
496488

497489
executable_application_instance.inputs[_FS] = {"input": {"target_dir": "."}}
498-
executable_application_instance.variables = {}
490+
executable_application_instance.set_variables(expansion_vars, None)
499491

500492
exec_graph = executable_application_instance._get_executable_graph("test_wl2")
501493

502-
executable_application_instance._set_default_experiment_variables()
503-
504494
executable_application_instance.chain_prepend = []
505495
executable_application_instance._define_commands(exec_graph)
506496
executable_application_instance._define_formatted_executables()

var/ramble/repos/builtin/package_managers/spack-lightweight/test/spack_runner.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -327,6 +327,8 @@ def test_new_compiler_installs(tmpdir, capsys, request):
327327
import os
328328

329329
with tmpdir.as_cwd():
330+
# compilers.yaml is written to support older spack versions.
331+
# The compiler information was changed to exist in the packages.yaml file around v1.0.0
330332
compilers_config = """
331333
compilers::
332334
- compiler:
@@ -352,6 +354,11 @@ def test_new_compiler_installs(tmpdir, capsys, request):
352354
externals:
353355
- spec: [email protected] languages=c,fortran
354356
prefix: {os.getcwd()}
357+
extra_attributes:
358+
compilers:
359+
c: /tmpdir_path/gcc
360+
cxx: /tmpdir_path/g++
361+
fortran: /tmpdir_path/gfortran
355362
buildable: false
356363
"""
357364

0 commit comments

Comments
 (0)