From 8b9cc5994bc05a47e335b1189a6b0b5cfbac7373 Mon Sep 17 00:00:00 2001
From: JayM0826 <maji91fan@gmail.com>
Date: Tue, 13 Feb 2024 11:04:47 +0100
Subject: [PATCH] implement leapfrog and stormer_verlet
---
jobs/tests/test_integrator.py | 3 +++
1 file changed, 3 insertions(+)
diff --git a/jobs/tests/test_integrator.py b/jobs/tests/test_integrator.py
index 9af0ca7..43f9265 100644
--- a/jobs/tests/test_integrator.py
+++ b/jobs/tests/test_integrator.py
@@ -297,5 +297,8 @@ class IntegratorTest(unittest.TestCase):
t=np.linspace(0, T, int(T // dt)),
force=force,
solver=stormer_verlet)
+ t, p_reverse, q_reverse = system.direct_simulation()
+ self.assertTrue(np.greater(1e-10 + np.zeros(4), q_reverse[-1] - q[0]).all())
+ self.assertTrue(np.greater(1e-10 + np.zeros(4), p_reverse[-1] - p[0]).all())
if __name__ == '__main__':
unittest.main()
--
GitLab