def test_closed_boundaries_in_middle(): X = -1 neighbors = sgrid.neighbor_node_array( (5, 4), closed_boundary_nodes=sgrid.boundary_nodes((5, 4)), inactive=X) expected = np.array([ [X, X, X, X], [X, X, X, X], [X, X, X, X], [X, X, X, X], [X, X, X, X], [6, 9, X, X], [X, 10, 5, X], [X, X, X, X], [X, X, X, X], [10, 13, X, 5], [X, 14, 9, 6], [X, X, X, X], [X, X, X, X], [14, X, X, 9], [X, X, 13, 10], [X, X, X, X], [X, X, X, X], [X, X, X, X], [X, X, X, X], [X, X, X, X], ]) assert_array_equal(neighbors, expected.T)
def test_closed_boundaries_in_middle(): X = -1 neighbors = sgrid.neighbor_node_array((5, 4), closed_boundary_nodes=sgrid.boundary_nodes((5, 4)), inactive=X) expected = np.array( [ [X, X, X, X], [X, X, X, X], [X, X, X, X], [X, X, X, X], [X, X, X, X], [6, 9, X, X], [X, 10, 5, X], [X, X, X, X], [X, X, X, X], [10, 13, X, 5], [X, 14, 9, 6], [X, X, X, X], [X, X, X, X], [14, X, X, 9], [X, X, 13, 10], [X, X, X, X], [X, X, X, X], [X, X, X, X], [X, X, X, X], [X, X, X, X], ] ) assert_array_equal(neighbors, expected.T)
def test_open_boundary_on_perimeter(): X = -1 neighbors = sgrid.neighbor_node_array( (5, 4), open_boundary_nodes=sgrid.boundary_nodes((5, 4)), inactive=X) 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(neighbors, expected.T)
def test_open_boundary_on_perimeter(): X = -1 neighbors = sgrid.neighbor_node_array((5, 4), open_boundary_nodes=sgrid.boundary_nodes((5, 4)), inactive=X) 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(neighbors, expected.T)