tests/testthat/test-get_results_data.R

context("get_results_data")

conn <- nfi_connect()

test_that("plot level works", {
  expect_s3_class(
    nfi_results_data(conn, 'nfi_2', 'none', FALSE, .collect = TRUE), 'tbl_df'
  )
  expect_s3_class(
    nfi_results_data(conn, 'nfi_2', 'none', FALSE, .collect = FALSE), 'tbl_sql'
  )
  expect_s3_class(
    nfi_results_data(conn, 'nfi_4', 'none', FALSE, .collect = TRUE), 'tbl_df'
  )
  expect_s3_class(
    nfi_results_data(conn, 'nfi_4', 'none', FALSE, .collect = FALSE), 'tbl_sql'
  )
  expect_s3_class(
    nfi_results_data(conn, 'nfi_2_nfi_3', 'none', FALSE, .collect = TRUE), 'tbl_df'
  )
  expect_s3_class(
    nfi_results_data(conn, 'nfi_2_nfi_3', 'none', FALSE, .collect = FALSE), 'tbl_sql'
  )
})

test_that("species level works", {
  expect_s3_class(
    nfi_results_data(conn, 'nfi_3', 'species', FALSE, .collect = TRUE), 'tbl_df'
  )
  expect_s3_class(
    nfi_results_data(conn, 'nfi_3', 'species', FALSE, .collect = FALSE), 'tbl_sql'
  )
  expect_s3_class(
    nfi_results_data(conn, 'nfi_4', 'species', FALSE, .collect = TRUE), 'tbl_df'
  )
  expect_s3_class(
    nfi_results_data(conn, 'nfi_4', 'species', FALSE, .collect = FALSE), 'tbl_sql'
  )
  expect_s3_class(
    nfi_results_data(conn, 'nfi_3_nfi_4', 'species', FALSE, .collect = TRUE), 'tbl_df'
  )
  expect_s3_class(
    nfi_results_data(conn, 'nfi_3_nfi_4', 'species', FALSE, .collect = FALSE), 'tbl_sql'
  )
})

test_that("simpspecies level works", {
  expect_s3_class(
    nfi_results_data(conn, 'nfi_3', 'simpspecies', FALSE, .collect = TRUE), 'tbl_df'
  )
  expect_s3_class(
    nfi_results_data(conn, 'nfi_3', 'simpspecies', FALSE, .collect = FALSE), 'tbl_sql'
  )
  expect_s3_class(
    nfi_results_data(conn, 'nfi_4', 'simpspecies', FALSE, .collect = TRUE), 'tbl_df'
  )
  expect_s3_class(
    nfi_results_data(conn, 'nfi_4', 'simpspecies', FALSE, .collect = FALSE), 'tbl_sql'
  )
  expect_s3_class(
    nfi_results_data(conn, 'nfi_3_nfi_4', 'simpspecies', FALSE, .collect = TRUE), 'tbl_df'
  )
  expect_s3_class(
    nfi_results_data(conn, 'nfi_3_nfi_4', 'simpspecies', FALSE, .collect = FALSE), 'tbl_sql'
  )
})

test_that("diamclass level works", {
  expect_s3_class(
    nfi_results_data(conn, 'nfi_3', 'genus', TRUE, .collect = FALSE), 'tbl_sql'
  )
  expect_s3_class(
    nfi_results_data(conn, 'nfi_3', 'simpspecies', TRUE, .collect = FALSE), 'tbl_sql'
  )
  expect_s3_class(
    nfi_results_data(conn, 'nfi_2', 'dec', TRUE, .collect = FALSE), 'tbl_sql'
  )
  expect_s3_class(
    nfi_results_data(conn, 'nfi_4', 'bc', TRUE, .collect = FALSE), 'tbl_sql'
  )
  expect_s3_class(
    nfi_results_data(conn, 'nfi_2_nfi_3', 'none', TRUE, .collect = FALSE), 'tbl_sql'
  )
  expect_s3_class(
    nfi_results_data(conn, 'nfi_3_nfi_4', 'species', TRUE, .collect = FALSE), 'tbl_sql'
  )
})

test_that("shrubs and regeneration tables works", {
  expect_s3_class(
    nfi_results_data(conn, 'nfi_2_shrub', .collect = FALSE), 'tbl_sql'
  )
  expect_s3_class(
    nfi_results_data(conn, 'nfi_3_shrub', .collect = FALSE), 'tbl_sql'
  )
  expect_s3_class(
    nfi_results_data(conn, 'nfi_4_shrub', .collect = FALSE), 'tbl_sql'
  )
  expect_s3_class(
    nfi_results_data(conn, 'nfi_2_regen', .collect = TRUE), 'tbl_df'
  )
  expect_s3_class(
    nfi_results_data(conn, 'nfi_3_regen', .collect = TRUE), 'tbl_df'
  )
  expect_s3_class(
    nfi_results_data(conn, 'nfi_4_regen', .collect = TRUE), 'tbl_df'
  )
})

test_that("attribute is set", {
  expect_identical(
    attr(nfi_results_data(conn, 'nfi_3', 'genus', FALSE, .collect = FALSE), 'nfi'), 'NFI_3'
  )
  expect_identical(
    attr(nfi_results_data(conn, 'nfi_2', 'genus', FALSE, .collect = TRUE), 'nfi'), 'NFI_2'
  )
  expect_identical(
    attr(nfi_results_data(conn, 'nfi_2_shrub', .collect = TRUE), 'nfi'), 'SHRUB_NFI_2_INFO'
  )
  expect_identical(
    attr(nfi_results_data(conn, 'nfi_2_regen', .collect = TRUE), 'nfi'), 'REGENERATION_NFI_2'
  )
  # expect_false(
  #   attr(nfi_results_data(conn, 'nfi_3', 'genus', FALSE, .collect = FALSE), 'diamclass')
  # )
  # expect_true(
  #   attr(nfi_results_data(conn, 'nfi_3', 'genus', TRUE, .collect = FALSE), 'diamclass')
  # )
})

nfi_close(conn)
MalditoBarbudo/tidyNFI documentation built on Sept. 7, 2019, 9:11 a.m.