tests/testthat/test_DataBackend.R

In mlr3: Machine Learning in R - Next Generation

test_that("Nested backends", {
  data = as.data.table(iris)
  data$Petal.Length[91:120] = NA
  data$id = 1:150

  b1 = as_data_backend(data[1:100, -"Sepal.Length"], primary_key = "id")
  b2 = as_data_backend(data[101:130, -"Sepal.Length"], primary_key = "id")
  b3 = DataBackendRbind$new(b1, b2)
  expect_backend(b3)

  b4 = as_data_backend(data[131:150, -"Sepal.Length"], primary_key = "id")
  b5 = DataBackendRbind$new(b3, b4)
  expect_backend(b5)

  b6 = as_data_backend(data[, c("id", "Sepal.Length")], primary_key = "id")
  b7 = DataBackendCbind$new(b5, b6)
  expect_backend(b7)

  expect_iris_backend(b7, n_missing = 30L)

  x = b7$missings(b7$rownames, c("Petal.Width", "Petal.Length"))
  expect_equal(x, set_names(c(0L, 30L), c("Petal.Width", "Petal.Length")))
})