tests/testthat/test_helpers.R

In iNZightVIT/iNZightTools: Tools for 'iNZight'

test_that("Vartype is correct", {
    expect_equal(vartype(as.POSIXct("2018-01-01 12:00:00")), "dt")
    expect_equal(vartype(as.Date("2018-01-01")), "dt")
    expect_equal(vartype(readr::parse_time("12:00:00")), "dt")

    expect_true(is_dt(as.POSIXct("2018-01-01 12:00:00")))

    iris_types <- c(
        Sepal.Length = "num",
        Sepal.Width = "num",
        Petal.Length = "num",
        Petal.Width = "num",
        Species = "cat"
    )
    expect_equal(vartypes(iris), iris_types)
    expect_equal(vartypes(as_tibble(iris)), iris_types)

    skip_if_not_installed("RSQLite")
    db <- tempfile(fileext = ".db")
    con <- DBI::dbConnect(RSQLite::SQLite(), db)
    on.exit({
        DBI::dbDisconnect(con)
        unlink(db)
    })
    DBI::dbWriteTable(con, "iris", iris)

    d <- inzdf(con, "iris", keep_con = TRUE)
    expect_equal(vartypes(d), iris_types)
    expect_equal(vartypes(dplyr::tbl(con, "iris")), iris_types)

    expect_error(
        vartypes(lm(Sepal.Width ~ Species, data = iris)),
        "Unsupported data object."
    )

    d <- structure(list(), vartypes = c("A", "B"), class = "inzdf_db")
    expect_equal(vartypes(d), c("A", "B"))

    expect_true(is_cat(iris$Species))
})

test_that("Survey objects identified correctly", {
    data(api, package = "survey")
    dclus2 <- svydesign(id = ~ dnum + snum, fpc = ~ fpc1 + fpc2, data = apiclus2)
    dclus2rep <- suppressWarnings(as.svrepdesign(dclus2))

    data(scd, package = "survey")
    repweights <- 2 *
        cbind(
            c(1, 0, 1, 0, 1, 0),
            c(1, 0, 0, 1, 0, 1),
            c(0, 1, 1, 0, 0, 1),
            c(0, 1, 0, 1, 1, 0)
        )
    scdrep <- suppressWarnings(
        svrepdesign(
            data = scd, type = "BRR", repweights = repweights,
            combined.weights = FALSE
        )
    )

    expect_false(is_survey(apiclus2))
    expect_true(is_survey(dclus2))
    expect_true(is_survey(dclus2rep))
    expect_false(is_survey(scd))
    expect_true(is_survey(scdrep))

    expect_false(is_svydesign(apiclus2))
    expect_true(is_svydesign(dclus2))
    expect_false(is_svydesign(dclus2rep))
    expect_false(is_svydesign(scd))
    expect_false(is_svydesign(scdrep))

    expect_false(is_svyrep(apiclus2))
    expect_false(is_svyrep(dclus2))
    expect_true(is_svyrep(dclus2rep))
    expect_false(is_svyrep(scd))
    expect_true(is_svyrep(scdrep))
})

test_that("Make names are unique", {
    expect_equal(make_names("var"), "var")
    expect_equal(make_names("var", "var_z"), "var")
    expect_equal(make_names("var", "var"), "var1")
    expect_equal(make_names(c("var", "var"), "var"), c("var1", "var2"))
})

test_that("Not in works", {
    expect_true("a" %notin% LETTERS)
    expect_false("A" %notin% LETTERS)
})

test_that("Or null helper", {
    x <- list(a = 1, b = NULL)
    expect_equal(orNULL(x), x)
    expect_equal(orNULL(x$a), x$a)
    expect_equal(orNULL(x$b), NULL)
    expect_equal(orNULL(x$c), NULL)
})

test_that("NULL OR operator works", {
    x <- list(a = 1, b = NULL)
    expect_equal(x$a %||% 2, 1)
    expect_equal(x$b %||% 2, 2)
    expect_equal(x$c %||% 2, 2)
})