print fix and tests

Author: dsweber2

R/step_adjust_latency.R

@@ -280,11 +280,6 @@ bake.step_adjust_latency <- function(object, new_data, ...) {
 #' @export
 print.step_adjust_latency <-
   function(x, width = max(20, options$width - 35), ...) {
-    if (length(x$terms) == 0) {
-      terms <- "all previous predictors"
-    } else {
-      terms <- x$terms
-    }
     if (!is.null(x$forecast_date)) {
       conj <- "with forecast date"
       extra_text <- x$forecast_date
@@ -299,9 +294,51 @@ print.step_adjust_latency <-
       conj <- "with latency"
       extra_text <- "set at train time"
     }
-    print_epi_step(terms, terms, x$trained, x$method,
-      conjunction = conj,
-      extra_text = extra_text
+    # what follows is a somewhat modified version of print_epi_step, since the case of no arguments for adjust_latency means apply to all relevant columns, and not none of them
+    theme_div_id <- cli::cli_div(
+      theme = list(.pkg = list(`vec-trunc` = Inf, `vec-last` = ", "))
+    )
+    # this is a slightly modified copy of
+    title <- trimws(x$method)
+    trained_text <- dplyr::if_else(x$trained, "Trained", "")
+    vline_seperator <- dplyr::if_else(trained_text == "", "", "|")
+    comma_seperator <- dplyr::if_else(
+      trained_text != "", true = ",", false = ""
     )
+    extra_text <- recipes::format_ch_vec(extra_text)
+    width_title <- nchar(paste0(
+      "* ", title, ":", " ", conj, " ", extra_text, " ", vline_seperator,
+      " ", trained_text, " "
+    ))
+    width_diff <- cli::console_width() * 1 - width_title
+    if (x$trained) {
+      elements <- x$columns
+    } else {
+      if (length(x$terms) == 0) {
+        elements <- "all previous predictors"
+      } else {
+        elements <- lapply(x$terms, function(x) {
+          rlang::expr_deparse(rlang::quo_get_expr(x), width = Inf)
+        })
+        elements <- vctrs::list_unchop(elements, ptype = character())
+      }
+    }
+
+    element_print_lengths <- cumsum(nchar(elements)) +
+      c(0L, cumsum(rep(2L, length(elements) - 1))) +
+      c(rep(5L, length(elements) - 1), 0L)
+    first_line <- which(width_diff >= element_print_lengths)
+    first_line <- unname(first_line)
+    first_line <- ifelse(
+      test = identical(first_line, integer(0)),
+      yes = length(element_print_lengths),
+      no = max(first_line)
+    )
+    more_dots <- ifelse(first_line == length(elements), "", ", ...")
+    cli::cli_bullets(
+      c("\n    {title}: \\\n    {.pkg {cli::cli_vec(elements[seq_len(first_line)])}}\\\n    {more_dots} \\\n    {conj} \\\n    {.pkg {extra_text}} \\\n    {vline_seperator} \\\n    {.emph {trained_text}}")
+    )
+
+    cli::cli_end(theme_div_id)
     invisible(x)
   }

man/epi_shift_single.Rd

@@ -0,0 +1,39 @@
+# printing step_adjust_latency results in expected output
+
+    Code
+      r5
+    Message
+      
+      -- Epi Recipe ------------------------------------------------------------------
+      
+      -- Inputs 
+      Number of variables by role
+      raw:        2
+      geo_value:  1
+      time_value: 1
+      
+      -- Operations 
+      1. Lagging: death_rate by 0, 6, 11
+      2. Lagging: case_rate by 1, 5
+      3. Leading: death_rate by 7
+      4. extend_lags: case_rate with latency set at train time
+
+---
+
+    Code
+      r
+    Message
+      
+      -- Epi Recipe ------------------------------------------------------------------
+      
+      -- Inputs 
+      Number of variables by role
+      raw:        2
+      geo_value:  1
+      time_value: 1
+      
+      -- Operations 
+      1. Leading: death_rate by 7
+      2. extend_ahead: all previous predictors with latency set at train time
+      3. Lagging: death_rate by 0, 7, 14
+

tests/testthat/_snaps/step_adjust_latency.md

@@ -245,4 +245,9 @@ test_that("printing step_adjust_latency results in expected output", {
     step_epi_ahead(death_rate, ahead = ahead) %>%
     step_adjust_latency(case_rate, method = "extend_lags")
   expect_snapshot(r5)
+  r <- epi_recipe(case_death_rate_subset) %>%
+    step_epi_ahead(death_rate, ahead = 7) %>%
+    step_adjust_latency(method = "extend_ahead") %>%
+    step_epi_lag(death_rate, lag = c(0, 7, 14))
+  expect_snapshot(r)
 })