def test_gravity(self): world = DiscreteDynamicsWorld() world.setGravity(Vector3(3, 2, 1)) gravity = world.getGravity() self.assertEqual(gravity.x, 3) self.assertEqual(gravity.y, 2) self.assertEqual(gravity.z, 1)