test tar_format_nanoparquet()

ropensci · Sep 25, 2024 · cf89e84 · cf89e84
1 parent 14606ee
commit cf89e84
Show file tree

Hide file tree

Showing 2 changed files with 34 additions and 1 deletion.
diff --git a/R/tar_format_nanoparquet.R b/R/tar_format_nanoparquet.R
@@ -23,7 +23,7 @@
 #'   list(
 #'     tar_target(
 #'       name = data,
-#'       command = airquality,
+#'       command = data.frame(x = 1),
 #'       format = tar_format_nanoparquet()
 #'     )
 #'   )

diff --git a/tests/testthat/test-tar_format_nanoparquet.R b/tests/testthat/test-tar_format_nanoparquet.R
@@ -0,0 +1,33 @@
+targets::tar_test("tar_format_nanoparquet()", {
+  skip_if_not_installed("nanoparquet")
+  targets::tar_script({
+    list(
+      tar_target(
+        name = data,
+        command = data.frame(x = 1),
+        format = tar_format_nanoparquet()
+      )
+    )
+  })
+  targets::tar_make(callr_function = NULL)
+  out <- targets::tar_read(data)
+  expect_equal(as.data.frame(out), data.frame(x = 1))
+  expect_equal(class(out), c("tbl", "data.frame"))
+})
+
+targets::tar_test("tar_format_nanoparquet() alternate options", {
+  skip_if_not_installed("nanoparquet")
+  targets::tar_script({
+    list(
+      tar_target(
+        name = data,
+        command = data.frame(x = 1),
+        format = tar_format_nanoparquet(compression = "gzip", class = "x")
+      )
+    )
+  })
+  targets::tar_make(callr_function = NULL)
+  out <- targets::tar_read(data)
+  expect_equal(as.data.frame(out), data.frame(x = 1))
+  expect_equal(class(out), c("x", "data.frame"))
+})