import pathlib
import pytest
import day01 as aoc

INPUTS_DIR = f"{pathlib.Path(__file__).parent.parent}/inputs"

@pytest.fixture
def example_data():
    input_path = f"{INPUTS_DIR}/day01-example.txt"
    return aoc.parse(pathlib.Path(input_path).read_text().strip())


@pytest.fixture
def day01_data():
    input_path = f"{INPUTS_DIR}/day01.txt"
    return aoc.parse(pathlib.Path(input_path).read_text().strip())


def test_example1(example_data):
    assert aoc.part1(example_data) == 7


def test_example2(example_data):
    assert aoc.part2(example_data) == 5


def test_part1(day01_data):
    assert aoc.part1(day01_data) == 1266


def test_part2(day01_data):
    assert aoc.part2(day01_data) == 1217