def test_stream_power_save_output(tmpdir):
mg = RasterModelGrid((3, 3), xy_spacing=10.0)
mg.set_status_at_node_on_edges(
right=mg.BC_NODE_IS_CLOSED,
top=mg.BC_NODE_IS_CLOSED,
left=mg.BC_NODE_IS_CLOSED,
bottom=mg.BC_NODE_IS_FIXED_VALUE,
)
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)
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, RasterModelGrid)
assert set(mg1.at_node.keys()) == set(keys)
assert_equal(mg1.status_at_node, mg.status_at_node)
def test_steady_sp_raster():
"""
Initialize HydrologySteadyStreamPower on a raster grid.
After one timestep it returns all recharge as discharge.
"""
mg = RasterModelGrid((3, 3), xy_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)
hm.run_step()
assert_almost_equal(hm.q[4], 1e-4)
assert_almost_equal(hm.q_an[4], 1e-5)
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_stream_power_run_model_subdivide():
mg = RasterModelGrid((3, 3), xy_spacing=10.0)
mg.set_status_at_node_on_edges(
right=mg.BC_NODE_IS_CLOSED,
top=mg.BC_NODE_IS_CLOSED,
left=mg.BC_NODE_IS_CLOSED,
bottom=mg.BC_NODE_IS_FIXED_VALUE,
)
z = mg.add_ones("node", "topographic__elevation")
z[1] = 1e-15
zb = 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)
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)
mdl = StreamPowerModel(
mg,
hydrology_model=hm,
diffusion_model=ld,
erosion_model=sp,
regolith_model=rm,
total_morphological_time=1e8,
maximum_morphological_dt=2e7,
)
mdl.run_step(1e5, dt_m_max=2e4)
assert z[4] < 1.0
assert_equal(z[4] - zb[4], 1.0)
assert_equal(mdl.num_substeps, 5)
grid,
porosity=n,
hydraulic_conductivity=ksat,
regularization_f=0.01,
recharge_rate=p,
courant_coefficient=0.1,
vn_coefficient=0.1,
)
ld = TaylorNonLinearDiffuser(grid,
linear_diffusivity=D,
slope_crit=Sc,
dynamic_dt=True)
hm = HydrologySteadyStreamPower(
grid,
groundwater_model=gdp,
hydrological_timestep=Th,
)
# surface_water_area_norm__discharge (Q/sqrt(A)) = Q* p v0 sqrt(a)
sp = FastscapeEroder(
grid,
K_sp=Ksp,
m_sp=1,
n_sp=1,
discharge_field="surface_water_area_norm__discharge",
)
rm = RegolithConstantThickness(grid, equilibrium_depth=b, uplift_rate=U)
mdl = StreamPowerModel(
grid,
dtg = df_params['dtg']
hg = df_params['hg']
gdp = GroundwaterDupuitPercolator(
mg,
porosity=n,
hydraulic_conductivity=Ks,
regularization_f=0.01,
recharge_rate=p,
courant_coefficient=0.01, #*Ks/1e-5,
vn_coefficient=0.01, #*Ks/1e-5,
)
hm = HydrologySteadyStreamPower(
mg,
groundwater_model=gdp,
hydrological_timestep=Th,
routing_method='D8' if isinstance(mg, RasterModelGrid) else 'Steepest',
)
#run model
for i in tqdm(range(500), desc="Completion"):
hm.run_step()
########## Analysis
#dataframe for output
df_output = {}
# Qstar
Q = mg.at_node['surface_water__discharge']
Qstar = mg.add_zeros('node', 'qstar')