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)
def test_isInWorld(self):
body = RigidBody()
self.assertFalse(body.isInWorld())
world = DiscreteDynamicsWorld()
world.addRigidBody(body)
self.assertTrue(body.isInWorld())
world.removeRigidBody(body)
self.assertFalse(body.isInWorld())
def test_cycle(self):
world = DiscreteDynamicsWorld()
class Cheat(RigidBody):
pass
body = Cheat(None, BoxShape(Vector3(3, 4, 5)))
body.cycle = world
world.addRigidBody(body)
del body, world
import gc
gc.collect()
def test_linearVelocity(self):
world = DiscreteDynamicsWorld()
world.setGravity(Vector3(0, 0, 0))
obj = RigidBody(None, None, 1)
obj.setLinearVelocity(Vector3(1, 2, 3))
world.addRigidBody(obj)
expectedSteps = 64
numSteps = world.stepSimulation(1.0, expectedSteps,
1.0 / expectedSteps)
self.assertEqual(numSteps, expectedSteps)
position = obj.getMotionState().getWorldTransform().getOrigin()
self.assertEquals(position.x, 1)
self.assertEquals(position.y, 2)
self.assertEquals(position.z, 3)
def test_stepSimulation(self):
world = DiscreteDynamicsWorld()
world.setGravity(Vector3(1, 2, 3))
obj = RigidBody(None, None, 1)
world.addRigidBody(obj)
expectedSteps = 64
numSteps = world.stepSimulation(1.0, expectedSteps,
1.0 / expectedSteps)
self.assertEqual(numSteps, expectedSteps)
position = obj.getMotionState().getWorldTransform().getOrigin()
# Unfortunately, there is some error (as compared to physical reality)
# in Bullet's results. My fault? Bullet's fault? I'm not sure.
self.assertEqual(position.x, 0.5 + 0.5 / expectedSteps)
self.assertEqual(position.y, 1.0 + 1.0 / expectedSteps)
self.assertEqual(position.z, 1.5 + 1.5 / expectedSteps)
def test_applyImpulse(self):
body = RigidBody(mass=1.0)
body.applyImpulse(Vector3(1, 2, 3), Vector3(1, 1, 1))
world = DiscreteDynamicsWorld()
world.setGravity(Vector3(0, 0, 0))
world.addRigidBody(body)
expectedSteps = 64
numSteps = world.stepSimulation(1.0, expectedSteps,
1.0 / expectedSteps)
self.assertEqual(numSteps, expectedSteps)
transform = body.getMotionState().getWorldTransform()
position = transform.getOrigin()
self.assertEqual(position.x, 1.0)
self.assertEqual(position.y, 2.0)
self.assertEqual(position.z, 3.0)
rot = transform.getRotation()
self.assertNotEqual((rot.getX(), rot.getY(), rot.getZ(), rot.getW()),
(0, 0, 0, 1))
def test_removeAction(self):
world = DiscreteDynamicsWorld()
action = KinematicCharacterController(SphereShape(1), 1.0, 1)
world.addAction(action)
world.removeAction(action)
def test_removeRigidBody(self):
world = DiscreteDynamicsWorld()
body = RigidBody()
world.addRigidBody(body)
world.removeRigidBody(body)
def test_empty(self):
world = DiscreteDynamicsWorld()
self.assertEqual(world.getNumCollisionObjects(), 0)
def setUp(self):
self.world = DiscreteDynamicsWorld()
self.recorder = DebugRecorder()
self.recorder.setDebugMode(DRAW_WIREFRAME | DRAW_CONTACT_POINTS)
self.world.setDebugDrawer(self.recorder)
