tests/testthat/test-lm2.R

In TKUmich96/biostat625hw4: Fit a linear regression and return related summary statistics

test_that("lm2: test default na.handle & res_display work", {
  expect_output(lm2(formula = Rape ~ Murder+Assault+UrbanPop, data = USArrests))
})

test_that("lm2: test na.fail na.handle works", {
  expect_error(lm2(formula = Temp ~ Wind + Solar.R, data = airquality, na.handle = "na.fail"))
})

test_that("lm2: test na.omit na.handle works", {
  expect_output(lm2(formula = Temp ~ Wind + Solar.R, data = airquality, na.handle = "na.omit"))
})

test_that("lm2: missing.N Correct", {
  expect_equal(lm2(formula = Rape ~ Murder+Assault+UrbanPop, data = USArrests, res_display = F)$missing.N, 0)
})

test_that("lm2: Residual Degrees of Freedom Correct", {
  expect_equal(lm2(formula = Rape ~ Murder+Assault+UrbanPop, data = USArrests, res_display = F)$df, 46)
})

test_that("lm2: Coefficent Match LM's output", {
  expect_equal(lm2(formula = Rape ~ Murder+Assault+UrbanPop, data = USArrests, res_display = F)$coefficients, lm(formula = Rape ~ Murder+Assault+UrbanPop, data = USArrests)$coefficients)
})

test_that("lm2: Residuals Match LM's output", {
  expect_equal(lm2(formula = Rape ~ Murder+Assault+UrbanPop, data = USArrests, res_display = F)$residuals, lm(formula = Rape ~ Murder+Assault+UrbanPop, data = USArrests)$residuals)
})

test_that("lm2:Fitted values Match LM's output", {
  expect_equal(lm2(formula = Rape ~ Murder+Assault+UrbanPop, data = USArrests, res_display = F)$fitted.values, lm(formula = Rape ~ Murder+Assault+UrbanPop, data = USArrests)$fitted.values)
})

test_that("lm2: Check that input for formula argument is required", {
  expect_error(lm2(data = USArrest))
})

test_that("lm2: Check that input for data argument is required", {
  expect_error(lm2(formula = Rape ~ Murder+Assault+UrbanPop))
})