day05 minor cleanup

This commit is contained in:
Ruediger Ludwig 2022-12-05 17:38:36 +01:00
parent 4814a87518
commit 743c1034b4
2 changed files with 14 additions and 20 deletions

View file

@ -34,7 +34,7 @@ def test_parse_line2():
def test_drawing():
data = utils.read_data(day_num, 'test01.txt')
expected = ["ZN", "MCD", "P"]
result = Crane.parse_drawing(data)
result = Crane.parse_stacks(data)
assert result == expected
@ -54,25 +54,17 @@ def test_parse_all():
assert result == expected
def test_step():
data = utils.read_data(day_num, 'test01.txt')
state = Crane.parse(data, True)
expected = ["ZND", "MC", "P"]
result = state.moves[0].do_move(state.stacks, False)
assert result == expected
def test_all_moves():
data = utils.read_data(day_num, 'test01.txt')
state = Crane.parse(data, False)
crane = Crane.parse(data, False)
expected = ["C", "M", "PDNZ"]
result = state.perform_all_moves()
result = crane.perform_all_moves()
assert result == expected
def test_all_moves9001():
data = utils.read_data(day_num, 'test01.txt')
state = Crane.parse(data, True)
crane = Crane.parse(data, True)
expected = ["M", "C", "PZND"]
result = state.perform_all_moves()
result = crane.perform_all_moves()
assert result == expected