Merge pull request #57 from memmett/feature/test-mpi

Add test for MPI version of advection/diffusion example.  This closes #41 as well.

Author: memmett

examples/advection_diffusion/mpi_pfasst.cpp

@@ -24,10 +24,8 @@ using namespace pfasst;
 using namespace pfasst::encap;
 using namespace pfasst::mpi;
 
-int main(int argc, char** argv)
+error_map run_mpi_pfasst()
 {
-  MPI_Init(&argc, &argv);
-
   const size_t nsteps = 4;
   const double dt     = 0.01;
   const size_t niters = 4;
@@ -64,7 +62,16 @@ int main(int argc, char** argv)
   pf.set_nsweeps({2, 1});
   pf.run();
 
-  fftw_cleanup();
+  auto fine = pf.get_finest<AdvectionDiffusionSweeper<>>();
+  return fine->get_errors();
+}
 
+#ifndef PFASST_UNIT_TESTING
+int main(int argc, char** argv)
+{
+  MPI_Init(&argc, &argv);
+  run_mpi_pfasst();
+  fftw_cleanup();
   MPI_Finalize();
 }
+#endif

tests/examples/advection_diffusion/CMakeLists.txt

@@ -9,6 +9,12 @@ set(TESTS
     test_advection_diffusion
 )
 
+if(${pfasst_WITH_MPI})
+    set(MPI_TESTS
+        test_mpi_advection_diffusion
+    )
+endif()
+
 foreach(test ${TESTS})
     message(STATUS "  ${test}")
     add_executable(${test} ${test}.cpp)
@@ -28,3 +34,33 @@ foreach(test ${TESTS})
         COMMAND ${CMAKE_RUNTIME_OUTPUT_DIRECTORY}/tests/examples/advection_diffusion/${test} --gtest_output=xml:${test}_out.xml
     )
 endforeach(test)
+
+if(${pfasst_WITH_MPI})
+    include_directories(${MPI_CXX_INCLUDE_PATH})
+    foreach(test ${MPI_TESTS})
+        message(STATUS "  ${test}")
+        add_executable(${test} ${test}.cpp)
+        add_dependencies(${test} googlemock)
+        if(NOT FFTW_FOUND)
+          add_dependencies(${test} googlemock fftw3)
+        endif()
+	if(MPI_COMPILE_FLAGS)
+            set_target_properties(${test} PROPERTIES COMPILE_FLAGS "${MPI_COMPILE_FLAGS}")
+        endif()
+        if(MPI_LINK_FLAGS)
+            set_target_properties(${test} PROPERTIES LINK_FLAGS "${MPI_LINK_FLAGS}")
+        endif()
+        target_link_libraries(${test}
+            ${3rdparty_DEPENDEND_LIBS}
+            ${FFTW_LIBRARIES}
+            ${pfasst_DEPENDED_LIBS}
+            ${MPI_CXX_LIBRARIES}
+        )
+        set_target_properties(${test}
+            PROPERTIES RUNTIME_OUTPUT_DIRECTORY ${CMAKE_RUNTIME_OUTPUT_DIRECTORY}/tests/examples/advection_diffusion
+        )
+        add_test(NAME ${test}
+            COMMAND ${MPIEXEC} ${MPIEXEC_NUMPROC_FLAG} 4 ${MPIEXEC_PREFLAGS} ${CMAKE_RUNTIME_OUTPUT_DIRECTORY}/tests/examples/advection_diffusion/${test} ${MPIEXEC_POSTFLAGS}
+        )
+    endforeach(test)
+endif()

tests/examples/advection_diffusion/test_mpi_advection_diffusion.cpp

@@ -0,0 +1,49 @@
+/*
+ * Tests for the advection-diffusion examples.
+ */
+
+#include <algorithm>
+#include <iostream>
+#include <tuple>
+#include <vector>
+
+#include <gtest/gtest.h>
+#include <gmock/gmock.h>
+
+#include <mpi.h>
+
+#define PFASST_UNIT_TESTING
+#include "../examples/advection_diffusion/mpi_pfasst.cpp"
+#undef PFASST_UNIT_TESTING
+
+
+using namespace std;
+
+TEST(ErrorTest, MPIPFASST)
+{
+  typedef error_map::value_type vtype;
+
+  auto errors = run_mpi_pfasst();
+  auto get_step  = [](const vtype x) { return get<0>(get<0>(x)); };
+  auto get_iter  = [](const vtype x) { return get<1>(get<0>(x)); };
+  auto get_error = [](const vtype x) { return get<1>(x); };
+
+  auto max_iter = get_iter(*std::max_element(errors.begin(), errors.end(),
+             [get_iter](const vtype p1, const vtype p2) { return get_iter(p1) < get_iter(p2); }));
+
+  vector<double> ex = { 1.1168e-13, 4.8849e-13, 5.3268e-13, 2.3059e-12 };
+  for (auto& x: errors) {
+    if (get_iter(x) == max_iter) {
+      EXPECT_NEAR(get_error(x), ex[get_step(x)], 1e-14);
+    }
+  }
+}
+
+int main(int argc, char** argv)
+{
+  testing::InitGoogleTest(&argc, argv);
+  MPI_Init(&argc, &argv);
+  auto result = RUN_ALL_TESTS();
+  MPI_Finalize();
+  return result;
+}