handle leading spaces in variable names with clean_names

closes #85
sfirke · Jan 8, 2017 · ec9916f · ec9916f
1 parent 53ca562
commit ec9916f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/R/clean_names.R b/R/clean_names.R
@@ -24,6 +24,7 @@ clean_names <- function(dat){
     gsub("'", "", .) %>% # remove quotation marks
     gsub("\"", "", .) %>% # remove quotation marks
     gsub("%", "percent", .) %>%
+    gsub("^[ ]+", "", .) %>%
     make.names(.) %>%
     gsub("[.]+", "_", .) %>% # convert 1+ periods to single _
     gsub("[_]+", "_", .) %>% # fix rare cases of multiple consecutive underscores

diff --git a/tests/testthat/test-clean-names.R b/tests/testthat/test-clean-names.R
@@ -3,9 +3,9 @@
 library(janitor)
 context("data.frame name cleaning")
 
-test_df <- data.frame(matrix(ncol = 10) %>% as.data.frame())
+test_df <- data.frame(matrix(ncol = 11) %>% as.data.frame())
 names(test_df) <- c("sp ace", "repeated", "a**#@", "%", "#", "!",
-                    "d(!)9", "REPEATED", "can\"'t", "hi_`there`")
+                    "d(!)9", "REPEATED", "can\"'t", "hi_`there`", "  leading spaces")
 
 clean <- clean_names(test_df)
 
@@ -20,6 +20,7 @@ test_that("Names are cleaned appropriately", {
   expect_equal(names(clean)[8], "repeated_2") # uppercase, 2nd instance of repeat
   expect_equal(names(clean)[9], "cant") # uppercase, 2nd instance of repeat
   expect_equal(names(clean)[10], "hi_there") # double-underscores to single
+  expect_equal(names(clean)[11], "leading_space") # leading spaces
 })
 
 test_that("Returns a data.frame", {