tests/testthat/test_extract_tables.R

In leeper/tabulizer: Extract Tables from PDF Documents

context("Minimum functionality of extract_tables")

sf <- system.file("examples", "mtcars.pdf", package = "tabulapdf")

test_that("It basically works", {
  tab1 <- extract_tables(sf)
  expect_true(is.list(tab1))
  expect_true(is.data.frame(tab1[[1]]))
})

test_that("Warning for ignored arguments", {
  expect_warning(extract_tables(sf, area = list(c(1, 2, 3, 4)), guess = TRUE))
  expect_warning(extract_tables(sf, column = list(c(1, 2, 3, 4)), guess = TRUE))
})

test_that("Test 'guess' argument", {
  tab1a <- extract_tables(sf, pages = 1, guess = TRUE)
  tab1b <- extract_tables(sf, pages = 1, guess = FALSE)
  expect_true(ncol(tab1a[[1]]) == 12)
  expect_true(ncol(tab1b[[1]]) == 12)

  tab2a <- extract_tables(sf, pages = 2, guess = TRUE)
  tab2b <- extract_tables(sf, pages = 2, guess = FALSE)
  expect_true(length(tab2a) == 1)
  expect_true(length(tab2b) == 1)
})

test_that("Import from remote file works", {
  f2 <- "https://github.com/tabulapdf/tabula-java/raw/master/src/test/resources/technology/tabula/12s0324.pdf"
  tab2 <- extract_tables(f2)
  expect_true(is.list(tab2))
  expect_true(length(tab2) == 2)
  expect_true(is.data.frame(tab2[[1]]))
})

test_that("Import from remote non-Western file", {
  f3 <- "https://github.com/tabulapdf/tabula-java/raw/master/src/test/resources/technology/tabula/arabic.pdf"
  tab3 <- extract_tables(f3)
  expect_true(is.list(tab3))
  expect_true(length(tab3) == 1)
  expect_true(is.data.frame(tab3[[1]]))
})

test_that("Test 'area' argument", {
  a4a <- list(c(122, 149, 536, 576))
  tab4a <- extract_tables(sf, pages = 1, area = a4a, guess = FALSE, output = "tibble")
  expect_true(is.list(tab4a))
  expect_true(is.data.frame(tab4a[[1]]))
  expect_true(nrow(tab4a[[1]]) == 1)
  expect_true(ncol(tab4a[[1]]) == 12)
  a4b <- list(c(122, 149, 251, 464))
  tab4b <- extract_tables(sf, pages = 1, area = a4b, guess = FALSE, output = "tibble")
  expect_true(is.list(tab4b))
  expect_true(is.data.frame(tab4b[[1]]))
  expect_true(nrow(tab4b[[1]]) == 1)
  expect_true(ncol(tab4b[[1]]) == 9)
})

test_that("Test 'columns' argument", {
  tab5 <- extract_tables(sf, pages = 1, columns = list(c(0, 612)), guess = FALSE)
  expect_true(is.list(tab5))
  expect_true(length(tab5) == 1)
  expect_true(ncol(tab5[[1]]) == 2)
  expect_true(nrow(tab5[[1]]) == 5)
})

test_that("Extract from encrypted PDF", {
  f6 <- "https://github.com/tabulapdf/tabula-java/raw/98957221950af4b90620b51a29e0bbe502eea9ad/src/test/resources/technology/tabula/encrypted.pdf"
  expect_error(extract_tables(f6, password = "wrongpassword"))
  tab6 <- extract_tables(f6, password = "userpassword")
  expect_true(is.list(tab6))
  expect_true(is.data.frame(tab6[[1]]))
})

test_that("Test 'copy' argument", {
  fls <- list.files(tempdir())
  filepath <- file.path(tempdir(), basename(sf))
  tab7 <- extract_tables(sf, copy = TRUE)
  fls2 <- list.files(tempdir())
  expect_identical(length(fls) + 1L, length(fls2))
  expect_true(file.exists(filepath))
  unlink(filepath)
})