def __init__(self, test_group, mesh, init, time_integrator):
"""
Create test case
Parameters
----------
test_group : compass.ocean.tests.global_ocean.GlobalOcean
The global ocean test group that this test case belongs to
mesh : compass.ocean.tests.global_ocean.mesh.Mesh
The test case that produces the mesh for this run
init : compass.ocean.tests.global_ocean.init.Init
The test case that produces the initial condition for this run
time_integrator : {'split_explicit', 'RK4'}
The time integrator to use for the forward run
"""
super().__init__(test_group=test_group,
mesh=mesh,
init=init,
time_integrator=time_integrator,
name='threads_test')
for threads in [1, 2]:
name = '{}thread'.format(threads)
step = ForwardStep(test_case=self,
mesh=mesh,
init=init,
time_integrator=time_integrator,
name=name,
subdir=name,
cores=4,
threads=threads)
step.add_output_file(filename='output.nc')
self.add_step(step)
def __init__(self, test_group, mesh, init, time_integrator):
"""
Create test case
Parameters
----------
test_group : compass.ocean.tests.global_ocean.GlobalOcean
The global ocean test group that this test case belongs to
mesh : compass.ocean.tests.global_ocean.mesh.Mesh
The test case that produces the mesh for this run
init : compass.ocean.tests.global_ocean.init.Init
The test case that produces the initial condition for this run
time_integrator : {'split_explicit', 'RK4'}
The time integrator to use for the forward run
"""
super().__init__(test_group=test_group,
mesh=mesh,
init=init,
time_integrator=time_integrator,
name='performance_test')
step = ForwardStep(test_case=self,
mesh=mesh,
init=init,
time_integrator=time_integrator)
step.add_output_file(filename='output.nc')
if mesh.with_ice_shelf_cavities:
module = self.__module__
step.add_namelist_file(module, 'namelist.wisc')
step.add_streams_file(module, 'streams.wisc')
step.add_output_file(filename='land_ice_fluxes.nc')
self.add_step(step)
def __init__(self, test_group, mesh, init, time_integrator):
"""
Create the test case
Parameters
----------
test_group : compass.ocean.tests.global_ocean.GlobalOcean
The global ocean test group that this test case belongs to
mesh : compass.ocean.tests.global_ocean.mesh.Mesh
The test case that produces the mesh for this run
init : compass.ocean.tests.global_ocean.init.Init
The test case that produces the initial condition for this run
time_integrator : {'split_explicit', 'RK4'}
The time integrator to use for the forward run
"""
if time_integrator != 'split_explicit':
raise ValueError('{} dynamic adjustment not defined for {}'.format(
mesh.mesh_name, time_integrator))
restart_times = [
'0001-01-01_06:00:00', '0001-01-01_12:00:00',
'0001-01-02_00:00:00', '0001-01-03_00:00:00',
'0001-01-04_00:00:00', '0001-01-07_00:00:00', '0001-01-31_00:00:00'
]
restart_filenames = [
'restarts/rst.{}.nc'.format(restart_time.replace(':', '.'))
for restart_time in restart_times
]
super().__init__(test_group=test_group,
mesh=mesh,
init=init,
time_integrator=time_integrator,
restart_filenames=restart_filenames)
module = self.__module__
global_stats = {
'config_AM_globalStats_enable': '.true.',
'config_AM_globalStats_compute_on_startup': '.true.',
'config_AM_globalStats_write_on_startup': '.true.'
}
# first step
step_name = 'damped_adjustment_1'
step = ForwardStep(test_case=self,
mesh=mesh,
init=init,
time_integrator=time_integrator,
name=step_name,
subdir=step_name)
namelist_options = {
'config_run_duration': "'00-00-00_06:00:00'",
'config_dt': "'00:00:30'",
'config_btr_dt': "'00:00:01.5'",
'config_Rayleigh_friction': '.true.',
'config_Rayleigh_damping_coeff': '1.0e-3'
}
namelist_options.update(global_stats)
step.add_namelist_options(namelist_options)
stream_replacements = {
'output_interval': '00-00-10_00:00:00',
'restart_interval': '00-00-00_06:00:00'
}
step.add_streams_file(module,
'streams.template',
template_replacements=stream_replacements)
step.add_output_file(filename='../{}'.format(restart_filenames[0]))
self.add_step(step)
# second step
step_name = 'damped_adjustment_2'
step = ForwardStep(test_case=self,
mesh=mesh,
init=init,
time_integrator=time_integrator,
name=step_name,
subdir=step_name)
namelist_options = {
'config_run_duration': "'00-00-00_06:00:00'",
'config_dt': "'00:01:00'",
'config_btr_dt': "'00:00:03'",
'config_Rayleigh_friction': '.true.',
'config_Rayleigh_damping_coeff': '4.0e-4',
'config_do_restart': '.true.',
'config_start_time': "'{}'".format(restart_times[0])
}
namelist_options.update(global_stats)
step.add_namelist_options(namelist_options)
stream_replacements = {
'output_interval': '00-00-10_00:00:00',
'restart_interval': '00-00-00_06:00:00'
}
step.add_streams_file(module,
'streams.template',
template_replacements=stream_replacements)
step.add_input_file(filename='../{}'.format(restart_filenames[0]))
step.add_output_file(filename='../{}'.format(restart_filenames[1]))
self.add_step(step)
# third step
step_name = 'damped_adjustment_3'
step = ForwardStep(test_case=self,
mesh=mesh,
init=init,
time_integrator=time_integrator,
name=step_name,
subdir=step_name)
namelist_options = {
'config_run_duration': "'00-00-00_12:00:00'",
'config_dt': "'00:02:00'",
'config_btr_dt': "'00:00:06'",
'config_Rayleigh_friction': '.true.',
'config_Rayleigh_damping_coeff': '1.0e-4',
'config_do_restart': '.true.',
'config_start_time': "'{}'".format(restart_times[1])
}
namelist_options.update(global_stats)
step.add_namelist_options(namelist_options)
stream_replacements = {
'output_interval': '00-00-10_00:00:00',
'restart_interval': '00-00-00_12:00:00'
}
step.add_streams_file(module,
'streams.template',
template_replacements=stream_replacements)
step.add_input_file(filename='../{}'.format(restart_filenames[1]))
step.add_output_file(filename='../{}'.format(restart_filenames[2]))
self.add_step(step)
# fourth step
step_name = 'damped_adjustment_4'
step = ForwardStep(test_case=self,
mesh=mesh,
init=init,
time_integrator=time_integrator,
name=step_name,
subdir=step_name)
namelist_options = {
'config_run_duration': "'00-00-01_00:00:00'",
'config_dt': "'00:03:00'",
'config_btr_dt': "'00:00:09'",
'config_Rayleigh_friction': '.true.',
'config_Rayleigh_damping_coeff': '4.0e-5',
'config_do_restart': '.true.',
'config_start_time': "'{}'".format(restart_times[2])
}
namelist_options.update(global_stats)
step.add_namelist_options(namelist_options)
stream_replacements = {
'output_interval': '00-00-10_00:00:00',
'restart_interval': '00-00-01_00:00:00'
}
step.add_streams_file(module,
'streams.template',
template_replacements=stream_replacements)
step.add_input_file(filename='../{}'.format(restart_filenames[2]))
step.add_output_file(filename='../{}'.format(restart_filenames[3]))
self.add_step(step)
# fifth step
step_name = 'damped_adjustment_5'
step = ForwardStep(test_case=self,
mesh=mesh,
init=init,
time_integrator=time_integrator,
name=step_name,
subdir=step_name)
namelist_options = {
'config_run_duration': "'00-00-01_00:00:00'",
'config_dt': "'00:05:00'",
'config_btr_dt': "'00:00:12'",
'config_Rayleigh_friction': '.true.',
'config_Rayleigh_damping_coeff': '2.0e-5',
'config_do_restart': '.true.',
'config_start_time': "'{}'".format(restart_times[3])
}
namelist_options.update(global_stats)
step.add_namelist_options(namelist_options)
stream_replacements = {
'output_interval': '00-00-10_00:00:00',
'restart_interval': '00-00-01_00:00:00'
}
step.add_streams_file(module,
'streams.template',
template_replacements=stream_replacements)
step.add_input_file(filename='../{}'.format(restart_filenames[3]))
step.add_output_file(filename='../{}'.format(restart_filenames[4]))
self.add_step(step)
# sixth step
step_name = 'damped_adjustment_6'
step = ForwardStep(test_case=self,
mesh=mesh,
init=init,
time_integrator=time_integrator,
name=step_name,
subdir=step_name)
namelist_options = {
'config_run_duration': "'00-00-03_00:00:00'",
'config_dt': "'00:07:30'",
'config_btr_dt': "'00:00:15'",
'config_do_restart': '.true.',
'config_start_time': "'{}'".format(restart_times[4])
}
namelist_options.update(global_stats)
step.add_namelist_options(namelist_options)
stream_replacements = {
'output_interval': '00-00-10_00:00:00',
'restart_interval': '00-00-03_00:00:00'
}
step.add_streams_file(module,
'streams.template',
template_replacements=stream_replacements)
step.add_input_file(filename='../{}'.format(restart_filenames[4]))
step.add_output_file(filename='../{}'.format(restart_filenames[5]))
self.add_step(step)
# final step
step_name = 'simulation'
step = ForwardStep(test_case=self,
mesh=mesh,
init=init,
time_integrator=time_integrator,
name=step_name,
subdir=step_name)
namelist_options = {
'config_run_duration': "'00-00-24_00:00:00'",
'config_do_restart': '.true.',
'config_start_time': "'{}'".format(restart_times[5])
}
namelist_options.update(global_stats)
step.add_namelist_options(namelist_options)
stream_replacements = {
'output_interval': '00-00-10_00:00:00',
'restart_interval': '00-00-06_00:00:00'
}
step.add_streams_file(module,
'streams.template',
template_replacements=stream_replacements)
step.add_input_file(filename='../{}'.format(restart_filenames[5]))
step.add_output_file(filename='../{}'.format(restart_filenames[6]))
step.add_output_file(filename='output.nc')
self.add_step(step)
self.restart_filenames = restart_filenames
def __init__(self, test_group, mesh, init, time_integrator):
"""
Create the test case
Parameters
----------
test_group : compass.ocean.tests.global_ocean.GlobalOcean
The global ocean test group that this test case belongs to
mesh : compass.ocean.tests.global_ocean.mesh.Mesh
The test case that produces the mesh for this run
init : compass.ocean.tests.global_ocean.init.Init
The test case that produces the initial condition for this run
time_integrator : {'split_explicit', 'RK4'}
The time integrator to use for the forward run
"""
restart_times = ['0001-01-02_00:00:00', '0001-01-03_00:00:00']
restart_filenames = [
'restarts/rst.{}.nc'.format(restart_time.replace(':', '.'))
for restart_time in restart_times
]
super().__init__(test_group=test_group,
mesh=mesh,
init=init,
time_integrator=time_integrator,
restart_filenames=restart_filenames)
module = self.__module__
global_stats = {
'config_AM_globalStats_enable': '.true.',
'config_AM_globalStats_compute_on_startup': '.true.',
'config_AM_globalStats_write_on_startup': '.true.'
}
# first step
step_name = 'damped_adjustment_1'
step = ForwardStep(test_case=self,
mesh=mesh,
init=init,
time_integrator=time_integrator,
name=step_name,
subdir=step_name)
namelist_options = {
'config_run_duration': "'00-00-01_00:00:00'",
'config_Rayleigh_friction': '.true.',
'config_Rayleigh_damping_coeff': '1.0e-4'
}
namelist_options.update(global_stats)
step.add_namelist_options(namelist_options)
stream_replacements = {
'output_interval': '00-00-01_00:00:00',
'restart_interval': '00-00-01_00:00:00'
}
step.add_streams_file(module,
'streams.template',
template_replacements=stream_replacements)
step.add_output_file(filename='../{}'.format(restart_filenames[0]))
self.add_step(step)
# final step
step_name = 'simulation'
step = ForwardStep(test_case=self,
mesh=mesh,
init=init,
time_integrator=time_integrator,
name=step_name,
subdir=step_name)
namelist_options = {
'config_run_duration': "'00-00-01_00:00:00'",
'config_do_restart': '.true.',
'config_start_time': "'{}'".format(restart_times[0])
}
namelist_options.update(global_stats)
step.add_namelist_options(namelist_options)
stream_replacements = {
'output_interval': '00-00-01_00:00:00',
'restart_interval': '00-00-01_00:00:00'
}
step.add_streams_file(module,
'streams.template',
template_replacements=stream_replacements)
step.add_input_file(filename='../{}'.format(restart_filenames[0]))
step.add_output_file(filename='../{}'.format(restart_filenames[1]))
step.add_output_file(filename='output.nc')
self.add_step(step)
def __init__(self, test_group, mesh, init, time_integrator):
"""
Create test case
Parameters
----------
test_group : compass.ocean.tests.global_ocean.GlobalOcean
The global ocean test group that this test case belongs to
mesh : compass.ocean.tests.global_ocean.mesh.Mesh
The test case that produces the mesh for this run
init : compass.ocean.tests.global_ocean.init.Init
The test case that produces the initial condition for this run
time_integrator : {'split_explicit', 'RK4'}
The time integrator to use for the forward run
"""
super().__init__(test_group=test_group,
mesh=mesh,
init=init,
time_integrator=time_integrator,
name='restart_test')
module = __name__
restart_time = {
'split_explicit': '0001-01-01_04:00:00',
'RK4': '0001-01-01_00:10:00'
}
restart_filename = '../restarts/rst.{}.nc'.format(
restart_time[time_integrator].replace(':', '.'))
input_file = {'restart': restart_filename}
output_file = {'full': restart_filename}
for part in ['full', 'restart']:
name = '{}_run'.format(part)
step = ForwardStep(test_case=self,
mesh=mesh,
init=init,
time_integrator=time_integrator,
name=name,
subdir=name,
cores=4,
threads=1)
suffix = '{}.{}'.format(time_integrator.lower(), part)
step.add_namelist_file(module, 'namelist.{}'.format(suffix))
step.add_streams_file(module, 'streams.{}'.format(suffix))
if part in input_file:
step.add_input_file(filename=input_file[part])
if part in output_file:
step.add_output_file(filename=output_file[part])
step.add_output_file(filename='output.nc')
self.add_step(step)
def __init__(self, test_group, mesh, init, time_integrator):
"""
Create test case
Parameters
----------
test_group : compass.ocean.tests.global_ocean.GlobalOcean
The global ocean test group that this test case belongs to
mesh : compass.ocean.tests.global_ocean.mesh.Mesh
The test case that produces the mesh for this run
init : compass.ocean.tests.global_ocean.init.Init
The test case that produces the initial condition for this run
time_integrator : {'split_explicit', 'RK4'}
The time integrator to use for the forward run
"""
super().__init__(test_group=test_group,
mesh=mesh,
init=init,
time_integrator=time_integrator,
name='analysis_test')
step = ForwardStep(test_case=self,
mesh=mesh,
init=init,
time_integrator=time_integrator,
cores=4,
threads=1)
self.variables = {
'output.nc':
['temperature', 'salinity', 'layerThickness', 'normalVelocity'],
'analysis_members/globalStats.0001-01-01_00.00.00.nc': [
'kineticEnergyCellMax', 'kineticEnergyCellMin',
'kineticEnergyCellAvg', 'temperatureAvg', 'salinityAvg'
],
'analysis_members/debugDiagnostics.0001-01-01.nc': ['rx1MaxCell'],
'analysis_members/highFrequencyOutput.0001-01-01.nc':
['temperatureAt250m'],
'analysis_members/mixedLayerDepths.0001-01-01.nc':
['dThreshMLD', 'tThreshMLD'],
'analysis_members/waterMassCensus.0001-01-01_00.00.00.nc':
['waterMassCensusTemperatureValues'],
'analysis_members/eliassenPalm.0001-01-01.nc': ['EPFT'],
'analysis_members/'
'layerVolumeWeightedAverage.0001-01-01_00.00.00.nc':
['avgVolumeTemperature', 'avgVolumeRelativeVorticityCell'],
'analysis_members/okuboWeiss.0001-01-01_00.00.00.nc':
['okuboWeiss'],
'analysis_members/zonalMeans.0001-01-01_00.00.00.nc':
['velocityZonalZonalMean', 'temperatureZonalMean'],
'analysis_members/'
'meridionalHeatTransport.0001-01-01_00.00.00.nc':
['meridionalHeatTransportLat'],
'analysis_members/'
'surfaceAreaWeightedAverages.0001-01-01_00.00.00.nc':
['avgSurfaceSalinity', 'avgSeaSurfacePressure'],
'analysis_members/'
'eddyProductVariables.0001-01-01.nc': [
'SSHSquared', 'velocityZonalSquared',
'velocityZonalTimesTemperature'
],
'analysis_members/oceanHeatContent.0001-01-01.nc': [
'oceanHeatContentSfcToBot', 'oceanHeatContentSfcTo700m',
'oceanHeatContent700mTo2000m', 'oceanHeatContent2000mToBot'
],
'analysis_members/mixedLayerHeatBudget.0001-01-01.nc': [
'temperatureHorAdvectionMLTend', 'salinityHorAdvectionMLTend',
'temperatureML', 'salinityML', 'bruntVaisalaFreqML'
]
}
for output in self.variables:
step.add_output_file(output)
module = self.__module__
step.add_namelist_file(module, 'namelist.forward')
step.add_streams_file(module, 'streams.forward')
self.add_step(step)