tests/testthat/ranger/completed/test_seed.R

In literanger: Random Forests for Multiple Imputation Based on 'ranger'

## Tests for using seeds

library(ranger)
context("ranger_seed")

## Initialize the random forests
ind = 1:150 %in% sample(150, 100)

set.seed(2)
mod1 = ranger(Species ~ ., data = iris[ind, ], write.forest = TRUE, num.trees = 50)
pred1 = predict(mod1, data = iris[!ind, ])

set.seed(2)
mod2 = ranger(Species ~ ., data = iris[ind, ], write.forest = TRUE, num.trees = 50)
pred2 = predict(mod2, data = iris[!ind, ])

set.seed(2)
mod3 = ranger(dependent.variable.name = "Species", data = iris[ind, ], write.forest = TRUE, num.trees = 50)
pred3 = predict(mod3, data = iris[!ind, ])

## Tests
test_that("same result with same seed", {
  expect_equal(pred1$predictions, pred2$predictions)
})

test_that("same result with same seed, different interface", {
  expect_equal(pred1$predictions, pred3$predictions)
})