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)
