tests/testthat/test-geovctr.R

In paleolimbot/geovctrs: Efficient Data Structures for Geometry Vectors

context("test-geovctr")

test_that("all geovctrs pass expect_geovctr", {
  expect_geovctr(wkt())
  expect_geovctr(wkb())
  expect_geovctr(geo_xy())
  expect_geovctr(geo_xyz())
  expect_geovctr(geo_segment())
  expect_geovctr(geo_rect())
})

test_that("geovctrs are left alone by restore_geovctr", {
  expect_identical(restore_geovctr(wkt(), wkb()), wkb())
})

test_that("character works with as_geovctr()", {
  expect_geovctr(as_geovctr("POINT (0 0)"))
})

test_that("data.frame works with as_geovctr()", {
  expect_geovctr(as_geovctr(geo_nc))
  expect_error(as_geovctr(tibble()), "Can't find geovctr")
  expect_error(as_geovctr(tibble(x = wkt(), y = wkt())), "More than one")

  # restore method replaces the geometry column
  expect_identical(restore_geovctr(tibble(x = wkt()), wkb()), tibble(x = wkb()))

  # check with transformation functions
  expect_is(geo_envelope(geo_nc), "data.frame")

  expect_is(geo_set_srid(geo_nc, 26920), "data.frame")

  # check with vectorization
  expect_identical(
    geo_set_z(geo_nc[1, ], 4:6),
    geo_set_z(geo_nc[c(1, 1, 1), ], 4:6)
  )
})