allow NA for set_label_attribute()

larmarange · Jan 7, 2025 · ee901d0 · ee901d0
1 parent 494a22a
commit ee901d0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/R/var_label.R b/R/var_label.R
@@ -299,7 +299,7 @@ get_label_attribute <- function(x) {
 #' @rdname var_label
 #' @export
 set_label_attribute <- function(x, value) {
-  check_string(value, allow_null = TRUE)
+  check_string(value, allow_null = TRUE, allow_na = TRUE)
   attr(x, "label") <- value
   x
 }

diff --git a/tests/testthat/test-labelled.r b/tests/testthat/test-labelled.r
@@ -8,6 +8,7 @@ test_that("var_label works properly", {
   var_label(x) <- NULL
   expect_null(attr(x, "label"))
   expect_null(var_label(x))
+  expect_no_error(var_label(x) <- NA_character_)
 
   x <- 1:3
   x <- set_variable_labels(x, "value")