tests/testthat/test-samplesubset.R

In wanjiag/ezPurrr: Makes functional programming with purrr easier to use

Name = c("Jon", "Bill", "Maria", "Ben", "Tina")
Age = c(23, 41, 32, 58, 26)
Gender = c('M', 'M', 'F', 'M', 'F')

df = data.frame(Name, Age, Gender)

test_that("Correct structure of the data frame", {
  expect_error(
    sample_row(df),
    'The dataframe does not contain a column named "data" with data frame type'
    )
})

nest_df = df %>% dplyr::group_by(Gender) %>%
  tidyr::nest()

test_that("correct index", {
  expect_error(
    sample_row(nest_df, index = 'error'),
    'Index must be a numeric integer and range between 1 and 2.'
  )
  expect_error(
    sample_row(nest_df, index = 1.2),
    'Index must be an integer.'
  )
})

test_df = nest_df[1,]$data[[1]]
test_list = list(test_df)
test_list = append(test_list, 'M')
names(test_list) = c('data','Gender')
test_df$group.Gender = 'M'

test_that("index sampling", {
  expect_equal(
    sample_row(nest_df, index = 1),
    test_list)
  expect_equal(
    sample_row(nest_df, index = 1, type = 'df'),
    test_df)
  expect_equal(
    sample_group(nest_df, index = 1),
    nest_df[1,c(1:ncol(nest_df)-1)])
  expect_equal(
    sample_data(nest_df, index = 2),
    nest_df[2,]$data[[1]])
})

sub_nest_df = nest_df[1,]
test_that("random sampling and warning message", {
  expect_equal(
    sample_row(sub_nest_df), test_list)
  expect_equal(
    sample_row(sub_nest_df, type = 'df'), test_df)
  expect_equal(
    sample_group(sub_nest_df),sub_nest_df[,c(1:ncol(sub_nest_df)-1)])
  expect_equal(
    sample_data(sub_nest_df), nest_df[1,]$data[[1]])
})