From a6aa25500fa4625d1757baab1a37debcb4a8fd61 Mon Sep 17 00:00:00 2001
From: eitsupi <ts1s1andn@gmail.com>
Date: Sat, 29 Oct 2022 09:31:57 +0000
Subject: [PATCH] add test for arrow dataset

---
 tests/testthat/test-arrow.R | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/tests/testthat/test-arrow.R b/tests/testthat/test-arrow.R
index 7dbe52f..5e675a3 100644
--- a/tests/testthat/test-arrow.R
+++ b/tests/testthat/test-arrow.R
@@ -47,6 +47,26 @@ test_that("Full example #1 returns expected result on ArrowTabular", {
   )
 })
 
+test_that("Simple SELECT example query #1 returns expected result on arrow Dataset", {
+  skip_if_not_installed("arrow")
+  skip_if_not(arrow::arrow_info()$capabilities["dataset"], message = "Arrow Datasets not available")
+
+  tmp <- tempfile()
+  on.exit(unlink(tmp))
+  arrow::write_parquet(iris, tmp)
+  iris_arrow <- arrow::open_dataset(tmp)
+  expect_equal(
+    query(
+      "SELECT Species, COUNT(*) AS n FROM iris_arrow GROUP BY Species"
+    ) %>% collect(),
+    iris_arrow %>%
+      group_by(Species) %>%
+      summarise(n = n()) %>%
+      ungroup() %>%
+      collect()
+  )
+})
+
 test_that("query() fails when input ArrowTabular is grouped", {
   skip_if_not_installed("arrow")
   skip_if_not_installed("nycflights13")