tests/testthat/test_randomForest.R

In swager/randomForestCI: Confidence Intervals for Random Forests

set.seed(1987)

library(randomForest)

n = 250
p = 100
X = matrix(rnorm(n * p), n, p)
Y = rnorm(n)
Y_multiclass = factor(rep(c('a','b','c'), length.out = n), c('a','b','c'))
rf = randomForest(X, Y, keep.inbag = TRUE)
pred = predict(rf, newdata = X, predict.all = TRUE)

rf.small = randomForest(X[1:20, ], Y[1:20], keep.inbag = TRUE)

rf.multiclass = randomForest(X, Y_multiclass, keep.inbag = TRUE)

test_that("sanity", {
  ij = randomForestInfJack(rf, X, calibrate = FALSE)
  expect_that(all(is.finite(ij$var.hat)), is_true())
  
  ij.calibrated = randomForestInfJack(rf, X, calibrate = TRUE)
  expect_that(all(ij.calibrated$var.hat > 0), is_true())

  expect_warning(randomForestInfJack(rf, X[1:20, ], calibrate = TRUE))

  ij.standalone = infJack(pred$individual, rf$inbag, calibrate = FALSE)
  expect_equal(ij.standalone, ij)

  ij.standalone.calibrated = infJack(pred$individual, rf$inbag, calibrate = TRUE)
  expect_equal(ij.standalone.calibrated, ij.calibrated, tolerance = .1)
  
  ij.multiclass <- randomForestInfJackMulticlass(rf.multiclass, X, calibrate = FALSE)
  expect_that(all(is.finite(ij.multiclass$var.hat_a)), is_true())
  expect_that(all(is.finite(ij.multiclass$var.hat_b)), is_true())
  expect_that(all(is.finite(ij.multiclass$var.hat_b)), is_true())
  
  ij.calibrated.multiclass <- randomForestInfJackMulticlass(rf.multiclass, X, calibrate = TRUE)
  expect_that(all(ij.calibrated.multiclass$var.hat_a > 0), is_true())
  expect_that(all(ij.calibrated.multiclass$var.hat_b > 0), is_true())
  expect_that(all(ij.calibrated.multiclass$var.hat_c > 0), is_true())
})