tests/testthat/test-prepare_data.R

In tjpajala/vaalikone: Scripts for analysing Finnish Election Machine datasets

context("test-prepare_data")

test_that("prepare data", {
  dataset_name <- "yle_2011"
  data <- get_dataset(dataset_name,filter=c("01 Helsingin vaalipiiri"))
  q_col=get_data_cols(dataset_name, data)
  party_col <- get_functional_column_name(data,alternative_spellings = c("puolue","Puolue","party"))
  data <- prepare_data(data, q_col, party_col, limit=10)
  nullrows <- data[rowSums(is.na(data[,q_col]))!=0,]
  
  expect_true(is.factor(data[[party_col]]))
  expect_equal(nrow(nullrows), 0)
})