Source code for compass.ocean.tests.sphere_transport.divergent_2d.init

from compass.model import run_model
from compass.step import Step


[docs]class Init(Step): """ A step for an initial condition for for the cosine bell test case """
[docs] def __init__(self, test_case, resolution): """ Create the step Parameters ---------- test_case : compass.ocean.tests.sphere_transport.divergent_2d.Divergent2D The test case this step belongs to resolution : int The resolution of the (uniform) mesh in km """ super().__init__(test_case=test_case, name='QU{}_init'.format(resolution), subdir='QU{}/init'.format(resolution), cores=36, min_cores=1) package = 'compass.ocean.tests.sphere_transport.divergent_2d' self.add_namelist_file(package, 'namelist.init', mode='init') self.add_streams_file(package, 'streams.init', mode='init') self.add_input_file(filename='mesh.nc', target='../mesh/mesh.nc') self.add_input_file(filename='graph.info', target='../mesh/graph.info') self.add_model_as_input() self.add_output_file(filename='namelist.ocean') self.add_output_file(filename='initial_state.nc')
[docs] def run(self): """ Run this step of the testcase """ run_model(self)