Source code for compass.ocean.tests.drying_slope.decomp

from compass.ocean.tests.drying_slope.forward import Forward
from compass.ocean.tests.drying_slope.initial_state import InitialState
from compass.testcase import TestCase
from compass.validate import compare_variables

[docs] class Decomp(TestCase): """ A decomposition test case for the baroclinic channel test group, which makes sure the model produces identical results on 1 and 12 cores. Attributes ---------- resolution : str The resolution of the test case coord_type : str The type of vertical coordinate (``sigma``, ``single_layer``, etc.) """
[docs] def __init__(self, test_group, resolution, coord_type, method): """ Create the test case Parameters ---------- test_group : compass.ocean.tests.drying_slope.DryingSlope The test group that this test case belongs to resolution : float The resolution of the test case in km coord_type : str The type of vertical coordinate (``sigma``, ``single_layer``) method : str The type of wetting-and-drying algorithm """ name = 'decomp' self.resolution = resolution self.coord_type = coord_type if resolution < 1.: res_name = f'{int(resolution*1e3)}m' else: res_name = f'{int(resolution)}km' subdir = f'{coord_type}/{method}/{res_name}/{name}' super().__init__(test_group=test_group, name=name, subdir=subdir) self.add_step(InitialState(test_case=self, resolution=resolution)) if coord_type == 'single_layer': damping_coeff = None else: damping_coeff = 0.01 for procs in [1, 12]: name = '{}proc'.format(procs) forward_step = Forward(test_case=self, name=name, subdir=name, resolution=resolution, ntasks=procs, openmp_threads=1, damping_coeff=damping_coeff, coord_type=coord_type) if method == 'ramp': forward_step.add_namelist_options( {'config_zero_drying_velocity_ramp': ".true."}) self.add_step(forward_step)
[docs] def validate(self): """ Test cases can override this method to perform validation of variables and timers """ variables = ['temperature', 'salinity', 'layerThickness', 'normalVelocity'] compare_variables(test_case=self, variables=variables, filename1='1proc/', filename2='12proc/')