tests/testthat/test_characterize.R

In leeper/rio: A Swiss-Army Knife for Data I/O

context("characterize()/factorize()")
x <- structure(1:4, labels = c("A" = 1, "B" = 2, "C" = 3))
xdf <- data.frame(v1 = structure(1:4, labels = c("A" = 1, "B" = 2, "C" = 3), label = "variable 1"),
                  v2 = structure(c(1,0,0,1), labels = c("foo" = 0, "bar" = 1)),
                  v3 = 4:1, label = "variable 2")


test_that("test characterize.default()", {
    expect_true(identical(characterize(x), c(LETTERS[1:3], NA)))
})

test_that("test characterize.default()", {
    expect_true(identical(characterize(xdf), {xdf[] <- lapply(xdf, characterize); xdf}))
})

test_that("test factorize.data.frame()", {
    expect_true(identical(factorize(x), factor(x, attributes(x)$labels, names(attributes(x)$labels))))
})

test_that("test factorize.data.frame()", {
    expect_true(identical(factorize(xdf), {xdf[] <- lapply(xdf, factorize); xdf}))
})

test_that("test factorize coerce_character", {
    expect_true(identical(letters[1:3], factorize(letters[1:3])))
    expect_true(
        identical(
            factorize(letters[3:1], coerce_character = TRUE),
            factor(letters[3:1], levels = letters[1:3])
        )
    )
})