tests/testthat/test_count_vars.R

In VectorPosse/datafinder: Find And Explore Currently Installed Data Sets

context("count_vars")
library(datafinder)

## Create longdata
longdata <<- data.frame(factor = factor(rep(1:10, 1001)),
                       numeric = rep(1:10, 1001),
                       logical = rep(c(TRUE, FALSE), 5005))
smallerdata <<- longdata[1:100, ]

longdata <- count_vars(longdata)

## Check if count_vars returns an object of class "data.frame"
test_that("count_vars: class data.frame", {
    expect_equal(class(longdata), "data.frame")
})

## Check if no input returns error
test_that("count_vars: count the number of missing dataframe", {
    expect_error(count_vars())
})

## Check if single data object returns data frame of length one with correct
## info
ref_single <- data.frame(
    Dataframe = "longdata",
    factor = 1,
    integer = 1,
    logical = 1,
    total_vars = 3,
    sample_size = 10010
)
test_that("count_vars: count the number of variables in a single dataframe", {
    expect_equal(longdata, ref_single)
})

## Check if multiple data objects returns data frame with correct length and
## info
smallerdata <- count_vars(smallerdata)

ref_multiple <- data.frame(
    Dataframe = c("longdata", "smallerdata"),
    factor = c(1, 1),
    integer = c(1, 1),
    logical = c(1, 1),
    total_vars = c(3, 3),
    sample_size = c(10010, 100)
)
test_that("count_vars: count the number of variables in multiple dataframes", {
              expect_equal(rbind(longdata, smallerdata), ref_multiple)
})

rm(longdata,smallerdata)