tests/testthat/test-hierarchy.R

In bayesRecon: Probabilistic Reconciliation via Conditioning

test_that("get_reconc_matrices produces expected aggregation and summing matrices", {
  aggrs <- c(1, 3, 6, 12)
  h <- 12

  sort_aggrs <- sort(aggrs, decreasing = TRUE)
  expected_rowSumsS <- rep(sort_aggrs, h / sort_aggrs)
  expectedLenRowSumsA <- sum(h / sort_aggrs[-length(sort_aggrs)])
  expected_rowSumsA <- expected_rowSumsS[1:expectedLenRowSumsA]

  out <- get_reconc_matrices(aggrs, h)

  diff <- max(abs(expected_rowSumsA - rowSums(out$A))) + max(abs(expected_rowSumsS - rowSums(out$S)))
  expect_equal(diff, 0)
})

test_that(".get_Au and .lowest_lev produce expected outcomes", {
  A <- matrix(
    data = c(
      1, 1, 1, 1, 1, 1,
      1, 1, 0, 0, 0, 0,
      0, 0, 1, 1, 0, 0,
      0, 0, 0, 0, 1, 1
    ), nrow = 4, byrow = TRUE
  )


  expect_equal(.get_Au(A), matrix(c(1, 1, 1), ncol = 3))
  expect_equal(.lowest_lev(A), c(2, 3, 4))
})

test_that(".get_Au behaves identically on A with different row order.", {
  A <- matrix(
    data = c(
      1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
      1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
      0, 0, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0,
      0, 0, 0, 0, 1, 1, 0, 0, 0, 0, 0, 0,
      0, 0, 0, 0, 0, 0, 1, 1, 0, 0, 0, 0,
      0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 0, 0,
      0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1,
      1, 1, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0,
      0, 0, 0, 0, 1, 1, 1, 1, 0, 0, 0, 0,
      0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1
    ),
    nrow = 10, byrow = TRUE
  )

  A1 <- matrix(
    data = c(
      1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
      1, 1, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0,
      0, 0, 0, 0, 1, 1, 1, 1, 0, 0, 0, 0,
      0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1,
      1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
      0, 0, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0,
      0, 0, 0, 0, 1, 1, 0, 0, 0, 0, 0, 0,
      0, 0, 0, 0, 0, 0, 1, 1, 0, 0, 0, 0,
      0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 0, 0,
      0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1
    ),
    nrow = 10, byrow = TRUE
  )

  expect_equal(.get_Au(A), .get_Au(A1))
})