tests/testthat/test-stats-dates.R

In AntonioJBT/episcout: Quickly Clean, Explore and Visualise Large Epidemiological Datasets

context("episcout stats function tests")

######################
library(testthat)
library(data.table) # Ensure data.table is available for IDate
######################

######################
# Working directory for informal tests, should be from pkg/tests/testthat/:
# setwd("")
######################

######################
# Set a test set:

# Descriptive stats for dates
# Range, min, max,etc
# Example data

# test_dates <- as.Date(c("2020-01-01", "2020-05-15", "2020-12-31", "2021-01-01", "2021-07-15"))
# Set seed for reproducibility
set.seed(42)

# Define the start and end dates
start_date <- as.Date("2020-01-01")
end_date <- as.Date("2023-12-31")

# Calculate the difference in days between start and end dates
days_between <- as.integer(end_date - start_date)
days_between

# Generate 300 random numbers within the range of days
random_days <- sample(0:days_between, 300, replace = TRUE)

# Add these random days to the start date to get random dates
random_dates <- start_date + random_days
random_dates

# Sort the dates
sorted_random_dates <- sort(random_dates)
sorted_random_dates

# Convert to character for display or further processing
sorted_random_dates_str <- as.character(sorted_random_dates)
sorted_random_dates_str

# Print the first 10 sorted random dates
sorted_random_dates_str[1:10]
str(sorted_random_dates_str)
sorted_random_dates_str <- as.Date(sorted_random_dates_str)
str(sorted_random_dates_str)

test_dates <- sorted_random_dates_str
######################

######################
# Test 1: Correct output for a simple Date input
print("Function being tested: epi_stats_dates")
test_that("Test with regular Date inputs", {
    # test_dates <- as.Date(c("2020-01-01", "2020-06-01", "2020-12-31"))
    result <- calculate_date_stats(test_dates)
    expect_is(result, "data.frame")
    expect_equal(nrow(result), 7)
    expect_equal(result$Value[1], as.character(min(test_dates)))
    expect_equal(result$Value[2], as.character(max(test_dates)))
})

# Test 2: Correct handling of IDate input
test_that("Test with IDate inputs", {
    test_dates <- IDate(as.Date(c("2021-01-01", "2021-06-01", "2021-12-31")))
    result <- calculate_date_stats(test_dates)
    expect_is(result, "data.frame")
    expect_equal(nrow(result), 7)
    expect_equal(result$Value[1], as.character(min(as.Date(test_dates))))
    expect_equal(result$Value[2], as.character(max(as.Date(test_dates))))
})

# Test 3: Function stops if non-date input is provided
test_that("Test with non-date input throws error", {
    non_date_input <- c(1, 2, 3)
    expect_error(calculate_date_stats(non_date_input))
})

# Test 4: Handling of NA values in date inputs
test_that("Test handling of NA values", {
    test_dates_with_na <- as.Date(c("2020-01-01", NA, "2020-12-31"))
    result <- calculate_date_stats(test_dates_with_na)
    expect_is(result, "data.frame")
    expect_equal(nrow(result), 7)
    # Expect NA not to affect the calculation of min, max, etc.
    expect_equal(result$Value[1], as.character(min(test_dates_with_na, na.rm = TRUE)))
})
######################