tests/testthat/test_define.R

In simpr: Flexible 'Tidyverse'-Friendly Simulations

context("simpr::define")
library(tibble)

## define can handle lists which can contain multiple matrices, etc.

test_that("define() can handle lists", {

  set.seed(100, kind = "L'Ecuyer-CMRG")
  possible_S = list(independent = diag(2), correlated = diag(2) + 2)
  possible_n = 10
  mat_refs = furrr::future_map(possible_S, ~ MASS::mvrnorm(possible_n, rep(0, 2), Sigma = .),
                               .options = furrr::furrr_options(seed = TRUE))

  set.seed(100, kind = "L'Ecuyer-CMRG")
  meta_list_out <- specify(x = ~ MASS::mvrnorm(n, rep(0, 2), Sigma = S)) %>%
    define(n = 10,
           S = possible_S) %>%
    generate(1)

  expect_equivalent(mat_refs, purrr::map(meta_list_out$sim, as.matrix))

})

## Define can handle functions as elements in a list
test_that("define can handle functions", {

  function_list = list(normal = rnorm,
                       lognormal = rlnorm)

  set.seed(100, kind = "L'Ecuyer-CMRG")
  out = specify(y = ~ myfun(10)) %>%
    define(myfun = function_list) %>%
    generate(1)

  set.seed(100, kind = "L'Ecuyer-CMRG")
  fun_refs = furrr::future_map(function_list, ~ .(10),
                               .options = furrr::furrr_options(seed = TRUE))

  expect_equivalent(fun_refs$normal, out$sim[[1]]$y)
  expect_equivalent(fun_refs$lognormal, out$sim[[2]]$y)

})

## Define can take an extra ".list" element and
## add that to the "..." elements

test_that("define with combo of ... and .list", {
  possible_S = list(independent = diag(2), correlated = diag(2) + 2)
  possible_n = c(10, 20)

  set.seed(100, kind = "L'Ecuyer-CMRG")
  meta_list_out_dots <- specify(x = ~ MASS::mvrnorm(n, rep(0, 2), Sigma = S)) %>%
    define(n = possible_n,
           S = possible_S) %>%
    generate(1)

  set.seed(100, kind = "L'Ecuyer-CMRG")
  meta_list_out_combo = specify(x = ~ MASS::mvrnorm(n, rep(0, 2), Sigma = S)) %>%
    define(n = possible_n,
           .list = list(S = possible_S)) %>%
    generate(1)

  expect_equivalent(meta_list_out_dots, meta_list_out_combo)
})