Skip to content

Commit d0331e4

Browse files
committed
one more test
1 parent 11aa5ce commit d0331e4

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

tests/testthat/test_GraphLearner.R

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -573,6 +573,7 @@ test_that("marshal", {
573573
task = tsk("iris")
574574
glrn = as_learner(as_graph(lrn("classif.debug")))
575575
glrn$train(task)
576+
p1 = glrn$predict(task)
576577
glrn$marshal()
577578
expect_true(glrn$marshaled)
578579
expect_true(is_marshaled_model(glrn$state$model$marshaled$classif.debug))
@@ -581,7 +582,8 @@ test_that("marshal", {
581582
expect_class(glrn$model, "graph_learner_model")
582583
expect_false(is_marshaled_model(glrn$state$model$marshaled$classif.debug$model))
583584

584-
glrn$predict(task)
585+
p2 = glrn$predict(task)
586+
expect_equal(p1$response, p2$response)
585587

586588
# checks that it is marshalable
587589
glrn$train(task)

0 commit comments

Comments
 (0)