test

Author: strengejacke

tests/testthat/test-estimate_expectation.R

@@ -43,6 +43,9 @@ test_that("estimate_expectation - data-grid", {
     colnames(estim),
     c("Predicted", "SE", "CI_low", "CI_high", "Residuals")
   )
+  estim2 <- estimate_expectation(m, by = "cyl")
+  expect_equal(estim$cyl, esimt2$cyl, tolerance = 1e-4)
+  expect_equal(estim$Predicted, esimt2$Predicted, tolerance = 1e-4)
 
   m <- lm(mpg ~ cyl, data = mtcars)
   estim <- estimate_expectation(m, data = "grid")
@@ -60,3 +63,12 @@ test_that("estimate_expectation - data-grid", {
     c("cyl", "Predicted", "SE", "CI_low", "CI_high")
   )
 })
+
+
+test_that("estimate_expectation - error", {
+  m <- lm(mpg ~ cyl, data = mtcars)
+  expect_error(
+    estimate_expectation(m, data = mtcars, by = "cyl"),
+    regex = "You can only"
+  )
+})