45 lines
956 B
Python
45 lines
956 B
Python
from advent.common import utils
|
|
|
|
from .solution import day_num, part1, part2, Shape, Result
|
|
|
|
|
|
def test_part1():
|
|
data = utils.read_data(day_num, 'test01.txt')
|
|
expected = 15
|
|
result = part1(data)
|
|
assert result == expected
|
|
|
|
|
|
def test_part2():
|
|
data = utils.read_data(day_num, 'test01.txt')
|
|
expected = 12
|
|
result = part2(data)
|
|
assert result == expected
|
|
|
|
|
|
def test_parse_line():
|
|
input = "A Y"
|
|
expected = Shape.Rock, Shape.Paper
|
|
result = Shape.parse(input)
|
|
assert result == expected
|
|
|
|
|
|
def test_round1():
|
|
input = "A Y"
|
|
expected = 8
|
|
opponent, player = Shape.parse(input)
|
|
assert player.score(opponent) == expected
|
|
|
|
|
|
def test_round2():
|
|
input = "A Y"
|
|
expected = 4
|
|
opponent, player = Result.parse(input)
|
|
assert player.score(opponent) == expected
|
|
|
|
|
|
def test_round3():
|
|
input = "B X"
|
|
expected = 1
|
|
opponent, player = Result.parse(input)
|
|
assert player.score(opponent) == expected
|