def test_get_edge_neighbors_with_history():
"""test the get neighbor function"""
volume = Volume(data=np.random.random((3, 3, 3)))
sponsor_history = [(0, 0, 1)]
neighbors = volume._get_neighbors((0, 0, 2), sponsor_history)
actual_neighbors = [(0, 1, 2), (1, 0, 2)]
npt.assert_array_equal(neighbors.sort(), actual_neighbors.sort())
def test_get_center_neighbors():
"""test the get neighbor function"""
volume = Volume(data=np.random.random((3, 3, 3)))
neighbors = volume._get_neighbors((1, 1, 1))
actual_neighbors = [(0, 1, 1), (2, 1, 1), (1, 2, 1), (1, 0, 1), (1, 1, 0), (1, 1, 2)]
npt.assert_array_equal(neighbors.sort(), actual_neighbors.sort())
