Exemple #1
0
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)
Exemple #3
0
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)