Simplify test, per PR feedback.

einsteinpy · Nov 14, 2022 · 31ddeb0 · 31ddeb0
1 parent 6545d89
commit 31ddeb0
Showing 1 changed file with 16 additions and 23 deletions.
diff --git a/tests/test_geodesic/test_geodesic.py b/tests/test_geodesic/test_geodesic.py
@@ -172,32 +172,25 @@ def test_kerr0_eq_kn00():
     assert_allclose(k.trajectory[1], kn.trajectory[1], atol=1e-6, rtol=1e-6)
 
 
-def test_kerr0_eq_custom():
+def test_custom_callable_metric():
     metric_params = (0.,)
     q0 = [4., np.pi / 2, 0.]
     p0 = [0., 0., 0.]
 
-    k = Timelike(
-        metric="Kerr",
-        metric_params=metric_params,
-        position=q0,
-        momentum=p0,
-        steps=50,
-        delta=0.5,
-        return_cartesian=True,
-        suppress_warnings=True,
-    )
+    try:
+        c = Timelike(
+            metric=_kerr,
+            metric_params=metric_params,
+            position=q0,
+            momentum=p0,
+            steps=50,
+            delta=0.5,
+            return_cartesian=True,
+            suppress_warnings=True,
+        )
+        assert c.metric_name == _kerr.__name__
+        assert c.metric == _kerr
 
-    c = Timelike(
-        metric=_kerr,
-        metric_params=metric_params,
-        position=q0,
-        momentum=p0,
-        steps=50,
-        delta=0.5,
-        return_cartesian=True,
-        suppress_warnings=True,
-    )
+    except:
+        assert False
 
-    assert_allclose(k.trajectory[0], c.trajectory[0], atol=1e-6, rtol=1e-6)
-    assert_allclose(k.trajectory[1], c.trajectory[1], atol=1e-6, rtol=1e-6)