tests/testthat/test-cumulate.R

In reconhub/incidence2: Compute, Handle and Plot Incidence of Dated Events

test_that("Fails with bad input", {
    expect_error(cumulate(mtcars))
})

test_that("Fails with good error for bad input", {
    expect_snapshot(error = TRUE, cumulate(mtcars))
})

test_that("cumulate works with groupings", {
    int <- sample(-3L:50L, 100, replace = TRUE)
    dates <- as.Date("2018-01-31") + int
    group_1 <- sample(letters[1:3], length(dates), replace = TRUE)
    group_2 <- sample(letters[1:3], length(dates), replace = TRUE)
    dat <- data.frame(dates, group_1, group_2)
    dat$dates <- as_period(dat$dates, n = 14L)
    x <- incidence(dat, date_index = "dates", groups = "group_1")

    xc <- x
    xc_dates <- seq(min(xc$date_index), max(xc$date_index))
    tmp <- expand.grid(date_index = xc_dates, group_1 = unique(group_1))
    tmp2 <- merge(tmp, xc, by = c("date_index", "group_1"), all.x = TRUE)
    tmp2[,"count"][is.na(tmp2[,"count"])] <- 0
    tmp2$count_variable <- "dates"
    tmp2$group_1 <- as.character(tmp2$group_1)
    for (gr in unique(group_1)) {
        idx <- tmp2$group_1 == gr
        tmp2[idx, "count"] <- cumsum(tmp2[idx, "count"])
    }

    res <- cumulate(complete_dates(x))
    res <- res[do.call(order, as.list(res)), ]
    tmp2 <- tmp2[do.call(order, as.list(tmp2)), ]

    attributes(tmp2) <- NULL
    attributes(res) <- NULL

    expect_equal(list2DF(res), list2DF(tmp2))

})

test_that("cumulate works without groupings", {
    dates <- as.Date("2018-01-31") + 1:100
    dat <- data.frame(dates)
    x <- incidence(dat, date_index = "dates")
    expected_count <- cumsum(rep(1,nrow(dat)))
    expect_equal(cumulate(x)$cumulative_count, expected_count)
})