day17 finished

advent/days/day17/test_solution.py

@@ -0,0 +1,34 @@
+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