tests/testthat/test-normality.R

In jentjr/manager: Tools for plotting and analyzing environmental data

# Normal case ------------------------------------------------------------------
set.seed(14)
df_norm1 <- data.frame(
  location_id = rep(1, 20),
  sample_date = seq(lubridate::ymd("2010-01-01"),
                    by = "quarter", length.out = 20),
  param_name = rep("test", 20),
  lt_measure = rep("", 20),
  analysis_result = rnorm(20, mean = 5, sd = 1),
  default_unit = rep("mg/L", 20)
)

test_that("data are normal", {
  expect_match(est_dist(df_norm1)$distribution, "Normal")
})

# Lognormal case ---------------------------------------------------------------
df_lnorm1 <- data.frame(
  location_id = rep(2, 20),
  sample_date = seq(lubridate::ymd("2010-01-01"),
                    by = "quarter", length.out = 20),
  param_name = rep("test", 20),
  lt_measure = rep("", 20),
  analysis_result = rlnorm(20, meanlog = 5, sdlog = 1),
  default_unit = rep("mg/L", 20)
)

test_that("data are lognormal", {
  expect_match(est_dist(df_lnorm1)$distribution, "Lognormal")
})

# Nonparametric case -----------------------------------------------------------
df_nonpar1 <- data.frame(
  location_id = rep(3, 20),
  sample_date = seq(lubridate::ymd("2010-01-01"),
                    by = "quarter", length.out = 20),
  param_name = rep("test", 20),
  lt_measure = rep("", 20),
  analysis_result = rhyper(20, m = 30, n = 1, k = 20),
  default_unit = rep("mg/L", 20)
  )

test_that("data are neither normal, nor lognormal", {
  expect_match(est_dist(df_nonpar1)$distribution, "Nonparametric")
})

# Data Frame test case ---------------------------------------------------------
df1 <- rbind(df_norm1, df_lnorm1, df_nonpar1)

df1_est_dist <- est_dist(df1)

norm_result <- df1_est_dist %>%
  filter(location_id == 1) %>%
  select(distribution)

lnorm_result <- df1_est_dist %>%
  filter(location_id == 2) %>%
  select(distribution)

npar_result <- df1_est_dist %>%
  filter(location_id == 3) %>%
  select(distribution)

test_that("normality check works on gw data.frame", {
  expect_match(norm_result$distribution, "Normal")
  expect_match(lnorm_result$distribution, "Lognormal")
  expect_match(npar_result$distribution, "Nonparametric")
})