def test_get_visits_with_robot(): assert get_visits("^v", 2) == 3 assert get_visits("^>v<", 2) == 3 assert get_visits("^v^v^v^v^v", 2) == 11
def test_get_visits(): assert get_visits(">") == 2 assert get_visits("^>v<") == 4 assert get_visits("^v^v^v^v^v") == 2 assert get_visits("^>v<>^") == 4 assert get_visits(">^^v^<>v") == 5