Source code for CIME.SystemTests.ers

"""
CIME restart test  This class inherits from SystemTestsCommon
"""
from CIME.XML.standard_module_setup import *
from CIME.SystemTests.system_tests_common import SystemTestsCommon
import glob

logger = logging.getLogger(__name__)


[docs] class ERS(SystemTestsCommon): def __init__(self, case, **kwargs): """ initialize an object interface to the ERS system test """ SystemTestsCommon.__init__(self, case, **kwargs) def _ers_first_phase(self): stop_n = self._case.get_value("STOP_N") stop_option = self._case.get_value("STOP_OPTION") rest_n = self._case.get_value("REST_N") expect(stop_n > 0, "Bad STOP_N: {:d}".format(stop_n)) expect(stop_n > 2, "ERROR: stop_n value {:d} too short".format(stop_n)) logger.info( "doing an {0} {1} initial test with restart file at {2} {1}".format( str(stop_n), stop_option, str(rest_n) ) ) self.run_indv() def _ers_second_phase(self): stop_n = self._case.get_value("STOP_N") stop_option = self._case.get_value("STOP_OPTION") rest_n = int(stop_n / 2 + 1) stop_new = stop_n - rest_n expect( stop_new > 0, "ERROR: stop_n value {:d} too short {:d} {:d}".format( stop_new, stop_n, rest_n ), ) rundir = self._case.get_value("RUNDIR") for pfile in glob.iglob(os.path.join(rundir, "PET*")): os.rename( pfile, os.path.join(os.path.dirname(pfile), "run1." + os.path.basename(pfile)), ) self._case.set_value("HIST_N", stop_n) self._case.set_value("STOP_N", stop_new) self._case.set_value("CONTINUE_RUN", True) self._case.set_value("REST_OPTION", "never") self._case.flush() logger.info("doing an {} {} restart test".format(str(stop_new), stop_option)) self._skip_pnl = False self.run_indv(suffix="rest") # Compare restart file self._component_compare_test("base", "rest")
[docs] def run_phase(self): self._ers_first_phase() self._ers_second_phase()