Add expand_bbox attribute (Fixes #153) (#154)

* Add expand_bbox attribute

* Update snaps

R/autoplot.R

@@ -119,11 +119,9 @@ autoplot.spatial_block_cv <- function(object, show_grid = TRUE, ..., alpha = 0.6
   data <- object$splits[[1]]$data
 
   plot_data <- data
-  if (is_longlat(data)) {
-    plot_data <- sf::st_bbox(data)
-    plot_data <- expand_grid(plot_data)
-    plot_data <- sf::st_as_sfc(plot_data)
-  }
+  plot_data <- sf::st_bbox(data)
+  plot_data <- expand_grid(plot_data, attr(object, "expand_bbox"))
+  plot_data <- sf::st_as_sfc(plot_data)
 
   grid_args <- list(x = plot_data)
   grid_args$cellsize <- attr(object, "cellsize", TRUE)

R/spatial_block_cv.R

@@ -166,6 +166,7 @@ spatial_block_cv <- function(data,
     radius = radius,
     buffer = buffer,
     repeats = repeats,
+    expand_bbox = expand_bbox,
     ...
   )
 

tests/testthat/test-autoplot.R

@@ -120,3 +120,12 @@ test_that("autoplot is stable", {
     autoplot(repeat_llo)
   )
 })
+
+test_that("autoplot respects expand_bbox", {
+  vdiffr::expect_doppelganger(
+    "expand_bbox",
+    autoplot(
+      spatial_block_cv(boston_canopy, expand_bbox = 0.5, v = 4)
+    )
+  )
+})

tests/testthat/test-spatial_block_cv.R

@@ -242,6 +242,22 @@ test_that("duplicated observations in assessment sets throws an error", {
   )
 })
 
+test_that("expand_bbox attribute is set", {
+  skip_if_not_installed("withr")
+  folds <- withr::with_seed(
+    123,
+    spatial_block_cv(boston_canopy, expand_bbox = 0.01)
+  )
+  expect_equal(attr(folds, "expand_bbox"), 0.01)
+  expect_identical(
+    folds,
+    withr::with_seed(
+      123,
+      rsample::reshuffle_rset(folds)
+    )
+  )
+})
+
 test_that("bad args", {
   skip_if_not(sf::sf_use_s2())
   set.seed(123)