Updated [.epi_df

Author: admin

NAMESPACE

@@ -1,5 +1,6 @@
 # Generated by roxygen2: do not edit by hand
 
+S3method("[",epi_df)
 S3method(arrange,epi_df)
 S3method(as_epi_df,data.frame)
 S3method(as_epi_df,epi_df)
@@ -79,5 +80,3 @@ importFrom(tidyr,unnest)
 importFrom(tidyselect,eval_select)
 importFrom(tidyselect,starts_with)
 importFrom(tsibble,as_tsibble)
-importFrom(utils,head)
-importFrom(utils,tail)

R/methods-epi_df.R

@@ -99,19 +99,14 @@ summary.epi_df = function(object, ...) {
 
   cn <- names(res)
   nr <- vctrs::vec_size(x) 
-  not_epi_df <- (!("time_value" %in% cn) || !("geo_value" %in% cn)
-                 || vctrs::vec_size(res) > nr || any(i > nr))
+  not_epi_df <- (!("time_value" %in% cn) || !("geo_value" %in% cn) || vctrs::vec_size(res) > nr || any(i > nr))
 
   if (not_epi_df) return(tibble::as_tibble(res))
 
   # Case when i is numeric and there are duplicate values in it
   if (is.numeric(i) && vctrs::vec_duplicate_any(i) > 0) 
     return(tibble::as_tibble(res))
 
-  # Column subsetting only, then return res as tibble
-  if (rlang::is_null(i) && !rlang::is_null(j)) 
-    return(tibble::as_tibble(res))
-  
   att_x = attr(x, "metadata")
   new_epi_df(tibble::as_tibble(res), 
              geo_type = att_x$geo_type, 

tests/testthat/test-epi_slide.R

@@ -27,8 +27,8 @@ test_that("`ref_time_values` + `align` that have some slide data, but generate t
 test_that("these doesn't produce an error; the error appears only if the ref time values are out of the range for every group", {
   expect_identical(edf %>% group_by(geo_value) %>% epi_slide(f, n=3L, ref_time_values=as.Date("2020-01-01")+200L) %>% 
                      dplyr::select("geo_value","slide_value_value"), 
-                   dplyr::tibble(geo_value = "ak", slide_value_value = 199) %>% group_by(geo_value)) # out of range for one group
+                   dplyr::tibble(geo_value = "ak", slide_value_value = 199)) # out of range for one group
   expect_identical(edf %>% group_by(geo_value) %>% epi_slide(f, n=3L, ref_time_values=as.Date("2020-01-04")) %>% 
                      dplyr::select("geo_value","slide_value_value"), 
-                   dplyr::tibble(geo_value = c("ak", "al"), slide_value_value = c(2, -2)) %>% group_by(geo_value)) # not out of range for either group
+                   dplyr::tibble(geo_value = c("ak", "al"), slide_value_value = c(2, -2))) # not out of range for either group
 })