tests/testthat/test-wbt_source.R

In whitebox: 'WhiteboxTools' R Frontend

test_that("wbt_source (raster) works", {

    skip_if_not_installed("terra")

    expect_error(wbt_source("does_not_exist.tif"))

    f <- sample_dem_data()

    skip_if(!file.exists(f))

    # raster source from geotiff path
    src <- wbt_source(f)

    x <- attr(src, "wbt_dsn")

    expect_true(grepl("\\.tif$", x))
    expect_true(file.exists(x))

    dem <- terra::rast(f)

    # raster source from spatraster (with source file)
    src <- wbt_source(dem)

    x <- attr(src, "wbt_dsn")

    expect_true(grepl("\\.tif$", x))
    expect_true(file.exists(x))

    dem2 <- dem*2

    # raster source from spatraster (in memory)
    src <- wbt_source(dem2)

    x <- attr(src, "wbt_dsn")

    expect_true(grepl("\\.tif$", x))
    expect_true(file.exists(x))

    tf <- tempfile(fileext = ".gpkg")
    terra::writeRaster(dem, tf, gdal = c("RASTER_TABLE=one"))
    terra::writeRaster(dem2, tf, gdal = c("RASTER_TABLE=two", "APPEND_SUBDATASET=YES"))

    # raster source from non-geotiff
    src <- wbt_source(tf)

    x <- attr(src, "wbt_dsn")

    expect_true(grepl("\\.tif$", x))
    expect_true(file.exists(x))

    # raster source from non-geotiff
    src <- wbt_source(tf, layer = "two")

    x <- attr(src, "wbt_dsn")

    expect_true(grepl("\\.tif$", x))
    expect_true(file.exists(x))

    unlink(tf)
})

test_that("wbt_source (vector) works", {

    skip_if_not_installed("terra")

    f <- sample_soils_data()

    skip_if(!file.exists(f))

    # vector source from shapefile
    src <- wbt_source(f)

    x <- attr(src, "wbt_dsn")

    expect_true(grepl("\\.shp$", x))
    expect_true(file.exists(x))

    vf <- terra::vect(f)

    # vector source from spatvector (in memory)
    src <- wbt_source(vf)

    x <- attr(src, "wbt_dsn")

    expect_true(grepl("\\.shp$", x))
    expect_true(file.exists(x))

    vf2 <- terra::vect(f, proxy = TRUE)

    # vector source from spatvectorproxy
    src <- wbt_source(vf2)

    x <- attr(src, "wbt_dsn")

    expect_true(grepl("\\.shp$", x))
    expect_true(file.exists(x))

    tf <- tempfile(fileext = ".gpkg")

    terra::writeVector(vf, tf, layer = "one")
    terra::writeVector(vf, tf, layer = "two", insert = TRUE)

    # vector source from non-shapefile
    src <- suppressWarnings(wbt_source(tf))
    # terra warning for multiple layers but layer unspecified

    x <- attr(src, "wbt_dsn")

    expect_true(grepl("\\.shp$", x))
    expect_true(file.exists(x))

    src <- wbt_source(tf, layer = "two")

    x <- attr(src, "wbt_dsn")

    expect_true(grepl("\\.shp$", x))
    expect_true(file.exists(x))

    unlink(tf)
})