tests/testthat/helper-tunable.R

In parsnip: A Common API to Modeling and Analysis Functions

# Helper function for displaying tunable call_info in snapshot tests
# This expands and displays all call_info elements in a format suitable for
# snapshot testing, capturing pkg, fun, range, and values.

display_tunable_call_info <- function(model_spec) {
  res <- tunable(model_spec)

  if (nrow(res) == 0) {
    cat("No tunable parameters.\n")
    return(invisible(res))
  }

  for (i in seq_len(nrow(res))) {
    param_name <- res$name[i]
    call_info <- res$call_info[[i]]
    component_id <- res$component_id[i]

    # Build the info string
    info_parts <- character(0)

    if (!is.null(call_info$pkg)) {
      info_parts <- c(info_parts, glue::glue("pkg: {call_info$pkg}"))
    }
    if (!is.null(call_info$fun)) {
      info_parts <- c(info_parts, glue::glue("fun: {call_info$fun}"))
    }
    if (!is.null(call_info$range)) {
      range_str <- glue::glue(
        "c({glue::glue_collapse(format(call_info$range, nsmall = 2), sep = ', ')})"
      )
      info_parts <- c(info_parts, glue::glue("range: {range_str}"))
    }
    if (!is.null(call_info$values)) {
      values_str <- glue::glue_collapse(call_info$values, sep = ", ")
      info_parts <- c(info_parts, glue::glue("values: {values_str}"))
    }

    info_str <- glue::glue_collapse(info_parts, sep = ", ")

    # Pad the parameter name for alignment
    padded_name <- format(param_name, width = 25)

    cat(glue::glue("{padded_name} | {info_str} | {component_id}\n\n"))
  }

  invisible(res)
}