from aoc2021 import day01


def test_day01_part1():
    data_path = "data/day01-sample.txt"
    assert day01.part1(data_path) == 7


def test_day01_part2():
    data_path = "data/day01-sample.txt"
    assert day01.part2(data_path) == 5
