Fix return end time twice in orbit integration (#61)

* fix error returning the end time twice Signed-off-by: nstarman <nstarman@users.noreply.github.com>
GalacticDynamics · Jan 18, 2024 · 0536100 · 0536100
1 parent 604f878
commit 0536100
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/src/galax/integrate/_builtin.py b/src/galax/integrate/_builtin.py
@@ -61,7 +61,7 @@ def run(
             y0=qp0,
             dt0=None,
             args=(),
-            saveat=DiffraxSaveAt(t0=False, t1=True, ts=ts, dense=False),
+            saveat=DiffraxSaveAt(t0=False, t1=False, ts=ts, dense=False),
             stepsize_controller=self.stepsize_controller,
             **self.diffeq_kw,
         )

diff --git a/tests/unit/potential/test_base.py b/tests/unit/potential/test_base.py
@@ -130,4 +130,5 @@ def test_integrate_orbit(self, pot, xv):
 
         orbit = pot.integrate_orbit(xv, t0=min(ts), t1=max(ts), ts=ts)
         assert isinstance(orbit, gd.Orbit)
-        assert orbit.shape == (101, 7)
+        assert orbit.shape == (len(ts), 7)
+        assert xp.array_equal(orbit.t, ts)