test_that("grouping_var selection works", {
  # test that function remains silent across all possible configurations
  config <- expand.grid(
    .mode = c("micro", "subj-avg", "doc-avg"),
    .doc_groups = list(
      NULL,
      tibble::tibble(
        doc_id = c("a", "b"),
        hsg = c("grp1", "grp2")
      ),
      tibble::tibble(
        doc_id = c("a", "b"),
        hsg = c("grp1", "grp2"),
        foo = c("foo1", "foo2")
      )
    ),
    .label_groups = list(
      NULL,
      tibble::tibble(
        label_id = c("A", "B"),
        label_grp = c("1", "2")
      )
    ),
    .var = list(NULL, c("abc"), c("abc", "def"))
  )

  messages <- config |>
    purrr::pmap(
      .f = function(.mode, .doc_groups, .label_groups, .var) {
        expect_silent(
          set_grouping_var(
            mode = .mode,
            doc_groups = .doc_groups,
            label_groups = .label_groups,
            var = .var
          )
        )
      }
    )
})

test_that("set_ps_flags is correct", {
  expect_equal(
    set_ps_flags(mode = "subj-avg", propensity_scored = TRUE),
    list(intermed = FALSE, summarise = TRUE)
  )

  expect_equal(
    set_ps_flags(mode = "doc-avg", propensity_scored = TRUE),
    list(intermed = TRUE, summarise = FALSE)
  )

  expect_equal(
    set_ps_flags(mode = "micro", propensity_scored = TRUE),
    list(intermed = TRUE, summarise = FALSE)
  )
})