tests/testthat/test_separate_var.R

In iNZightTools: Tools for 'iNZight'

data <- data.frame(
    "B" = c("hi", "hello", "bye"),
    "A" = c("ada.sd\\d-sd^fdsf", "B_37K", "C"),
    "C" = c("2019M02", "2019M03", "2019M04"),
    "D" = c("x.x", "x.x.x.x.x", "x.x.x")
)


test_that("Basic separation works", {
    expect_silent(separate_var(data, "C", "M", c("year", "month")))
    expect_silent(separate_var(data, "C", 4, c("year", "month")))
    expect_silent(separate_var(data, "C", "M", into = "rows"))
    expect_silent(separate_var(data, "C", 4, into = "rows"))
})

test_that("Multi-column separation works", {
    expect_silent(separate_var(data, "D", ".", c("a", "b", "c")))
    expect_silent(separate_var(data, "A", c(1, 3, 10), c("a", "b", "c", "d")))
})

test_that("Condition checks", {
    expect_error(separate_var(data, "C", factor("M"), c("year", "month")))
    expect_warning(separate_var(data, "C", 4.1, c("year", "month")))
    expect_error(separate_var(data, "C", 11, c("year", "month")))
    expect_error(separate_var(data, "C", 0, c("year", "month")))
    expect_error(suppressWarnings(separate_var(data, "C", 0.9, c("year", "month"))))
    expect_error(separate_var(data, "A", c(1, 3, 10), c("a", "b", "c")))
    expect_error(separate_var(data, "C", 1:4, into = "rows"))
    expect_error(separate_var(data, "C", c("M", "N"), c("year", "month")))
})


library(survey)
data(api)
svy <- svydesign(~ dnum + snum, weights = ~pw, fpc = ~ fpc1 + fpc2, data = apiclus2)

test_that("Survey designs work", {
    svy$variables$avg.ed <- format(svy$variables$avg.ed)
    suppressWarnings(
        d <- separate_var(svy, "avg.ed", by = ".", c("ed.a", "ed.b"))
    )
    expect_s3_class(d, "survey.design2")
    expect_true(all(c("ed.a", "ed.b") %in% names(d$variables)))
    expect_false("avg.ed" %in% names(d$variables))
    check_eval(d)
    expect_error(separate_var(svy, into = "rows"))
})