def test_no_reroute():
mg = RasterModelGrid((5, 5), 2.)
z = mg.add_zeros('node', 'topographic__elevation', dtype=float)
z[1] = -1.
z[6] = -2.
z[19] = -2.
z[18] = -1.
z[17] = -3.
fd = FlowDirectorSteepest(mg)
fa = FlowAccumulator(mg)
lmb = LakeMapperBarnes(mg,
method='Steepest',
fill_flat=True,
redirect_flow_steepest_descent=True,
track_lakes=True)
lake_dict = {
1: deque([
6,
]),
18: deque([
17,
])
}
fd.run_one_step() # fill the director fields
fa.run_one_step() # get a drainage_area
orig_surf = lmb._track_original_surface()
lmb._redirect_flowdirs(orig_surf, lake_dict)
assert mg.at_node['flow__receiver_node'][6] == 1
assert mg.at_node['flow__receiver_node'][17] == 18
assert mg.at_node['flow__receiver_node'][18] == 19
def test_no_reroute():
mg = RasterModelGrid((5, 5), xy_spacing=2.)
z = mg.add_zeros("node", "topographic__elevation", dtype=float)
z[1] = -1.
z[6] = -2.
z[19] = -2.
z[18] = -1.
z[17] = -3.
fd = FlowDirectorSteepest(mg)
fa = FlowAccumulator(mg)
lmb = LakeMapperBarnes(
mg,
method="Steepest",
fill_flat=True,
redirect_flow_steepest_descent=True,
track_lakes=True,
)
lake_dict = {1: deque([6]), 18: deque([17])}
fd.run_one_step() # fill the director fields
fa.run_one_step() # get a drainage_area
orig_surf = lmb._track_original_surface()
lmb._redirect_flowdirs(orig_surf, lake_dict)
assert mg.at_node["flow__receiver_node"][6] == 1
assert mg.at_node["flow__receiver_node"][17] == 18
assert mg.at_node["flow__receiver_node"][18] == 19
def test_no_reroute():
mg = RasterModelGrid((5, 5), xy_spacing=2.0)
z = mg.add_zeros("topographic__elevation", at="node", dtype=float)
z[1] = -1.0
z[6] = -2.0
z[19] = -2.0
z[18] = -1.0
z[17] = -3.0
fd = FlowDirectorSteepest(mg)
fa = FlowAccumulator(mg)
lmb = LakeMapperBarnes(
mg,
method="Steepest",
fill_flat=True,
redirect_flow_steepest_descent=True,
track_lakes=True,
)
openq = StablePriorityQueue()
lake_dict = {1: deque([6]), 18: deque([17])}
fd.run_one_step() # fill the director fields
fa.run_one_step() # get a drainage_area
orig_surf = lmb._track_original_surface()
lmb._redirect_flowdirs(orig_surf, lake_dict, openq)
assert mg.at_node["flow__receiver_node"][6] == 1
assert mg.at_node["flow__receiver_node"][17] == 18
assert mg.at_node["flow__receiver_node"][18] == 19
def test_route_to_multiple_error_raised_run():
mg = RasterModelGrid((10, 10))
z = mg.add_zeros("node", "topographic__elevation")
z += mg.x_of_node + mg.y_of_node
lmb = LakeMapperBarnes(mg)
fa = FlowAccumulator(mg, flow_director="MFD")
fa.run_one_step()
with pytest.raises(NotImplementedError):
lmb.run_one_step()
def test_route_to_multiple_error_raised_run():
mg = RasterModelGrid((10, 10))
z = mg.add_zeros("node", "topographic__elevation")
z += mg.x_of_node + mg.y_of_node
lmb = LakeMapperBarnes(mg)
fa = FlowAccumulator(mg, flow_director="MFD")
fa.run_one_step()
with pytest.raises(NotImplementedError):
lmb.run_one_step()
def __init__(self, grid, routing_method):
self._grid = grid
if routing_method == "D8":
self.fd = FlowDirectorD8(self._grid)
elif routing_method == "Steepest":
self.fd = FlowDirectorSteepest(self._grid)
else:
raise ValueError("routing_method must be either D8 or Steepest.")
self.fa = FlowAccumulator(
self._grid,
surface="topographic__elevation",
flow_director=self.fd,
runoff_rate="average_surface_water__specific_discharge",
)
self.lmb = LakeMapperBarnes(
self._grid,
method=routing_method,
fill_flat=False,
surface="topographic__elevation",
fill_surface="topographic__elevation",
redirect_flow_steepest_descent=False,
reaccumulate_flow=False,
track_lakes=False,
ignore_overfill=True,
)
self.dfr = DepressionFinderAndRouter(self._grid)
def test_closed_up_grid():
mg = RasterModelGrid((5, 5))
for edge in ("left", "right", "top", "bottom"):
mg.status_at_node[mg.nodes_at_edge(edge)] = CLOSED_BOUNDARY
mg.add_zeros("node", "topographic__elevation", dtype=float)
with pytest.raises(ValueError):
LakeMapperBarnes(mg)
def test_with_lake_mapper_barnes():
hmg_hole = HexModelGrid((9, 5))
z = hmg_hole.add_field(
"topographic__elevation",
hmg_hole.x_of_node + np.round(hmg_hole.y_of_node),
at="node",
)
hole_nodes = [21, 22, 23, 30, 31, 39, 40]
z[hole_nodes] = z[hole_nodes] * 0.1
hmg_hole.add_zeros("filled__elevation", at="node")
fa = FlowAccumulator(
hmg_hole,
flow_director="Steepest",
depression_finder="LakeMapperBarnes",
fill_flat=False,
redirect_flow_steepest_descent=True,
reaccumulate_flow=True,
fill_surface="filled__elevation",
)
fa.run_one_step()
rcvrs = hmg_hole.at_node["flow__receiver_node"]
assert_array_equal(rcvrs[[13, 21, 46]], [6, 13, 39])
from landlab.components import LakeMapperBarnes
fa = FlowAccumulator(hmg_hole, depression_finder=LakeMapperBarnes)
lmb = LakeMapperBarnes(hmg_hole)
fa = FlowAccumulator(hmg_hole, depression_finder=lmb)
def test_closed_up_grid():
mg = RasterModelGrid((5, 5), dx=1.)
for edge in ('left', 'right', 'top', 'bottom'):
mg.status_at_node[mg.nodes_at_edge(edge)] = CLOSED_BOUNDARY
mg.add_zeros('node', 'topographic__elevation', dtype=float)
with pytest.raises(ValueError):
LakeMapperBarnes(mg)
def test_route_to_multiple_error_raised_init():
mg = RasterModelGrid((10, 10))
z = mg.add_zeros('node', 'topographic__elevation')
z += mg.x_of_node + mg.y_of_node
fa = FlowAccumulator(mg, flow_director='MFD')
fa.run_one_step()
with pytest.raises(NotImplementedError):
LakeMapperBarnes(mg)
def test_closed_up_grid():
mg = RasterModelGrid((5, 5))
for edge in ("left", "right", "top", "bottom"):
mg.status_at_node[mg.nodes_at_edge(edge)] = mg.BC_NODE_IS_CLOSED
mg.add_zeros("topographic__elevation", at="node", dtype=float)
_ = FlowAccumulator(mg, flow_director="D8")
with pytest.raises(ValueError):
LakeMapperBarnes(mg)
def test_route_to_many():
mg = RasterModelGrid((5, 5))
mg.add_zeros("node", "topographic__elevation", dtype=float)
fd = FlowDirectorDINF(mg, "topographic__elevation")
fd.run_one_step()
assert mg.at_node["flow__receiver_node"].shape == (mg.number_of_nodes, 2)
with pytest.raises(NotImplementedError):
LakeMapperBarnes(mg, method="D8", redirect_flow_steepest_descent=True)
def test_neighbor_shaping_no_creation():
mg = RasterModelGrid((5, 5))
mg.add_zeros("topographic__elevation", at="node", dtype=float)
_ = FlowAccumulator(mg)
lmb = LakeMapperBarnes(mg,
method="D8",
redirect_flow_steepest_descent=False)
with pytest.raises(AttributeError):
lmb._neighbor_arrays
def test_neighbor_shaping_hex():
hmg = HexModelGrid((6, 5), spacing=1.0)
hmg.add_zeros("topographic__elevation", at="node", dtype=float)
_ = FlowAccumulator(hmg)
lmb = LakeMapperBarnes(hmg, redirect_flow_steepest_descent=True)
for arr in (lmb._neighbor_arrays, lmb._link_arrays):
assert len(arr) == 1
assert arr[0].shape == (hmg.number_of_nodes, 6)
assert len(lmb._neighbor_lengths) == hmg.number_of_links
def test_redirect_no_lakes():
mg = RasterModelGrid((5, 5))
mg.add_zeros("topographic__elevation", at="node", dtype=float)
_ = FlowAccumulator(mg)
with pytest.raises(ValueError):
LakeMapperBarnes(mg,
method="D8",
track_lakes=False,
redirect_flow_steepest_descent=True)
def test_accum_wo_reroute():
mg = RasterModelGrid((5, 5))
mg.add_zeros("topographic__elevation", at="node", dtype=float)
_ = FlowAccumulator(mg)
with pytest.raises(ValueError):
LakeMapperBarnes(
mg,
method="Steepest",
redirect_flow_steepest_descent=False,
reaccumulate_flow=True,
)
def test_accum_wo_reroute():
mg = RasterModelGrid((5, 5), dx=1.)
mg.add_zeros('node', 'topographic__elevation', dtype=float)
mg.add_zeros('node', 'topographic__steepest_slope', dtype=float)
mg.add_zeros('node', 'flow__receiver_node', dtype=int)
mg.add_zeros('node', 'flow__link_to_receiver_node', dtype=int)
with pytest.raises(ValueError):
LakeMapperBarnes(mg,
method='Steepest',
redirect_flow_steepest_descent=False,
reaccumulate_flow=True)
def test_neighbor_shaping_hex():
hmg = HexModelGrid(6, 5, dx=1.)
hmg.add_zeros("node", "topographic__elevation", dtype=float)
hmg.add_zeros("node", "topographic__steepest_slope", dtype=float)
hmg.add_zeros("node", "flow__receiver_node", dtype=int)
hmg.add_zeros("node", "flow__link_to_receiver_node", dtype=int)
lmb = LakeMapperBarnes(hmg, redirect_flow_steepest_descent=True)
for arr in (lmb._neighbor_arrays, lmb._link_arrays):
assert len(arr) == 1
assert arr[0].shape == (hmg.number_of_nodes, 6)
assert len(lmb._neighbor_lengths) == hmg.number_of_links
def test_neighbor_shaping_D4():
mg = RasterModelGrid((5, 5))
mg.add_zeros("topographic__elevation", at="node", dtype=float)
_ = FlowAccumulator(mg)
lmb = LakeMapperBarnes(mg,
method="Steepest",
redirect_flow_steepest_descent=True)
for arr in (lmb._neighbor_arrays, lmb._link_arrays):
assert len(arr) == 1
assert arr[0].shape == (25, 4)
assert len(lmb._neighbor_lengths) == mg.number_of_links
def test_redirect_no_lakes():
mg = RasterModelGrid((5, 5))
mg.add_zeros("node", "topographic__elevation", dtype=float)
mg.add_zeros("node", "topographic__steepest_slope", dtype=float)
mg.add_zeros("node", "flow__receiver_node", dtype=int)
mg.add_zeros("node", "flow__link_to_receiver_node", dtype=int)
with pytest.raises(ValueError):
LakeMapperBarnes(mg,
method="D8",
track_lakes=False,
redirect_flow_steepest_descent=True)
def test_permitted_overfill():
mg = RasterModelGrid((3, 7))
for edge in ("top", "right", "bottom"):
mg.status_at_node[mg.nodes_at_edge(edge)] = CLOSED_BOUNDARY
z = mg.add_zeros("node", "topographic__elevation", dtype=float)
z.reshape(mg.shape)[1, 1:-1] = [1., 0.2, 0.1, 1.0000000000000004, 1.5]
lmb = LakeMapperBarnes(mg, method="Steepest")
lmb._closed = mg.zeros("node", dtype=bool)
lmb._closed[mg.status_at_node == CLOSED_BOUNDARY] = True
edges = np.array([7])
for edgenode in edges:
lmb._open.add_task(edgenode, priority=z[edgenode])
lmb._closed[edges] = True
while True:
try:
lmb._fill_one_node_to_slant(
z, mg.adjacent_nodes_at_node, lmb._pit, lmb._open, lmb._closed, True
)
except KeyError:
break
def test_neighbor_shaping_no_creation():
mg = RasterModelGrid((5, 5))
mg.add_zeros("node", "topographic__elevation", dtype=float)
mg.add_zeros("node", "topographic__steepest_slope", dtype=float)
mg.add_zeros("node", "flow__receiver_node", dtype=int)
mg.add_zeros("node", "flow__link_to_receiver_node", dtype=int)
lmb = LakeMapperBarnes(mg,
method="D8",
redirect_flow_steepest_descent=False)
with pytest.raises(AttributeError):
lmb._neighbor_arrays
def test_neighbor_shaping_D4():
mg = RasterModelGrid((5, 5), dx=1.)
mg.add_zeros('node', 'topographic__elevation', dtype=float)
mg.add_zeros('node', 'topographic__steepest_slope', dtype=float)
mg.add_zeros('node', 'flow__receiver_node', dtype=int)
mg.add_zeros('node', 'flow__link_to_receiver_node', dtype=int)
lmb = LakeMapperBarnes(mg,
method='Steepest',
redirect_flow_steepest_descent=True)
for arr in (lmb._neighbor_arrays, lmb._link_arrays):
assert len(arr) == 1
assert arr[0].shape == (25, 4)
assert len(lmb._neighbor_lengths) == mg.number_of_links
def test_neighbor_shaping_D8():
mg = RasterModelGrid((5, 5))
mg.add_zeros("node", "topographic__elevation", dtype=float)
mg.add_zeros("node", "topographic__steepest_slope", dtype=float)
mg.add_zeros("node", "flow__receiver_node", dtype=int)
mg.add_zeros("node", "flow__link_to_receiver_node", dtype=int)
lmb = LakeMapperBarnes(mg,
method="D8",
redirect_flow_steepest_descent=True)
for arr in (lmb._neighbor_arrays, lmb._link_arrays):
assert len(arr) == 2
assert arr[0].shape == (25, 4)
assert arr[1].shape == (25, 4)
assert len(lmb._neighbor_lengths) == mg.number_of_d8
def test_permitted_overfill():
mg = RasterModelGrid((3, 7))
for edge in ("top", "right", "bottom"):
mg.status_at_node[mg.nodes_at_edge(edge)] = CLOSED_BOUNDARY
z = mg.add_zeros("node", "topographic__elevation", dtype=float)
z.reshape(mg.shape)[1, 1:-1] = [1., 0.2, 0.1, 1.0000000000000004, 1.5]
lmb = LakeMapperBarnes(mg, method="Steepest")
lmb._closed = mg.zeros("node", dtype=bool)
lmb._closed[mg.status_at_node == CLOSED_BOUNDARY] = True
edges = np.array([7])
for edgenode in edges:
lmb._open.add_task(edgenode, priority=z[edgenode])
lmb._closed[edges] = True
while True:
try:
lmb._fill_one_node_to_slant(z, mg.adjacent_nodes_at_node, lmb._pit,
lmb._open, lmb._closed, True)
except KeyError:
break
top=grid.BC_NODE_IS_CLOSED,
left=grid.BC_NODE_IS_FIXED_VALUE,
bottom=grid.BC_NODE_IS_CLOSED,
)
z = grid.add_zeros('node', 'topographic__elevation')
z[:] = 0.1 * hg * np.random.rand(len(z))
fa = FlowAccumulator(grid,
surface='topographic__elevation',
flow_director='D8')
lmb = LakeMapperBarnes(
grid,
method='D8',
fill_flat=False,
surface="topographic__elevation",
fill_surface="topographic__elevation",
redirect_flow_steepest_descent=False,
reaccumulate_flow=False,
track_lakes=False,
ignore_overfill=True,
)
dfr = DepressionFinderAndRouter(grid)
ld = LinearDiffuser(grid, D)
sp = FastscapeEroder(grid, K_sp=Ksp, m_sp=0.5, n_sp=1.0, threshold_sp=E0)
for i in range(N):
dfr._find_pits()
if dfr._number_of_pits > 0:
lmb.run_one_step()
def test_bad_init_method2():
rmg = RasterModelGrid((5, 5), dx=2.)
rmg.add_zeros('node', 'topographic__elevation', dtype=float)
with pytest.raises(ValueError):
LakeMapperBarnes(rmg, method='d8')
def test_neighbor_shaping_no_fldir():
mg = RasterModelGrid((5, 5), dx=1.)
mg.add_zeros('node', 'topographic__elevation', dtype=float)
with pytest.raises(FieldError):
LakeMapperBarnes(mg, method='D8', redirect_flow_steepest_descent=True)
def test_bad_init_method2():
rmg = RasterModelGrid((5, 5), xy_spacing=2.)
rmg.add_zeros("node", "topographic__elevation", dtype=float)
with pytest.raises(ValueError):
LakeMapperBarnes(rmg, method="d8")
def test_bad_init_gridmethod():
hmg = HexModelGrid(30, 29, dx=3.)
hmg.add_zeros("node", "topographic__elevation", dtype=float)
with pytest.raises(ValueError):
LakeMapperBarnes(hmg, method="D8")
def test_neighbor_shaping_no_fldir():
mg = RasterModelGrid((5, 5))
mg.add_zeros("topographic__elevation", at="node", dtype=float)
with pytest.raises(FieldError):
LakeMapperBarnes(mg, method="D8", redirect_flow_steepest_descent=True)
def test_bad_init_method2():
rmg = RasterModelGrid((5, 5), xy_spacing=2.0)
rmg.add_zeros("topographic__elevation", at="node", dtype=float)
_ = FlowAccumulator(rmg, flow_director="D8")
with pytest.raises(ValueError):
LakeMapperBarnes(rmg, method="d8")
def test_bad_init_gridmethod():
hmg = HexModelGrid((30, 29), spacing=3.0)
hmg.add_zeros("topographic__elevation", at="node", dtype=float)
_ = FlowAccumulator(hmg, flow_director="Steepest")
with pytest.raises(ValueError):
LakeMapperBarnes(hmg, method="D8")