文件
gprMax/reframe_tests/tests/test_snapshots.py
2025-03-17 12:17:13 +00:00

92 行
2.5 KiB
Python

import reframe as rfm
from reframe.core.builtins import parameter
from reframe_tests.tests.mixins import MpiMixin
from reframe_tests.tests.standard_tests import GprMaxSnapshotTest
@rfm.simple_test
class Test2DSnapshot(GprMaxSnapshotTest):
tags = {"test", "serial", "2d", "waveform", "hertzian_dipole", "snapshot"}
sourcesdir = "src/snapshot_tests"
model = parameter(["whole_domain_2d"])
snapshots = ["snapshot_0.h5", "snapshot_1.h5", "snapshot_2.h5", "snapshot_3.h5"]
@rfm.simple_test
class TestSnapshot(GprMaxSnapshotTest):
tags = {"test", "serial", "2d", "waveform", "hertzian_dipole", "snapshot"}
sourcesdir = "src/snapshot_tests"
model = parameter(["whole_domain"])
snapshots = ["snapshot_0.h5", "snapshot_1.h5", "snapshot_2.h5", "snapshot_3.h5"]
@rfm.simple_test
class Test2DSliceSnapshot(GprMaxSnapshotTest):
tags = {"test", "serial", "2d", "waveform", "hertzian_dipole", "snapshot"}
sourcesdir = "src/snapshot_tests"
model = parameter(["2d_slices"])
snapshots = [
"snapshot_x_05.h5",
"snapshot_x_35.h5",
"snapshot_x_65.h5",
"snapshot_x_95.h5",
"snapshot_y_15.h5",
"snapshot_y_40.h5",
"snapshot_y_45.h5",
"snapshot_y_50.h5",
"snapshot_y_75.h5",
"snapshot_z_25.h5",
"snapshot_z_55.h5",
"snapshot_z_85.h5",
]
"""Test MPI Functionality
"""
@rfm.simple_test
class Test2DSnapshotMpi(MpiMixin, Test2DSnapshot):
tags = {"test", "mpi", "2d", "waveform", "hertzian_dipole", "snapshot"}
mpi_layout = parameter([[2, 2, 1], [3, 3, 1], [4, 4, 1]])
test_dependency = Test2DSnapshot
@rfm.simple_test
class TestSnapshotMpi(MpiMixin, TestSnapshot):
tags = {"test", "mpi", "2d", "waveform", "hertzian_dipole", "snapshot"}
mpi_layout = parameter(
[
[2, 1, 1],
[1, 2, 1],
[1, 1, 2],
[3, 1, 1],
[1, 3, 1],
[1, 1, 3],
[2, 2, 2],
[3, 3, 3],
[4, 4, 4],
]
)
test_dependency = TestSnapshot
@rfm.simple_test
class Test2DSliceSnapshotMpi(MpiMixin, Test2DSliceSnapshot):
tags = {"test", "mpi", "2d", "waveform", "hertzian_dipole", "snapshot"}
mpi_layout = parameter(
[
[2, 1, 1],
[1, 2, 1],
[1, 1, 2],
[3, 1, 1],
[1, 3, 1],
[1, 1, 3],
[2, 2, 2],
[3, 3, 3],
[4, 4, 4],
]
)
test_dependency = Test2DSliceSnapshot