tests/testthat/test-format_period.R

In bayesiandemography/demprep: Prepare Demographic Data

## format_period_year ---------------------------------------------------------

test_that("format_period_year works with valid input", {
    expect_identical(format_period_year(x = c("2000", "2010", NA, "2004")),
                     factor(c(2000, 2010, NA, 2004),
                            levels = c(as.character(2000:2010), NA),
                            exclude = NULL))
    expect_identical(format_period_year(x = character()),
                     factor())
    expect_identical(format_period_year(x = c(NA, NA)),
                     factor(c(NA, NA),
                            levels = NA_character_,
                            exclude = NULL))
})

test_that("format_period_year gives correct error with invalid inputs", {
    expect_error(format_period_year(x = c("2000", "2010", NA, "wrong")),
                 "'x' has invalid label \\[\"wrong\"\\]")
})


## format_period_multi --------------------------------------------------------

test_that("format_period_multi works with valid input", {
    expect_identical(format_period_multi(x = c("2000-2001", "2010-2015", NA, "2004-2005")),
                     factor(c("2000-2005", "2010-2015", NA, "2000-2005"),
                            levels = c("2000-2005", "2005-2010", "2010-2015", NA),
                            exclude = NULL))
    expect_identical(format_period_multi(x = c("2000", "2000", NA, "2010-2015", NA, "2004-2005")),
                     factor(c("2000-2005", "2000-2005", NA, "2010-2015", NA, "2000-2005"),
                            levels = c("2000-2005", "2005-2010", "2010-2015", NA),
                            exclude = NULL))
    expect_identical(format_period_multi(x = c("2000-2001", "2010-2015", NA, "2004-2005"),
                                         origin = 2004,
                                         width = 20),
                     factor(c("1984-2004", "2004-2024", NA, "2004-2024"),
                            levels = c("1984-2004", "2004-2024", NA),
                            exclude = NULL))
    expect_identical(format_period_multi(x = character()),
                     factor())
    expect_identical(format_period_multi(x = c(NA, NA)),
                     factor(c(NA, NA),
                            levels = NA_character_,
                            exclude = NULL))
})

test_that("format_period_multi gives correct error with invalid inputs", {
    expect_error(format_period_multi(x = c("<2000", "2010-2015", NA, "2005")),
                 "'x' has open interval \\[\"<2000\"\\]")
    expect_error(format_period_multi(x = c("2000-2001", "2010-2015", NA, "2005+")),
                 "'x' has open interval \\[\"2005\\+\"\\]")
    expect_error(format_period_multi(x = c("2000-2001", "2010-2015", NA, "wrong")),
                 "'x' has invalid label \\[\"wrong\"\\]")
    expect_error(format_period_multi(x = c("2000-2001", "2010-2016")),
                 "label \"2010-2016\" from 'x' intersects two or more intervals formed using origin = 2000 and width = 5")
})


## format_period_custom -------------------------------------------------------

test_that("format_period_custom works with valid input", {
    expect_identical(format_period_custom(x = c("2000-2001", "2010-2015", NA, "2004-2005"),
                                          breaks = c(2000, 2003, 2006, 2020)),
                     factor(c("2000-2003", "2006-2020", NA, "2003-2006"),
                            levels = c("2000-2003", "2003-2006", "2006-2020", NA),
                            exclude = NULL))
    expect_identical(format_period_custom(x = c("2000-2001", "2000-2001", NA, NA),
                                          breaks = c(2000, 2003, 2006, 2020)),
                     factor(c("2000-2003", "2000-2003", NA, NA),
                            levels = c("2000-2003", "2003-2006", "2006-2020", NA),
                            exclude = NULL))
    expect_identical(format_period_custom(x = character(), breaks = integer()),
                     factor())
})

test_that("format_period_custom gives correct error with invalid inputs", {
    expect_error(format_period_custom(x = "2000-2010", breaks = integer()),
                 "'breaks' has length 0")
    expect_error(format_period_custom(x = c("2000-2001", "2010-2015", NA, "2000+"),
                                      breaks = c(2000, 2020)),
                 "'x' has open interval \\[\"2000\\+\"\\]")
    expect_error(format_period_custom(x = c("2000-2001", "2010-2015", NA, "<1990"),
                                      breaks = c(2000, 2020)),
                 "'x' has open interval \\[\"<1990\"\\]")
    expect_error(format_period_custom(x = c("2000-2001", "2010-2015", NA, "wrong"),
                                      breaks = c(2000, 2020)),
                 "'x' has invalid label \\[\"wrong\"\\]")
    expect_error(format_period_custom(x = c("2000-2001", "2010-2015", NA, "2019-2021"),
                                      breaks = c(2000, 2020, 2030)),
                 "label \"2019-2021\" from 'x' intersects two or more intervals formed from 'breaks'")
})


## format_period_quarter ------------------------------------------------------

test_that("format_period_quarter works with valid input", {
    expect_identical(format_period_quarter(x = c("2000 Q4", "2010 Q1", NA, "2004 Q2")),
                     factor(c("2000 Q4", "2010 Q1", NA, "2004 Q2"),
                            levels = c("2000 Q4",
                                       paste(rep(2001:2009, each = 4),
                                             paste0("Q", 1:4)),
                                       "2010 Q1",
                                       NA),
                            exclude = NULL))
    expect_identical(format_period_quarter(x = character()),
                     factor())
    expect_identical(format_period_quarter(x = c(NA, NA)),
                     factor(c(NA, NA),
                            levels = NA_character_,
                            exclude = NULL))
})


## format_period_month --------------------------------------------------------

test_that("format_period_month works with valid input", {
    expect_identical(format_period_month(x = c("2000 Jan", "2001 Feb", NA, "2001 Mar")),
                     factor(c("2000 Jan", "2001 Feb", NA, "2001 Mar"),
                            levels = c(paste("2000", month.abb),
                                       "2001 Jan", "2001 Feb", "2001 Mar",
                                       NA),
                            exclude = NULL))
    expect_identical(format_period_month(x = character()),
                     factor())
    expect_identical(format_period_month(x = c(NA, NA)),
                     factor(c(NA, NA),
                            levels = NA_character_,
                            exclude = NULL))
})