Added Horizons test (+10 sec runtime)

alphaparrot · Nov 22, 2015 · fec16c5 · fec16c5
1 parent 2f08b45
commit fec16c5
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/rebound/tests/test_horizons.py b/rebound/tests/test_horizons.py
@@ -0,0 +1,21 @@
+import rebound
+import unittest
+import datetime
+
+class TestHorizons(unittest.TestCase):
+    def setUp(self):
+        self.sim = rebound.Simulation()
+
+    def tearDown(self):
+        self.sim = None
+
+    def test_earth(self):
+        self.sim.add("Earth",date="2000-01-01 00:00")
+        self.assertAlmostEqual(self.sim.particles[0].x,-0.16855044899082616,delta=1e-10)
+        self.assertAlmostEqual(self.sim.particles[0].m,3.0404326480226416e-06,delta=1e-15)
+
+    def test_notfound(self):
+        with self.assertRaises(Exception):
+            self.sim.add("BogusPlanet",date="2000-01-01 00:00")
+
+