tests/testthat/test_running_environment.R

In jatherrien/largeRCRF: Large Random Competing Risks Forests

context("Train and predict without error using an environment")

test_that("Training with environment works", {

  sampleData <- data.frame(x=rnorm(100))
  sampleData$T <- rexp(100) + abs(sampleData$x)
  sampleData$delta <- sample(0:2, size = 100, replace=TRUE)
  
  testData <- sampleData[1:5,]
  trainingData <- sampleData[6:100,]
  
  e <- new.env()
  e$data <- trainingData
  rm(trainingData)
  
  forest <- train(CR_Response(delta, T) ~ x, e, ntree=50, numberOfSplits=0, mtry=1, nodeSize=5, cores=2, displayProgress=FALSE)
  
  expect_null(e$data)
  
  predictions <- predict(forest, testData)
  
  forest <- addTrees(forest, 50, displayProgress = FALSE)
  
  predictions <- predict(forest, testData)
  
  expect_true(T) # show Ok if we got this far
  
})