tests/testthat/test-parameters_type-2.R

In parameters: Processing of Model Parameters

dat <- iris
m <- lm(Sepal.Length ~ Species, data = dat)
test_that("parameters_type default contrasts", {
  p_type <- parameters_type(m)
  expect_equal(p_type$Type, c("intercept", "factor", "factor"))
  expect_equal(p_type$Level, c(NA, "versicolor", "virginica"))
})


dat <- iris
dat$Species <- as.ordered(dat$Species)
m <- lm(Sepal.Length ~ Species, data = dat)
test_that("parameters_type ordered factor", {
  p_type <- parameters_type(m)
  expect_equal(p_type$Type, c("intercept", "ordered", "ordered"))
  expect_equal(p_type$Level, c(NA, "[linear]", "[quadratic]"))
})


dat <- iris
dat$Species <- as.ordered(dat$Species)
contrasts(dat$Species) <- contr.treatment(3)
m <- lm(Sepal.Length ~ Species, data = dat)
test_that("parameters_type ordered factor", {
  p_type <- parameters_type(m)
  expect_equal(p_type$Type, c("intercept", "factor", "factor"))
  expect_equal(p_type$Level, c(NA, "2", "3"))
})


dat <- iris
contrasts(dat$Species) <- contr.poly(3)
m <- lm(Sepal.Length ~ Species, data = dat)
test_that("parameters_type poly contrasts", {
  p_type <- parameters_type(m)
  expect_equal(p_type$Type, c("intercept", "factor", "factor"))
  expect_equal(p_type$Level, c(NA, ".L", ".Q"))
})


dat <- iris
contrasts(dat$Species) <- contr.treatment(3)
m <- lm(Sepal.Length ~ Species, data = dat)
test_that("parameters_type treatment contrasts", {
  p_type <- parameters_type(m)
  expect_equal(p_type$Type, c("intercept", "factor", "factor"))
  expect_equal(p_type$Level, c(NA, "2", "3"))
})


dat <- iris
contrasts(dat$Species) <- contr.sum(3)
m <- lm(Sepal.Length ~ Species, data = dat)
test_that("parameters_type sum contrasts", {
  p_type <- parameters_type(m)
  expect_equal(p_type$Type, c("intercept", "factor", "factor"))
  expect_equal(p_type$Level, c(NA, "1", "2"))
})


dat <- iris
contrasts(dat$Species) <- contr.helmert(3)
m <- lm(Sepal.Length ~ Species, data = dat)
test_that("parameters_type helmert contrasts", {
  p_type <- parameters_type(m)
  expect_equal(p_type$Type, c("intercept", "factor", "factor"))
  expect_equal(p_type$Level, c(NA, "1", "2"))
})


dat <- iris
contrasts(dat$Species) <- contr.SAS(3)
m <- lm(Sepal.Length ~ Species, data = dat)
test_that("parameters_type SAS contrasts", {
  p_type <- parameters_type(m)
  expect_equal(p_type$Type, c("intercept", "factor", "factor"))
  expect_equal(p_type$Level, c(NA, "1", "2"))
})