def test_new_room_verts_tri_on_final_wall(self): builder = LevelBuilder() room = Listener() room.verts = [(1, 11), (2, 22), (3, 33), (4, 44)] wall = 3 verts = builder.new_room_verts(room, wall, 3) self.assertEquals(len(verts), 3) self.assertEquals(verts[0], (1, 11)) self.assertEquals(verts[1], (4, 44))
def test_new_room_verts(self): builder = LevelBuilder() room = Listener() room.verts = [(1, 11), (2, 22), (3, 33), (4, 44)] wall = 1 for num_verts in xrange(3, 5): verts = builder.new_room_verts(room, wall, num_verts) self.assertEquals(len(verts), num_verts) self.assertEquals(verts[0], (3, 33)) self.assertEquals(verts[1], (2, 22))
def test_select_branch_wall(self): builder = LevelBuilder() room = Listener() room.verts = range(5) room.neighbours = {0: object(), 2:object(), 3:object()} walls = set() for _ in xrange(32): walls.add(builder.select_branch_wall(room)) self.assertTrue(0 not in walls) self.assertTrue(2 not in walls) self.assertTrue(3 not in walls)
def test_select_branch_wall_no_free_walls(self): builder = LevelBuilder() room = Listener() room.verts = range(3) room.neighbours = range(3) self.assertNone(builder.select_branch_wall(room))