tests/testthat/test-collect_options.R

In duckspatial: R Interface to 'DuckDB' Database with Spatial Extension

testthat::skip_on_cran()

test_that("collect.duckspatial_df respects global options", {
  skip_if_not_installed("sf")
  skip_if_not_installed("duckdb")
  
  conn <- ddbs_create_conn()
  on.exit(ddbs_stop_conn(conn))
  
  # Setup data
  nc_sf <- sf::st_read(system.file("shape/nc.shp", package = "sf"), quiet = TRUE)[1:5,]
  ddbs_write_table(conn, nc_sf, "nc_test_opts")
  
  # Create duckspatial_df
  # x <- dplyr::tbl(conn, "nc_test_opts") |> 
  #    as_duckspatial_df(geom_col = "geometry", crs = sf::st_crs(nc_sf))
  x <- as_duckspatial_df("nc_test_opts", conn)
  
  # Save original option
  op_orig <- options()
  on.exit(options(op_orig), add = TRUE)
  
  # 1. Default (sf) behavior when option is "duckspatial_df" (default)
  ddbs_options(output_type = "sf")
  res <- dplyr::collect(x)
  expect_s3_class(res, "sf")
  
  # 2. Option = "tibble"
  ddbs_options(output_type = "tibble")
  res_tbl <- dplyr::collect(x)
  expect_false(inherits(res_tbl, "sf"))
  expect_s3_class(res_tbl, "tbl_df")
  expect_false("geometry" %in% names(res_tbl))
  
  # 3. Option = "raw"
  ddbs_options(output_type = "raw")
  res_raw <- dplyr::collect(x)
  expect_true("geometry" %in% names(res_raw))
  expect_true(is.list(res_raw$geometry))
  
  # 4. Explicit override wins
  ddbs_options(output_type = "tibble")
  res_sf <- dplyr::collect(x, as = "sf")
  expect_s3_class(res_sf, "sf")
})



test_that("duckspatial.mode works as expected", {
  skip_if_not_installed("sf")
  
  # Setup data
  x <- ddbs_open_dataset(
    system.file("spatial/argentina.geojson", 
    package = "duckspatial")
  )
  
  # Save original option
  op_orig <- options()
  on.exit(options(op_orig), add = TRUE)
  
  # 1. Mode = "sf"
  ddbs_options(mode = "sf")
  res <- ddbs_centroid(x)
  expect_s3_class(res, "sf")
  
  # 2. Mode = "sf" on operations that return a vector
  res <- ddbs_area(x)
  expect_s3_class(res, "units")

  # 3. Mode = "sf" on operations that return a list
  res <- ddbs_intersects(x, x)
  expect_type(res, "list")
  
  # 4. Default mode (duckspatial)
  ddbs_options(mode = "duckspatial")
  res <- ddbs_centroid(x)
  expect_s3_class(res, "duckspatial_df")

})