def test_inactive_boundaries():
    rmg = RasterModelGrid(3, 4, 1.0)
    assert_array_equal(rmg.active_links, np.array([4, 5, 7, 8, 9, 11, 12]))
    assert_array_equal(
        rmg._node_active_inlink_matrix2,
        np.array([[-1, -1, -1, -1, -1, 4, 5, -1, -1, 11, 12, -1], [-1, -1, -1, -1, -1, 7, 8, 9, -1, -1, -1, -1]]),
    )

    rmg.set_inactive_boundaries(True, True, True, True)
    assert_array_equal(rmg.active_links, np.array([8]))
    assert_array_equal(
        rmg._node_active_inlink_matrix2,
        np.array([[-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1], [-1, -1, -1, -1, -1, -1, 8, -1, -1, -1, -1, -1]]),
    )
    assert_array_equal(
        rmg._node_active_outlink_matrix2,
        np.array([[-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1], [-1, -1, -1, -1, -1, 8, -1, -1, -1, -1, -1, -1]]),
    )
    assert_array_equal(
        rmg._active_links_at_node(),
        np.array(
            [
                [-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1],
                [-1, -1, -1, -1, -1, -1, 8, -1, -1, -1, -1, -1],
                [-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1],
                [-1, -1, -1, -1, -1, 8, -1, -1, -1, -1, -1, -1],
            ]
        ),
    )
Exemple #2
0
def test__active_links_at_node_with_no_args():
    rmg = RasterModelGrid((4, 5), spacing=1.)
    assert_array_equal(
        rmg._active_links_at_node(),
        np.array([[-1, -1, -1, -1, -1, -1,  5,  6,  7, -1,
                   -1, 14, 15, 16, -1, -1, 23, 24, 25, -1],
                  [-1, -1, -1, -1, -1, -1,  9, 10, 11, 12,
                   -1, 18, 19, 20, 21, -1, -1, -1, -1, -1],
                  [-1,  5,  6,  7, -1, -1, 14, 15, 16, -1,
                   -1, 23, 24, 25, -1, -1, -1, -1, -1, -1],
                  [-1, -1, -1, -1, -1,  9, 10, 11, 12, -1,
                   18, 19, 20, 21, -1, -1, -1, -1, -1, -1]]))
def test_inactive_boundaries():
    rmg = RasterModelGrid(3, 4, 1.)
    assert_array_equal(rmg.active_links,
                       np.array([4,  5,  7,  8,  9, 11, 12]))

    rmg.set_inactive_boundaries(True, True, True, True)
    assert_array_equal(rmg.active_links, np.array([8]))
    assert_array_equal(
        rmg._active_links_at_node(),
        np.array([[-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1],
                  [-1, -1, -1, -1, -1, -1,  8, -1, -1, -1, -1, -1],
                  [-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1],
                  [-1, -1, -1, -1, -1,  8, -1, -1, -1, -1, -1, -1]]))
def test_inactive_interiors():
    rmg = RasterModelGrid(4, 5, 1.0)
    rmg.set_closed_nodes([6, 12])
    assert_array_equal(
        rmg._active_links_at_node(),
        np.array(
            [
                [-1, -1, -1, -1, -1, -1, -1, 6, 7, -1, -1, -1, -1, 16, -1, -1, 23, -1, 25, -1],
                [-1, -1, -1, -1, -1, -1, -1, -1, 11, 12, -1, 18, -1, -1, 21, -1, -1, -1, -1, -1],
                [-1, -1, 6, 7, -1, -1, -1, -1, 16, -1, -1, 23, -1, 25, -1, -1, -1, -1, -1, -1],
                [-1, -1, -1, -1, -1, -1, -1, 11, 12, -1, 18, -1, -1, 21, -1, -1, -1, -1, -1, -1],
            ]
        ),
    )
Exemple #5
0
def test_inactive_boundaries():
    rmg = RasterModelGrid(3, 4)
    assert_array_equal(rmg.active_links, np.array([4, 5, 7, 8, 9, 11, 12]))

    rmg.set_inactive_boundaries(True, True, True, True)
    assert_array_equal(rmg.active_links, np.array([8]))
    assert_array_equal(
        rmg._active_links_at_node(),
        np.array([
            [-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1],
            [-1, -1, -1, -1, -1, -1, 8, -1, -1, -1, -1, -1],
            [-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1],
            [-1, -1, -1, -1, -1, 8, -1, -1, -1, -1, -1, -1],
        ]),
    )
def test_inactive_boundaries():
    rmg = RasterModelGrid(3, 4)
    assert_array_equal(rmg.active_links, np.array([4, 5, 7, 8, 9, 11, 12]))
    with pytest.deprecated_call():
        rmg.set_inactive_boundaries(True, True, True, True)
    assert_array_equal(rmg.active_links, np.array([8]))
    assert_array_equal(
        rmg._active_links_at_node(),
        np.array(
            [
                [-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1],
                [-1, -1, -1, -1, -1, -1, 8, -1, -1, -1, -1, -1],
                [-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1],
                [-1, -1, -1, -1, -1, 8, -1, -1, -1, -1, -1, -1],
            ]
        ),
    )
Exemple #7
0
def test__active_links_at_node_with_no_args():
    rmg = RasterModelGrid((4, 5), spacing=1.)
    assert_array_equal(
        rmg._active_links_at_node(),
        np.array([[
            -1, -1, -1, -1, -1, -1, 5, 6, 7, -1, -1, 14, 15, 16, -1, -1, 23,
            24, 25, -1
        ],
                  [
                      -1, -1, -1, -1, -1, -1, 9, 10, 11, 12, -1, 18, 19, 20,
                      21, -1, -1, -1, -1, -1
                  ],
                  [
                      -1, 5, 6, 7, -1, -1, 14, 15, 16, -1, -1, 23, 24, 25, -1,
                      -1, -1, -1, -1, -1
                  ],
                  [
                      -1, -1, -1, -1, -1, 9, 10, 11, 12, -1, 18, 19, 20, 21,
                      -1, -1, -1, -1, -1, -1
                  ]]))
Exemple #8
0
def test_inactive_interiors():
    rmg = RasterModelGrid(4, 5, 1.)
    rmg.set_closed_nodes([6, 12])
    assert_array_equal(
        rmg._active_links_at_node(),
        np.array([[
            -1, -1, -1, -1, -1, -1, -1, 6, 7, -1, -1, -1, -1, 16, -1, -1, 23,
            -1, 25, -1
        ],
                  [
                      -1, -1, -1, -1, -1, -1, -1, -1, 11, 12, -1, 18, -1, -1,
                      21, -1, -1, -1, -1, -1
                  ],
                  [
                      -1, -1, 6, 7, -1, -1, -1, -1, 16, -1, -1, 23, -1, 25, -1,
                      -1, -1, -1, -1, -1
                  ],
                  [
                      -1, -1, -1, -1, -1, -1, -1, 11, 12, -1, 18, -1, -1, 21,
                      -1, -1, -1, -1, -1, -1
                  ]]))
Exemple #9
0
def test_inactive_boundaries():
    rmg = RasterModelGrid(3, 4, 1.)
    assert_array_equal(rmg.active_links, np.array([4, 5, 7, 8, 9, 11, 12]))
    assert_array_equal(
        rmg._node_active_inlink_matrix2,
        np.array([[-1, -1, -1, -1, -1, 4, 5, -1, -1, 11, 12, -1],
                  [-1, -1, -1, -1, -1, 7, 8, 9, -1, -1, -1, -1]]))

    rmg.set_inactive_boundaries(True, True, True, True)
    assert_array_equal(rmg.active_links, np.array([8]))
    assert_array_equal(
        rmg._node_active_inlink_matrix2,
        np.array([[-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1],
                  [-1, -1, -1, -1, -1, -1, 8, -1, -1, -1, -1, -1]]))
    assert_array_equal(
        rmg._node_active_outlink_matrix2,
        np.array([[-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1],
                  [-1, -1, -1, -1, -1, 8, -1, -1, -1, -1, -1, -1]]))
    assert_array_equal(
        rmg._active_links_at_node(),
        np.array([[-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1],
                  [-1, -1, -1, -1, -1, -1, 8, -1, -1, -1, -1, -1],
                  [-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1],
                  [-1, -1, -1, -1, -1, 8, -1, -1, -1, -1, -1, -1]]))
Exemple #10
0
def test_active_node_with_array_arg():
    rmg = RasterModelGrid((4, 5), spacing=1.)
    assert_array_equal(rmg._active_links_at_node([6, 7]),
                       np.array([[5, 9, 14, 10], [6, 10, 15, 11]]).T)
Exemple #11
0
def test__active_links_at_node_scalar_boundary():
    rmg = RasterModelGrid((4, 5), spacing=1.)
    assert_array_equal(rmg._active_links_at_node([1]),
                       np.array([[-1, -1, 5, -1]]).T)
Exemple #12
0
def test__active_links_at_node_scalar_interior():
    rmg = RasterModelGrid((4, 5), spacing=1.)
    assert_array_equal(rmg._active_links_at_node([6]),
                       np.array([[5, 9, 14, 10]]).T)
Exemple #13
0
def test_active_node_with_array_arg():
    rmg = RasterModelGrid((4, 5), spacing=1.)
    assert_array_equal(rmg._active_links_at_node([6, 7]),
                       np.array([[5,  9, 14, 10],
                                 [6, 10, 15, 11]]).T)
Exemple #14
0
def test__active_links_at_node_scalar_boundary():
    rmg = RasterModelGrid((4, 5), spacing=1.)
    assert_array_equal(rmg._active_links_at_node([1]),
                       np.array([[-1, -1,  5, -1]]).T)
Exemple #15
0
def test__active_links_at_node_scalar_interior():
    rmg = RasterModelGrid((4, 5), spacing=1.)
    assert_array_equal(rmg._active_links_at_node([6]),
                       np.array([[5, 9, 14, 10]]).T)