equivalence_test, estimate_density

[skip]

R/equivalence_test.R

@@ -145,8 +145,24 @@ equivalence_test.numeric <- function(x, range = "default", ci = 0.95, verbose =
 
 
 #' @rdname equivalence_test
+#' @inheritParams p_direction
 #' @export
-equivalence_test.data.frame <- function(x, range = "default", ci = 0.95, verbose = TRUE, ...) {
+equivalence_test.data.frame <- function(x, range = "default", ci = 0.95, rvar_col = NULL, verbose = TRUE, ...) {
+
+  obj_name <- insight::safe_deparse_symbol(substitute(x))
+
+  if (length(x_rvar <- .possibly_extract_rvar_col(x, rvar_col)) > 0L) {
+    cl <- match.call()
+    cl[[1]] <- bayestestR::equivalence_test
+    cl$x <- x_rvar
+    cl$rvar_col <- NULL
+    out <- eval.parent(cl)
+
+    attr(out, "object_name") <- sprintf('%s[["%s"]]', obj_name, rvar_col)
+
+    return(.append_datagrid(out, x))
+  }
+
   l <- insight::compact_list(lapply(
     x,
     equivalence_test,
@@ -163,7 +179,7 @@ equivalence_test.data.frame <- function(x, range = "default", ci = 0.95, verbose
   )
   row.names(out) <- NULL
 
-  attr(out, "object_name") <- insight::safe_deparse_symbol(substitute(x))
+  attr(out, "object_name") <- obj_name
   class(out) <- unique(c("equivalence_test", "see_equivalence_test_df", class(out)))
 
   out

R/estimate_density.R

@@ -221,6 +221,7 @@ estimate_density.numeric <- function(x,
 
 
 #' @rdname estimate_density
+#' @inheritParams p_direction
 #' @export
 estimate_density.data.frame <- function(x,
                                         method = "kernel",
@@ -232,7 +233,29 @@ estimate_density.data.frame <- function(x,
                                         select = NULL,
                                         by = NULL,
                                         at = NULL,
+                                        rvar_col = NULL,
                                         ...) {
+
+  if (length(x_rvar <- .possibly_extract_rvar_col(x, rvar_col)) > 0L) {
+    cl <- match.call()
+    cl[[1]] <- bayestestR::estimate_density
+    cl$x <- x_rvar
+    cl$rvar_col <- NULL
+    out <- eval.parent(cl)
+
+    obj_name <- insight::safe_deparse_symbol(substitute(x))
+    attr(out, "object_name") <- sprintf('%s[["%s"]]', obj_name, rvar_col)
+
+    # This doesn't use .append_datagrid because we get a non-grid output
+    dgrid <- x[,vapply(x, function(col) !inherits(col, "rvar"), FUN.VALUE = logical(1)), drop = FALSE]
+    dgrid$Parameter <- unique(out$Parameter)
+    out <- datawizard::data_join(dgrid, out, by = "Parameter")
+    out$Parameter <- NULL
+    class(out) <- .set_density_class(out)
+    return(out)
+  }
+
+
   # Sanity
   if (!is.null(at)) {
     insight::format_warning(paste0(