tests/testthat/test-statistics_of_imputed.R

In stephematician/miForang: Single or multiple imputation of missing data using random forests

context('statistics_of_imputed')

test_that('empty', {

    expect_null(statistics_of_imputed(list()))

})

test_that('numeric data', {

    imputed <- data.frame(x=as.numeric(1:5))

    expect_identical(statistics_of_imputed(imputed),
                     rbind(data.frame(variable='x', value=var(imputed$x),
                                      measure='var'),
                           data.frame(variable='x', value=mean(imputed$x),
                                      measure='mean')))
})

test_that('categorical data', {

    imputed <- data.frame(x=factor(letters[1:5]))

    expect_identical(statistics_of_imputed(imputed),
                     data.frame(variable='x',
                                value=-sum(rep(log2(0.2), 5)) / 5,
                                measure='entropy'))
})