tests/testthat/test-5_so2_3_three_yr_avg.R

In bcgov/rcaaqs: Tools to implement CAAQS analysis for air quality

context("so2 three year rolling average")

so1 <- readRDS("so2_yearly99_1.rds")
so2 <- readRDS("so2_yearly99_2.rds")

test_that("Runs silently", {
  expect_silent(r1 <- so2_three_yr_avg(so1))
  expect_silent(r2 <- so2_three_yr_avg(so2, by = c("ems_id", "site")))
  
  saveRDS(r1, "so2_3yr_1.rds")
  saveRDS(r2, "so2_3yr_2.rds")
})

ret1 <- readRDS("so2_3yr_1.rds")
ret2 <- readRDS("so2_3yr_2.rds")

test_that("has correct classes", {
  for(r in list(ret1, ret2)){
    expect_is(r, "data.frame")
    expect_is(r$year, "numeric")
    expect_is(r$valid_in_year, "numeric")
    expect_is(r$quarter_1, "numeric")
    expect_is(r$quarter_2, "numeric")
    expect_is(r$quarter_3, "numeric")
    expect_is(r$quarter_4, "numeric")
    expect_is(r$ann_99_percentile, "numeric")
    expect_is(r$exceed, "logical")
    expect_is(r$flag_daily_incomplete, "logical")
    expect_is(r$so2_metric, "numeric")
    expect_is(r$n_val, "numeric")
    expect_is(r$flag_two_of_three_years, "logical")
  }
})

test_that("has correct dimensions", {
  nrows <- nrow(so1)
  expect_equal(dim(ret1), c(nrows, 18))
  
  nrows <- nrow(so2)
  expect_equal(dim(ret2), c(nrows, 20))
})

test_that("has correct data", {
  for(r in list(ret1, ret2[ret2$ems_id == "E231866",])){
    expect_equivalent(r$valid_in_year[1:3], c(0.951, 0.989, 0.992), tolerance = 0.001)
    expect_equivalent(r$quarter_1[1:3], c(0.889, 1.00, 0.989), tolerance = 0.001)
    expect_equivalent(r$quarter_2[1:3], c(1, 1, 1), tolerance = 0.001)
    expect_equivalent(r$quarter_3[1:3], c(0.913, 0.978, 1), tolerance = 0.001)
    expect_equivalent(r$quarter_4[1:3], c(1, 0.978, 0.978), tolerance = 0.001)
    expect_equivalent(r$ann_99_percentile[1:3], c(14.8, 19.4, 17.1), tolerance = 0.001)
    expect_equivalent(r$n_val[1:3], c(1, 2, 3))
    expect_equal(r$flag_two_of_three_years[1:3], c(FALSE, TRUE, FALSE))
    expect_equal(r$so2_metric[1:3], as.numeric(c(NA, 17.1, 17.1)))
  }
})

test_that("performs data completeness accurately", {
  for(r in list(ret1, ret2)){
    expect_true(all(is.na(r$so2_metric[r$n_val < 2])))
    expect_true(all(r$flag_two_of_three_years[r$n_val == 2]))
  }
})