def test_set_goal(bomb):
"""Test setting the goal for the maze."""
goal = (0, 1)
maze = Maze(bomb)
maze.set_goal(goal)
actual = maze.goal
expected = goal
assert actual == expected
def test_find_path(bomb, test_mazes):
"""Test finding paths through each maze."""
maze = Maze(bomb)
for test_maze in test_mazes:
maze.set_start(test_maze['start'])
maze.set_goal(test_maze['goal'])
maze.set_greens(test_maze['greens'])
actual = maze.find_path()
expected = test_maze['path']
assert actual == expected