From 6220e514eefced87d869d80b68dce790d64c9df2 Mon Sep 17 00:00:00 2001
From: nmannall <n.mannall@epcc.ed.ac.uk>
Date: Mon, 6 Jan 2025 16:15:11 +0000
Subject: [PATCH] Add MPI tests for example models

---
 reframe_tests/tests/test_example_models.py | 36 +++++++++++++++++++++-
 1 file changed, 35 insertions(+), 1 deletion(-)

diff --git a/reframe_tests/tests/test_example_models.py b/reframe_tests/tests/test_example_models.py
index 3029ad48..e669064b 100644
--- a/reframe_tests/tests/test_example_models.py
+++ b/reframe_tests/tests/test_example_models.py
@@ -1,7 +1,7 @@
 import reframe as rfm
 from reframe.core.builtins import parameter
 
-from reframe_tests.tests.mixins import BScanMixin
+from reframe_tests.tests.mixins import BScanMixin, MpiMixin
 from reframe_tests.tests.standard_tests import GprMaxRegressionTest
 
 """Reframe regression tests for example models in gprMax documentation
@@ -25,6 +25,23 @@ class TestAscan(GprMaxRegressionTest):
     model = parameter(["cylinder_Ascan_2D"])
 
 
+@rfm.simple_test
+class TestAscanMPI(MpiMixin, TestAscan):
+    tags = {
+        "test",
+        "mpi",
+        "ascan",
+        "2d",
+        "hertzian_dipole",
+        "waveform",
+        "material",
+        "box",
+        "cylinder",
+    }
+    mpi_layout = parameter([[2, 2, 1]])
+    test_dependency = TestAscan
+
+
 @rfm.simple_test
 class TestBscan(BScanMixin, GprMaxRegressionTest):
     tags = {
@@ -41,3 +58,20 @@ class TestBscan(BScanMixin, GprMaxRegressionTest):
     sourcesdir = "src/bscan_tests"
     model = parameter(["cylinder_Bscan_2D"])
     num_models = parameter([64])
+
+
+@rfm.simple_test
+class TestBscanMPI(MpiMixin, TestBscan):
+    tags = {
+        "test",
+        "mpi",
+        "bscan",
+        "steps",
+        "waveform",
+        "hertzian_dipole",
+        "material",
+        "box",
+        "cylinder",
+    }
+    mpi_layout = parameter([[2, 2, 1]])
+    test_dependency = TestBscan