@@ -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 ()
0 commit comments