from compass.ocean.mesh.cull import CullMeshStep
from compass.ocean.tests.hurricane.configure import configure_hurricane
from compass.ocean.tests.hurricane.lts.mesh.lts_regions import LTSRegionsStep
from compass.ocean.tests.hurricane.mesh.dequ120at30cr10rr2 import (
DEQU120at30cr10rr2BaseMesh,
)
from compass.testcase import TestCase
[docs]
class Mesh(TestCase):
"""
A test case for creating a global MPAS-Ocean mesh
"""
[docs]
def __init__(self, test_group, mesh_name, use_lts):
"""
Create test case for creating a global MPAS-Ocean mesh
Parameters
----------
test_group : compass.ocean.tests.hurricane.Hurricane
The test group that this test case belongs to
mesh_name : str
The name of the mesh
use_lts: bool
Whether local time-stepping is used
"""
self.mesh_name = mesh_name
self.use_lts = use_lts
if use_lts:
name = 'mesh_lts'
else:
name = 'mesh'
subdir = '{}/{}'.format(mesh_name, name)
super().__init__(test_group=test_group, name=name, subdir=subdir)
name = 'base_mesh'
if mesh_name == 'DEQU120at30cr10rr2':
base_mesh_step = DEQU120at30cr10rr2BaseMesh(
self, name=name, preserve_floodplain=False)
mesh_lower = 'dequ120at30cr10rr2'
elif mesh_name == 'DEQU120at30cr10rr2WD':
base_mesh_step = DEQU120at30cr10rr2BaseMesh(
self, name=name, preserve_floodplain=True)
mesh_lower = 'dequ120at30cr10rr2'
else:
raise ValueError(f'Unexpected mesh name {mesh_name}')
self.package = f'compass.ocean.tests.hurricane.mesh.{mesh_lower}'
self.mesh_config_filename = f'{mesh_lower}.cfg'
self.add_step(base_mesh_step)
cull_mesh_step = CullMeshStep(
test_case=self, base_mesh_step=base_mesh_step,
with_ice_shelf_cavities=False, do_inject_bathymetry=True,
preserve_floodplain=True)
self.add_step(cull_mesh_step)
if use_lts:
self.add_step(LTSRegionsStep(
test_case=self, cull_mesh_step=cull_mesh_step))