day05 finished

advent/days/day05/test_solution.py

@@ -0,0 +1,76 @@
+from advent.common import utils
+
+from .solution import day_num, part1, part2, State
+
+
+def test_part1():
+    data = utils.read_data(day_num, 'test01.txt')
+    expected = "CMZ"
+    result = part1(data)
+    assert result == expected
+
+
+def test_part2():
+    data = utils.read_data(day_num, 'test01.txt')
+    expected = "MCD"
+    result = part2(data)
+    assert result == expected
+
+
+def test_parse_line():
+    input = "    [D]    "
+    expected = [None, 'D', None]
+    result = State.parse_crate_line(input)
+    assert result == expected
+
+
+def test_parse_line2():
+    input = "[Z] [M] [P]"
+    expected = ["Z", 'M', "P"]
+    result = State.parse_crate_line(input)
+    assert result == expected
+
+
+def test_drawing():
+    data = utils.read_data(day_num, 'test01.txt')
+    expected = ["ZN", "MCD", "P"]
+    result = State.parse_drawing(data)
+    assert result == expected
+
+
+def test_parse_move():
+    input = "move 1 from 2 to 1"
+    expected = 1, 1, 0
+    result = State.parse_move(input)
+    assert result == expected
+
+
+def test_parse_all():
+    data = utils.read_data(day_num, 'test01.txt')
+    expected = State(["ZN", "MCD", "P"], [(1, 1, 0), (3, 0, 2), (2, 1, 0), (1, 0, 1)])
+    result = State.parse(data)
+    assert result == expected
+
+
+def test_step():
+    data = utils.read_data(day_num, 'test01.txt')
+    state = State.parse(data)
+    expected = ["ZND", "MC", "P"]
+    result = State.do_move9000(state.crates, state.moves[0])
+    assert result == expected
+
+
+def test_all_moves():
+    data = utils.read_data(day_num, 'test01.txt')
+    state = State.parse(data)
+    expected = ["C", "M", "PDNZ"]
+    result = state.all_moves9000()
+    assert result == expected
+
+
+def test_all_moves9001():
+    data = utils.read_data(day_num, 'test01.txt')
+    state = State.parse(data)
+    expected = ["M", "C", "PZND"]
+    result = state.all_moves9001()
+    assert result == expected