Skip to content

Commit 8a40d6a

Browse files
committed
Updated unit test (testConverterJoints.py).
1 parent 1deb5b2 commit 8a40d6a

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

tests/testConverterJoints.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,7 @@ def test_fixed_revolute_joints(self):
3939

4040
physics_fixed_joint_prim = stage.GetPrimAtPath(f"{physics_scope_prim.GetPath()}/joint_root")
4141
self.assertIsNotNone(physics_fixed_joint_prim)
42-
self.assertEqual(physics_fixed_joint_prim.GetTypeName(), "PhysicsFixedJoint")
42+
self.assertTrue(physics_fixed_joint_prim.IsA(UsdPhysics.FixedJoint))
4343
fixed_joint = UsdPhysics.FixedJoint(physics_fixed_joint_prim)
4444
self.assertEqual(fixed_joint.GetBody0Rel().GetTargets(), ["/simple_fixed_revolute_joints/Geometry/BaseLink"])
4545
self.assertEqual(fixed_joint.GetBody1Rel().GetTargets(), ["/simple_fixed_revolute_joints/Geometry/BaseLink/tn__Arm1_j8"])
@@ -50,7 +50,7 @@ def test_fixed_revolute_joints(self):
5050

5151
physics_revolute_joint_prim = stage.GetPrimAtPath(f"{physics_scope_prim.GetPath()}/tn__joint_arm1_wJ")
5252
self.assertIsNotNone(physics_revolute_joint_prim)
53-
self.assertEqual(physics_revolute_joint_prim.GetTypeName(), "PhysicsRevoluteJoint")
53+
self.assertTrue(physics_revolute_joint_prim.IsA(UsdPhysics.RevoluteJoint))
5454
revolute_joint = UsdPhysics.RevoluteJoint(physics_revolute_joint_prim)
5555
self.assertEqual(revolute_joint.GetBody0Rel().GetTargets(), ["/simple_fixed_revolute_joints/Geometry/BaseLink/tn__Arm1_j8"])
5656
self.assertEqual(revolute_joint.GetBody1Rel().GetTargets(), ["/simple_fixed_revolute_joints/Geometry/BaseLink/tn__Arm1_j8/tn__Arm2_j8"])
@@ -86,7 +86,7 @@ def test_fixed_continuous_joints(self):
8686

8787
physics_fixed_joint_prim = stage.GetPrimAtPath(f"{physics_scope_prim.GetPath()}/joint_root")
8888
self.assertIsNotNone(physics_fixed_joint_prim)
89-
self.assertEqual(physics_fixed_joint_prim.GetTypeName(), "PhysicsFixedJoint")
89+
self.assertTrue(physics_fixed_joint_prim.IsA(UsdPhysics.FixedJoint))
9090
fixed_joint = UsdPhysics.FixedJoint(physics_fixed_joint_prim)
9191
self.assertEqual(fixed_joint.GetBody0Rel().GetTargets(), ["/simple_fixed_continuous_joints/Geometry/BaseLink"])
9292
self.assertEqual(fixed_joint.GetBody1Rel().GetTargets(), ["/simple_fixed_continuous_joints/Geometry/BaseLink/tn__Arm1_j8"])
@@ -97,7 +97,7 @@ def test_fixed_continuous_joints(self):
9797

9898
physics_revolute_joint_prim = stage.GetPrimAtPath(f"{physics_scope_prim.GetPath()}/tn__joint_arm1_wJ")
9999
self.assertIsNotNone(physics_revolute_joint_prim)
100-
self.assertEqual(physics_revolute_joint_prim.GetTypeName(), "PhysicsRevoluteJoint")
100+
self.assertTrue(physics_revolute_joint_prim.IsA(UsdPhysics.RevoluteJoint))
101101
revolute_joint = UsdPhysics.RevoluteJoint(physics_revolute_joint_prim)
102102
self.assertEqual(revolute_joint.GetBody0Rel().GetTargets(), ["/simple_fixed_continuous_joints/Geometry/BaseLink/tn__Arm1_j8"])
103103
self.assertEqual(revolute_joint.GetBody1Rel().GetTargets(), ["/simple_fixed_continuous_joints/Geometry/BaseLink/tn__Arm1_j8/tn__Arm2_j8"])
@@ -133,7 +133,7 @@ def test_fixed_prismatic_joints(self):
133133

134134
physics_fixed_joint_prim = stage.GetPrimAtPath(f"{physics_scope_prim.GetPath()}/joint_root")
135135
self.assertIsNotNone(physics_fixed_joint_prim)
136-
self.assertEqual(physics_fixed_joint_prim.GetTypeName(), "PhysicsFixedJoint")
136+
self.assertTrue(physics_fixed_joint_prim.IsA(UsdPhysics.FixedJoint))
137137
fixed_joint = UsdPhysics.FixedJoint(physics_fixed_joint_prim)
138138
self.assertEqual(fixed_joint.GetBody0Rel().GetTargets(), ["/simple_fixed_prismatic_joints/Geometry/BaseLink"])
139139
self.assertEqual(fixed_joint.GetBody1Rel().GetTargets(), ["/simple_fixed_prismatic_joints/Geometry/BaseLink/tn__Arm1_j8"])
@@ -144,7 +144,7 @@ def test_fixed_prismatic_joints(self):
144144

145145
physics_prismatic_joint_prim = stage.GetPrimAtPath(f"{physics_scope_prim.GetPath()}/tn__joint_arm1_wJ")
146146
self.assertIsNotNone(physics_prismatic_joint_prim)
147-
self.assertEqual(physics_prismatic_joint_prim.GetTypeName(), "PhysicsPrismaticJoint")
147+
self.assertTrue(physics_prismatic_joint_prim.IsA(UsdPhysics.PrismaticJoint))
148148
prismatic_joint = UsdPhysics.PrismaticJoint(physics_prismatic_joint_prim)
149149
self.assertEqual(prismatic_joint.GetBody0Rel().GetTargets(), ["/simple_fixed_prismatic_joints/Geometry/BaseLink/tn__Arm1_j8"])
150150
self.assertEqual(prismatic_joint.GetBody1Rel().GetTargets(), ["/simple_fixed_prismatic_joints/Geometry/BaseLink/tn__Arm1_j8/tn__Arm2_j8"])

0 commit comments

Comments
 (0)