34 lines
823 B
Python
34 lines
823 B
Python
from advent.common import input
|
|
|
|
from .solution import Cave, day_num, part1, part2
|
|
|
|
|
|
def test_part1():
|
|
lines = input.read_lines(day_num, 'example01.txt')
|
|
expected = 3068
|
|
result = part1(lines)
|
|
assert result == expected
|
|
|
|
|
|
def test_part2():
|
|
lines = input.read_lines(day_num, 'example01.txt')
|
|
expected = 1514285714288
|
|
result = part2(lines)
|
|
assert result == expected
|
|
|
|
|
|
def test_first():
|
|
input = ">>><<><>><<<>><>>><<<>>><<<><<<>><>><<>>"
|
|
cave = Cave.create(7, input)
|
|
cave.process_one_rock()
|
|
expected = " #### "
|
|
assert cave.cave[0] == expected
|
|
|
|
|
|
def test_third():
|
|
input = ">>><<><>><<<>><>>><<<>>><<<><<<>><>><<>>"
|
|
cave = Cave.create(7, input)
|
|
height = cave.process_many_rocks(3)
|
|
expected = "#### "
|
|
assert cave.cave[3] == expected
|
|
assert height == 6
|