Python horizontal_active_link_ids Examples

Example #1

File: test_structured_grid.py Project: cmshobe/landlab

def test_active_east_links():
    links = sgrid.active_east_links((3, 4))
    assert_array_equal(
        links, np.array([[-1, -1, -1, -1], [4, 5, 6, -1], [-1, -1, -1, -1]])
    )

    links = sgrid.active_east_links((1, 4))
    assert_array_equal(links, np.array([[-1, -1, -1, -1]]))

    links = sgrid.active_east_links((4, 1))
    assert_array_equal(links, np.array([[-1, -1, -1, -1]]).T)

    links = sgrid.horizontal_active_link_ids((4, 5))
    assert_array_equal(np.array([[9, 10, 11, 12], [13, 14, 15, 16]]), links)

    links = sgrid.active_east_links((4, 5))
    assert_array_equal(
        np.array(
            [
                [-1, -1, -1, -1, -1],
                [9, 10, 11, 12, -1],
                [13, 14, 15, 16, -1],
                [-1, -1, -1, -1, -1],
            ]
        ),
        links,
    )

Example #2

File: test_structured_grid.py Project: pfeiffea/landlab

def test_active_east_links():
    links = sgrid.active_east_links((3, 4))
    assert_array_equal(
        links, np.array([[-1, -1, -1, -1], [4, 5, 6, -1], [-1, -1, -1, -1]])
    )

    links = sgrid.active_east_links((1, 4))
    assert_array_equal(links, np.array([[-1, -1, -1, -1]]))

    links = sgrid.active_east_links((4, 1))
    assert_array_equal(links, np.array([[-1, -1, -1, -1]]).T)

    links = sgrid.horizontal_active_link_ids((4, 5))
    assert_array_equal(np.array([[9, 10, 11, 12], [13, 14, 15, 16]]), links)

    links = sgrid.active_east_links((4, 5))
    assert_array_equal(
        np.array(
            [
                [-1, -1, -1, -1, -1],
                [9, 10, 11, 12, -1],
                [13, 14, 15, 16, -1],
                [-1, -1, -1, -1, -1],
            ]
        ),
        links,
    )

Example #3

File: test_structured_grid.py Project: sklgp2016/landlab

def test_horizontal_active_link_ids():
    links = sgrid.horizontal_active_link_ids((3, 4))
    assert_array_equal(links, np.array([[4, 5, 6]]))

    links = sgrid.horizontal_active_link_ids((1, 4))
    expected = np.array([], ndmin=2, dtype=np.int64)
    expected.shape = (0, 3)
    assert_array_equal(expected, links)

    links = sgrid.horizontal_active_link_ids((4, 1))
    expected.shape = (2, 0)
    assert_array_equal(expected, links)

    node_status = np.ones((4, 5), dtype=int)
    links = sgrid.horizontal_active_link_ids((4, 5), node_status=node_status)
    assert_array_equal(links, np.array([[9, 10, 11, 12], [13, 14, 15, 16]]))

    node_status = np.ones((4, 5), dtype=int)
    node_status[1, 1] = 0
    links = sgrid.horizontal_active_link_ids((4, 5), node_status=node_status)
    assert_array_equal(links, np.array([[-1, -1, 7, 8], [9, 10, 11, 12]]))

    node_status[2, 1] = 0
    links = sgrid.horizontal_active_link_ids((4, 5), node_status=node_status)
    assert_array_equal(links, np.array([[-1, -1, 6, 7], [-1, -1, 8, 9]]))

    node_status[0, 0] = 0
    links = sgrid.horizontal_active_link_ids((4, 5), node_status=node_status)
    assert_array_equal(links, np.array([[-1, -1, 6, 7], [-1, -1, 8, 9]]))

Example #4

File: test_structured_grid.py Project: cmshobe/landlab

def test_horizontal_active_link_ids():
    links = sgrid.horizontal_active_link_ids((3, 4))
    assert_array_equal(links, np.array([[4, 5, 6]]))

    links = sgrid.horizontal_active_link_ids((1, 4))
    expected = np.array([], ndmin=2, dtype=np.int64)
    expected.shape = (0, 3)
    assert_array_equal(expected, links)

    links = sgrid.horizontal_active_link_ids((4, 1))
    expected.shape = (2, 0)
    assert_array_equal(expected, links)

    node_status = np.ones((4, 5), dtype=int)
    links = sgrid.horizontal_active_link_ids((4, 5), node_status=node_status)
    assert_array_equal(links, np.array([[9, 10, 11, 12], [13, 14, 15, 16]]))

    node_status = np.ones((4, 5), dtype=int)
    node_status[1, 1] = 0
    links = sgrid.horizontal_active_link_ids((4, 5), node_status=node_status)
    assert_array_equal(links, np.array([[-1, -1, 7, 8], [9, 10, 11, 12]]))

    node_status[2, 1] = 0
    links = sgrid.horizontal_active_link_ids((4, 5), node_status=node_status)
    assert_array_equal(links, np.array([[-1, -1, 6, 7], [-1, -1, 8, 9]]))

    node_status[0, 0] = 0
    links = sgrid.horizontal_active_link_ids((4, 5), node_status=node_status)
    assert_array_equal(links, np.array([[-1, -1, 6, 7], [-1, -1, 8, 9]]))