tests/testthat/test-incidence_project_beds.R

In reconhub/occupancy: Forecast Hospital Bed Occupancy

# Integration test for project_beds on the incidence class

## contant admissions and one day length of stay, default last_date
test_that("project_beds, incidence class - test 1 ", {

    # parameters
    dates <- Sys.Date() - 1:10
    admissions <- rep(1, 10)
    rlos <- function(n) {
        rep(1, n)
    }

    # setup
    x <- incidence(rep(dates, admissions))

    # simulation results
    beds <- project_beds(x, rlos, n_sim = 10)

    # check daily occupancies
    expected <- matrix(1, 10, 10)
    expect_true(all(expected == beds))

    # check rownames
    beds_dates <- rownames(beds)
    expected <- as.character(sort(dates))
    expect_equal(beds_dates, expected)
})

## contant admissions and long length of stay, default last_date
test_that("project_beds, incidence class - test 2", {

    # parameters
    dates <- Sys.Date() - 1:10
    admissions <- rep(1, 10)
    rlos <- function(n) {
        rep(10, n)
    }

    # setup
    x <- incidence(rep(dates, admissions))

    # simulation results
    beds <- project_beds(x, rlos, n_sim = 10)

    # check daily occupancies
    expected <- matrix(1:10, nrow = 10, ncol = 10)
    expect_true(all(expected == beds))

    # check rownames
    beds_dates <- rownames(beds)
    expected <- as.character(sort(dates))
    expect_equal(beds_dates, expected)
})

## contant admissions and long length of stay, last_date way ahead
test_that("project_beds, incidence class - test 3", {

    # parameters
    dates <- Sys.Date() - 1:10
    admissions <- rep(1, 10)
    rlos <- function(n) {
        rep(10, n)
    }
    last_date <- Sys.Date() + 100

    # setup
    x <- incidence(rep(dates, admissions))

    # simulation results
    beds <- project_beds(x, rlos, n_sim = 10, last_date = last_date)

    # check daily occupancies
    expected <- matrix(c(1:10, 9:1), nrow = 19, ncol = 10)
    expect_true(all(expected == beds))

    # check rownames
    beds_dates <- rownames(beds)
    expected <- c(as.character(sort(dates)), as.character(Sys.Date() + 0:8))
    expect_equal(beds_dates, expected)
})

## check output class is correct
test_that("project_beds, incidence class - test 4 ", {

    # parameters
    dates <- Sys.Date() - 1:10
    admissions <- rep(1, 10)
    rlos <- function(n) {
        rep(1, n)
    }

    # setup
    x <- incidence(rep(dates, admissions))

    # simulation results
    beds <- project_beds(x, rlos, n_sim = 10)

    # check class
    expect_true(inherits(beds, "projections"))
})