one more test

mlr-org · Apr 10, 2024 · d0331e4 · d0331e4
1 parent 11aa5ce
commit d0331e4
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/testthat/test_GraphLearner.R b/tests/testthat/test_GraphLearner.R
@@ -573,6 +573,7 @@ test_that("marshal", {
   task = tsk("iris")
   glrn = as_learner(as_graph(lrn("classif.debug")))
   glrn$train(task)
+  p1 = glrn$predict(task)
   glrn$marshal()
   expect_true(glrn$marshaled)
   expect_true(is_marshaled_model(glrn$state$model$marshaled$classif.debug))
@@ -581,7 +582,8 @@ test_that("marshal", {
   expect_class(glrn$model, "graph_learner_model")
   expect_false(is_marshaled_model(glrn$state$model$marshaled$classif.debug$model))
 
-  glrn$predict(task)
+  p2 = glrn$predict(task)
+  expect_equal(p1$response, p2$response)
 
   # checks that it is marshalable
   glrn$train(task)