context("test-create_total")


####################
# DECLARAR DISEÑOS #
# ##################


options(survey.lonely.psu = "certainty")

# Diseño complejo con varstrat y varunit
dc <- survey::svydesign(ids = ~varunit,
                        data = epf_personas %>%
                          dplyr::group_by(folio) %>%
                          dplyr::slice(1) %>%
                          dplyr::ungroup() %>%
                          dplyr::mutate(
                            metro = dplyr::if_else(zona == 1, 1, 0),
                            metro_na = dplyr::if_else(dplyr::row_number() <= 10, NA_real_, metro )),
                        strata = ~varstrat,
                        weights = ~fe)

# Diseño sin varunit
dc_sin_varunit <- survey::svydesign(ids = ~1,
                                    data = epf_personas %>%
                                      dplyr::group_by(folio) %>%
                                      dplyr::slice(1) %>%
                                      dplyr::ungroup() %>%
                                      dplyr::mutate(
                                        metro = dplyr::if_else(zona == 1, 1, 0),
                                        metro_na = dplyr::if_else(dplyr::row_number() <= 10, NA_real_, metro )),
                                    weights = ~fe)


#####################
# Testear estimación
#####################

# Probar suma con desagregación
suma <- create_total("gastot_hd", domains = "zona", design = dc) %>%
  dplyr::pull(stat)

# valor real de la suma
suma_real <- dc$variables %>%
  dplyr::group_by(zona) %>%
  dplyr::summarise(stat = sum(fe * gastot_hd), 3  )   %>%
  dplyr::pull(stat)

test_that("suma del gasto", {
  expect_equal(suma, suma_real)
})


suma <- expect_warning( create_total("gastot_hd", domains = "zona", design = dc_sin_varunit) %>%
                          dplyr::pull(stat),
                        "se calculated without complex design")


test_that("suma del gasto sin ", {
  expect_equal(suma, suma_real)
})

# Test without complex design and domains
suma2 <-  expect_warning(
  create_total("gastot_hd", domains = "zona", design = dc_sin_varunit) %>%
    dplyr::pull(stat)
)

test_that("suma del gasto", {
  expect_equal(suma2, suma_real)
})

# Test without complex design at national level
suma <- expect_warning(
  create_total("gastot_hd", design = dc_sin_varunit) %>%
    dplyr::pull(stat)
)

suma_real <- sum(dc_sin_varunit$variables$fe * dc_sin_varunit$variables$gastot_hd)

test_that("suma del gasto nivel nacional", {
  expect_equal(suma , suma_real)
})


################################################
# Testear grados de libertad y otros parámetros
################################################

all <- create_total("gastot_hd", domains = "zona+sexo", design = dc, ci = T, ess = T, deff = T, rm.na = T, unweighted = T, rel_error = T)

# Check column names
waited_output <- c("stat", "se", "n", "cv", "deff", "lower", "upper", "relative_error", "ess", "unweighted")

test_that("suma del gasto nivel nacional", {
  expect_equal(sum(names(all) %in% waited_output), length(waited_output))
})

# Check df
true_upm <- dc$variables %>%
  dplyr::group_by(sexo, zona, varunit) %>%
  dplyr::mutate(upm = dplyr::if_else(dplyr::row_number() == 1, 1, 0 )) %>%
  dplyr::group_by(sexo, zona) %>%
  dplyr::summarise(upm = sum(upm))

true_strata <- dc$variables %>%
  dplyr::group_by(sexo, zona, varstrat) %>%
  dplyr::mutate(strata = dplyr::if_else(dplyr::row_number() == 1, 1, 0 )) %>%
  dplyr::group_by(sexo, zona) %>%
  dplyr::summarise(strata = sum(strata))

true_df <- true_upm %>%
  dplyr::left_join(true_strata, by = c("sexo", "zona")) %>%
  dplyr::mutate(df = upm - strata)

test_that("conteo df diseño complejo", {
  expect_equal(true_df$df, all$df)
})