from card import Card mock_input = [ "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_parse(): assert Card.parse(mock_input[0]) == Card( {41, 48, 83, 86, 17}, {83, 86, 6, 31, 17, 9, 48, 53}, ) def test_match_count(): assert [Card.parse(desc).match_count for desc in mock_input] == [4, 2, 2, 1, 0, 0] def test_points(): assert [Card.parse(desc).points for desc in mock_input] == [8, 2, 2, 1, 0, 0]