File: test_base_makeTask.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 (39 lines) | stat: -rwxr-xr-x 1,373 bytes parent folder | download | duplicates (3)
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
test_that("makeXXTask ignores bad columns if check.data = FALSE", {
  constructTask = function(data, target, type, id) {
    constructor = switch(type,
      classif = makeClassifTask,
      multilabel = makeMultilabelTask,
      regr = makeRegrTask,
      surv = makeSurvTask)
    constructor(id = id, data = data, target = target, fixup.data = "no", check.data = FALSE)
  }

  badcoldf = data.frame(x = c(TRUE, TRUE, FALSE), y = c("x", "y", "z"),
    stringsAsFactors = FALSE)
  expect_logical(badcoldf[[1]])
  expect_character(badcoldf[[2]])

  constructors = list(
    classif = function(...) {
      makeClassifTask("test", cbind(badcoldf, target = "x"),
        target = "target", ...)
    },
    multilabel = function(...) {
      makeMultilabelTask("test", cbind(badcoldf,
        target1 = TRUE, target2 = FALSE), target = c("target1", "target2"), ...)
    },
    regr = function(...) {
      makeRegrTask("test", cbind(badcoldf, target = 1),
        target = "target", ...)
    },
    surv = function(...) {
      makeSurvTask("test", cbind(badcoldf, target1 = 1,
        target2 = FALSE), target = c("target1", "target2"), ...)
    },
    cluster = function(...) makeClusterTask("test", badcoldf, ...))

  for (type in names(constructors)) {
    expect_error(constructors[[type]](),
      "Unsupported feature type (logical) in column 'x'", fixed = TRUE)
  }
})