tests/testthat/test-dstar.R

In zabore/ethet: Functions to Study Etiologic Heterogeneity

context("test-dstar")

test_that("dstar produces expected value", {
  expect_equal(
    round(dstar("subtype", 4, list("x1", "x2", "x3"), subtype_data[subtype_data$subtype > 0, ]), 3),
    0.402
  )
})


test_that("dstar prints message when label variable is not numeric", {
  expect_error(
    dstar("subtype_name", 4, list("x1", "x2", "x3"), subtype_data[subtype_data$subtype > 0, ]),
    "The argument to label must be numeric or integer. Arguments of type character and factor are not supported, please see the documentation.",
    fixed = TRUE
  )
})


test_that("dstar prints message when subtype variable doesn't start with 1", {
  expect_error(
    dstar("subtype", 4, list("x1", "x2", "x3"), subtype_data[subtype_data$subtype > 1, ]),
    "The argument to label should start with 1. Cases should be labeled with subtypes 1 through M, the total number of subtypes.",
    fixed = TRUE
  )
})


test_that("dstar prints message when there are fewer than 2 subtype levels", {
  expect_error(
    dstar("subtype", 1, list("x1", "x2", "x3"), subtype_data[subtype_data$subtype > 0, ]),
    "The argument to M, the total number of subtypes, must be a numeric value >=2.",
    fixed = TRUE
  )
})


test_that("dstar prints message when M is not equal to the number of subtypes in the data", {
  expect_error(
    dstar("subtype", 2, list("x1", "x2", "x3"), subtype_data[subtype_data$subtype > 0, ]),
    "M is not equal to the number of levels in the variable supplied to label. Please make sure M reflects the number of subtypes in the data.",
    fixed = TRUE
  )
})