tests: Add AdaptiveErrorTest for AD example to MPI PFASST.

Signed-off-by: Matthew Emmett <[email protected]>

Author: memmett

examples/advection_diffusion/mpi_pfasst.cpp

@@ -31,12 +31,10 @@ namespace pfasst
   {
     namespace advection_diffusion
     {
-      error_map run_mpi_pfasst()
+      error_map run_mpi_pfasst(double abs_residual_tol, size_t niters=4)
       {
-        // const size_t nsteps = 8;
         const size_t nsteps = 4;
         const double dt     = 0.01;
-        const size_t niters = 4;
 
         vector<pair<size_t, quadrature::QuadratureType>> nodes = {
           { 3, quadrature::QuadratureType::GaussLobatto },
@@ -68,7 +66,7 @@ namespace pfasst
         pf.set_comm(&comm);
         pf.set_duration(0.0, nsteps * dt, dt, niters);
         pf.set_nsweeps({2, 1});
-        // pf.get_finest<AdvectionDiffusionSweeper<>>()->set_residual_tolerances(1e-8, 0.0);
+        pf.get_finest<AdvectionDiffusionSweeper<>>()->set_residual_tolerances(abs_residual_tol, 0.0);
         pf.run();
 
         auto fine = pf.get_finest<AdvectionDiffusionSweeper<>>();
@@ -83,7 +81,7 @@ namespace pfasst
 int main(int argc, char** argv)
 {
   MPI_Init(&argc, &argv);
-  pfasst::examples::advection_diffusion::run_mpi_pfasst();
+  pfasst::examples::advection_diffusion::run_mpi_pfasst(0.0);
   fftw_cleanup();
   MPI_Finalize();
 }

tests/examples/advection_diffusion/test_mpi_advection_diffusion.cpp

@@ -24,7 +24,7 @@ TEST(ErrorTest, MPIPFASST)
 {
   typedef error_map::value_type vtype;
 
-  auto errors = run_mpi_pfasst();
+  auto errors = run_mpi_pfasst(0.0);
   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); };
@@ -40,6 +40,31 @@ TEST(ErrorTest, MPIPFASST)
   }
 }
 
+TEST(AdaptiveErrorTest, MPIPFASST)
+{
+  typedef error_map::value_type vtype;
+
+  auto errors = run_mpi_pfasst(1.e-8, 12);
+  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> ub = { 5e-8, 5e-8, 5e-8, 5e-8 };
+  for (auto& x: errors) {
+    if (get_iter(x) == max_iter) {
+      EXPECT_LE(get_error(x), ub[get_step(x)]);
+    }
+  }
+
+  int rank;
+  MPI_Comm_rank(MPI_COMM_WORLD, &rank);
+  vector<size_t> ic = { 1, 1, 2, 2 };
+  ASSERT_EQ(max_iter, (size_t) ic[rank]);
+}
+
 int main(int argc, char** argv)
 {
   testing::InitGoogleTest(&argc, argv);