tests/testthat/test-balancedSampling.R

In sylvainloiseau/balancedSampling: Cross-balanced sampling according to one or more qualitative variables

context("balancedSampling")

test_that("try.counterbalanced.sample.index.matrix", {
  data <- data.frame(
    A=rep(LETTERS[1:4], times=2),
    B=rep(letters[1:2], each=4)
  )
  maxtry <- 50;
  success <- FALSE;
  ntry <- 0;
  while(!success) {
    res <- try.counterbalanced.sample.index.matrix(data, c("A", "B"), c(4, 2), c(2, 4), 8, 2, 4);
    ntry <- ntry + 1
    if (ntry == maxtry) {
      break;
    }
    success <- res[1]
  }
})

test_that("balancedSampling: wrong arguments 1", {
  expect_error(counterbalanced.sample.index.matrix(1, 3, 4));
})

test_that("balancedSampling: wrong arguments 2", {
  expect_error(counterbalanced.sample.index.matrix(1, 1, 4));
})

test_that("balancedSampling: wrong arguments 3", {
  expect_error(counterbalanced.sample.index.matrix(5, 5, 4));
})

test_that("balancedSampling 1", {
  data <- data.frame(
    A=rep(LETTERS[1:4], times=2),
    B=rep(letters[1:2], each=4)
  )

  res <- balancedSampling(data, vars=c("A", "B"), sample.nb=2, c(2,2))

  expect_equal(length(res), 2);

  sample1.A <- table(res[[1]][,"A"])
  expect_true(all(sample1.A == 1))

  sample1.B <- table(res[[1]][,"B"])
  expect_true(all(sample1.B == 2))

  sample2.A <- table(res[[2]][,"A"])
  expect_true(all(sample2.A == 1))

  sample2.B <- table(res[[2]][,"B"])
  expect_true(all(sample2.B == 2))
})

test_that("balancedSampling 2", {
  data <- expand.grid(A=factor(as.character(1:60)), B=factor(letters[1:3]));
  res <- balancedSampling(data, c("A", "B"), 3);
  expect_equal(length(res), 3)
})