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], ] ), )
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], ] ), )
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], ] ), )
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_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 ]]))
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]]))
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)
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)
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)