From bbab5baf50c313bed60edb03398bc21e6d6f2002 Mon Sep 17 00:00:00 2001
From: Cristian C Lalescu <Cristian.Lalescu@mpcdf.mpg.de>
Date: Wed, 16 Dec 2020 09:27:03 +0100
Subject: [PATCH] tweaks Heun test

---
 TurTLE/test/test_Heun_p2p.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/TurTLE/test/test_Heun_p2p.py b/TurTLE/test/test_Heun_p2p.py
index 8c381c1a..ab8bbc3b 100644
--- a/TurTLE/test/test_Heun_p2p.py
+++ b/TurTLE/test/test_Heun_p2p.py
@@ -328,7 +328,7 @@ def main():
     df = bla.get_particle_file()
     x = []
     v = []
-    for ii in range(200):
+    for ii in range(bla.parameters['niter_out']):
         x.append(df['tracers0/position/{0}'.format(ii)][()])
         v.append(df['tracers0/velocity/{0}'.format(ii)][()])
     df.close()
@@ -354,6 +354,7 @@ def main():
         f.savefig('traj.pdf')
     except ImportError:
         print('couldn\'t import matplotlib, no figure')
+    print(np.max(relative_error))
     assert(np.max(relative_error) < 1e-4)
     return None
 
-- 
GitLab