Unit tests for transformations now check all particles and not only t…

…he first. (#806)
hannorein · Nov 23, 2024 · 9fc283a · 9fc283a
1 parent 0b5c85d
commit 9fc283a
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/rebound/tests/test_transformations.py b/rebound/tests/test_transformations.py
@@ -6,14 +6,14 @@
 def getc(sim):
     c = []
     for i in range(sim.N):
-        c.append(sim.particles[0].x)
-        c.append(sim.particles[0].y)
-        c.append(sim.particles[0].z)
-        c.append(sim.particles[0].vx)
-        c.append(sim.particles[0].vy)
-        c.append(sim.particles[0].vz)
-        c.append(sim.particles[0].m)
-        c.append(sim.particles[0].r)
+        c.append(sim.particles[i].x)
+        c.append(sim.particles[i].y)
+        c.append(sim.particles[i].z)
+        c.append(sim.particles[i].vx)
+        c.append(sim.particles[i].vy)
+        c.append(sim.particles[i].vz)
+        c.append(sim.particles[i].m)
+        c.append(sim.particles[i].r)
     return c
 
 class TestTransformations(unittest.TestCase):