From: Wolfgang Bangerth <bangerth@colostate.edu>
Date: Thu, 18 Jan 2018 22:45:56 +0000 (-0700)
Subject: Add a test for LinearAlgebra::distributed::Vector::add_and_dot().
X-Git-Tag: v9.0.0-rc1~552^2
X-Git-Url: https://gitweb.dealii.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=refs%2Fpull%2F5760%2Fhead;p=dealii.git

Add a test for LinearAlgebra::distributed::Vector::add_and_dot().
---

diff --git a/tests/lac/la_parallel_vector_add_and_dot_complex.cc b/tests/lac/la_parallel_vector_add_and_dot_complex.cc
new file mode 100644
index 0000000000..ae740d55a2
--- /dev/null
+++ b/tests/lac/la_parallel_vector_add_and_dot_complex.cc
@@ -0,0 +1,78 @@
+// ---------------------------------------------------------------------
+//
+// Copyright (C) 2012 - 2015 by the deal.II authors
+//
+// This file is part of the deal.II library.
+//
+// The deal.II library is free software; you can use it, redistribute
+// it, and/or modify it under the terms of the GNU Lesser General
+// Public License as published by the Free Software Foundation; either
+// version 2.1 of the License, or (at your option) any later version.
+// The full text of the license can be found in the file LICENSE at
+// the top level of the deal.II distribution.
+//
+// ---------------------------------------------------------------------
+
+
+// check that LinearAlgebra::distributed::Vector::add_and_dot works correctly for complex-valued
+// vectors
+
+#include "../tests.h"
+#include <deal.II/lac/la_parallel_vector.h>
+
+
+
+
+template <typename number>
+void check ()
+{
+  for (unsigned int test=0; test<5; ++test)
+    {
+      const unsigned int size = 17 + test*1101;
+      const IndexSet complete_set = complete_index_set(size);
+      LinearAlgebra::distributed::Vector<std::complex<number>> v1 (complete_set, MPI_COMM_SELF), v2(complete_set, MPI_COMM_SELF), v3(complete_set, MPI_COMM_SELF), check(complete_set, MPI_COMM_SELF);
+      for (unsigned int i=0; i<size; ++i)
+        {
+          v1(i) = std::complex<number>(0.1 + 0.005 * i, 1.234 + 12 * i);
+          v2(i) = std::complex<number>(-5.2 + 0.18 * i, 42.4242 + 42 * i);
+          v3(i) = std::complex<number>(3.14159 + 2.7183/(1.+i), 13.);
+        }
+      check = v1;
+      const std::complex<number> factor = std::complex<number>(0.01432);
+
+      // do things by hand once
+      v1.add(factor, v2);
+      const std::complex<number> prod = v1 * v3;
+
+      // then do it a second time with the add_and_dot function
+      const std::complex<number> prod_check = check.add_and_dot(factor, v2, v3);
+      if (test == 0 && std::is_same<number,double>::value)
+        {
+          deallog << "Vector add reference:   ";
+          v1.print(deallog.get_file_stream());
+          deallog << std::endl;
+          deallog << "Vector check reference: ";
+          check.print(deallog.get_file_stream());
+          deallog << std::endl;
+        }
+
+      deallog << "Add and dot should be " << prod/static_cast<number>(size)
+              << ", is " << prod_check/static_cast<number>(size)
+              << std::endl;
+    }
+}
+
+
+int main()
+{
+  std::ofstream logfile("output");
+  deallog << std::fixed;
+  deallog << std::setprecision(2);
+  deallog.attach(logfile);
+
+  check<float>();
+  check<double>();
+  deallog << "OK" << std::endl;
+}
+
+
diff --git a/tests/lac/la_parallel_vector_add_and_dot_complex.output b/tests/lac/la_parallel_vector_add_and_dot_complex.output
new file mode 100644
index 0000000000..0dc50c403e
--- /dev/null
+++ b/tests/lac/la_parallel_vector_add_and_dot_complex.output
@@ -0,0 +1,22 @@
+
+DEAL::Add and dot should be (1334.79,349.71), is (1334.79,349.71)
+DEAL::Add and dot should be (91530.09,22094.70), is (91530.09,22094.70)
+DEAL::Add and dot should be (181725.41,43834.06), is (181725.41,43834.06)
+DEAL::Add and dot should be (271920.72,65573.37), is (271920.72,65573.37)
+DEAL::Add and dot should be (362116.03,87312.66), is (362116.03,87312.66)
+Process #0
+Local range: [0, 17), global size: 17
+Vector data:
+(2.554e-02,1.842e+00) (3.311e-02,1.444e+01) (4.069e-02,2.704e+01) (4.827e-02,3.965e+01) (5.585e-02,5.225e+01) (6.342e-02,6.485e+01) (7.100e-02,7.745e+01) (7.858e-02,9.005e+01) (8.616e-02,1.027e+02) (9.373e-02,1.153e+02) (1.013e-01,1.279e+02) (1.089e-01,1.405e+02) (1.165e-01,1.531e+02) (1.240e-01,1.657e+02) (1.316e-01,1.783e+02) (1.392e-01,1.909e+02) (1.468e-01,2.035e+02) 
+DEAL::Vector add reference:   
+Process #0
+Local range: [0, 17), global size: 17
+Vector data:
+(2.554e-02,1.842e+00) (3.311e-02,1.444e+01) (4.069e-02,2.704e+01) (4.827e-02,3.965e+01) (5.585e-02,5.225e+01) (6.342e-02,6.485e+01) (7.100e-02,7.745e+01) (7.858e-02,9.005e+01) (8.616e-02,1.027e+02) (9.373e-02,1.153e+02) (1.013e-01,1.279e+02) (1.089e-01,1.405e+02) (1.165e-01,1.531e+02) (1.240e-01,1.657e+02) (1.316e-01,1.783e+02) (1.392e-01,1.909e+02) (1.468e-01,2.035e+02) 
+DEAL::Vector check reference: 
+DEAL::Add and dot should be (1334.79,349.71), is (1334.79,349.71)
+DEAL::Add and dot should be (91530.09,22094.70), is (91530.09,22094.70)
+DEAL::Add and dot should be (181725.40,43834.06), is (181725.40,43834.06)
+DEAL::Add and dot should be (271920.71,65573.37), is (271920.71,65573.37)
+DEAL::Add and dot should be (362116.02,87312.66), is (362116.02,87312.66)
+DEAL::OK