def test_neighbours_3d(): neighbour2d = lambda (x, y, z): [ (x - 1, y - 1, z - 1), (x - 1, y - 1, z), (x - 1, y - 1, z + 1), (x - 1, y, z - 1), (x - 1, y, z), (x - 1, y, z + 1), (x - 1, y + 1, z - 1), (x - 1, y + 1, z), (x - 1, y + 1, z + 1), (x, y - 1, z - 1), (x, y - 1, z), (x, y - 1, z + 1), (x, y, z - 1), (x, y, z + 1), (x, y + 1, z - 1), (x, y + 1, z), (x, y + 1, z + 1), (x + 1, y - 1, z - 1), (x + 1, y - 1, z), (x + 1, y - 1, z + 1), (x + 1, y, z - 1), (x + 1, y, z), (x + 1, y, z + 1), (x + 1, y + 1, z - 1), (x + 1, y + 1, z), (x + 1, y + 1, z + 1), ] for coordinates2d in [(5, 5, 5), (10, 10, 0), (0, 0, 0), (-1, 5, 0), (-1, 43, -5)]: np.testing.assert_array_equal(neighbours(coordinates2d), neighbour2d(coordinates2d))
def test_neighbours_1d(): neighbour1d = lambda x: [[x[0] - 1], [x[0] + 1]] for coordinates1d in [[5], [10], [0], [-1]]: np.testing.assert_array_equal(neighbours(coordinates1d), neighbour1d(coordinates1d))