Implement expect_all_true() and expect_all_false()

Author: hadley

R/expect-all.R

@@ -2,19 +2,44 @@ expect_all_equal <- function(object, expected) {
   act <- quasi_label(enquo(object))
   exp <- quasi_label(enquo(expected))
 
-  check_vector(act$val, error_arg = "object")
+  expect_all_equal_(act, exp)
+  invisible(act$val)
+}
+
+expect_all_true <- function(object) {
+  act <- quasi_label(enquo(object))
+  exp <- labelled_value(TRUE, "TRUE")
+
+  expect_all_equal_(act, exp)
+  invisible(act$val)
+}
+
+expect_all_false <- function(object) {
+  act <- quasi_label(enquo(object))
+  exp <- labelled_value(FALSE, "FALSE")
+
+  expect_all_equal_(act, exp)
+  invisible(act$val)
+}
+
+
+expect_all_equal_ <- function(act, exp, trace_env = caller_env()) {
+  check_vector(act$val, error_call = trace_env, error_arg = "object")
   if (length(act$val) == 0) {
-    cli::cli_abort("{.arg object} must not be empty.")
+    cli::cli_abort("{.arg object} must not be empty.", call = trace_env)
   }
 
-  check_vector(exp$val, error_arg = "expected")
+  check_vector(exp$val, error_call = trace_env, error_arg = "expected")
   if (length(exp$val) != 1) {
-    cli::cli_abort("{.arg expected} must be length 1.")
+    cli::cli_abort("{.arg expected} must be length 1.", call = trace_env)
   }
 
   exp$val <- rep(exp$val, length(act$val))
   names(exp$val) <- names(act$val)
-  expect_waldo_equal_("Expected every element of %s to equal %s.", act, exp)
-
-  invisible(act$val)
+  expect_waldo_equal_(
+    "Expected every element of %s to equal %s.",
+    act,
+    exp,
+    trace_env = trace_env
+  )
 }

tests/testthat/test-expect-all.R

@@ -24,3 +24,10 @@ test_that("truncates very long differences", {
   x <- rep(TRUE, 10)
   expect_snapshot_failure(expect_all_equal(x, FALSE))
 })
+
+test_that("has TRUE and FALSE helpers", {
+  x1 <- rep(TRUE, 10)
+  x2 <- rep(FALSE, 10)
+  expect_success(expect_all_true(x1))
+  expect_success(expect_all_false(x2))
+})