tests/testthat/test_local_convertOMLDataSetToMlr.R

In openml/openml-r: Open Machine Learning and Open Data Platform

test_that("convertOMLDataSetToMlr", {
  with_test_cache({
    ds = getOMLDataSet(10)

    expect_is_mlr_task = function(mlr.task, ds) {
      expect_equal(mlr.task$type, "classif")
      expect_equal(mlr.task$task.desc$size, nrow(ds$data))
      expect_equal(ds$desc$default.target.attribute, mlr.task$task.desc$target)
    }

    # now create the task
    mlr.task = convertOMLDataSetToMlr(ds)
    expect_equal(getTaskType(mlr.task), "classif")

    # now modify dataset by hand (no more server calls) to check
    # ignore attributes stuff:
    # Define the first two attributes as ignored attributes
    ds$desc$ignore.attribute = colnames(ds$data[, 1:2])

    mlr.task = convertOMLDataSetToMlr(ds, ignore.flagged.attributes = TRUE)
    expect_is_mlr_task(mlr.task, ds)
    # we removed two attributes (and the target column is not considered here)
    expect_equal(sum(mlr.task$task.desc$n.feat), ncol(ds$data) - 3L)

    # pass faulty parameters
    expect_error(convertOMLDataSetToMlr(ds, task.type = "Nonexistent task type"), "element of")

    # check setting mlr task id
    expect_equal(getTaskId(convertOMLDataSetToMlr(ds)), ds$desc$name)
    expect_equal(getTaskId(convertOMLDataSetToMlr(ds, mlr.task.id = "<oml.data.name>.<oml.data.id>")),
      sprintf("%s.%s", ds$desc$name, ds$desc$id))
    expect_equal(getTaskId(convertOMLDataSetToMlr(ds, mlr.task.id = "test")), "test")
    expect_equal(getTaskId(convertOMLDataSetToMlr(ds, mlr.task.id = "<oml.data.id>")), as.character(ds$desc$id))
    expect_equal(getTaskId(convertOMLDataSetToMlr(ds, mlr.task.id = "<oml.data.name>")), as.character(ds$desc$name))
    expect_equal(getTaskId(convertOMLDataSetToMlr(ds, mlr.task.id = "<oml.data.version>")), as.character(ds$desc$version))
    expect_equal(getTaskId(convertOMLDataSetToMlr(ds, mlr.task.id = "<oml.task.id>")), "<oml.task.id>")

    # check if conversion to regression task works
    ds$desc$target.features = ds$desc$default.target.attribute = "no_of_nodes_in"
    expect_equal(getTaskType(convertOMLDataSetToMlr(ds)), "regr")

    # check if conversion to multilabel task works
    ds$desc$target.features = ds$desc$default.target.attribute = c("bl_of_lymph_c", "bl_of_lymph_s")
    expect_error(convertOMLDataSetToMlr(ds), "logical")
    ds$data$bl_of_lymph_c = as.logical(as.numeric(ds$data$bl_of_lymph_c) - 1)
    ds$data$bl_of_lymph_s = as.logical(as.numeric(ds$data$bl_of_lymph_s) - 1)
    multilab.task = convertOMLDataSetToMlr(ds)
    expect_equal(getTaskType(multilab.task), "multilabel")
  })
})