use assertApprox from TestCase

stack-of-tasks · May 30, 2024 · 7ee9e8f · 7ee9e8f
1 parent 04a18d9
commit 7ee9e8f
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/unittest/python/bindings_contact_inverse_dynamics.py b/unittest/python/bindings_contact_inverse_dynamics.py
@@ -3,10 +3,11 @@
 import numpy as np
 
 import pinocchio as pin
+from test_case import PinocchioTestCase as TestCase
 
 
 @unittest.skipUnless(pin.WITH_URDFDOM, "Needs URDFDOM")
-class TestContactInverseDynamics(unittest.TestCase):
+class TestContactInverseDynamics(TestCase):
     def setUp(self):
         self.current_file = os.path.dirname(str(os.path.abspath(__file__)))
         self.model_dir = os.path.abspath(
@@ -132,8 +133,8 @@ def test_call_to_contact_inverse_dynamics(self):
             prox_settings,
             lambda_guess,
         )
-        assert np.allclose(tau1, tau2)
-        assert np.allclose(tau1, tau3)
+        self.assertApprox(tau1, tau2)
+        self.assertApprox(tau1, tau3)
 
 
 if __name__ == "__main__":