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