use dplyr::as_tibble() in arrow tests

EmilHvitfeldt · EmilHvitfeldt · commit 230d0a40d8fb · 2025-10-15T13:46:39.000-07:00
diff --git a/tests/testthat/test-adjust_numeric_range.R b/tests/testthat/test-adjust_numeric_range.R
@@ -190,7 +190,7 @@ test_that("arrow - adjust_predictions_custom works", {
   mtcars_tbl <- arrow::as_arrow_table(mtcars)
 
   expect_identical(
-    dplyr::collect(predict(res, mtcars_tbl)),
-    predict(tlr_fit, mtcars)
+    dplyr::as_tibble(dplyr::collect(predict(res, mtcars_tbl))),
+    dplyr::as_tibble(predict(tlr_fit, mtcars))
   )
 })
diff --git a/tests/testthat/test-adjust_predictions_custom.R b/tests/testthat/test-adjust_predictions_custom.R
@@ -174,7 +174,7 @@ test_that("arrow - adjust_predictions_custom works", {
   mtcars_tbl <- arrow::as_arrow_table(mtcars)
 
   expect_identical(
-    dplyr::collect(predict(res, mtcars_tbl)),
-    predict(tlr_fit, mtcars)
+    dplyr::as_tibble(dplyr::collect(predict(res, mtcars_tbl))),
+    dplyr::as_tibble(predict(tlr_fit, mtcars))
   )
 })

Original file line number	Diff line number	Diff line change
`@@ -190,7 +190,7 @@ test_that("arrow - adjust_predictions_custom works", {`
`190`	`190`	`mtcars_tbl <- arrow::as_arrow_table(mtcars)`
`191`	`191`
`192`	`192`	`expect_identical(`
`193`		`- dplyr::collect(predict(res, mtcars_tbl)),`
`194`		`- predict(tlr_fit, mtcars)`
	`193`	`+ dplyr::as_tibble(dplyr::collect(predict(res, mtcars_tbl))),`
	`194`	`+ dplyr::as_tibble(predict(tlr_fit, mtcars))`
`195`	`195`	`)`
`196`	`196`	`})`
Original file line number	Diff line number	Diff line change
`@@ -174,7 +174,7 @@ test_that("arrow - adjust_predictions_custom works", {`
`174`	`174`	`mtcars_tbl <- arrow::as_arrow_table(mtcars)`
`175`	`175`
`176`	`176`	`expect_identical(`
`177`		`- dplyr::collect(predict(res, mtcars_tbl)),`
`178`		`- predict(tlr_fit, mtcars)`
	`177`	`+ dplyr::as_tibble(dplyr::collect(predict(res, mtcars_tbl))),`
	`178`	`+ dplyr::as_tibble(predict(tlr_fit, mtcars))`
`179`	`179`	`)`
`180`	`180`	`})`