from pathlib import Path

from day1 import first

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


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


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