advent-of-code/2020/template.py

def main(filename: str, expected_part_1: int = None, expected_part_2: int = None):
    print(f"\n+ Running on {filename}")
    with open(filename) as f:
        data = f.read().strip().split("\n")

    counter_part_1 = solve_part_1(data)

    print(f"1. Found {counter_part_1}")
    if expected_part_1:
        assert expected_part_1 == counter_part_1

    counter_part_2 = solve_part_2(data)
    print(f"2. Found {counter_part_2}")
    if expected_part_2:
        assert expected_part_2 == counter_part_2


def solve_part_1(data):
    return 0


def solve_part_2(data):
    return 0


if __name__ == "__main__":
    main("inputs/dayXX-test1")
    main("inputs/dayXX")