def test_not_implemented(): """Test that private run_one_step is not implemented""" mg = RasterModelGrid((10, 10), spacing=(1, 1)) z = mg.add_field('topographic__elevation', mg.node_x**2 + mg.node_y**2, at='node') fd0 = _FlowDirector(mg, 'topographic__elevation') assert_raises(NotImplementedError, fd0.run_one_step) fd1 = _FlowDirectorToMany(mg, 'topographic__elevation') assert_raises(NotImplementedError, fd1.run_one_step) fd2 = _FlowDirectorToOne(mg, 'topographic__elevation') assert_raises(NotImplementedError, fd2.run_one_step)
def test_not_implemented(): """Test that private run_one_step is not implemented""" mg = RasterModelGrid((10, 10), xy_spacing=(1, 1)) mg.add_field("topographic__elevation", mg.node_x ** 2 + mg.node_y ** 2, at="node") fd0 = _FlowDirector(mg, "topographic__elevation") with pytest.raises(NotImplementedError): fd0.run_one_step() fd1 = _FlowDirectorToMany(mg, "topographic__elevation") with pytest.raises(NotImplementedError): fd1.run_one_step() fd2 = _FlowDirectorToOne(mg, "topographic__elevation") with pytest.raises(NotImplementedError): fd2.run_one_step()
def test_not_implemented(): """Test that private run_one_step is not implemented""" mg = RasterModelGrid((10, 10), xy_spacing=(1, 1)) mg.add_field("topographic__elevation", mg.node_x**2 + mg.node_y**2, at="node") fd0 = _FlowDirector(mg, "topographic__elevation") with pytest.raises(NotImplementedError): fd0.run_one_step() fd1 = _FlowDirectorToMany(mg, "topographic__elevation") with pytest.raises(NotImplementedError): fd1.run_one_step() fd2 = _FlowDirectorToOne(mg, "topographic__elevation") with pytest.raises(NotImplementedError): fd2.run_one_step()
def test_check_fields(): """Check to make sure the right fields have been created. Check that the sizes of at least one are also correct (they are at node so if one is the right size, then they all should be) """ mg0 = RasterModelGrid((10, 10), spacing=(1, 1)) z0 = mg0.add_field('topographic__elevation', mg0.node_x**2 + mg0.node_y**2, at='node') fd0 = _FlowDirector(mg0, 'topographic__elevation') assert list(mg0.at_node.keys()) == ['topographic__elevation'] assert np.size( mg0.at_node['topographic__elevation']) == mg0.number_of_nodes mg1 = RasterModelGrid((10, 10), spacing=(1, 1)) z1 = mg1.add_field('topographic__elevation', mg1.node_x**2 + mg1.node_y**2, at='node') fd1 = _FlowDirectorToMany(mg1, 'topographic__elevation') assert sorted(list(mg1.at_node.keys())) == [ 'flow__link_to_receiver_node', 'flow__receiver_node', 'flow__sink_flag', 'topographic__elevation', 'topographic__steepest_slope', ] assert np.size( mg1.at_node['topographic__elevation']) == mg1.number_of_nodes mg2 = RasterModelGrid((10, 10), spacing=(1, 1)) z2 = mg2.add_field('topographic__elevation', mg2.node_x**2 + mg2.node_y**2, at='node') fd2 = _FlowDirectorToOne(mg2, 'topographic__elevation') assert sorted(list(mg2.at_node.keys())) == [ 'flow__link_to_receiver_node', 'flow__receiver_node', 'flow__sink_flag', 'topographic__elevation', 'topographic__steepest_slope', ] assert np.size( mg2.at_node['topographic__elevation']) == mg2.number_of_nodes mg3 = RasterModelGrid((10, 10), spacing=(1, 1)) z3 = mg3.add_field('topographic__elevation', mg3.node_x**2 + mg3.node_y**2, at='node') fd3 = FlowDirectorMFD(mg3, 'topographic__elevation') assert sorted(list(mg3.at_node.keys())) == [ 'flow__link_to_receiver_node', 'flow__links_to_receiver_nodes', 'flow__receiver_node', 'flow__receiver_nodes', 'flow__receiver_proportions', 'flow__sink_flag', 'topographic__elevation', 'topographic__steepest_slope', ] assert np.size( mg3.at_node['topographic__elevation']) == mg3.number_of_nodes mg4 = RasterModelGrid((10, 10), spacing=(1, 1)) z4 = mg4.add_field('topographic__elevation', mg4.node_x**2 + mg4.node_y**2, at='node') fd4 = FlowDirectorDINF(mg4, 'topographic__elevation') assert sorted(list(mg4.at_node.keys())) == [ 'flow__link_to_receiver_node', 'flow__links_to_receiver_nodes', 'flow__receiver_node', 'flow__receiver_nodes', 'flow__receiver_proportions', 'flow__sink_flag', 'topographic__elevation', 'topographic__steepest_slope', ] assert np.size( mg4.at_node['topographic__elevation']) == mg4.number_of_nodes mg5 = RasterModelGrid((10, 10), spacing=(1, 1)) z5 = mg5.add_field('topographic__elevation', mg5.node_x**2 + mg5.node_y**2, at='node') fd5 = FlowDirectorSteepest(mg5, 'topographic__elevation') assert sorted(list(mg5.at_node.keys())) == [ 'flow__link_to_receiver_node', 'flow__receiver_node', 'flow__sink_flag', 'topographic__elevation', 'topographic__steepest_slope', ] assert np.size( mg5.at_node['topographic__elevation']) == mg5.number_of_nodes mg6 = RasterModelGrid((10, 10), spacing=(1, 1)) z6 = mg6.add_field('topographic__elevation', mg6.node_x**2 + mg6.node_y**2, at='node') fd6 = FlowDirectorD8(mg6, 'topographic__elevation') assert sorted(list(mg6.at_node.keys())) == [ 'flow__link_to_receiver_node', 'flow__receiver_node', 'flow__sink_flag', 'topographic__elevation', 'topographic__steepest_slope', ] assert np.size( mg6.at_node['topographic__elevation']) == mg6.number_of_nodes
def test_check_fields(): """Check to make sure the right fields have been created. Check that the sizes of at least one are also correct (they are at node so if one is the right size, then they all should be) """ mg0 = RasterModelGrid((10, 10), xy_spacing=(1, 1)) mg0.add_field( "topographic__elevation", mg0.node_x ** 2 + mg0.node_y ** 2, at="node" ) _FlowDirector(mg0, "topographic__elevation") assert sorted(list(mg0.at_node.keys())) == [ "flow__sink_flag", "topographic__elevation", ] assert np.size(mg0.at_node["topographic__elevation"]) == mg0.number_of_nodes mg1 = RasterModelGrid((10, 10), xy_spacing=(1, 1)) mg1.add_field( "topographic__elevation", mg1.node_x ** 2 + mg1.node_y ** 2, at="node" ) _FlowDirectorToMany(mg1, "topographic__elevation") assert sorted(list(mg1.at_node.keys())) == [ "flow__sink_flag", "topographic__elevation", ] assert np.size(mg1.at_node["topographic__elevation"]) == mg1.number_of_nodes mg2 = RasterModelGrid((10, 10), xy_spacing=(1, 1)) mg2.add_field( "topographic__elevation", mg2.node_x ** 2 + mg2.node_y ** 2, at="node" ) _FlowDirectorToOne(mg2, "topographic__elevation") assert sorted(list(mg2.at_node.keys())) == [ "flow__link_to_receiver_node", "flow__receiver_node", "flow__sink_flag", "topographic__elevation", "topographic__steepest_slope", ] assert np.size(mg2.at_node["topographic__elevation"]) == mg2.number_of_nodes mg3 = RasterModelGrid((10, 10), xy_spacing=(1, 1)) mg3.add_field( "topographic__elevation", mg3.node_x ** 2 + mg3.node_y ** 2, at="node" ) FlowDirectorMFD(mg3, "topographic__elevation") assert sorted(list(mg3.at_node.keys())) == [ "flow__link_to_receiver_node", "flow__receiver_node", "flow__receiver_proportions", "flow__sink_flag", "topographic__elevation", "topographic__steepest_slope", ] assert np.size(mg3.at_node["topographic__elevation"]) == mg3.number_of_nodes mg4 = RasterModelGrid((10, 10), xy_spacing=(1, 1)) mg4.add_field( "topographic__elevation", mg4.node_x ** 2 + mg4.node_y ** 2, at="node" ) FlowDirectorDINF(mg4, "topographic__elevation") assert sorted(list(mg4.at_node.keys())) == [ "flow__link_to_receiver_node", "flow__receiver_node", "flow__receiver_proportions", "flow__sink_flag", "topographic__elevation", "topographic__steepest_slope", ] assert np.size(mg4.at_node["topographic__elevation"]) == mg4.number_of_nodes mg5 = RasterModelGrid((10, 10), xy_spacing=(1, 1)) mg5.add_field( "topographic__elevation", mg5.node_x ** 2 + mg5.node_y ** 2, at="node" ) FlowDirectorSteepest(mg5, "topographic__elevation") assert sorted(list(mg5.at_node.keys())) == [ "flow__link_to_receiver_node", "flow__receiver_node", "flow__sink_flag", "topographic__elevation", "topographic__steepest_slope", ] assert np.size(mg5.at_node["topographic__elevation"]) == mg5.number_of_nodes mg6 = RasterModelGrid((10, 10), xy_spacing=(1, 1)) mg6.add_field( "topographic__elevation", mg6.node_x ** 2 + mg6.node_y ** 2, at="node" ) FlowDirectorD8(mg6, "topographic__elevation") assert sorted(list(mg6.at_node.keys())) == [ "flow__link_to_receiver_node", "flow__receiver_node", "flow__sink_flag", "topographic__elevation", "topographic__steepest_slope", ] assert np.size(mg6.at_node["topographic__elevation"]) == mg6.number_of_nodes
def test_check_fields(): """Check to make sure the right fields have been created. Check that the sizes of at least one are also correct (they are at node so if one is the right size, then they all should be) """ mg0 = RasterModelGrid((10, 10), spacing=(1, 1)) mg0.add_field("topographic__elevation", mg0.node_x**2 + mg0.node_y**2, at="node") _FlowDirector(mg0, "topographic__elevation") assert sorted(list(mg0.at_node.keys())) == [ "flow__sink_flag", "topographic__elevation", ] assert np.size( mg0.at_node["topographic__elevation"]) == mg0.number_of_nodes mg1 = RasterModelGrid((10, 10), spacing=(1, 1)) mg1.add_field("topographic__elevation", mg1.node_x**2 + mg1.node_y**2, at="node") _FlowDirectorToMany(mg1, "topographic__elevation") assert sorted(list(mg1.at_node.keys())) == [ "flow__sink_flag", "topographic__elevation", ] assert np.size( mg1.at_node["topographic__elevation"]) == mg1.number_of_nodes mg2 = RasterModelGrid((10, 10), spacing=(1, 1)) mg2.add_field("topographic__elevation", mg2.node_x**2 + mg2.node_y**2, at="node") _FlowDirectorToOne(mg2, "topographic__elevation") assert sorted(list(mg2.at_node.keys())) == [ "flow__link_to_receiver_node", "flow__receiver_node", "flow__sink_flag", "topographic__elevation", "topographic__steepest_slope", ] assert np.size( mg2.at_node["topographic__elevation"]) == mg2.number_of_nodes mg3 = RasterModelGrid((10, 10), spacing=(1, 1)) mg3.add_field("topographic__elevation", mg3.node_x**2 + mg3.node_y**2, at="node") FlowDirectorMFD(mg3, "topographic__elevation") assert sorted(list(mg3.at_node.keys())) == [ "flow__link_to_receiver_node", "flow__receiver_node", "flow__receiver_proportions", "flow__sink_flag", "topographic__elevation", "topographic__steepest_slope", ] assert np.size( mg3.at_node["topographic__elevation"]) == mg3.number_of_nodes mg4 = RasterModelGrid((10, 10), spacing=(1, 1)) mg4.add_field("topographic__elevation", mg4.node_x**2 + mg4.node_y**2, at="node") FlowDirectorDINF(mg4, "topographic__elevation") assert sorted(list(mg4.at_node.keys())) == [ "flow__link_to_receiver_node", "flow__receiver_node", "flow__receiver_proportions", "flow__sink_flag", "topographic__elevation", "topographic__steepest_slope", ] assert np.size( mg4.at_node["topographic__elevation"]) == mg4.number_of_nodes mg5 = RasterModelGrid((10, 10), spacing=(1, 1)) mg5.add_field("topographic__elevation", mg5.node_x**2 + mg5.node_y**2, at="node") FlowDirectorSteepest(mg5, "topographic__elevation") assert sorted(list(mg5.at_node.keys())) == [ "flow__link_to_receiver_node", "flow__receiver_node", "flow__sink_flag", "topographic__elevation", "topographic__steepest_slope", ] assert np.size( mg5.at_node["topographic__elevation"]) == mg5.number_of_nodes mg6 = RasterModelGrid((10, 10), spacing=(1, 1)) mg6.add_field("topographic__elevation", mg6.node_x**2 + mg6.node_y**2, at="node") FlowDirectorD8(mg6, "topographic__elevation") assert sorted(list(mg6.at_node.keys())) == [ "flow__link_to_receiver_node", "flow__receiver_node", "flow__sink_flag", "topographic__elevation", "topographic__steepest_slope", ] assert np.size( mg6.at_node["topographic__elevation"]) == mg6.number_of_nodes
def test_check_fields(): """Check to make sure the right fields have been created. Check that the sizes of at least one are also correct (they are at node so if one is the right size, then they all should be) """ mg0 = RasterModelGrid((10,10), spacing=(1, 1)) z0 = mg0.add_field('topographic__elevation', mg0.node_x**2 + mg0.node_y**2, at = 'node') fd0 = _FlowDirector(mg0, 'topographic__elevation') assert list(mg0.at_node.keys()) == ['topographic__elevation'] assert np.size(mg0.at_node['topographic__elevation']) == mg0.number_of_nodes mg1 = RasterModelGrid((10,10), spacing=(1, 1)) z1 = mg1.add_field('topographic__elevation', mg1.node_x**2 + mg1.node_y**2, at = 'node') fd1 = _FlowDirectorToMany(mg1, 'topographic__elevation') assert sorted(list(mg1.at_node.keys())) == [ 'flow__link_to_receiver_node', 'flow__receiver_node', 'flow__sink_flag', 'topographic__elevation', 'topographic__steepest_slope', ] assert np.size(mg1.at_node['topographic__elevation']) == mg1.number_of_nodes mg2 = RasterModelGrid((10,10), spacing=(1, 1)) z2 = mg2.add_field('topographic__elevation', mg2.node_x**2 + mg2.node_y**2, at = 'node') fd2 = _FlowDirectorToOne(mg2, 'topographic__elevation') assert sorted(list(mg2.at_node.keys())) == [ 'flow__link_to_receiver_node', 'flow__receiver_node', 'flow__sink_flag', 'topographic__elevation', 'topographic__steepest_slope', ] assert np.size(mg2.at_node['topographic__elevation']) == mg2.number_of_nodes mg3 = RasterModelGrid((10,10), spacing=(1, 1)) z3 = mg3.add_field('topographic__elevation', mg3.node_x**2 + mg3.node_y**2, at = 'node') fd3 = FlowDirectorMFD(mg3, 'topographic__elevation') assert sorted(list(mg3.at_node.keys())) == [ 'flow__link_to_receiver_node', 'flow__links_to_receiver_nodes', 'flow__receiver_node', 'flow__receiver_nodes', 'flow__receiver_proportions', 'flow__sink_flag', 'topographic__elevation', 'topographic__steepest_slope', ] assert np.size(mg3.at_node['topographic__elevation']) == mg3.number_of_nodes mg4 = RasterModelGrid((10,10), spacing=(1, 1)) z4 = mg4.add_field('topographic__elevation', mg4.node_x**2 + mg4.node_y**2, at = 'node') fd4 = FlowDirectorDINF(mg4, 'topographic__elevation') assert sorted(list(mg4.at_node.keys())) == [ 'flow__link_to_receiver_node', 'flow__links_to_receiver_nodes', 'flow__receiver_node', 'flow__receiver_nodes', 'flow__receiver_proportions', 'flow__sink_flag', 'topographic__elevation', 'topographic__steepest_slope', ] assert np.size(mg4.at_node['topographic__elevation']) == mg4.number_of_nodes mg5 = RasterModelGrid((10,10), spacing=(1, 1)) z5 = mg5.add_field('topographic__elevation', mg5.node_x**2 + mg5.node_y**2, at = 'node') fd5 = FlowDirectorSteepest(mg5, 'topographic__elevation') assert sorted(list(mg5.at_node.keys())) == [ 'flow__link_to_receiver_node', 'flow__receiver_node', 'flow__sink_flag', 'topographic__elevation', 'topographic__steepest_slope', ] assert np.size(mg5.at_node['topographic__elevation']) == mg5.number_of_nodes mg6 = RasterModelGrid((10,10), spacing=(1, 1)) z6 = mg6.add_field('topographic__elevation', mg6.node_x**2 + mg6.node_y**2, at = 'node') fd6 = FlowDirectorD8(mg6, 'topographic__elevation') assert sorted(list(mg6.at_node.keys())) == [ 'flow__link_to_receiver_node', 'flow__receiver_node', 'flow__sink_flag', 'topographic__elevation', 'topographic__steepest_slope', ] assert np.size(mg6.at_node['topographic__elevation']) == mg6.number_of_nodes