Skip to content

Commit 6dcb2de

Browse files
Update tests/testthat/test_postprocessing_helpers.R
Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>
1 parent a2663d4 commit 6dcb2de

File tree

1 file changed

+10
-1
lines changed

1 file changed

+10
-1
lines changed

tests/testthat/test_postprocessing_helpers.R

Lines changed: 10 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -86,7 +86,16 @@ test_that("keras_postprocess_probs handles multi-output (named list) correctly",
8686
names(results) <- c("output1", "output2")
8787
processed <- keras_postprocess_probs(results, mock_object_multi_output)
8888
expect_s3_class(processed, "tbl_df")
89-
expect_equal(names(processed), c(".pred_output1_classA", ".pred_output1_classB", ".pred_output2_typeX", ".pred_output2_typeY", ".pred_output2_typeZ"))
89+
expect_equal(
90+
names(processed),
91+
c(
92+
".pred_output1_classA",
93+
".pred_output1_classB",
94+
".pred_output2_typeX",
95+
".pred_output2_typeY",
96+
".pred_output2_typeZ"
97+
)
98+
)
9099
expect_equal(processed$.pred_output1_classA, c(0.1, 0.2))
91100
expect_equal(processed$.pred_output2_typeX, c(0.3, 0.5))
92101
})

0 commit comments

Comments
 (0)