R/check_stratification_columns_available.R

In vpc: Create Visual Predictive Checks

#' Check whether stratification columns are available
#' 
#' @param data `data.frame` with observation or simulation data
#' @param stratify vector of stratification columns
#' @param type either `observation` or `simulation`
check_stratification_columns_available <- function(data, stratify, type = "observation") {
  diffs <- setdiff(stratify, names(data))
  if(length(diffs) >=1) {
    stop(paste0("The following specified stratification columns were NOT found in ",type, " data: \n", 
                  paste(diffs, collapse = ",")))
  }
  return(TRUE)
}