def test_get_directions():
grid = spiral_grid.SpiralGrid()
assert list(itertools.islice(grid.direction_generator(), 25)) == [(1, 0),
(0, -1),
(-1, 0),
(-1, 0),
(0, 1),
(0, 1),
(1, 0),
(1, 0),
(1, 0),
(0, -1),
(0, -1),
(0, -1),
(-1, 0),
(-1, 0),
(-1, 0),
(-1, 0),
(0, 1),
(0, 1),
(0, 1),
(0, 1),
(1, 0),
(1, 0),
(1, 0),
(1, 0),
(1, 0)]
def test_create_grid():
grid = spiral_grid.SpiralGrid()
grid.create_grid(reach_value=9, increment=True)
assert next(reversed(grid.grid)) == (2, 2)
assert grid.grid[1, 1] == 1
assert grid.grid[0, 0] == 5
assert grid.grid[2, 2] == 9
def test_get_manhattan_distance_examples():
grid = spiral_grid.SpiralGrid()
grid.create_grid(reach_value=1024, increment=True)
assert grid.get_manhattan_distance(1) == 0
assert grid.get_manhattan_distance(12) == 3
assert grid.get_manhattan_distance(23) == 2
assert grid.get_manhattan_distance(1024) == 31
def test_accumulate_examples():
grid = spiral_grid.SpiralGrid()
grid.create_grid(reach_value=54, increment=False)
assert grid.get_val_at_index(0) == 1
assert grid.get_val_at_index(1) == 1
assert grid.get_val_at_index(2) == 2
assert grid.get_val_at_index(4) == 5
assert grid.get_val_at_index(10) == 54
assert grid.get_max_val() == 54
def test_part_2_answer():
grid = spiral_grid.SpiralGrid()
grid.create_grid(reach_value=325489, increment=False)
assert grid.get_max_val() == 330785
def test_part_1_answer():
grid = spiral_grid.SpiralGrid()
grid.create_grid(reach_value=325489, increment=True)
assert grid.get_manhattan_distance(325489) == 552