tests/testthat/test-viraltab.R

In viralmodels: Viral Load and CD4 Lymphocytes Regression Models

test_that("viraltab() works", {
  library(dplyr)
  library(magrittr)
  library(baguette)
  library(kernlab)
  library(kknn)
  library(ranger)
  library(rules)
  library(glmnet)
  # Define the function to impute values in the undetectable range
  impute_undetectable <- function(column) {
    set.seed(123)
    ifelse(column <= 40,
           rexp(sum(column <= 40), rate = 1/13) + 1,
           column)
  }
  # Apply the function to all vl columns using purrr's map_dfc
  library(viraldomain)
  data("viral", package = "viraldomain")
  viral_imputed <- viral %>%
    mutate(across(starts_with("vl"), ~impute_undetectable(.x)))
  traindata <- viral_imputed
  semilla <- 1501
  target <- "cd_2022"
  viralvars <- c("vl_2019", "vl_2021", "vl_2022")
  logbase <- 10
  pliegues <- 2
  repeticiones <- 1
  rejilla <- 1
  set.seed(123)
  expect_snapshot(viraltab(traindata, semilla, target, viralvars, logbase, 
                           pliegues, repeticiones, rejilla, rank_output = TRUE))
})