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_room_works_on_branch_room_and_wall_zero(self): builder = LevelBuilder() branch_room = Room([(1, 0), (0, 0), (0, 1)]) branch_room.id = 0 branch_wall = 0 new_room = Room([(1, 0), (0, 0), (0, 1)]) new_room.neighbours = {} builder.add_room(new_room, branch_room, 0) self.assertEquals(len(builder.rooms), 1) added_room = builder.rooms.values()[0] self.assertEquals(added_room.neighbours, {0: branch_room}) self.assertEquals(branch_room.neighbours, {branch_wall: added_room})