File: test_base_getTaskFormula.R

package info (click to toggle)
r-cran-mlr 2.19.1%2Bdfsg-1
  • links: PTS, VCS
  • area: main
  • in suites: bookworm
  • size: 8,392 kB
  • sloc: ansic: 65; sh: 13; makefile: 5
file content (47 lines) | stat: -rwxr-xr-x 1,591 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47

test_that("getTaskFormula", {
  ## binaryclass
  expect_equal(binaryclass.formula, getTaskFormula(binaryclass.task),
    ignore_formula_env = TRUE)
  my.binaryclass.formula = paste(
    binaryclass.target, "~",
    collapse(colnames(binaryclass.df[, -binaryclass.class.col]), sep = " + "))
  expect_equal(
    as.formula(my.binaryclass.formula),
    getTaskFormula(binaryclass.task, explicit.features = TRUE),
    ignore_formula_env = TRUE)

  ## multiclass
  expect_equal(multiclass.formula, getTaskFormula(multiclass.task),
    ignore_formula_env = TRUE)
  my.multiclass.formula = paste(
    multiclass.target, "~",
    collapse(colnames(multiclass.df[, -multiclass.class.col]), sep = " + "))
  expect_equal(
    as.formula(my.multiclass.formula),
    getTaskFormula(multiclass.task, explicit.features = TRUE))

  ## regr
  expect_equal(regr.formula, getTaskFormula(regr.task),
    ignore_formula_env = TRUE)
  my.regr.formula = paste(
    regr.target, "~",
    collapse(colnames(regr.df[, -regr.class.col]), sep = " + "))
  expect_equal(
    as.formula(my.regr.formula),
    getTaskFormula(regr.task, explicit.features = TRUE))

  ## regr
  expect_equal(regr.num.formula, getTaskFormula(regr.num.task),
    ignore_formula_env = TRUE)
  my.regr.num.formula = paste(
    regr.num.target, "~",
    collapse(colnames(regr.num.df[, -regr.num.class.col]), sep = " + "))
  expect_equal(
    as.formula(my.regr.num.formula),
    getTaskFormula(regr.num.task, explicit.features = TRUE))
})

test_that("issue #1467", {
  expect_error(getTaskFormula(unclass(iris.task)), "no applicable method")
})