tests/testthat/test-regroup.R

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

test_that("regroup works", {
    set.seed(1)
    int <- sample(-3:50, 100L, 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_epiweek(dat$dates)
    x <- incidence(dat, date_index = "dates", groups = c("group_1", "group_2"))

    # regroup to know groups
    expected <- incidence(dat, date_index = "dates")
    expect_equal(regroup(x), expected)

    # regroup to one group
    expected <- incidence(dat, date_index = "dates", groups = "group_1")
    expect_equal(regroup(x, "group_1"), expected)

    # no groups
    expected <- incidence(dat, date_index = "dates")
    expect_equal(regroup(expected), expected)

    # regroup none-incidence object
    expect_error(regroup("test"))
    expect_snapshot(error = TRUE, regroup("test"))

    # other errors
    expect_error(
        regroup(
            incidence(dat, date_index = "dates", groups = "group_1"),
            groups = character()
        )
    )
    expect_snapshot(
        error = TRUE,
        regroup(
            incidence(dat, date_index = "dates", groups = "group_1"),
            groups = character()
        )
    )

    expect_error(
        regroup(
            incidence(dat, date_index = "dates", groups = "group_1"),
            groups = 1
        )
    )
    expect_snapshot(
        error = TRUE,
        regroup(
            incidence(dat, date_index = "dates", groups = "group_1"),
            groups = 1
        )
    )

    expect_error(
        regroup(
            incidence(dat, date_index = "dates", groups = "group_1"),
            groups = "bob"
        )
    )
    expect_snapshot(
        error = TRUE,
        regroup(
            incidence(dat, date_index = "dates", groups = "group_1"),
            groups = "bob"
        )
    )

})