def test_add_to_body(self): color = (100, 150, 200) v1, v2, v3 = (0.0, 0.0), (0.0, 1.0), (1.0, 0.0) verts = [v1, v2, v3] room = Room(verts) space = Space() body = Body(inf, inf) room.add_to_body(space, body) segs = set([ ((seg.a[0], seg.a[1]), (seg.b[0], seg.b[1])) for seg in space.static_shapes ]) self.assertEquals(segs, set([(v1, v2), (v2, v3), (v3, v1), (v1, v2)]), "room walls not added to space") for seg in space.static_shapes: self.assertEquals(seg.friction, 0.5, "bad wall friction") self.assertEquals(seg.elasticity, 0.5, "bad wall elasticity")
def test_add_to_pymunk(self): color = (50, 100, 200) v1, v2, v3 = (0, 0), (0, 100), (100, 0) verts = [v1, v2, v3] room1 = Room(verts) room1.id = 999 room1.add_to_body = Listener() room2 = Room(verts) room2.id = 1000 room2.add_to_body = Listener() world = World() world.rooms = {999: room1, 1000: room2} world.add_to_pymunk() expected = (world.space, world.static_body) self.assertEquals(room1.add_to_body.args, expected, "room1 walls not added to space") self.assertEquals(room2.add_to_body.args, expected, "room2 walls not added to space")
def test_add_to_body_should_not_add_segments_for_neighbours(self): color = (100, 150, 200) v1, v2, v3 = (0,1), (1,2), (2,0) verts = [v1, v2, v3] room = Room(verts) room.neighbours = {1: object(), 2: object()} space = Space() body = Body(inf, inf) room.add_to_body(space, body) segs = set([ ((seg.a[0], seg.a[1]), (seg.b[0], seg.b[1])) for seg in space.static_shapes ]) self.assertEquals(segs, set([(v1, v2)]), "wrong room walls added to space") for seg in space.static_shapes: self.assertEquals(seg.friction, 0.5, "bad wall friction") self.assertEquals(seg.elasticity, 0.5, "bad wall elasticity")