diff --git a/R/padronizar_bairros.R b/R/padronizar_bairros.R index c1dbb5a..f2685d7 100644 --- a/R/padronizar_bairros.R +++ b/R/padronizar_bairros.R @@ -29,7 +29,7 @@ padronizar_bairros <- function(bairros) { # identificamos o indice dos bairros vazios para "reesvazia-los" ao final, # ja que a sequencia de operacoes abaixo acabaria atribuindo um valor a eles - indice_bairro_vazio <- which(bairros == "" | is.na(bairros)) + indice_bairro_vazio <- which(is.na(bairros)) bairros_padrao <- stringr::str_squish(bairros) bairros_padrao <- toupper(bairros_padrao) @@ -138,5 +138,7 @@ padronizar_bairros <- function(bairros) { ) ) + bairros_padrao[indice_bairro_vazio] <- "" + return(bairros_padrao) } diff --git a/tests/testthat/test-padronizar_bairros.R b/tests/testthat/test-padronizar_bairros.R index b697d35..d8957c3 100644 --- a/tests/testthat/test-padronizar_bairros.R +++ b/tests/testthat/test-padronizar_bairros.R @@ -10,6 +10,7 @@ test_that("padroniza corretamente", { expect_equal(padronizar_bairros("JARDIM BOTÂNICO"), "JARDIM BOTANICO") expect_equal(padronizar_bairros("jardim botanico"), "JARDIM BOTANICO") expect_equal(padronizar_bairros("jd..botanico"), "JARDIM BOTANICO") + expect_equal(padronizar_bairros(NA_character_), "") }) test_that("lida com vetores vazios corretamente", {