def transition_function(mouse, path, goals): mouse = Mouse(path[0]) for i in range(len(path)): if i == 0: continue previous = path[i - 1] position = path[i] if position == (previous[0] - 1, previous[1]): mouse.goNorth() if position == (previous[0], previous[1] + 1): mouse.goEast() if position == (previous[0] + 1, previous[1]): mouse.goSouth() if position == (previous[0], previous[1] - 1): mouse.goWest() if position in goals: mouse.get_goal(position) return mouse