Add reframe tests for Snapshots

reframe_tests/tests/test_snapshots.py

@@ -0,0 +1,42 @@
+import reframe as rfm
+from reframe.core.builtins import parameter
+
+from reframe_tests.tests.base_tests import GprMaxMPIRegressionTest, GprMaxRegressionTest
+
+
+@rfm.simple_test
+class Test2DSnapshot(GprMaxRegressionTest):
+    tags = {"test", "serial", "2d", "waveform", "hertzian_dipole", "snapshot"}
+    sourcesdir = "src/snapshot_tests"
+    model = parameter(["whole_domain_2d"])
+    has_receiver_output = False
+    snapshots = ["snapshot.h5"]
+
+
+@rfm.simple_test
+class TestSnapshot(GprMaxRegressionTest):
+    tags = {"test", "serial", "2d", "waveform", "hertzian_dipole", "snapshot"}
+    sourcesdir = "src/snapshot_tests"
+    model = parameter(["whole_domain"])
+    has_receiver_output = False
+    snapshots = ["snapshot.h5"]
+
+
+"""Test MPI Functionality
+"""
+
+
+@rfm.simple_test
+class Test2DSnapshotMpi(GprMaxMPIRegressionTest):
+    tags = {"test", "mpi", "2d", "waveform", "hertzian_dipole", "snapshot"}
+    mpi_layout = parameter([[2, 2, 1], [3, 3, 1], [4, 4, 1]])
+    serial_dependency = Test2DSnapshot
+    model = serial_dependency.model
+
+
+@rfm.simple_test
+class TestSnapshotMpi(GprMaxMPIRegressionTest):
+    tags = {"test", "mpi", "2d", "waveform", "hertzian_dipole", "snapshot"}
+    mpi_layout = parameter([[2, 2, 2], [3, 3, 3], [4, 4, 4]])
+    serial_dependency = TestSnapshot
+    model = serial_dependency.model