def test_stream_power_save_output_hex(tmpdir):
mg = HexModelGrid((3, 3), node_layout="rect", spacing=10.0)
mg.status_at_node[mg.status_at_node == 1] = 4
mg.status_at_node[1] = 1
mg.add_ones("node", "topographic__elevation")
mg.add_zeros("node", "aquifer_base__elevation")
mg.add_ones("node", "water_table__elevation")
gdp = GroundwaterDupuitPercolator(mg, recharge_rate=1e-4)
hm = HydrologySteadyStreamPower(mg,
groundwater_model=gdp,
routing_method="Steepest")
sp = FastscapeEroder(
mg,
K_sp=1e-10,
m_sp=1,
n_sp=1,
discharge_field="surface_water_area_norm__discharge",
)
ld = LinearDiffuser(mg, linear_diffusivity=1e-10)
rm = RegolithConstantThickness(mg, uplift_rate=0.0)
output = {}
output["output_interval"] = 1000
output["output_fields"] = [
"at_node:topographic__elevation",
"at_node:aquifer_base__elevation",
"at_node:water_table__elevation",
]
output["base_output_path"] = tmpdir.strpath + "/"
output["run_id"] = 0 # make this task_id if multiple runs
mdl = StreamPowerModel(
mg,
hydrology_model=hm,
diffusion_model=ld,
erosion_model=sp,
regolith_model=rm,
total_morphological_time=1e8,
output_dict=output,
)
mdl.run_model()
file = tmpdir.join("0_grid_0.nc")
mg1 = from_netcdf(file.strpath)
keys = [
"topographic__elevation",
"aquifer_base__elevation",
"water_table__elevation",
]
assert isinstance(mg1, HexModelGrid)
assert set(mg1.at_node.keys()) == set(keys)
assert_equal(mg1.status_at_node, mg.status_at_node)
def test_stoch_sp_threshold_hex():
"""
Initialize HydrologyEventStreamPower on a hex grid.
Use single storm-interstorm pair and make sure it returns the
quantity calculated. This is not an analytical solution, just
the value that is returned when using gdp and adaptive
timestep solver. Confirms that hex grid returns the same value
as raster grid, adjusted for cell area. Confirms that when streampower
threshold is zero (Default), returns the same values as
HydrologyEventStreamPower.
"""
mg = HexModelGrid((3, 3), node_layout="rect", spacing=10.0)
mg.status_at_node[mg.status_at_node == 1] = 4
mg.status_at_node[0] = 1
mg.add_ones("node", "topographic__elevation")
mg.add_zeros("node", "aquifer_base__elevation")
mg.add_ones("node", "water_table__elevation")
gdp = GroundwaterDupuitPercolator(mg, recharge_rate=1e-6)
pd = PrecipitationDistribution(
mg,
mean_storm_duration=10,
mean_interstorm_duration=100,
mean_storm_depth=1e-3,
total_t=100,
)
pd.seed_generator(seedval=1)
hm = HydrologyEventThresholdStreamPower(mg,
precip_generator=pd,
groundwater_model=gdp,
routing_method="Steepest")
hm.run_step()
assert_almost_equal(hm.q_eff[4], 0.00017614 * np.sqrt(3) / 2)
assert_almost_equal(
hm.q_an[4],
0.00017614 * np.sqrt(3) / 2 / np.sqrt(np.sqrt(3) / 2 * 100))
def test_steady_sp_hex():
"""
Initialize HydrologySteadyStreamPower on a hex grid.
After one timestep it returns all recharge as discharge.
"""
mg = HexModelGrid((3, 3), node_layout="rect", spacing=10.0)
mg.status_at_node[mg.status_at_node == 1] = 4
mg.status_at_node[0] = 1
mg.add_ones("node", "topographic__elevation")
mg.add_zeros("node", "aquifer_base__elevation")
mg.add_ones("node", "water_table__elevation")
gdp = GroundwaterDupuitPercolator(mg, recharge_rate=1e-6)
hm = HydrologySteadyStreamPower(mg,
groundwater_model=gdp,
routing_method="Steepest")
hm.run_step()
assert_almost_equal(hm.q[4], (np.sqrt(3) / 2) * 10**(-4))
assert_almost_equal(hm.q_an[4], np.sqrt((np.sqrt(3) / 2)) * 10**(-5))
def test_outlet_lowering_rate_no_scaling_bedrock():
"""Test using an rate lowering object with no scaling and bedrock."""
mg = HexModelGrid(5, 5)
z = mg.add_ones("node", "topographic__elevation")
b = mg.add_zeros("node", "bedrock__elevation")
node_id = 27
bh = SingleNodeBaselevelHandler(mg, outlet_id=node_id, lowering_rate=-0.1)
for _ in range(240):
bh.run_one_step(10)
assert z[1] == 1.0
assert b[1] == 0.0
assert z[node_id] == -239.0
assert b[node_id] == -240.0
def test_outlet_lowering_rate_on_not_outlet():
"""Test using an rate lowering object with no scaling and bedrock."""
mg = HexModelGrid(5, 5)
z = mg.add_ones("node", "topographic__elevation")
b = mg.add_zeros("node", "bedrock__elevation")
node_id = 27
bh = SingleNodeBaselevelHandler(
mg, outlet_id=node_id, lowering_rate=-0.1, modify_outlet_id=False
)
for _ in range(240):
bh.run_one_step(10)
assert z[node_id] == 1.0
assert b[node_id] == 0.0
not_outlet = mg.nodes != node_id
assert np.all(z[not_outlet] == 241.0)
assert np.all(b[not_outlet] == 240.0)
def test_outlet_lowering_object_no_scaling_bedrock():
"""Test using an outlet lowering object with no scaling and bedrock."""
mg = HexModelGrid(5, 5)
z = mg.add_ones("node", "topographic__elevation")
b = mg.add_zeros("node", "bedrock__elevation")
node_id = 27
file = os.path.join(_TEST_DATA_DIR, "outlet_history.txt")
bh = SingleNodeBaselevelHandler(
mg, outlet_id=node_id, lowering_file_path=file
)
for _ in range(241):
bh.run_one_step(10)
assert z[1] == 1.0
assert b[1] == 0.0
assert z[node_id] == -46.5
assert b[node_id] == -47.5
def test_outlet_lowering_object_no_scaling():
"""Test using an outlet lowering object with no scaling."""
mg = HexModelGrid((5, 5))
z = mg.add_ones("node", "topographic__elevation")
file = os.path.join(_TEST_DATA_DIR, "outlet_history.txt")
bh = NotCoreNodeBaselevelHandler(
mg, modify_core_nodes=False, lowering_file_path=file
)
for _ in range(241):
bh.run_one_step(10)
closed = mg.status_at_node != 0
not_closed = mg.status_at_node == 0
# not closed should have stayed the same
assert_array_equal(z[not_closed], np.ones(np.sum(not_closed)))
# closed should lowered by 47.5 to -46.5
assert_array_equal(z[closed], -46.5 * np.ones(np.sum(closed)))
@author: nicgaspar
"""
## Linear diffusion model on a hex grid
## First import what you need
from landlab import HexModelGrid
from landlab.components import LinearDiffuser
from landlab.plot import imshow_grid
from matplotlib import pyplot as plt
## Make a grid that is 50 by 50 with dx=dy=20. m,
## except that doesn't work for Hex! so make it 51 by 50
hmg1 = HexModelGrid(51, 50, 20.)
## Add elevation data to the grid.
z1 = hmg1.add_ones('node', 'topographic__elevation')
## Instantiate linear diffusion object
ld1 = LinearDiffuser(hmg1, linear_diffusivity=0.1)
## Set some variables
rock_up_rate = 5e-4 #m/yr
dt = 1000 # yr, time step
rock_up_len = dt * rock_up_rate # m
## Time loop where the evolution happens.
for i in range(1500):
z1[hmg1.core_nodes] += rock_up_len #uplift only the core nodes
ld1.run_one_step(dt) #diffusion happens
## Plot the topography to see what comes out.
