tests/testthat/test-mutate_geocode.R

In HughParsonage/PSMA: Geocoding and Reverse-Geocoding of Australian Locations

context("test-mutate_geocode.R")

test_that("Error handling", {
  expect_error(mutate_geocode(list()),
               "`DT` .* data.frame")
  if (exists("bne_addresses")) {
    skip_if_not(is.data.frame(bne_addresses) &&
                  identical(dim(bne_addresses),
                            c(10L, 12L)))
  } else {
    load(system.file("extdata", "bne_addresses.rda", package = "PSMA"))
  }
  expect_error(mutate_geocode(bne_addresses,
                              number_first = house_number,
                              new_names = ""),
               regexp = "two names are required")
  expect_error(mutate_geocode(bne_addresses,
                              number_first = house_number,
                              new_names = 1:2),
               regexp = "character vector")
  expect_error(mutate_geocode(mutate_geocode(bne_addresses,
                                             number_first = house_number),
                              number_first = house_number,
                              overwrite = FALSE),
               regexp = "First entry.*but this is already a column name in DT")
  expect_error(mutate_geocode(mutate_geocode(bne_addresses,
                                             number_first = house_number),
                              number_first = house_number,
                              new_names = c("foo", "lon"),
                              overwrite = FALSE),
               regexp = "Second entry.*but this is already a column name in DT")


})

test_that("tibbles", {
  skip_if_not_installed("tibble")
  out <-
    mutate_geocode(tibble::tibble(flat_number = NA_character_,
                                  number_first = 1L,
                                  building_name = NA_character_,
                                  street_name = "George",
                                  street_type = "Street",
                                  postcode = 2000L))
  expect_true(tibble::is_tibble(out))
  expect_true("lat" %in% names(out))
  expect_true("lon" %in% names(out))
  expect_gt(out$lat[1L], -34)

})

test_that("Formals match", {
  skip_on_cran()
  req_formals <-
    c("flat_number", "number_first", "building_name", "street_name",
      "street_type", "postcode")
  expect_true(all(req_formals %in% names(formals(geocode))),
              label = "geocode() has consistent arguments.")
  expect_true(all(req_formals %in% names(formals(mutate_geocode))),
              label = "mutate_geocode() has consistent arguments.")
})

test_that("mutate works with Nick's example", {
  if (exists("bne_addresses")) {
    skip_if_not(is.data.frame(bne_addresses) &&
                  identical(dim(bne_addresses),
                            c(10L, 12L)))
  } else {
    load(system.file("extdata", "bne_addresses.rda", package = "PSMA"))
  }

  M <- mutate_geocode(bne_addresses, number_first = house_number)
  M4 <- M[4L, ]
  expect_equal(M4[["lat"]], -27.467, tol = 0.01)
  expect_equal(M4[["lon"]], 153.06, tol = 0.01)


  BNE <- bne_addresses
  BNE$flat_number <- NA_character_
  BNE$building_name <- NA_character_

  A <- mutate_geocode(bne_addresses,
                      number_first = house_number)
  A4 <- A[4L, ]
  expect_equal(A4[["lat"]], -27.467, tol = 0.01)
  expect_equal(A4[["lon"]], 153.06, tol = 0.01)




  load(system.file("extdata", "bne_addresses.rda", package = "PSMA"))

  BNE <- copy(bne_addresses)
  BNE$flat_number <- NA_character_
  BNE$building_name <- NA_character_

  A <- add_geocode(bne_addresses,
                   number_first = house_number)
  A4 <- A[4L, ]
  expect_equal(A4[["latitude"]], -27.467, tol = 0.01)
  expect_equal(A4[["longitude"]], 153.06, tol = 0.01)


})

test_that("Building names", {
  library(data.table)
  library(magrittr)
  dt <-
    data.table(flat_number = 3L,
               building_name = "AINSLIE FLATS",
               number_first = 11L,
               street_name = "QUICK",
               street_type = "Street",
               postcode = 2602L) %>%
    add_geocode
  dt2 <-
    data.table(flat_number = 3L,
               building_name = "AINSLIE FLATS",
               number_first = 11L,
               street_name = "QUICK",
               street_type = "Street",
               postcode = 2602L) %>%
    mutate_geocode(new_names = c("latitude", "longitude"))

  expect_identical(dt, dt2)
  expect_gt(dt[, latitude], -35.5)
  expect_lt(dt[, latitude], -35.2)
  expect_lt(dt[, longitude], 149.15)
  expect_gt(dt[, longitude], 149.14)

  setnames(dt, "flat_number", "unit_no")
  dt3 <- mutate_geocode(dt, flat_number = unit_no)
  dt4 <- mutate_geocode(dt, flat_number = 'unit_no', overwrite = TRUE)
  expect_lt(dt[, longitude], 149.15)
  expect_gt(dt[, longitude], 149.14)


  dt5 <-
    data.table(flat = 3L,
               bldg = "AINSLIE FLATS",
               number = 11L,
               street = "QUICK",
               type = "Street",
               Postcode = 2602L) %>%
    mutate_geocode(flat_number = flat,
                   number_first = number,
                   building_name = bldg,
                   street_name = street,
                   street_type = type,
                   postcode = Postcode)
  expect_lt(dt5[, lon], 149.15)
  expect_gt(dt5[, lon], 149.14)

})