tests/testthat/test-inputdata.R

In mfrdixon/OSTSC: Over Sampling for Time Series Classification

library(xts)
context("input data")

test_that("input data could be in xts format", {
  # create xts data
  data("Dataset_Synthetic_Control")
  train.label <- Dataset_Synthetic_Control$train.y
  train.sample <- Dataset_Synthetic_Control$train.x
  train <- cbind(train.label, train.sample)
  df.xts <- as.xts(train, order.by=Sys.time() + 1:dim(train)[1])
  df.label <- df.xts[, c(1)]
  df.sample <- df.xts[, -1]
  # oversampling
  MyData <- OSTSC(df.sample, df.label, parallel = FALSE)
  label <- MyData$label
  expect_equal( length(label[which(label == 0)]), length(label[which(label == 1)]) )
})

test_that("input data could be in array format", {
  # create array
  data("Dataset_Synthetic_Control")
  train.label <- Dataset_Synthetic_Control$train.y
  train.sample <- Dataset_Synthetic_Control$train.x
  df.label <- as.array(train.label)
  df.sample <- as.array(train.sample)
  # oversampling
  MyData <- OSTSC(df.sample, df.label, parallel = FALSE)
  label <- MyData$label
  expect_equal( length(label[which(label == 0)]), length(label[which(label == 1)]) )
})

test_that("input data could be in data frame format", {
  # create data frame
  data("Dataset_Synthetic_Control")
  train.label <- Dataset_Synthetic_Control$train.y
  train.sample <- Dataset_Synthetic_Control$train.x
  df.label <- as.data.frame(train.label)
  df.sample <- as.data.frame(train.sample)
  # oversampling
  MyData <- OSTSC(df.sample, df.label, parallel = FALSE)
  label <- MyData$label
  expect_equal( length(label[which(label == 0)]), length(label[which(label == 1)]) )
})

test_that("input sample and label data have same first dimension", {
  # create samples and labels with different records
  data("Dataset_Synthetic_Control")
  train.label <- Dataset_Synthetic_Control$train.y
  train.sample <- Dataset_Synthetic_Control$train.x
  df.label <- train.label
  df.sample <- train.sample[-1, ]
  # oversampling
  expect_error( OSTSC(df.sample, df.label, parallel = FALSE), "Number of time series sequences provided" )
})

test_that("OSTSC handles datasets containing NA, NaN, or accidental strings", {
  # create data containing NA, NaN, and accidental strings
  data("Dataset_Synthetic_Control")
  train.label <- Dataset_Synthetic_Control$train.y
  train.sample <- Dataset_Synthetic_Control$train.x
  train.sample[1, 3] <- NaN
  train.sample[4, 1] <- NA 
  train.sample[5, 5] <- 'a'
  # oversampling
  MyData <- OSTSC(train.sample, train.label, parallel = FALSE)
  label <- MyData$label
  expect_equal( length(label[which(label == 0)]), length(label[which(label == 1)]) )
})