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 +++++++++++++++++++++++++++++++++++++
 1 file changed, 37 insertions(+)
 create mode 100644 tests/Heun.cpp

(limited to 'tests/Heun.cpp')

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;
+  }
+}
-- 
cgit v1.2.3