tests/testthat/test_TaskClassif.R

In mlr-org/mlrng: Machine Learning in R - Next Generation

context("TaskClassif")

test_that("Basic ops on iris task", {
  dd = as.data.table(iris)
  task = mlr.tasks$get("iris")
  expect_classiftask(task)
  expect_class(task, "TaskClassif")
  expect_equal(task$target, "Species")

  expect_set_equal(task$classes, levels(iris$Species))
  expect_identical(task$nclasses, nlevels(iris$Species))

  f = task$formula
  expect_class(f, "formula")
  expect_set_equal(attr(terms(f), "term.labels"), setdiff(names(dd), "Species"))
})