advent-of-code/tests/aoc_2024/day1/test_part2.py

14 lines
343 B
Python

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