from pathlib import Path

from day1 import part1

TEST_FILE = Path(__file__).parent / "test-data"


def test_first():
    assert part1.compute(TEST_FILE) == 11


def test_get_lists():
    assert part1.get_ordered_lists(TEST_FILE)[0] == [1, 2, 3, 3, 3, 4]
    assert part1.get_ordered_lists(TEST_FILE)[1] == [3, 3, 3, 4, 5, 9]