Clearer error messages for invalid aesthetics

NEWS.md

@@ -8,6 +8,8 @@
 
 * `coord_map()` now can have axes on the top and right (@karawoo, #3042).
 
+* Clearer error messages for innappropriate aesthetics. (@clairemcwhite, #3060).
+
 * `geom_rug()` gains an "outside" option to allow for moving the rug tassels to outside the plot area. (@njtierney, #3085)
 
 * `geom_rug()` now works with `coord_flip()` (@has2k1, #2987).

R/layer.r

@@ -229,6 +229,15 @@ Layer <- ggproto("Layer", NULL,
     evaled <- lapply(aesthetics, rlang::eval_tidy, data = data)
     evaled <- compact(evaled)
 
+    nondata_cols <- check_nondata_cols(evaled)
+    if(length(nondata_cols) > 0){
+      msg <- paste0(
+        "Aesthetics must be valid data columns: ", nondata_cols,
+        ". Did you forget to add stat()?"
+      )
+      stop(msg, call. = FALSE)
+      }
+
     n <- nrow(data)
     if (n == 0) {
       # No data, so look at longest evaluated aesthetic
@@ -280,6 +289,17 @@ Layer <- ggproto("Layer", NULL,
     env$stat <- stat
 
     stat_data <- new_data_frame(lapply(new, rlang::eval_tidy, data, env))
+
+    # Check that all columns in aesthetic stats are valid data
+    nondata_stat_cols <- check_nondata_cols(stat_data)
+    if(length(nondata_stat_cols) > 0){
+      msg <- paste0(
+        "Aesthetics must be valid computed stats: ", nondata_stat_cols,
+        ". Did you map your stat in the wrong layer?"
+      )
+      stop(msg, call. = FALSE)
+    }
+
     names(stat_data) <- names(new)
 
     # Add any new scales, if needed

R/utilities.r

@@ -354,6 +354,14 @@ is.discrete <- function(x) {
   is.factor(x) || is.character(x) || is.logical(x)
 }
 
+# This function checks that all columns of a dataframe `x` are data and
+# returns the names of any columns that are not.
+# We define "data" as atomic types or lists, not functions or otherwise
+check_nondata_cols <- function(x) {
+  idx <- (vapply(x, function(x) rlang::is_atomic(x) || rlang::is_list(x), logical(1)))
+  names(x)[which(!idx)]
+}
+
 compact <- function(x) {
   null <- vapply(x, is.null, logical(1))
   x[!null]

tests/testthat/test-layer.r

@@ -38,6 +38,22 @@ test_that("missing aesthetics trigger informative error", {
   )
 })
 
+test_that("function aesthetics are wrapped with stat()", {
+  df <- data_frame(x = 1:10)
+  expect_error(
+    ggplot_build(ggplot(df, aes(density)) + geom_tile(stat = "density")),
+    "Aesthetics must be valid data columns:"
+  )
+})
+
+test_that("computed stats are in appropriate layer", {
+  df <- data_frame(x = 1:10)
+  expect_error(
+    ggplot_build(ggplot(df, aes(x = x, stat(density))) + geom_tile(stat = "density") + geom_point()),
+    "Aesthetics must be valid computed stats:"
+  )
+})
+
 test_that("if an aes is mapped to a function that returns NULL, it is removed", {
   df <- data_frame(x = 1:10)
   null <- function(...) NULL