Test out adding in filenames

apache · Jul 4, 2022 · 9911ea6 · 9911ea6
1 parent 897a4c0
commit 9911ea6
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 1 deletion.
diff --git a/r/R/dataset.R b/r/R/dataset.R
@@ -438,3 +438,10 @@ stop_if_no_datasets <- function() {
     stop("This build of the arrow package does not support Datasets", call. = FALSE)
   }
 }
+
+#' Add filename to the dataset
+#'
+#' @export
+add_filenames <- function(){
+  Expression$field_ref("__filename")
+}
diff --git a/r/tests/testthat/test-dataset.R b/r/tests/testthat/test-dataset.R
@@ -1015,7 +1015,6 @@ test_that("dataset to C-interface to arrow_dplyr_query with proj/filter", {
   delete_arrow_array_stream(stream_ptr)
 })
 
-
 test_that("Filter parquet dataset with is.na ARROW-15312", {
   ds_path <- make_temp_dir()
 
@@ -1233,3 +1232,12 @@ test_that("FileSystemFactoryOptions input validation", {
     fixed = TRUE
   )
 })
+
+test_that("can add in augmented fields", {
+  ds <- open_dataset(hive_dir)
+
+  ds <- ds %>%
+    mutate(file_name = add_filenames()) %>%
+    collect()
+
+})