Error if wrong schema in pl$DataFrame() (#486)

pola-rs · Nov 9, 2023 · c97f6bd · c97f6bd
1 parent 5aa0603
commit c97f6bd
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 4 deletions.
diff --git a/NEWS.md b/NEWS.md
@@ -4,6 +4,8 @@
 
 - The argument `quote_style` in `$write_csv()` and `$sink_csv()` can now take 
   the value `"never"` (#483).
+- `pl$DataFrame()` now errors if the variables specified in `schema` do not exist
+  in the data (#486).
 
 # polars 0.10.0
 

diff --git a/R/dataframe__frame.R b/R/dataframe__frame.R
@@ -136,6 +136,13 @@ DataFrame
 pl$DataFrame = function(..., make_names_unique = TRUE, schema = NULL) {
   largs = unpack_list(...)
 
+  uw = \(res) unwrap(res, "in $DataFrame():")
+
+  if (!is.null(schema) && !all(names(schema) %in% names(largs))) {
+    Err_plain("Some columns in `schema` are not in the DataFrame.") |>
+      uw()
+  }
+
   # no args crete empty DataFrame
   if (length(largs) == 0L) {
     return(.pr$DataFrame$default())
@@ -188,7 +195,7 @@ pl$DataFrame = function(..., make_names_unique = TRUE, schema = NULL) {
     }) |>
       do.call(what = pl$select)
   }) |>
-    unwrap("in pl$DataFrame()")
+    uw()
 }
 
 

diff --git a/tests/testthat/test-dataframe.R b/tests/testthat/test-dataframe.R
@@ -160,15 +160,18 @@ test_that("DataFrame, custom schema", {
   expect_no_error(
     pl$DataFrame(list(schema = 1), schema = list(schema = pl$Float32))
   )
-  # errors if incorrect datatype
+  # incorrect datatype
   expect_error(pl$DataFrame(x = 1, schema = list(schema = foo)))
   expect_error(
     pl$DataFrame(x = 1, schema = list(x = "foo")),
     "expected RPolarsDataType"
   )
 
-  # TODO: why doesn't this error?
-  # expect_error(pl$DataFrame(x = 1, schema = list(schema = pl$foo)))
+  # wrong variable name in schema
+  expect_error(
+    pl$DataFrame(x = 1, schema = list(schema = pl$Float32)),
+    "Some columns in `schema` are not in the DataFrame"
+  )
 })