tests/testthat/test_reshape_data.R

In iNZightTools: Tools for 'iNZight'

dt1 <- data.frame(
    "Country" = c("A", "B", "C"),
    "v1999" = c("0.7K", "37K", "212K"),
    "v2000" = c("2K", "80K", "213K")
)
dt2 <- data.frame(
    "Country" = c(
        "Afghanistan", "Afghanistan", "Afghanistan", "Afghanistan",
        "Brazil", "Brazil", "Brazil", "Brazil", "China", "China", "China", "China"
    ),
    "Year" = c(1999, 1999, 2000, 2000, 1999, 1999, 2000, 2000, 1999, 1999, 2000, 2000),
    "Type" = c(
        "cases", "population", "cases", "population", "cases", "population",
        "cases", "population", "cases", "population", "cases", "population"
    ),
    "Count" = c(
        745, 19987071, 2666, 20595360, 37737, 172006362, 80488, 174504898,
        212258, 1272915272, 213766, 1280428583
    )
)

stripattr <- function(x, attr = "code") {
    attributes(x)[[attr]] <- NULL
    x
}

test_that("Reshape wide to long works", {
    expect_equal(
        stripattr(reshape_data(dt1,
            cols = c("v1999", "v2000"), names_to = "Year", values_to = "Count"
        )),
        tidyr::pivot_longer(
            dt1, c(v1999, v2000),
            names_to = "Year", values_to = "Count"
        )
    )
    expect_equal(
        stripattr(reshape_data(dt2,
            cols = "Count", names_to = "xx", values_to = "yy"
        )),
        tidyr::pivot_longer(
            dt2, Count,
            names_to = "xx", values_to = "yy"
        )
    )
})

test_that("Reshape long to wide works", {
    expect_equal(
        stripattr(reshape_data(dt2,
            data_to = "wide",
            names_from = "Type", values_from = "Count"
        )),
        tidyr::pivot_wider(dt2,
            names_from = Type, values_from = Count
        )
    )
})