fix tests

mlr-org · Nov 10, 2021 · 2381eae · 2381eae
1 parent 54c14da
commit 2381eae
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 7 deletions.
diff --git a/R/operators.R b/R/operators.R
@@ -81,8 +81,6 @@
 #' o1 %>>!% o2
 #'
 #' o1  # not changed, becuase not a Graph.
-#'
-#' concat_graphs(glist = list(o1, o2, o3))
 concat_graphs = function(g1, g2, in_place = FALSE) {
   assert_flag(in_place)
   # neutral elements handling

diff --git a/man/concat_graphs.Rd b/man/concat_graphs.Rd
diff --git a/tests/testthat/test_pipeop_subsample.R b/tests/testthat/test_pipeop_subsample.R
@@ -50,7 +50,7 @@ test_that("PipeOpSubsample works stratified", {
   expect_true(tnew[[1]]$nrow == ceiling(task$nrow * 0.6))
   # Proportions as expected
   expect_equal(table(tnew[[1]]$data(cols = tnew[[1]]$target_names)),
-    table(rep(c("setosa", "versicolor", "virginica"), 30)))
+    table(list(Species = rep(c("setosa", "versicolor", "virginica"), 30))))
 
   po = PipeOpSubsample$new()
   po$param_set$values = list(stratify = TRUE, frac = 0.6, replace = TRUE)
@@ -60,7 +60,7 @@ test_that("PipeOpSubsample works stratified", {
   expect_true(tnew[[1]]$nrow == ceiling(task$nrow * 0.6))
   # Proportions as expected
   expect_equal(table(tnew[[1]]$data(cols = tnew[[1]]$target_names)),
-    table(rep(c("setosa", "versicolor", "virginica"), 30)))
+    table(list(Species = rep(c("setosa", "versicolor", "virginica"), 30))))
 
   po = PipeOpSubsample$new()
   po$param_set$values = list(stratify = TRUE, frac = 2, replace = TRUE)
@@ -70,7 +70,7 @@ test_that("PipeOpSubsample works stratified", {
   expect_true(tnew[[1]]$nrow == ceiling(task$nrow * 2))
   # Proportions as expected
   expect_equal(table(tnew[[1]]$data(cols = tnew[[1]]$target_names)),
-    table(rep(c("setosa", "versicolor", "virginica"), 100)))
+    table(list(Species = rep(c("setosa", "versicolor", "virginica"), 100))))
 })