def test_active_neighbor_list_with_scalar_arg():
rmg = RasterModelGrid(5, 4)
assert_array_equal(rmg.active_neighbors_at_node(6), np.array([7, 10, 5,
2]))
assert_array_equal(rmg.active_neighbors_at_node(-1), np.array([X, X, X,
X]))
assert_array_equal(rmg.active_neighbors_at_node(-2),
np.array([X, X, X, 14]))
def test_active_neighbor_list_with_scalar_arg():
rmg = RasterModelGrid(5, 4)
assert_array_equal(rmg.active_neighbors_at_node(6),
np.array([7, 10, 5, 2]))
assert_array_equal(rmg.active_neighbors_at_node(-1),
np.array([X, X, X, X]))
assert_array_equal(rmg.active_neighbors_at_node(-2),
np.array([X, X, X, 14]))
def test_all_active_neighbors():
rmg = RasterModelGrid(5, 4)
expected = np.array([
[X, X, X, X],
[X, 5, X, X],
[X, 6, X, X],
[X, X, X, X],
[5, X, X, X],
[6, 9, 4, 1],
[7, 10, 5, 2],
[X, X, 6, X],
[9, X, X, X],
[10, 13, 8, 5],
[11, 14, 9, 6],
[X, X, 10, X],
[13, X, X, X],
[14, 17, 12, 9],
[15, 18, 13, 10],
[X, X, 14, X],
[X, X, X, X],
[X, X, X, 13],
[X, X, X, 14],
[X, X, X, X],
])
assert_array_equal(rmg.active_neighbors_at_node(), expected)
def test_active_neighbor_list_boundary():
"""All of the neighbor IDs for a boundary cell are -1."""
rmg = RasterModelGrid(5, 4)
import landlab.utils.structured_grid as sgrid
rmg.set_closed_nodes([0, 1, 2, 3, 4, 7, 8, 11, 12, 15, 16, 17, 18, 19])
for node_id in sgrid.boundary_iter(rmg.shape):
assert_array_equal(rmg.active_neighbors_at_node(node_id),
np.array([X, X, X, X]))
def test_active_neighbor_list_boundary():
"""All of the neighbor IDs for a boundary cell are -1."""
rmg = RasterModelGrid(5, 4)
import landlab.utils.structured_grid as sgrid
rmg.set_closed_nodes([0, 1, 2, 3, 4, 7, 8, 11, 12, 15, 16, 17, 18, 19])
for node_id in sgrid.boundary_iter(rmg.shape):
assert_array_equal(rmg.active_neighbors_at_node(node_id),
np.array([X, X, X, X]))
def test_all_active_neighbors():
rmg = RasterModelGrid(5, 4)
expected = np.array([
[X, X, X, X], [X, 5, X, X], [X, 6, X, X], [X, X, X, X],
[5, X, X, X], [6, 9, 4, 1], [7, 10, 5, 2], [X, X, 6, X],
[9, X, X, X], [10, 13, 8, 5], [11, 14, 9, 6], [X, X, 10, X],
[13, X, X, X], [14, 17, 12, 9], [15, 18, 13, 10], [X, X, 14, X],
[X, X, X, X], [X, X, X, 13], [X, X, X, 14], [X, X, X, X],
])
assert_array_equal(rmg.active_neighbors_at_node(), expected)
def test_active_neighbor_list_with_array_arg():
rmg = RasterModelGrid(5, 4)
assert_array_equal(rmg.active_neighbors_at_node([6, -1]),
np.array([[7, 10, 5, 2], [X, X, X, X]]))
def test_active_neighbor_list_with_array_arg():
rmg = RasterModelGrid(5, 4)
assert_array_equal(rmg.active_neighbors_at_node([6, -1]),
np.array([[7, 10, 5, 2], [X, X, X, X]]))
