source("../deprivation-profile.R")

test_that("adjusted headcount ratio works correctly", {
  dm <- df_household |>
    create_deprivation_matrix(deprivation_profile, .mpi_specs = mpi_specs)

  ahr_uncensored <- dm$uncensored |>
    compute_headcount_ratio_adjusted(.aggregation = attr_spec$aggregation)

  expect_equal(nrow(ahr_uncensored), 1)
  expect_equal(ncol(ahr_uncensored), 4)

  ahr_censored <- dm$censored |>
    compute_headcount_ratio_adjusted(.aggregation = attr_spec$aggregation)

  expect_equal(nrow(ahr_censored), 1)
  expect_equal(ncol(ahr_censored), 4)
})


test_that("adjusted headcount ratio works correctly with aggregation", {
  mpi_specs <- use_global_mpi_specs(.uid = "uuid", .aggregation = "class")
  attr_spec <- attributes(mpi_specs)
  dm <- df_household |>
    create_deprivation_matrix(deprivation_profile, .mpi_specs = mpi_specs)

  ahr_uncensored <- dm$uncensored |>
    compute_headcount_ratio_adjusted(.aggregation = attr_spec$aggregation)

  expect_equal(nrow(ahr_uncensored), 2)
  expect_equal(ncol(ahr_uncensored), 5)

  ahr_censored <- dm$censored |>
    compute_headcount_ratio_adjusted(.aggregation = attr_spec$aggregation)

  expect_equal(nrow(ahr_censored), 2)
  expect_equal(ncol(ahr_censored), 5)

  expect_contains(names(ahr_censored), "class")
})