Files
advent-of-code-2023/tests/scratchcards_test.py
2023-12-18 20:29:08 -08:00

54 lines
1.6 KiB
Python

import pytest
from advent_of_code.scratchcards import Scratchcard, solve_part_1, solve_part_2
mock_inputs = [
"Card 1: 41 48 83 86 17 | 83 86 6 31 17 9 48 53",
"Card 2: 13 32 20 16 61 | 61 30 68 82 17 32 24 19",
"Card 3: 1 21 53 59 44 | 69 82 63 72 16 21 14 1",
"Card 4: 41 92 73 84 69 | 59 84 76 51 58 5 54 83",
"Card 5: 87 83 26 28 32 | 88 30 70 12 93 22 82 36",
"Card 6: 31 18 13 56 72 | 74 77 10 23 35 67 36 11",
]
def test_scratchcard_parse():
assert Scratchcard.parse(mock_inputs[0]) == Scratchcard(
1,
{41, 48, 83, 86, 17},
{83, 86, 6, 31, 17, 9, 48, 53},
)
with pytest.raises(ValueError):
Scratchcard.parse("invalid")
with pytest.raises(ValueError):
Scratchcard.parse("Card 4: invalid | invalid")
assert Scratchcard.parse("Card 5:") == Scratchcard(5, set(), set())
assert Scratchcard.parse("Card 6: | ") == Scratchcard(6, set(), set())
def test_scratchcard_match_count():
assert Scratchcard(1, {1, 2}, {3, 4}).match_count == 0
assert Scratchcard(1, {1, 2}, {2, 3}).match_count == 1
assert Scratchcard(1, set(), set()).match_count == 0
def test_scratchcard_points():
card = Scratchcard(1, {41, 48, 83, 86, 17}, {83, 86, 6, 31, 17, 9, 48, 53})
assert card.points == 8
def test_scratchcard_resolve_copies():
cards = [Scratchcard.parse(mock_input) for mock_input in mock_inputs]
assert Scratchcard.resolve_copies(cards) == [1, 2, 4, 8, 14, 1]
def test_solve_pt_1():
assert solve_part_1("\n".join(mock_inputs)) == 13
def test_solve_pt_2():
assert solve_part_2("\n".join(mock_inputs)) == 30