gprMax/reframe_tests/reframe_tests.py

import reframe as rfm
from base_tests import (
    GprMaxAPIRegressionTest,
    GprMaxBScanRegressionTest,
    GprMaxMPIRegressionTest,
    GprMaxRegressionTest,
    GprMaxTaskfarmRegressionTest,
)
from reframe.core.builtins import parameter, run_after, skip

"""ReFrame tests for basic functionality

    Usage:
        cd gprMax/reframe_tests
        reframe -C configuraiton/{CONFIG_FILE} -c reframe_tests.py -c base_tests.py -r
"""


class TestAscan(GprMaxRegressionTest):
    tags = {
        "test",
        "serial",
        "ascan",
        "2d",
        "hertzian_dipole",
        "waveform",
        "material",
        "box",
        "cylinder",
    }
    model = "cylinder_Ascan_2D"


@rfm.simple_test
class TestBscan(GprMaxBScanRegressionTest):
    tags = {
        "test",
        "serial",
        "bscan",
        "steps",
        "waveform",
        "hertzian_dipole",
        "material",
        "box",
        "cylinder",
    }
    model = "cylinder_Bscan_2D"
    num_models = 64


@rfm.simple_test
class TestSingleNodeTaskfarm(GprMaxTaskfarmRegressionTest):
    tags = {
        "test",
        "mpi",
        "taskfarm",
        "steps",
        "waveform",
        "hertzian_dipole",
        "material",
        "box",
        "cylinder",
    }
    model = "cylinder_Bscan_2D"
    num_tasks = 8
    num_tasks_per_node = 8
    num_models = 64
    serial_dependency = TestBscan


@rfm.simple_test
class TestMultiNodeTaskfarm(GprMaxTaskfarmRegressionTest):
    tags = {
        "test",
        "mpi",
        "taskfarm",
        "steps",
        "waveform",
        "hertzian_dipole",
        "material",
        "box",
        "cylinder",
    }
    model = "cylinder_Bscan_2D"
    num_tasks = 32
    num_tasks_per_node = 8
    num_models = 64
    serial_dependency = TestBscan


@rfm.simple_test
class Test2DModelXY(GprMaxRegressionTest):
    tags = {"test", "serial", "2d", "waveform", "hertzian_dipole"}
    model = "2D_EzHxHy"


@rfm.simple_test
class Test2DModelXZ(GprMaxRegressionTest):
    tags = {"test", "serial", "2d", "waveform", "hertzian_dipole"}
    model = "2D_EyzHxHz"


@rfm.simple_test
class Test2DModelYZ(GprMaxRegressionTest):
    tags = {"test", "serial", "2d", "waveform", "hertzian_dipole"}
    model = "2D_ExzHyHz"


@rfm.simple_test
class TestHertzianDipoleSource(GprMaxRegressionTest):
    tags = {"test", "serial", "hertzian_dipole", "waveform"}
    model = "hertzian_dipole_fs"


@rfm.simple_test
class TestMagneticDipoleSource(GprMaxRegressionTest):
    tags = {"test", "serial", "magnetic_dipole", "waveform"}
    model = "magnetic_dipole_fs"


@rfm.simple_test
class TestDispersiveMaterials(GprMaxRegressionTest):
    tags = {"test", "serial", "hertzian_dipole", "waveform", "material", "dispersive", "box"}
    model = "hertizian_dipole_dispersive"


@rfm.simple_test
class TestTransmissionLineSource(GprMaxRegressionTest):
    tags = {"test", "serial", "transmission_line", "waveform"}
    model = "transmission_line_fs"


@rfm.simple_test
class TestEdgeGeometry(GprMaxRegressionTest):
    tags = {"test", "serial", "geometry", "edge", "transmission_line", "waveform", "antenna"}
    model = "antenna_wire_dipole_fs"


@rfm.simple_test
class TestSubgrids(GprMaxAPIRegressionTest):
    tags = {
        "test",
        "api",
        "serial",
        "subgrid",
        "hertzian_dipole",
        "waveform",
        "material",
        "dispersive",
        "cylinder",
    }
    model = "cylinder_fs"


@rfm.simple_test
@skip("Takes ~1hr 30m on ARCHER2")
class TestSubgridsWithAntennaModel(GprMaxAPIRegressionTest):
    tags = {
        "test",
        "api",
        "serial",
        "subgrid",
        "antenna",
        "material",
        "box",
        "fractal_box",
        "add_surface_roughness",
    }
    model = "gssi_400_over_fractal_subsurface"


@rfm.simple_test
class Test2DModelXYMpi(GprMaxMPIRegressionTest):
    tags = {"test", "mpi", "2d", "waveform", "hertzian_dipole"}
    model = "2D_EzHxHy"
    mpi_layout = [4, 4, 1]
    serial_dependency = Test2DModelXY


@rfm.simple_test
class Test2DModelXZMpi(GprMaxMPIRegressionTest):
    tags = {"test", "mpi", "2d", "waveform", "hertzian_dipole"}
    model = "2D_EyzHxHz"
    mpi_layout = [4, 1, 4]
    serial_dependency = Test2DModelXZ


@rfm.simple_test
class Test2DModelYZMpi(GprMaxMPIRegressionTest):
    tags = {"test", "mpi", "2d", "waveform", "hertzian_dipole"}
    model = "2D_ExzHyHz"
    mpi_layout = [1, 4, 4]
    serial_dependency = Test2DModelYZ


@rfm.simple_test
class TestHertzianDipoleSourceMpi(GprMaxMPIRegressionTest):
    tags = {"test", "mpi", "hertzian_dipole", "waveform"}
    model = "hertzian_dipole_fs"
    mpi_layout = [3, 3, 3]
    serial_dependency = TestHertzianDipoleSource


@rfm.simple_test
class TestMagneticDipoleSourceMpi(GprMaxMPIRegressionTest):
    tags = {"test", "mpi", "magnetic_dipole", "waveform"}
    model = "magnetic_dipole_fs"
    mpi_layout = [3, 3, 3]
    serial_dependency = TestMagneticDipoleSource


@rfm.simple_test
class TestDispersiveMaterialsMpi(GprMaxMPIRegressionTest):
    tags = {"test", "mpi", "hertzian_dipole", "waveform", "material", "dispersive", "box"}
    model = "hertizian_dipole_dispersive"
    mpi_layout = [3, 3, 3]
    serial_dependency = TestDispersiveMaterials


@rfm.simple_test
class TestTransmissionLineSourceMpi(GprMaxMPIRegressionTest):
    tags = {"test", "mpi", "transmission_line", "waveform"}
    model = "transmission_line_fs"
    mpi_layout = [3, 3, 3]
    serial_dependency = TestTransmissionLineSource


@rfm.simple_test
class TestEdgeGeometryMpi(GprMaxMPIRegressionTest):
    tags = {"test", "mpi", "geometry", "edge", "transmission_line", "waveform", "antenna"}
    model = "antenna_wire_dipole_fs"
    mpi_layout = [3, 3, 3]
    serial_dependency = TestEdgeGeometry


@rfm.simple_test
class TestBoxGeometryNoPml(GprMaxRegressionTest):
    tags = {"test", "serial", "geometery", "box"}
    sourcesdir = "src/box_geometry_tests"
    models = parameter(["box_full_model", "box_half_model"])

    @run_after("init", always_last=True)
    def add_gprmax_commands(self):
        self.prerun_cmds.append(f"echo '#pml_cells: 0' >> {self.input_file}")


@rfm.simple_test
class TestBoxGeometryDefaultPml(GprMaxRegressionTest):
    tags = {"test", "serial", "geometery", "box"}
    sourcesdir = "src/box_geometry_tests"
    models = parameter(["box_full_model", "box_half_model"])


@rfm.simple_test
class TestBoxGeometryNoPmlMpi(GprMaxMPIRegressionTest):
    tags = {"test", "mpi", "geometery", "box"}
    models = TestBoxGeometryNoPml.models
    mpi_layout = [2, 2, 2]
    serial_dependency = TestBoxGeometryNoPml

    @run_after("init", always_last=True)
    def add_gprmax_commands(self):
        self.prerun_cmds.append(f"echo '#pml_cells: 0' >> {self.input_file}")


@rfm.simple_test
class TestBoxGeometryDefaultPmlMpi(GprMaxMPIRegressionTest):
    tags = {"test", "mpi", "geometery", "box"}
    models = TestBoxGeometryDefaultPml.models
    mpi_layout = [2, 2, 2]
    serial_dependency = TestBoxGeometryDefaultPml