tests/testthat/test_annual_load.R

In USGS-CIDA/nar_data: NAR Data

library(testthat)
library(validate)
context("annual load")
options(scipen=999)
temp_aloads<-aloads
temp_aloads$TONS_N<-as.numeric(temp_aloads$TONS)
temp_aloads$TONS_L95_N<-as.numeric(temp_aloads$TONS_L95)
temp_aloads$TONS_U95_N<-as.numeric(temp_aloads$TONS_U95)
temp_aloads$FWC_N<-as.numeric(temp_aloads$FWC)
temp_aloads$YIELD_N<-as.numeric(temp_aloads$YIELD)
temp_aloads$mod1<-as.character(temp_aloads$MODTYPE)
temp_aloads[temp_aloads$mod1%in%"REGHIST","mod1"]<-"REG"
temp_aloads<-temp_aloads[!is.na(temp_aloads$TONS_L95_N),]
temp_aloads<-temp_aloads[!is.na(temp_aloads$TONS_U95_N),]


test_that("annual load has the correct columns", {
	expect_has_names(aloads, c(
		"CONSTIT",
		"FWC",
		"MODTYPE",
		"SITE_ABB",
		"SITE_FLOW_ID",
		"SITE_QW_ID",
		"TONS",
		"TONS_L95",
		"TONS_U95",
		"YIELD",
		"WY"
	))
})

test_that("annual load's columns are correctly typed", {
	result <- validate::check_that(aloads,
		is.integer(WY),
		is.character(c(
			SITE_ABB,
			SITE_QW_ID,
			SITE_FLOW_ID,TONS, TONS_L95, TONS_U95, FWC, YIELD
		)),
		is.factor(CONSTIT),
		is.factor(MODTYPE)
	)
	
	expect_no_errors(result)
})


test_that("annual load has a reasonable range of values", {
	result <- validate::check_that(temp_aloads, 
		TONS_N > 0,
		TONS_N < 5E8,
		TONS_L95_N < TONS_U95_N,
		TONS_L95_N < TONS_N,
		TONS_N < TONS_U95_N,
		nchar(SITE_ABB) == 4,
		WY < 2020,
		WY > 1950
	)
	expect_no_errors(result)
})



test_that("annual loads for the MISS site are included", {
	miss_sites <- subset(aloads, SITE_ABB == 'MISS')
	expect_gt(nrow(miss_sites), 0)
	
	})


test_that("annual loads for the GULF are included", {
  gulf_sites <- subset(aloads, SITE_ABB == 'GULF')
  expect_gt(nrow(gulf_sites), 0)
})


test_that("the expected modtypes are present", {
  expected <- sort(c("REG","REG_2","REG_3","REG_4","REGHIST","DAILY"))
  actual <- sort(unique(as.character(aloads$MODTYPE)))
  expect_equal(actual, expected)
  
})

test_that("Load data have the correct number of significant digits", {
  result <- validate::check_that(temp_aloads, 
                                 
                                 count_sig_figs(temp_aloads$TONS_N/1E8) <= 3,
                                 count_sig_figs(temp_aloads$TONS_L95_N/1E8) <= 3,
                                 count_sig_figs(temp_aloads$TONS_U95_N/1E8) <= 3,
                                 count_sig_figs(temp_aloads$FWC_N/1E8) <= 3,
                                 count_sig_figs(temp_aloads$YIELD_N/1E8) <= 3
                                 
                                 )
  expect_no_errors(result) 
})

test_that("There are no duplicate values", {
  aloads_without_ignored_modtypes <- subset(aloads, !(MODTYPE %in% c('COMP', 'CONTIN')))
  unique_columns <- aloads_without_ignored_modtypes[c('SITE_QW_ID', 'CONSTIT', 'WY')]
  expect_no_duplicates(unique_columns)
  
})

test_that("Most recent water year has all of the necessary sites ", {
  temp_aloads_recent<-temp_aloads[temp_aloads$WY%in%max(temp_aloads$WY),] 
  expected <- sort(c("HAZL","PADU","GRAN","CLIN","WAPE","KEOS","VALL","GRAF","SIDN","OMAH","ELKH","LOUI","DESO","HERM","THEB","SEDG","HARR","KERS","MORG","BELL",
                     "STFR","MELV","SUMN","STTH","GULF","NEWH","CANN","MISS"))
  actual <- sort(unique(temp_aloads_recent[temp_aloads_recent$SITE_ABB%in%expected,"SITE_ABB"]))

  
  expect_equal(actual, expected)
})