예제 #1
0
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