lots of cleanup
This commit is contained in:
parent
7d0d3e504e
commit
0385cbd62e
26 changed files with 337 additions and 417 deletions
|
|
@ -1,41 +1,41 @@
|
|||
from advent.common import utils
|
||||
from advent.common import input
|
||||
|
||||
from .solution import Command, day_num, part1, part2
|
||||
from .solution import Command, day_num, part1, part2, simulate
|
||||
|
||||
|
||||
def test_part1():
|
||||
data = utils.read_data(day_num, 'test01.txt')
|
||||
data = input.read_lines(day_num, 'test01.txt')
|
||||
expected = 13
|
||||
result = part1(data)
|
||||
assert result == expected
|
||||
|
||||
|
||||
def test_part2():
|
||||
data = utils.read_data(day_num, 'test02.txt')
|
||||
data = input.read_lines(day_num, 'test02.txt')
|
||||
expected = 36
|
||||
result = part2(data)
|
||||
assert result == expected
|
||||
|
||||
|
||||
def test_short():
|
||||
data = utils.read_data(day_num, 'test01.txt')
|
||||
data = input.read_lines(day_num, 'test01.txt')
|
||||
expected = 13
|
||||
lst = [Command.parse(line) for line in data]
|
||||
result = Command.walk(lst, 2)
|
||||
lst = (Command.parse(line) for line in data)
|
||||
result = simulate(lst, 2)
|
||||
assert result == expected
|
||||
|
||||
|
||||
def test_long1():
|
||||
data = utils.read_data(day_num, 'test01.txt')
|
||||
data = input.read_lines(day_num, 'test01.txt')
|
||||
expected = 1
|
||||
lst = [Command.parse(line) for line in data]
|
||||
result = Command.walk(lst, 10)
|
||||
lst = (Command.parse(line) for line in data)
|
||||
result = simulate(lst, 10)
|
||||
assert result == expected
|
||||
|
||||
|
||||
def test_long2():
|
||||
data = utils.read_data(day_num, 'test02.txt')
|
||||
data = input.read_lines(day_num, 'test02.txt')
|
||||
expected = 36
|
||||
lst = [Command.parse(line) for line in data]
|
||||
result = Command.walk(lst, 10)
|
||||
lst = (Command.parse(line) for line in data)
|
||||
result = simulate(lst, 10)
|
||||
assert result == expected
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue