def test_find_inletoutlet_sw(self, known_river):
"""Test using south and west exit sides."""
nodes = known_river.nodes
links = known_river.links
known_river.skeletonize()
Iskel = known_river.Iskel
exit_sides = ['s', 'w']
nodes = ru.find_inlet_outlet_nodes(links, nodes, exit_sides, Iskel)
# assert that correct nodes were assigned as inlets/outlets
assert nodes['inlets'] == [1919]
assert nodes['outlets'] == [150]
def test_find_inletoutlet_ne(self, known_river):
"""Test using north and east exit sides."""
nodes = known_river.nodes
links = known_river.links
known_river.skeletonize()
Iskel = known_river.Iskel
exit_sides = ['n', 'e']
nodes = ru.find_inlet_outlet_nodes(links, nodes, exit_sides, Iskel)
# assert that correct nodes were assigned as inlets/outlets
assert nodes['inlets'] == [247]
assert nodes['outlets'] == [872]