tests/testthat/test_add_derived_column.r

In roqua/autovar: Automated Vector Autoregression Model Creation

context('add_derived_column')

test_that('ln only accepts existing columns',{
  av_state <- load_test_data()
  expect_error(add_derived_column(av_state,'new','nonexistant',operation='LN'),'does not exist')
})

test_that('ln only accepts numeric columns',{
  av_state <- load_test_data()
  expect_error(add_derived_column(av_state,'new','home',operation='LN'),'is not numeric')
})

test_that('ln gives the correct output',{
  av_state <- load_test_data()
  av_state <- add_derived_column(av_state,'new','tijdstip',operation='LN')
  expect_equivalent(av_state$data[[1]],cbind(generate_test_data(),log(generate_test_data()$tijdstip)))
})

test_that('ln does not change NA values',{
  av_state <- load_test_data()
  av_state$data[[1]]$tijdstip[2] <- NA
  av_state <- add_derived_column(av_state,'new','tijdstip',operation='LN')
  modded_test_data <- generate_test_data()
  modded_test_data$tijdstip[2] <- NA
  expect_equivalent(av_state$data[[1]],cbind(modded_test_data,log(modded_test_data$tijdstip)))
})

test_that('ln translates values to a minimum of 1',{
  av_state <- load_test_data()
  av_state$data[[1]]$tijdstip[2] <- -2
  expect_output({av_state <- add_derived_column(av_state,'new','tijdstip',operation='LN')},'increasing')
  modded_test_data <- generate_test_data()
  modded_test_data$tijdstip[2] <- -2
  expect_equivalent(av_state$data[[1]],cbind(modded_test_data,log(modded_test_data$tijdstip+3)))
})

test_that('mtoh only accepts existing columns',{
  av_state <- load_test_data()
  expect_error(add_derived_column(av_state,'new','nonexistant',operation='MINUTES_TO_HOURS'),'does not exist')
})

test_that('mtoh only accepts numeric columns',{
  av_state <- load_test_data()
  expect_error(add_derived_column(av_state,'new','home',operation='MINUTES_TO_HOURS'),'is not numeric')
})

test_that('mtoh gives the correct output',{
  av_state <- load_test_data()
  av_state <- add_derived_column(av_state,'new','tijdstip',operation='MINUTES_TO_HOURS')
  expect_equivalent(av_state$data[[1]],cbind(generate_test_data(),(generate_test_data()$tijdstip/60)))
})

test_that('sum only accepts existing columns',{
  av_state <- load_test_data()
  expect_error(add_derived_column(av_state,'new','nonexistant',operation='SUM'),'does not exist')
})

test_that('sum gives the correct output for numeric columns',{
  av_state <- load_test_data()
  av_state <- add_derived_column(av_state,'new',c('id','tijdstip'),operation='SUM')
  expect_equivalent(av_state$data[[1]],cbind(generate_test_data(),generate_test_data()$tijdstip+1))
})

test_that('sum gives the correct output for nominal columns',{
  av_state <- load_test_data()
  expect_output({av_state <- add_derived_column(av_state,'new',c('home','home'),operation='SUM')},'converting')
  expect_equivalent(av_state$data[[1]],cbind(generate_test_data(),c(2,0,2,0,2)))
})