tests/testthat/test-eval_cv.R

In ncullen93/abaR: Automated Biomarker Analysis

test_that("eval_cv works", {
  data <- adnimerge %>% dplyr::filter(VISCODE == 'bl')
  model <- aba_model() %>%
    set_data(data) %>%
    set_groups(everyone()) %>%
    set_outcomes(ConvertedToAlzheimers, CSF_ABETA_STATUS_bl) %>%
    set_predictors(
      PLASMA_ABETA_bl, PLASMA_PTAU181_bl, PLASMA_NFL_bl,
      c(PLASMA_ABETA_bl, PLASMA_PTAU181_bl, PLASMA_NFL_bl)
    ) %>%
    set_stats('lm') %>%
    set_evals(
      eval_cv(nfolds = 3, ntrials=5)
    )

  expect_error(
    model <- model %>% fit(),
    NA
  )

  expect_error(
    s <- model %>% summary(),
    NA
  )
})


test_that("eval_cv with glm works", {
  data <- adnimerge %>% dplyr::filter(VISCODE == 'bl')
  model <- aba_model() %>%
    set_data(data) %>%
    set_groups(everyone()) %>%
    set_outcomes(ConvertedToAlzheimers) %>%
    set_predictors(
      PLASMA_ABETA_bl, PLASMA_PTAU181_bl, PLASMA_NFL_bl,
      c(PLASMA_ABETA_bl, PLASMA_PTAU181_bl, PLASMA_NFL_bl)
    ) %>%
    set_stats('glm') %>%
    set_evals(
      eval_cv(nfolds = 3, ntrials=5)
    )

  expect_error(
    model <- model %>% fit(),
    NA
  )

  expect_error(
    s <- model %>% summary(),
    NA
  )
})