Source code for compass.ocean.tests.baroclinic_channel
from compass.ocean.tests.baroclinic_channel.decomp_test import DecompTest
from compass.ocean.tests.baroclinic_channel.default import Default
from compass.ocean.tests.baroclinic_channel.restart_test import RestartTest
from compass.ocean.tests.baroclinic_channel.rpe_test import RpeTest
from compass.ocean.tests.baroclinic_channel.threads_test import ThreadsTest
from compass.testgroup import TestGroup
[docs]
class BaroclinicChannel(TestGroup):
"""
A test group for baroclinic channel test cases
"""
[docs]
def __init__(self, mpas_core):
"""
mpas_core : compass.MpasCore
the MPAS core that this test group belongs to
"""
super().__init__(mpas_core=mpas_core, name='baroclinic_channel')
for resolution in ['1km', '4km', '10km']:
self.add_test_case(
RpeTest(test_group=self, resolution=resolution))
for resolution in ['10km']:
self.add_test_case(
DecompTest(test_group=self, resolution=resolution))
self.add_test_case(
Default(test_group=self, resolution=resolution))
self.add_test_case(
RestartTest(test_group=self, resolution=resolution))
self.add_test_case(
ThreadsTest(test_group=self, resolution=resolution))
[docs]
def configure(resolution, config):
"""
Modify the configuration options for one of the baroclinic test cases
Parameters
----------
resolution : str
The resolution of the test case
config : compass.config.CompassConfigParser
Configuration options for this test case
"""
res_params = {'10km': {'nx': 16,
'ny': 58, # NB. 2/sqrt(3) due to hexagons
'dc': 10e3},
'4km': {'nx': 40,
'ny': 144,
'dc': 4e3},
'1km': {'nx': 160,
'ny': 578,
'dc': 1e3}}
comment = {'nx': 'the number of mesh cells in the x direction',
'ny': 'the number of mesh cells in the y direction',
'dc': 'the distance between adjacent cell centers'}
if resolution not in res_params:
raise ValueError(f'Unsupported resolution {resolution}. Supported '
f'values are: {list(res_params)}')
res_params = res_params[resolution]
for param in res_params:
config.set('baroclinic_channel', param, str(res_params[param]),
comment=comment[param])