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))