def test_incorrect_endpoints_type():
mg = RasterModelGrid((3, 5))
core_nodes = mg.core_nodes
endpoints = [core_nodes.min()]
with pytest.raises(ValueError):
Profiler(mg, endpoints)
endpoints = [core_nodes.min(), (0, 1, 2)]
with pytest.raises(TypeError):
Profiler(mg, endpoints)
def test_out_of_bounds_point():
mg = RasterModelGrid((3, 5))
core_nodes = mg.core_nodes
endpoints = [core_nodes.min(), 2 * core_nodes.max()]
with pytest.raises(IndexError):
Profiler(mg, endpoints)
ep0 = mg.xy_of_node[core_nodes.min()]
ep1 = 2 * mg.xy_of_node[core_nodes.max()]
with pytest.raises(ValueError):
Profiler(mg, [ep0, ep1])
def test_endpoint_options():
mg = RasterModelGrid((3, 5))
core_nodes = mg.core_nodes
endpoints = [core_nodes.min(), core_nodes.max()]
profiler_by_nodes = Profiler(mg, endpoints)
profiler_by_nodes.run_one_step()
ep0 = mg.xy_of_node[core_nodes.min()]
ep1 = mg.xy_of_node[core_nodes.max()]
profiler_by_coords = Profiler(mg, [ep0, ep1])
profiler_by_coords.run_one_step()
np.testing.assert_array_equal(profiler_by_nodes.nodes[0],
profiler_by_coords.nodes[0])
def test_steep_profile():
mg = RasterModelGrid((5, 3))
core_nodes = mg.core_nodes
endpoints = [core_nodes.min(), core_nodes.max()]
profiler = Profiler(mg, endpoints)
profiler.run_one_step()
np.testing.assert_array_equal(profiler.nodes[0], core_nodes)
def test_positive_ystep_profile():
mg = RasterModelGrid((4, 5))
core_nodes = mg.core_nodes
endpoints = [core_nodes.min(), core_nodes.max()]
profiler = Profiler(mg, endpoints)
profiler.run_one_step()
np.testing.assert_array_equal(profiler.nodes[0], np.array([6, 7, 13]))
def test_flipped_single_segment_profile():
mg = RasterModelGrid((3, 5))
core_nodes = mg.core_nodes
endpoints = [core_nodes.max(), core_nodes.min()]
profiler = Profiler(mg, endpoints)
profiler.run_one_step()
np.testing.assert_array_equal(profiler.nodes[0], np.flip(core_nodes))
def test_multi_segment_profile_structure():
mg = RasterModelGrid((5, 5))
core_nodes = mg.core_nodes
endpoints = [core_nodes.min(), core_nodes.min() + 2, core_nodes.max()]
profiler = Profiler(mg, endpoints)
profiler.run_one_step()
np.testing.assert_array_equal(list(profiler.data_structure.keys()), [0, 1])
ds = profiler.data_structure
np.testing.assert_array_equal(ds[0]["ids"], [6, 7, 8])
np.testing.assert_array_equal(ds[1]["ids"], [8, 13, 18])
np.testing.assert_array_equal(ds[0]["distances"], [0, 1, 2])
np.testing.assert_array_equal(ds[1]["distances"], [2, 3, 4])