Add tests for if_else()

voltrondata · Jan 24, 2023 · 0662fae · 0662fae
1 parent 8c7da3e
commit 0662fae
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 0 deletions.
diff --git a/tests/testthat/test-pkg-arrow.R b/tests/testthat/test-pkg-arrow.R
@@ -632,3 +632,18 @@ test_that("arrow translation for is.na() works", {
     )
   )
 })
+
+test_that("arrow translation for if_else() works", {
+  skip_if_not(has_arrow_with_substrait())
+
+  expect_equal(
+    example_data[1:5, "dbl"] %>%
+      arrow_substrait_compiler() %>%
+      substrait_project(dbl, gt_five = if_else(dbl > 5, "over", "under")) %>%
+      dplyr::collect(),
+    tibble::tibble(
+      dbl = c(-999, -99, -9, 0, 9),
+      gt_five = c("under", "under", "under", "under", "over")
+    )
+  )
+})
diff --git a/tests/testthat/test-pkg-duckdb.R b/tests/testthat/test-pkg-duckdb.R
@@ -309,3 +309,18 @@ test_that("duckdb translation for >= works", {
     tibble::tibble(dbl = c(0, 9))
   )
 })
+
+test_that("duckdb translation for if_else() works", {
+  skip_if_not(has_duckdb_with_substrait())
+
+  expect_equal(
+    example_data[1:5, "dbl"] %>%
+      duckdb_substrait_compiler() %>%
+      substrait_project(dbl, gt_five = if_else(dbl > 5, "over", "under")) %>%
+      dplyr::collect(),
+    tibble::tibble(
+      dbl = c(-999, -99, -9, 0, 9),
+      gt_five = c("under", "under", "under", "under", "over")
+    )
+  )
+})