consolidate test_acceleration (#82)

Signed-off-by: nstarman <nstarman@users.noreply.github.com> # Conflicts: # src/galax/potential/_potential/composite.py # tests/unit/potential/builtin/test_mwpotential.py
GalacticDynamics · Jan 24, 2024 · 144c9b9 · 144c9b9
1 parent 0ed3206
commit 144c9b9
Show file tree

Hide file tree

Showing 7 changed files with 4 additions and 18 deletions.
diff --git a/tests/unit/potential/builtin/test_bar.py b/tests/unit/potential/builtin/test_bar.py
@@ -67,6 +67,3 @@ def test_hessian(self, pot, x):
                 ]
             ),
         )
-
-    def test_acceleration(self, pot, x):
-        assert xp.allclose(pot.acceleration(x, t=0), -pot.gradient(x, t=0))
diff --git a/tests/unit/potential/builtin/test_hernquist.py b/tests/unit/potential/builtin/test_hernquist.py
@@ -47,6 +47,3 @@ def test_hessian(self, pot, x):
                 ]
             ),
         )
-
-    def test_acceleration(self, pot, x):
-        assert xp.allclose(pot.acceleration(x, t=0), -pot.gradient(x, t=0))
diff --git a/tests/unit/potential/builtin/test_isochrone.py b/tests/unit/potential/builtin/test_isochrone.py
@@ -47,6 +47,3 @@ def test_hessian(self, pot, x):
                 ]
             ),
         )
-
-    def test_acceleration(self, pot, x):
-        assert xp.allclose(pot.acceleration(x, t=0), -pot.gradient(x, t=0))
diff --git a/tests/unit/potential/builtin/test_miyamotonagai.py b/tests/unit/potential/builtin/test_miyamotonagai.py
@@ -50,6 +50,3 @@ def test_hessian(self, pot, x):
                 ]
             ),
         )
-
-    def test_acceleration(self, pot, x):
-        assert xp.allclose(pot.acceleration(x, t=0), -pot.gradient(x, t=0))
diff --git a/tests/unit/potential/builtin/test_nfw.py b/tests/unit/potential/builtin/test_nfw.py
@@ -95,6 +95,3 @@ def test_hessian(self, pot, x):
                 ]
             ),
         )
-
-    def test_acceleration(self, pot, x):
-        assert xp.allclose(pot.acceleration(x, t=0), -pot.gradient(x, t=0))
diff --git a/tests/unit/potential/builtin/test_null.py b/tests/unit/potential/builtin/test_null.py
@@ -33,6 +33,3 @@ def test_hessian(self, pot, x):
             pot.hessian(x, t=0),
             xp.array([[0.0, 0.0, 0.0], [0.0, 0.0, 0.0], [0.0, 0.0, 0.0]]),
         )
-
-    def test_acceleration(self, pot, x):
-        assert xp.allclose(pot.acceleration(x, t=0), -pot.gradient(x, t=0))
diff --git a/tests/unit/potential/test_core.py b/tests/unit/potential/test_core.py
@@ -63,3 +63,7 @@ def _potential_energy(self, q, t):
 
         pot = TestPotential()
         assert isinstance(pot, gp.AbstractPotentialBase)
+
+    # TODO: move up the hierarchy?
+    def test_acceleration(self, pot, x):
+        assert xp.allclose(pot.acceleration(x, t=0), -pot.gradient(x, t=0))