Add test for lower dimensional vr()

jobovy · Jul 22, 2020 · e6be6a0 · e6be6a0
1 parent a3c1c99
commit e6be6a0
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/test_orbit.py b/tests/test_orbit.py
@@ -1958,6 +1958,7 @@ def test_orbit_setup_linear():
     assert o.dim() == 1, 'linearOrbit does not have dim == 1'
     assert numpy.fabs(o.x()-1.) < 10.**-16., 'linearOrbit x setup does not agree with o.x()'
     assert numpy.fabs(o.vx()-0.1) < 10.**-16., 'linearOrbit vx setup does not agree with o.vx()'
+    assert numpy.fabs(o.vr()-0.1) < 10.**-16., 'linearOrbit vx setup does not agree with o.vr()'
     if False:
         # setphi was deprecated when moving to Orbits
         try: