def generate_maze(algorithm, height, width, path, displayMaze): if algorithm == "backtracking": backtracking = Backtracking(height, width, path, displayMaze) backtracking.createMaze() elif algorithm == "aldous_broder": aldous_broder = Aldous_Broder(height, width, path, displayMaze) aldous_broder.createMaze() elif algorithm == "hunt_and_kill": hunt_and_kill = Hunt_and_Kill(height, width, path, displayMaze) hunt_and_kill.createMaze() elif algorithm == "prims": prims = Prims(height, width, path, displayMaze) prims.createMaze()
def test_aldous_broder(): aldous_broder = Aldous_Broder(201, 201, "maze.png", False) assert aldous_broder.createMaze() == 0