def test_get_node_endpoints_one_min_rectangular(): num_of_cols = 2 num_of_rows = 3 downslope_neighbors = np.array([3, 3, 5, 5, 5, -1]) result_terminal_nodes = np.array([5, 5, 5, 5, 5, 5]) terminal_nodes = trap_analysis.get_node_endpoints_alternative(num_of_cols, num_of_rows, downslope_neighbors) assert np.array_equal(terminal_nodes, result_terminal_nodes)
def test_get_node_endpoints_four_mins_rectangular(): num_of_cols = 4 num_of_rows = 3 downslope_neighbors = np.array([1, 6, 6, 7, 5, 6, -1, -1, 9, 10, -1, -1]) result_terminal_nodes = np.array([6, 6, 6, 7, 6, 6, 6, 7, 10, 10, 10, 11]) terminal_nodes = trap_analysis.get_node_endpoints_alternative(num_of_cols, num_of_rows, downslope_neighbors) assert np.array_equal(terminal_nodes, result_terminal_nodes)
def test_get_node_endpoints_two_mins(): num_of_cols = 3 num_of_rows = 3 downslope_neighbors = np.array([-1, 0, 5, 0, 0, 8, 7, 8, -1]) result_terminal_nodes = np.array([0, 0, 8, 0, 0, 8, 8, 8, 8]) terminal_nodes = trap_analysis.get_node_endpoints_alternative(num_of_cols, num_of_rows, downslope_neighbors) assert np.array_equal(terminal_nodes, result_terminal_nodes)