from pathlib import Path

from aoc_2024.day1 import part2


def test_compute_similarity_score():
    dict1 = {1: 1, 2: 1, 3: 3, 4: 1}
    dict2 = {3: 3, 4: 1, 5: 1, 9: 1}
    assert part2.compute_similarity_score(dict1, dict2) == 31


def test_main():
    test_file = Path(__file__).parent / "test-data"
    assert part2.main(test_file) == 31