def test_neighbors_edge_right_bot():
empty_matrix = numpy.zeros([3, 3])
valued_matrix = calculate_neighbors_given_a_bomb(empty_matrix, 2, 2)
assert valued_matrix.tolist() == [[0, 0, 0],
[0, 1, 1],
[0, 1, 0]]
def test_neighbors_edge_mid_mid():
empty_matrix = numpy.zeros([3, 3])
valued_matrix = calculate_neighbors_given_a_bomb(empty_matrix, 1, 1)
assert valued_matrix.tolist() == [[1, 1, 1],
[1, 0, 1],
[1, 1, 1]]
def test_bombs_dont_get_changed():
empty_matrix = numpy.zeros([3, 3])
empty_matrix[0, 0] = BOMB
valued_matrix = calculate_neighbors_given_a_bomb(empty_matrix, 1, 0)
assert valued_matrix.tolist() == [[BOMB, 1, 0],
[0, 1, 0],
[1, 1, 0]]