tests/testthat/test_select.R

In slwu89/GA: Genetic Algorithm-based Variable Selection for Generalized Linear Models

library(GA)
context("select")

y <- as.matrix(mtcars$mpg)
x <- as.matrix(mtcars[2:11])

test_that('select works',
          expect_that(select(y = y, x = x, family = "gaussian", P=10, maxIter = 100)$count, is_less_than(100)))

test_that('select fails',
          expect_that(select(y = y, x = x, family = "gaussian", P=10, maxIter = 2)$warning, equals("Reached Maximum Iterations prior to convergence")))

test_that('select input error', {
          expect_error(select(y = y,x = "something", family = "gaussian", maxIter = 10))
          expect_error(select(y = rep(-1,10), x = x, family = "gamma", maxIter = 10))
          expect_error(select(y = y,x = x, P = c(10,20), family = "gaussian", maxIter = 10))
          expect_error(select(y = y,x = x, fitness_function= "something", family = "gaussian", maxIter = 10))
          expect_error(select(y = y,x = x, fitness= "other that weight or rank", family = "gaussian", maxIter = 10))
})