From d5639168232d1c111d6924b10c6abd74204abec1 Mon Sep 17 00:00:00 2001
From: Tavian Barnes <tavianator@gmail.com>
Date: Wed, 6 Oct 2010 20:44:32 -0400
Subject: Add Heun's method.

---
 tests/Heun.cpp    | 37 +++++++++++++++++++++++++++++++++++++
 tests/Makefile.am |  2 ++
 2 files changed, 39 insertions(+)
 create mode 100644 tests/Heun.cpp

(limited to 'tests')

diff --git a/tests/Heun.cpp b/tests/Heun.cpp
new file mode 100644
index 0000000..758524b
--- /dev/null
+++ b/tests/Heun.cpp
@@ -0,0 +1,37 @@
+#include "vZ.hpp"
+#include <cmath>
+#include <cstdlib>
+#include <iostream>
+#include <iomanip>
+
+// y' = y (y == C*exp(t))
+double
+f(double t, double y)
+{
+  return y;
+}
+
+int
+main()
+{
+  vZ::HeunIntegrator integrator(f);
+  integrator.y(1.0).x(0.0).h(0.02);
+
+  integrator.integrate(2.0);
+
+  double actual   = integrator.y();
+  double expected = std::exp(2.0);
+
+  std::cout << std::setprecision(10)
+            << "Numerical: " << actual << std::endl
+            << "Expected:  " << expected  << std::endl;
+
+  double error = std::fabs(expected - actual)/expected;
+  if (error > 1.4e-4) {
+    std::cerr << "Error:     " << 100.0*error << "%" << std::endl;
+    return EXIT_FAILURE;
+  } else {
+    std::cout << "Error:     " << 100.0*error << "%" << std::endl;
+    return EXIT_SUCCESS;
+  }
+}
diff --git a/tests/Makefile.am b/tests/Makefile.am
index edef1ae..0baa463 100644
--- a/tests/Makefile.am
+++ b/tests/Makefile.am
@@ -21,9 +21,11 @@ INCLUDES = -I$(top_srcdir)/src
 
 check_PROGRAMS = Euler-test                                                    \
                  Midpoint-test                                                 \
+                 Heun-test                                                     \
                  RK4-test
 TESTS          = $(check_PROGRAMS)
 
 Euler_test_SOURCES    = Euler.cpp
 Midpoint_test_SOURCES = Midpoint.cpp
+Heun_test_SOURCES     = Heun.cpp
 RK4_test_SOURCES      = RK4.cpp
-- 
cgit v1.2.3