day06 finished

advent/days/day06/test_solution.py

@@ -0,0 +1,31 @@
+from advent.common import utils
+
+from .solution import day_num, marker, part1, part2
+
+
+def test_part1():
+    data = utils.read_data(day_num, 'test01.txt')
+    expected = 7
+    result = part1(data)
+    assert result == expected
+
+
+def test_part2():
+    data = utils.read_data(day_num, 'test01.txt')
+    expected = 19
+    result = part2(data)
+    assert result == expected
+
+
+def test_marker04():
+    input = "mjqjpqmgbljsphdztnvjfqwrcgsmlb"
+    expected = 7
+    result = marker(input, 4)
+    assert result == expected
+
+
+def test_marker14():
+    input = "mjqjpqmgbljsphdztnvjfqwrcgsmlb"
+    expected = 19
+    result = marker(input, 14)
+    assert result == expected