daft.functions.date_add#

def date_add(expr: Expression, days: Expression) -> Expression:
    """Adds a number of days to a date.

    Args:
        expr: A date expression.
        days: An integer expression representing the number of days to add.

    Returns:
        Expression: a Date expression.

    Examples:
        >>> import daft
        >>> from daft.functions import date_add
        >>> df = daft.from_pydict({"d": ["2021-01-01", "2021-06-15"], "n": [10, 5]})
        >>> df = df.with_column("d", df["d"].cast(daft.DataType.date()))
        >>> df = df.with_column("result", date_add(df["d"], df["n"]))
        >>> df.schema()["result"].dtype == daft.DataType.date()
        True
    """
    return Expression._call_builtin_scalar_fn("date_add", expr, days)