54 lines
1.6 KiB
Python
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
|