tests/testthat/test-zoo.R

In poissonconsulting/nrp: Kootenay Lake Nutrient Restoration Program Data Loading

test_that("nrp_read_zooplankton_file works", {

  conn <- nrp_create_db(path = ":memory:", ask = FALSE)
  teardown(DBI::dbDisconnect(conn))

  path <- system.file("extdata", "zooplankton/Arzp20.xlsx",
                      package = "nrp", mustWork = TRUE)

  wrong_path <- system.file("extdata", "ar-empty.rtf", package = "nrp", mustWork = TRUE)

  data <- nrp_read_zooplankton_file(path = path, db_path = conn) %>%
    suppressWarnings()

  expect_is(data, "tbl_df")
  expect_identical(nrow(data), 60L)
  check_zoo_raw_data(data)

  expect_error(nrp_read_zooplankton_file(path = path, db_path = conn, system = "columbia"),
               "'system' must be one of 'arrow', 'kootenay'.")

  expect_error(nrp_read_zooplankton_file(path = wrong_path, db_path = conn),
               "Please ensure input data is a valid excel spreadsheet \\(.xlsx\\).")
})

test_that("nrp_read_zooplankton works", {

  conn <- nrp_create_db(path = ":memory:", ask = FALSE)
  teardown(DBI::dbDisconnect(conn))

  path <- system.file("extdata", "zooplankton",
                      package = "nrp", mustWork = TRUE)
  data <- nrp_read_zooplankton(path, db_path = conn)

  expect_is(data, "tbl_df")
  expect_identical(length(data), 164L)
  expect_identical(nrow(data), 132L)

  expect_error(nrp_read_zooplankton("not-a-path", db_path = conn),
               "path 'not-a-path' must exist")

  path <- system.file("extdata",
                      package = "nrp", mustWork = TRUE)
  data <- nrp_read_zooplankton(path, db_path = conn)
  expect_identical(data,  list(x = 1)[-1])
})

test_that("nrp_upload_zooplankton and nrp_download_zoo_sample works", {

  conn <- nrp_create_db(path = ":memory:", ask = FALSE)
  teardown(DBI::dbDisconnect(conn))

  path <- system.file("extdata", "zooplankton/Arzp20.xlsx",
                      package = "nrp", mustWork = TRUE)

  data <- nrp_read_zooplankton_file(path = path, db_path = conn) %>%
    suppressWarnings()

  nrp_upload_zooplankton(data = data, db_path = conn)

  db_data <- readwritesqlite::rws_read_table("Zooplankton", conn = conn)
  expect_identical(length(db_data), 7L)
  expect_identical(nrow(db_data), 8760L)

  db_sample <- readwritesqlite::rws_read_table("ZooplanktonSample", conn = conn)
  expect_identical(length(db_sample), 12L)
  expect_identical(nrow(db_sample), 60L)

  nrp_upload_zooplankton(data = data, db_path = conn, replace = TRUE)

  db_data <- readwritesqlite::rws_read_table("Zooplankton", conn = conn)
  expect_identical(length(db_data), 7L)
  expect_identical(nrow(db_data), 8760L)

  expect_error(nrp_upload_zooplankton(data = data, db_path = conn),
               "UNIQUE constraint failed: ZooplanktonSample.Date, ZooplanktonSample.SiteID, ZooplanktonSample.Replicate, ZooplanktonSample.FileName")
  db_data <- nrp_download_zooplankton(start_date = "2020-04-01",
                                      end_date = "2020-04-25", db_path = conn)

  expect_identical(length(db_data), 150L)
  expect_identical(nrow(db_data), 12L)

  db_data <- nrp_download_zooplankton(start_date = "2020-04-01",
                                      end_date = "2020-04-25",
                                      counts = TRUE,
                                      db_path = conn)

  expect_identical(length(db_data), 150L)
  expect_identical(nrow(db_data), 12L)
  expect_true(all(is.na(db_data$BBosm)))

  expect_error(nrp_download_zooplankton(start_date = "2020-04-01",
                                        end_date = "2020-04-25", db_path = conn,
                                        sites = "wrong"),
               "1 or more invalid site names")

  expect_error(nrp_download_zooplankton(start_date = "2020-04-01",
                                        end_date = "2020-04-25", db_path = conn,
                                        parameters = "wrong"),
               "1 or more invalid parameter names")

})