tests/testthat/test-class.R

In overtureR: Load 'Overture' Datasets as 'dbplyr' and 'sf'-Ready Data Frames

test_that("Counties download works as expected", {
  skip_if_offline()
  skip_on_cran()

  counties <- open_curtain("division_area", bbox = NULL) %>%
    # in R, filtering on variables must come before removing them via select
    dplyr::filter(subtype == "county" & region == "US-PA") %>%
    dplyr::transmute(
      id,
      division_id,
      # STRUCT/MAP columns can be accessed as column[["subcolumn"]] in transmute
      primary = names[["primary"]],
      geometry
    )

  counties_collected <- dplyr::collect(counties)
  expect_true(class(counties)[[1]] == "overture_call")
  playbill <- attr(counties, "overture_playbill")

  expect_equal(playbill[["type"]], "division_area")
  expect_equal(playbill[["theme"]], "divisions")

  expect_true("tbl_lazy" %in% class(counties))

  counties_sf <- dplyr::collect(counties)

  expect_false("overture_call" %in% class(counties_sf))
  expect_true("sf" %in% class(counties_sf))
  expect_equal(dplyr::pull(dplyr::count(counties), n), 67)

  expect_true(all(grepl("County", counties_sf$primary)))
  expect_true(all(sf::st_is_valid(counties_sf$geometry)))
  expect_true(sf::st_crs(counties_sf$geometry) == sf::st_crs(4326))
})

test_that("class assignment works", {
  skip_on_cran()
  skip_if_offline()

  conn <- DBI::dbConnect(duckdb::duckdb())
  division <- open_curtain("division", conn = conn, tablename = "test")

  # convert arbitrary sql into `overture_call`
  division2 <- dplyr::tbl(conn, "test")
  division2 <- as_overture(division2, "division")

  expect_true(class(division2)[[1]] == "overture_call")
  expect_error(as_overture(mtcars), "Input must be a tbl_sql object")

  DBI::dbDisconnect(conn, shutdown = TRUE)
})