def test_lower_neighbors_partial(): assert _lower_neighbors(dist_mat_1, 1) == reference_lower_neighbors(dist_mat_1, 1)
def test_lower_neighbors_zero(): d = np.array(dist_mat_1) d[3][0] = 0 d[0][3] = 0 assert _lower_neighbors(d, inf) == reference_lower_neighbors(d, inf)
def test_lower_neighbors_inf(): assert _lower_neighbors(dist_mat_1, inf) == reference_lower_neighbors(dist_mat_1, inf)