R/sanitize_numderiv.R

In marginaleffects: Predictions, Comparisons, Slopes, Marginal Means, and Hypothesis Tests

sanitize_numderiv <- function(numderiv) {
    checkmate::assert(
        checkmate::check_choice(numderiv, c("richardson", "fdforward", "fdcenter")),
        checkmate::check_list(numderiv, min.len = 1)
    )

    if (isTRUE(checkmate::check_string(numderiv))) {
        numderiv <- list(numderiv)
    }

    if (length(numderiv) > 1) {
        if (numderiv[[1]] %in% c("fdforward", "fdcenter")) {
            if (any(!names(numderiv)[2:length(numderiv)] %in% "eps")) {
                stop("The only valid argument for this numeric differentiation method is `eps`.")
            }
        } else if (numderiv[[1]] == "richardson") {
            valid <- c("eps", "d", "zero_tol", "size", "r", "v")
            if (any(!names(numderiv)[2:length(numderiv)] %in% valid)) {
                stop(sprintf("The only valid arguments for this numeric differentiation method are: %s. See `?numDeriv::grad` for details.", paste(valid, collapse = ", ")), call. = FALSE)
            }
        }
    }

    return(numderiv)
}