day01.py

import more_itertools


def get_data(filename: str) -> list:
    with open(filename, 'r') as f:
        return [int(line.rstrip('\n')) for line in f]


def make_couples(singles: list) -> list:
    return zip(singles, singles[1:])


def count_increases(couples: list) -> int:
    return sum(next > prev for prev, next in couples)


if __name__ == "__main__":
    data = get_data("day01.txt")

    depths_couples = make_couples(data)
    num_increases = count_increases(depths_couples)
    print(f'Day 1 Star 1 answer: {num_increases}')

    triples = [sum(triple) for triple in more_itertools.triplewise(data)]

    windowed_couples = make_couples(triples)
    num_increases = count_increases(windowed_couples)
    print(f'Day 1 Star 2 answer: {num_increases}')