def setup_method(self, method):
self.bs1 = RasterBoundaryStatus([
NodeStatus.CORE, NodeStatus.FIXED_VALUE_BOUNDARY, NodeStatus.CORE,
NodeStatus.FIXED_VALUE_BOUNDARY
])
self.bs2 = RasterBoundaryStatus(NodeStatus.CORE)
self.bs3 = RasterBoundaryStatus(NodeStatus.LOOPED_BOUNDARY)
def setup_class(cls):
profile_grid = ProfileGrid(8, 2.2, [NodeStatus.FIXED_VALUE_BOUNDARY] * 2, [])
cls.profile_flow_graph = FlowGraph(profile_grid, SingleFlowRouter())
cls.profile_elevation = np.r_[0.82, 0.16, 0.14, 0.20, 0.71, 0.97, 0.41, 0.09]
cls.result_profile_elevation = cls.profile_flow_graph.update_routes(
cls.profile_elevation
)
raster_grid = RasterGrid(
[4, 4],
[1.1, 1.2],
RasterBoundaryStatus(NodeStatus.FIXED_VALUE_BOUNDARY),
[],
)
cls.raster_flow_graph = FlowGraph(raster_grid, SingleFlowRouter())
cls.raster_elevation = np.array(
[
[0.82, 0.16, 0.14, 0.20],
[0.71, 0.97, 0.41, 0.09],
[0.49, 0.01, 0.19, 0.38],
[0.29, 0.82, 0.09, 0.88],
]
)
cls.result_raster_elevation = cls.raster_flow_graph.update_routes(
cls.raster_elevation
)
def test_raster_grid(self, func, k):
# Test on a tiny (2x2) 2-d square grid with a planar surface
# tilted in y (rows) and with all outlets on the 1st row.
spacing = 300.
grid = RasterGrid([2, 2], [spacing, spacing],
RasterBoundaryStatus(
NodeStatus.FIXED_VALUE_BOUNDARY), [])
flow_graph = FlowGraph(grid, SingleFlowRouter())
h = 1.
elevation = np.array([[0., 0.], [h, h]], dtype='d')
graph_elevation = flow_graph.update_routes(elevation)
drainage_area = flow_graph.accumulate(1.)
erosion = np.zeros_like(elevation)
m_exp = 0.5
n_exp = 1.
dt = 1 # use small time step (compare with explicit scheme)
tolerance = 1e-3
n_corr = func(erosion, elevation, drainage_area, flow_graph, k, m_exp,
n_exp, dt, tolerance)
slope = h / spacing
a = spacing**2
k_coef = 1e-3
err = dt * k_coef * a**m_exp * slope**n_exp
expected_erosion = np.array([[0., 0.], [err, err]], dtype='d')
np.testing.assert_allclose(erosion, expected_erosion, atol=1e-5)
assert n_corr == 0
def grid(grid_type):
if grid_type == "profile":
bs = ProfileBoundaryStatus(NodeStatus.FIXED_VALUE_BOUNDARY)
return ProfileGrid(4, 1.0, bs, [])
elif grid_type == "raster":
bs = RasterBoundaryStatus(NodeStatus.FIXED_VALUE_BOUNDARY)
return RasterGrid([3, 3], np.array([1.0, 1.0]), bs, [])
def test___init__(self):
profile_grid = ProfileGrid(8, 2.2,
[NodeStatus.FIXED_VALUE_BOUNDARY] * 2, [])
raster_grid = RasterGrid([5, 10], [2.2, 2.4],
RasterBoundaryStatus(
NodeStatus.FIXED_VALUE_BOUNDARY), [])
FlowGraph(profile_grid, DummyFlowRouter())
FlowGraph(profile_grid, MultipleFlowRouter(1., 1.1))
FlowGraph(profile_grid, SingleFlowRouter())
def test_accumulate(self):
grid = ProfileGrid(8, 2.2, [NodeStatus.FIXED_VALUE_BOUNDARY] * 2, [])
flow_graph = FlowGraph(grid, SingleFlowRouter())
elevation = np.r_[0.82, 0.16, 0.14, 0.20, 0.71, 0.97, 0.41, 0.09]
graph_elevation = flow_graph.update_routes(elevation)
npt.assert_almost_equal(
flow_graph.accumulate(np.ones(elevation.shape)),
np.r_[2.2, 4.4, 11.0, 4.4, 2.2, 2.2, 4.4, 6.6],
)
grid = RasterGrid(
[4, 4],
[1.1, 1.2],
RasterBoundaryStatus(NodeStatus.FIXED_VALUE_BOUNDARY),
[],
)
flow_graph = FlowGraph(grid, SingleFlowRouter())
elevation = np.array([
[0.82, 0.16, 0.14, 0.20],
[0.71, 0.97, 0.41, 0.09],
[0.49, 0.01, 0.19, 0.38],
[0.29, 0.82, 0.09, 0.88],
])
graph_elevation = flow_graph.update_routes(elevation)
expected = np.array([
[1.32, 2.64, 3.96, 1.32],
[1.32, 1.32, 1.32, 9.24],
[1.32, 11.88, 1.32, 1.32],
[1.32, 1.32, 2.64, 1.32],
])
npt.assert_almost_equal(
flow_graph.accumulate(np.ones(elevation.shape)), expected)
npt.assert_almost_equal(flow_graph.accumulate(1.0), expected)
npt.assert_almost_equal(flow_graph.accumulate(5.0), 5 * expected)
data = np.array([
[1.1, 1.0, 1.1, 1.0],
[1.1, 1.0, 1.1, 1.0],
[1.1, 1.0, 1.1, 1.0],
[1.1, 1.0, 1.1, 1.0],
])
expected = np.array([
[1.452, 2.772, 4.224, 1.32],
[1.452, 1.32, 1.452, 9.636],
[1.452, 12.54, 1.452, 1.32],
[1.452, 1.32, 2.772, 1.32],
])
npt.assert_almost_equal(flow_graph.accumulate(data), expected)
def setup_method(self, method):
self.bs = bs = RasterBoundaryStatus(NodeStatus.FIXED_VALUE_BOUNDARY)
self.g = RasterGrid(
[5, 10], [2.2, 2.4], bs,
[RasterNode(0, 5, NodeStatus.FIXED_VALUE_BOUNDARY)])