day09 finished

advent/days/day09/test_solution.py

@@ -0,0 +1,41 @@
+from advent.common import utils
+
+from .solution import Command, day_num, part1, part2
+
+
+def test_part1():
+    data = utils.read_data(day_num, 'test01.txt')
+    expected = 13
+    result = part1(data)
+    assert result == expected
+
+
+def test_part2():
+    data = utils.read_data(day_num, 'test02.txt')
+    expected = 36
+    result = part2(data)
+    assert result == expected
+
+
+def test_short():
+    data = utils.read_data(day_num, 'test01.txt')
+    expected = 13
+    lst = [Command.parse(line) for line in data]
+    result = Command.walk(lst, 2)
+    assert result == expected
+
+
+def test_long1():
+    data = utils.read_data(day_num, 'test01.txt')
+    expected = 1
+    lst = [Command.parse(line) for line in data]
+    result = Command.walk(lst, 10)
+    assert result == expected
+
+
+def test_long2():
+    data = utils.read_data(day_num, 'test02.txt')
+    expected = 36
+    lst = [Command.parse(line) for line in data]
+    result = Command.walk(lst, 10)
+    assert result == expected