Skip to content

Commit

Permalink
pequena correcao na padronizar_bairros()
Browse files Browse the repository at this point in the history
  • Loading branch information
dhersz committed Feb 7, 2024
1 parent cde4cf8 commit 17697f8
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 1 deletion.
4 changes: 3 additions & 1 deletion R/padronizar_bairros.R
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down Expand Up @@ -138,5 +138,7 @@ padronizar_bairros <- function(bairros) {
)
)

bairros_padrao[indice_bairro_vazio] <- ""

return(bairros_padrao)
}
1 change: 1 addition & 0 deletions tests/testthat/test-padronizar_bairros.R
Original file line number Diff line number Diff line change
Expand Up @@ -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", {
Expand Down

0 comments on commit 17697f8

Please sign in to comment.