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_backtracking():
backtracking = Backtracking(201, 201, "maze.png", False)
assert backtracking.createMaze() == 0
