tests/testthat/test-select_cont.R

In saeSim: Simulation Tools for Small Area Estimation

context("select_cont")
test_that(desc = "select_cont", {
  dat1 <- sim_base(base_id(nDomains=5, nUnits = 2)) %>% sim_gen_e() %>% as.data.frame
  expect_equal(sum(select_cont(dat1, 1L, "unit", "idD", TRUE)$e == 0), 5)
  expect_equal(sum(select_cont(dat1, 1L, "unit", "idD", FALSE)$e == 0), 5)
  expect_equal(sum(select_cont(dat1, 1L, "area", "idD", TRUE)$e == 0), 8)
  expect_equal(sum(select_cont(dat1, 1L, "area", "idD", FALSE)$e == 0), 8)
  expect_equal(sum(select_cont(dat1, 1L, "unit", NULL, TRUE)$e == 0), 9)
  expect_equal(sum(select_cont(dat1, 1L, "unit", NULL, FALSE)$e == 0), 9)
  
  expect_true(sum(select_cont(dat1, 0.5, "unit", "idD", TRUE)$e == 0) > 0)
  expect_true(sum(select_cont(dat1, 0.5, "area", "idD", TRUE)$e == 0) > 0)
  expect_true(sum(select_cont(dat1, 0.5, "unit", NULL, TRUE)$e == 0) > 0)
  
  dat2 <- sim_base(base_id(nDomains = 5, nUnits = 1:5)) %>% sim_gen_e() %>% as.data.frame
  expect_equal(sum(select_cont(dat2, 1L, "unit", "idD", TRUE)$e == 0), 10)
  expect_equal(sum(select_cont(dat2, 1L, "unit", "idD", FALSE)$e == 0), 10)
  expect_equal(sum(select_cont(dat2, 1L, "area", "idD", TRUE)$e == 0), 10)
  expect_equal(sum(select_cont(dat2, 1L, "unit", NULL, TRUE)$e == 0), 14)
  expect_equal(sum(select_cont(dat2, 1L, "unit", NULL, FALSE)$e == 0), 14)
  
  numberOfContObs <- (select_cont(dat2, nCont = c(0, 0, 0, 0.2, 1), "unit", "idD", TRUE)$idC %>% sum)
  (numberOfContObs >= 5 & numberOfContObs <= 9) %>%
    expect_true
  
  testthat::expect_equal(sum(select_cont(dat2, c(1, 4), "area", "idD", TRUE)$e == 0), 10)
  
})