tests/testthat/test-rep_sample_n.R

In andrewpbray/infer: Tidy Statistical Inference

context("rep_sample_n")

N <- 5
population <- tibble::data_frame(
  ball_ID = 1:N,
  color = factor(c(rep("red", 3), rep("white", N - 3)))
)

test_that("rep_sample_n works", {
  expect_silent(population %>% rep_sample_n(size = 2, reps = 10))
  expect_error(
    population %>%
      rep_sample_n(size = 2, reps = 10, prob = rep(x = 1/5, times = 100))
  )
  expect_error(
    population %>%
      rep_sample_n(size = 2, reps = 10, prob = c(1/2, 1/2))
  )
  expect_error(
    population %>%
      rep_sample_n(size = 2, reps = 10, prob = c(0.25, 1/5, 1/5, 1/5, 0.15))
  )
  
  test_rep <- population %>% rep_sample_n(size = 2, reps = 10)
  expect_equal(c("replicate", names(population)), names(test_rep))
})