130 lines
3.8 KiB
Python
130 lines
3.8 KiB
Python
#!/usr/bin/env python3
|
|
|
|
"""System tests for gen_mksurfdata_jobscript_single
|
|
|
|
"""
|
|
|
|
import os
|
|
|
|
import unittest
|
|
import tempfile
|
|
import shutil
|
|
import sys
|
|
|
|
from ctsm.path_utils import path_to_ctsm_root
|
|
from ctsm.toolchain.gen_mksurfdata_jobscript_single import main
|
|
from ctsm import unit_testing
|
|
|
|
# Allow test names that pylint doesn't like; otherwise hard to make them
|
|
# readable
|
|
# pylint: disable=invalid-name
|
|
|
|
|
|
class TestSysGenMkSurfJSSingle(unittest.TestCase):
|
|
"""System tests for gen_mksurfdata_jobscript_single"""
|
|
|
|
def setUp(self):
|
|
"""Setp temporary directory to make the files in"""
|
|
self._original_wd = os.getcwd()
|
|
self._tempdir = tempfile.mkdtemp()
|
|
os.chdir(self._tempdir)
|
|
self.outfile = "jobscript.sh"
|
|
self.namelist = "res.namelist"
|
|
bld_path = os.path.join(
|
|
path_to_ctsm_root(), "python", "ctsm", "test", "testinputs", "mksurfdata_esmf_bld"
|
|
)
|
|
sys.argv = [
|
|
"gen_mksurfdata_jobscript_single",
|
|
"--bld-path",
|
|
bld_path,
|
|
"--namelist",
|
|
self.namelist,
|
|
"--jobscript-file",
|
|
self.outfile,
|
|
]
|
|
|
|
def tearDown(self):
|
|
"""
|
|
Remove temporary directory
|
|
"""
|
|
os.chdir(self._original_wd)
|
|
shutil.rmtree(self._tempdir, ignore_errors=True)
|
|
|
|
def createJS(self, nodes, tasks_per_node, option_list=None):
|
|
"""
|
|
Create a JobScript by sending a list of options in
|
|
"""
|
|
if option_list is None:
|
|
option_list = []
|
|
if len(option_list) > 1:
|
|
sys.argv.extend(option_list)
|
|
sys.argv.extend(
|
|
[
|
|
"--number-of-nodes",
|
|
nodes,
|
|
"--tasks-per-node",
|
|
tasks_per_node,
|
|
]
|
|
)
|
|
print(sys.argv)
|
|
main()
|
|
self.assertTrue(os.path.exists(self.outfile), "Output jobscript file should exist")
|
|
|
|
def test_simple_jobscript_single(self):
|
|
"""
|
|
Test that a standard simple namelist works
|
|
"""
|
|
# pylint: disable=no-self-use
|
|
self.createJS(nodes="4", tasks_per_node="12")
|
|
|
|
def test_casper_jobscript_single(self):
|
|
"""
|
|
Test that a standard simple namelist works for casper
|
|
"""
|
|
# pylint: disable=no-self-use
|
|
opt_list = ["--machine", "casper"]
|
|
self.createJS(nodes="4", tasks_per_node="12", option_list=opt_list)
|
|
|
|
def test_izumi_jobscript_single(self):
|
|
"""
|
|
Test that a standard simple namelist works for asper
|
|
"""
|
|
# pylint: disable=no-self-use
|
|
opt_list = ["--machine", "izumi"]
|
|
self.createJS(nodes="4", tasks_per_node="12", option_list=opt_list)
|
|
|
|
def test_bad_bld_path(self):
|
|
"""
|
|
Test aborts if the input bld-path does NOT exist
|
|
"""
|
|
# pylint: disable=no-self-use
|
|
with self.assertRaisesRegex(SystemExit, "Input Build path"):
|
|
self.createJS(nodes="4", tasks_per_node="12", option_list=["--bld-path", "zztop"])
|
|
|
|
def test_neg_nodes(self):
|
|
"""
|
|
Test aborts if the input node count is negative
|
|
"""
|
|
# pylint: disable=no-self-use
|
|
with self.assertRaisesRegex(
|
|
SystemExit,
|
|
"Input argument --number_of_nodes is zero or negative and needs to be positive",
|
|
):
|
|
self.createJS(nodes="-4", tasks_per_node="12")
|
|
|
|
def test_neg_tasks(self):
|
|
"""
|
|
Test aborts if the input tasks_per_node is zero or negative
|
|
"""
|
|
# pylint: disable=no-self-use
|
|
with self.assertRaisesRegex(
|
|
SystemExit,
|
|
"Input argument --tasks_per_node is zero or negative and needs to be positive",
|
|
):
|
|
self.createJS(nodes="4", tasks_per_node="0")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unit_testing.setup_for_tests()
|
|
unittest.main()
|