def test_create_xy_of_node_one_segement(nodes): grid = RasterModelGrid((16, 64), xy_spacing=(2.0, 3.0)) network = ChannelSegmentConnector(nodes) xy_of_node = create_xy_of_node(network.root, grid) assert np.allclose(xy_of_node[:, 0], grid.x_of_node[nodes]) assert np.allclose(xy_of_node[:, 1], grid.y_of_node[nodes])
def test_create_xy_of_node_with_branch(): grid = RasterModelGrid((3, 4), xy_spacing=(2.0, 3.0)) network = ChannelSegmentConnector([4, 5], [5, 2, 3], [5, 10, 11]) xy_of_node = create_xy_of_node(network.root, grid) assert np.allclose(xy_of_node, [[0, 3], [2, 3], [4, 0], [6, 0], [4, 6], [6, 6]])
def test_xy_of_node_if_not_network_root(): grid = RasterModelGrid((3, 4), xy_spacing=(2.0, 3.0)) network = ChannelSegmentConnector([4, 5], [5, 2], [5, 10, 11], [2, 3], [2, 7]) base = network.root.upstream[0] xy_of_node = create_xy_of_node(base, grid) assert np.allclose(xy_of_node, [[4, 0], [6, 0], [6, 3]])