Remove all-NA variable_level column from regression ARD output

R/ard_regression.R

@@ -49,6 +49,7 @@ ard_regression.default <- function(x, tidy_fun = broom.helpers::tidy_with_broom_
       values_to = "stat"
     ) |>
     dplyr::filter(map_lgl(.data$stat, Negate(is.na))) |>
+    dplyr::select(-(cards::all_ard_variables("levels") & dplyr::where(\(x) all(is.na(x))))) |>
     dplyr::mutate(
       fmt_fn =
         lapply(

R/ard_regression_basic.R

@@ -55,5 +55,6 @@ ard_regression_basic <- function(x, tidy_fun = broom.helpers::tidy_with_broom_or
     utils::modifyList(val = rlang::dots_list(...))
 
   rlang::inject(ard_regression(x = x, tidy_fun = tidy_fun, !!!args)) |>
-    dplyr::filter(!.data$stat_name %in% stats_to_remove)
+    dplyr::filter(!.data$stat_name %in% stats_to_remove) |>
+    dplyr::select(-(cards::all_ard_variables("levels") & dplyr::where(\(x) all(is.na(x)))))
 }

tests/testthat/test-ard_regression.R

@@ -20,3 +20,7 @@ test_that("ard_regression() works", {
     "Treatment Arm"
   )
 })
+
+test_that("ard_regression() does not produce `variable_level` column where not applicable", {
+  expect_true(!"variable_level" %in% names(lm(mpg ~ hp, mtcars) |> ard_regression()))
+})

tests/testthat/test-ard_regression_basic.R

@@ -9,3 +9,7 @@ test_that("ard_regression_basic() works", {
 
   expect_snapshot(as.data.frame(ard) |> dplyr::select(-fmt_fn))
 })
+
+test_that("ard_regression_basic() does not produce `variable_level` column where not applicable", {
+  expect_true(!"variable_level" %in% names(lm(mpg ~ hp, mtcars) |> ard_regression_basic()))
+})