tests/testthat/test-matching.R

In aknandi/disaggregation: Disaggregation Modelling

context("Match points to polygons")

test_that("Getting start and end index errors when it should", {

  covs <- data.frame(area_id = c(1, 1, 1, 2, 2, 3, 3, 3, 3), cov1 = c(3, 9, 5, 2, 3, 6, 7, 3, 5))
  response <- data.frame(area_id = c(1, 2, 3), response = c(4, 7, 2), N = c(NA, NA, NA))

  expect_error(getStartendindex(covs, response, 'id'))
  expect_error(getStartendindex(response, covs, 'id'))
})

test_that("Getting start and end index returns the right object", {

  covs <- data.frame(id = c(1, 1, 1, 2, 2, 3, 3, 3, 3, 4), cov1 = c(3, 9, 5, 2, 3, 6, 7, 3, 5, 6))
  response <- data.frame(area_id = c(1, 2, 3), response = c(4, 7, 2), N = c(NA, NA, NA))

  result <- getStartendindex(covs, response, 'id')
  save(result, file = paste0(tempdir(), '/test_startendindex.RData'))

  expect_is(result, "matrix")
  expect_equal(nrow(result), nrow(response))
  expect_equal(ncol(result), 2)
  expect_equal(result, matrix(c(0, 3, 5, 2, 4, 8), nrow = 3, ncol = 2))
})