feat: add a epi_df attribute that avoids the decay. See #467

cmu-delphi · Jun 20, 2024 · b7f36e0 · b7f36e0
1 parent 7fd2119
commit b7f36e0
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/R/methods-epi_df.R b/R/methods-epi_df.R
@@ -12,6 +12,7 @@ as_tibble.epi_df <- function(x, ...) {
   # Decaying drops the class and metadata. `as_tibble.grouped_df` drops the
   # grouping and should be called by `NextMethod()` in the current design.
   # See #223 for discussion of alternatives.
+  if (attr(x, "no_decay_to_tibble") %||% FALSE) return(x)
   decay_epi_df(NextMethod())
 }
 

diff --git a/tests/testthat/test-as_tibble-decay.R b/tests/testthat/test-as_tibble-decay.R
@@ -0,0 +1,6 @@
+test_that("as_tibble checks an attr to avoid decay to tibble", {
+  edf <- jhu_csse_daily_subset
+  expect_s3_class(as_tibble(edf), c("tbl_df", "tbl", "data.frame"))
+  attr(edf, "no_decay_to_tibble") <- TRUE
+  expect_s3_class(as_tibble(edf), c("epi_df", "tbl_df", "tbl", "data.frame"))
+})