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