tests/testthat/test-rf.R

In andybega/demspaces: Functions for the Democratic Spaces Project

library(modeldata)
data(credit_data)

credit_data <- credit_data[complete.cases(credit_data), ]

credit_features <- credit_data[, setdiff(colnames(credit_data), "Status")]

test_that("rf works", {

  expect_error(
    mdl <- rf(credit_features, credit_data$Status),
    NA
  )
  expect_s3_class(mdl, "rf")

})

test_that("predict.rf works", {

  mdl <- rf(credit_features, credit_data$Status)

  expect_error(
    preds <- predict(mdl, new_data = credit_data),
    NA
  )

  expect_equal(
    colnames(preds),
    paste0("p_", levels(credit_data$Status))
  )

})


data("states")
states <- states %>%
  dplyr::filter(complete.cases(.))


test_that("ds_rf work", {

  expect_error(
    mdl <- ds_rf("v2x_veracc_osp", states),
    NA
  )

})

test_that("predict.ds_rf works", {

  mdl <- ds_rf("v2x_veracc_osp", states)

  expect_error(
    preds <- predict(mdl, new_data = states),
    NA
  )

})