if search > 2 or search < 0:
print("Wrong selection")
sys.exit()
# run DFS
if search == 0:
puzzle.DFS()
puzzle.generateSolution("puzzleDFS")
# Pick heuristic
print("Pick heuristic A or B")
heuristic_choice = input()
# run best first search
if search == 1:
if heuristic_choice == 'A':
puzzle.BFS(puzzle.heuristicA)
puzzle.generateSolution("puzzleBFS-h1")
else:
puzzle.BFS(puzzle.heuristicB)
puzzle.generateSolution("puzzleBFS-h2")
# run a* algorithm
if search == 2:
if heuristic_choice == 'A':
puzzle.Astar(puzzle.heuristicA)
puzzle.generateSolution("puzzleAs-h1")
else:
puzzle.Astar(puzzle.heuristicB)
puzzle.generateSolution("puzzleAs-h2")
