update tests

mlr-org · Feb 7, 2025 · dac498f · dac498f
1 parent 3f9da4b
commit dac498f
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/tests/testthat/test_TaskClassif.R b/tests/testthat/test_TaskClassif.R
@@ -119,7 +119,7 @@ test_that("offset column role works with binary tasks", {
 
   task$set_col_roles("age", "offset")
   expect_subset("offset", task$properties)
-  expect_data_table(task$offset, nrows = task$nrow, ncols = 1)
+  expect_data_table(task$offset, nrows = task$nrow, ncols = 2)
   expect_subset("age", names(task$offset))
 
   expect_error({
@@ -138,8 +138,8 @@ test_that("offset column role works with multiclass tasks", {
   expect_null(task$offset)
   task$set_col_roles("year", "offset")
   expect_subset("offset", task$properties)
-  expect_data_table(task$offset, nrows = task$nrow, ncols = 1)
-  expect_subset("year", names(task$offset))
+  expect_data_table(task$offset, nrows = task$nrow, ncols = 2)
+  expect_subset(c("row_id", "year"), names(task$offset))
 
   expect_error({
     task$col_roles$offset = "bill_length"
@@ -159,6 +159,6 @@ test_that("offset column role works with multiclass tasks", {
   task$set_col_roles(c("offset_1", "offset_2"), "offset")
 
   expect_subset("offset", task$properties)
-  expect_data_table(task$offset, nrows = task$nrow, ncols = 2)
-  expect_subset(c("offset_1", "offset_2"), names(task$offset))
+  expect_data_table(task$offset, nrows = task$nrow, ncols = 3)
+  expect_subset(c("row_id", "offset_1", "offset_2"), names(task$offset))
 })
diff --git a/tests/testthat/test_TaskRegr.R b/tests/testthat/test_TaskRegr.R
@@ -56,8 +56,8 @@ test_that("offset column role works", {
   task$set_col_roles("am", "offset")
 
   expect_subset("offset", task$properties)
-  expect_data_table(task$offset, nrows = task$nrow, ncols = 1)
-  expect_subset("am", names(task$offset))
+  expect_data_table(task$offset, nrows = task$nrow, ncols = 2)
+  expect_subset(c("row_id", "am"), names(task$offset))
 
   expect_error({
     task$col_roles$offset = c("am", "gear")