⚡ Refactor rpois_dispersion()

heavywatal · Jun 2, 2024 · 7988e28 · 7988e28
1 parent b1740c1
commit 7988e28
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/R/dispersion.R b/R/dispersion.R
@@ -38,9 +38,17 @@ mutate_chisq = function(.data, ...) {
 #' @rdname dispersion
 #' @export
 rpois_dispersion = function(n, lambda, nrep = 1L) {
-  seq_len(nrep) |>
-    purrr::map(\(.i) summary_row(stats::rpois(n, lambda))) |>
-    purrr::list_rbind(names_to = "repl") |>
+  list(
+    repl = rep(seq_len(nrep), each = n),
+    x = stats::rpois(n * nrep, lambda)
+  ) |>
+    tibble::new_tibble() |>
+    dplyr::summarize(
+      nsam = length(!!as.name("x")),
+      mean = mean(!!as.name("x")),
+      var = stats::var(!!as.name("x")),
+      .by = !!as.name("repl")
+    ) |>
     mutate_chisq()
 }