tests/testthat/test-data.R

In mbannick/RoboCar:

test_that("Data", {

  df <- data.frame(treatment=rbinom(10, size=1, prob=0.5),
                   response=rnorm(10),
                   cov_1=rnorm(10),
                   cov_2=rnorm(10),
                   cov_3=as.factor(rbinom(10, size=1, prob=0.5)),
                   s_1=rbinom(10, size=1, prob=0.5),
                   s_2=rbinom(10, size=1, prob=0.5))
  df$treatment <- as.factor(df$treatment)
  
  data <- .make.data(df, "RoboDataLinear",
                      treat_col="treatment",
                      response_col="response",
                      covariate_cols=c("cov_1", "cov_2", "cov_3"),
                      strata_cols=c("s_1", "s_2"))
  expect_equal(class(data), "RoboDataLinear")
  
  validate(data)
})

test_that("Data with missing attributes", {
  
  df <- data.frame(treatment=rbinom(10, size=1, prob=0.5),
                   response=rnorm(10),
                   cov_1=rnorm(10),
                   cov_2=rnorm(10),
                   s_1=rbinom(10, size=1, prob=0.5),
                   s_2=rbinom(10, size=1, prob=0.5))
  df$treatment <- as.factor(df$treatment)
  
  data <- .make.data(df, "RoboDataLinear",
                      response_col="response",
                      covariate_cols=c("cov_1", "cov_2"),
                      strata_cols=c("s_1", "s_2"))
  expect_error(validate(data))
  
})

test_that("Data with non-numeric response", {
  
  df <- data.frame(treatment=rbinom(10, size=1, prob=0.5),
                   response=rep(c("x", "y"), each=5),
                   cov_1=rnorm(10),
                   cov_2=rnorm(10),
                   s_1=rbinom(10, size=1, prob=0.5),
                   s_2=rbinom(10, size=1, prob=0.5))
  df$treatment <- as.factor(df$treatment)
  
  data <- .make.data(df, "RoboDataLinear",
                      treat_col="treatment",
                      response_col="response",
                      covariate_cols=c("cov_1", "cov_2"),
                      strata_cols=c("s_1", "s_2"))
  expect_error(validate(data))
  
})