tests for percentile estimation function

Author: ernestguevarra

R/boot_bw_estimate.R

@@ -25,22 +25,27 @@ boot_bw_estimate <- function(boot_df) {
   est <- lapply(
     X = boot_df,
     FUN = boot_percentile
-  ) |>
+  )
+  
+  ## Structure list names ----
+  if (nrow(est[[1]]) == 1) {
+    names(est) <- paste(
+      names(est),
+      lapply(X = est, FUN = row.names) |> unlist(), 
+      sep = "."
+    )
+  }
+  
+  ## Flatten list ----
+  est <- est |>
     do.call(rbind, args = _)
 
   ## Re-structure results ----
-  if (all(grepl("\\.", row.names(est)))) {
-    est <- data.frame(
-      strata = gsub("\\.[^\\.]{1,}", "", row.names(est)),
-      indicator = gsub("[^\\.]{1,}\\.", "", row.names(est)),
-      est
-    )
-  } else {
-    est <- data.frame(
-      indicator = row.names(est),
-      est
-    )
-  }
+  est <- data.frame(
+    strata = gsub("\\.[^\\.]{1,}", "", row.names(est)),
+    indicator = gsub("[^\\.]{1,}\\.", "", row.names(est)),
+    est
+  )
 
   ## Tidy up row names ----
   row.names(est) <- NULL

R/utils.R

@@ -118,6 +118,7 @@ tidy_boot <- function(boot, w, strata, outputColumns) {
   boot
 }
 
+
 #'
 #' Boot estimate
 #' 
@@ -135,14 +136,11 @@ boot_percentile <- function(boot_df) {
       do.call(rbind, args = _) |>
       data.frame()
 
-    # est <- data.frame(
-    #   indicator = row.names(est),
-    #   est
-    # )
-
     names(est) <- c("est", "lcl", "ucl")
   } else {
-    est <- quantile(x = boot_df, probs = c(0.5, 0.025, 0.975), na.rm = TRUE) |>
+    est <- stats::quantile(
+      x = boot_df, probs = c(0.5, 0.025, 0.975), na.rm = TRUE
+    ) |>
       rbind() |>
       data.frame()
 

tests/testthat/test-boot_bw_estimate.R

@@ -0,0 +1,17 @@
+# Tests for boot_bw_estimate function/s ----------------------------------------
+
+test_that("boot_bw_estimate works as expected", {
+  boot_df <- boot_bw(
+    x = indicatorsHH, w = villageData, statistic = bootClassic,
+    params = "anc1", replicates = 99
+  )
+
+  expect_s3_class(boot_bw_estimate(boot_df), "data.frame")
+
+  boot_df <- boot_bw(
+    x = indicatorsHH, w = villageData, statistic = bootClassic,
+    params = "anc1", replicates = 99, strata = "region"
+  )
+
+  expect_s3_class(boot_bw_estimate(boot_df), "data.frame")
+})