context("Test that level accepts vector correctly")

suppressPackageStartupMessages(library(survey))
data(api)

# Overall
## srvyr
dstrata_srvyr <- apistrat %>%
  as_survey(strata = stype, weights = pw)

srvyr_results <- dstrata_srvyr %>%
  summarise(api99_mn = survey_mean(api99, vartype = "ci", level=c(.75,.9,.95)),
            api99_tot = survey_total(api99, vartype = "ci", level=c(.75,.9,.95)))


## survey
dstrata_survey <- svydesign(ids = ~1, strata = ~stype, weights = ~pw,
                            data = apistrat)

survey_mn <- svymean(~api99, dstrata_survey)
survey_tot <- svytotal(~api99, dstrata_survey)

test_that("Test that CIs are correct for .75",
          expect_equal(confint(survey_mn, level=.75,
                               df = degf(dstrata_survey))[1:2],
                       c(srvyr_results[[2]][[1]], srvyr_results[[3]][[1]])))
test_that("Test that CIs are correct for .9",
          expect_equal(confint(survey_mn, level=.9,
                               df = degf(dstrata_survey))[1:2],
                       c(srvyr_results[[4]][[1]], srvyr_results[[5]][[1]])))
test_that("Test that CIs are correct for .95",
          expect_equal(confint(survey_mn, level=.95,
                               df = degf(dstrata_survey))[1:2],
                       c(srvyr_results[[6]][[1]],srvyr_results[[7]][[1]])))
test_that("Test that CIs are correct for .75",
          expect_equal(confint(survey_tot, level=.75,
                               df = degf(dstrata_survey))[1:2],
                       c(srvyr_results[[9]][[1]],srvyr_results[[10]][[1]])))
test_that("Test that CIs are correct for .9",
          expect_equal(confint(survey_tot, level = .9,
                               df = degf(dstrata_survey))[1:2],
                       c(srvyr_results[[11]][[1]], srvyr_results[[12]][[1]])))
test_that("Test that CIs are correct for .95",
          expect_equal(confint(survey_tot, level=.95,
                               df = degf(dstrata_survey))[1:2],
                       c(srvyr_results[[13]][[1]], srvyr_results[[14]][[1]])))


# Grouped data
srvyr_grouped_results <- dstrata_srvyr %>%
  group_by(stype) %>%
  summarise(api99_mn = survey_mean(api99, vartype = "ci", level = c(.75, .9, .95)),
            api99_tot = survey_total(api99, vartype ="ci", level = c(.75, .9, .95)))

survey_grouped_results_mn <- svyby(~api99, ~stype, dstrata_survey, svymean)

survey_grouped_results_tot <- svyby(~api99, ~stype, dstrata_survey, svytotal)

test_that("Test that CIs are correct for .75",
          expect_equal(c(unname(confint(survey_grouped_results_mn, level=.75,
                                        df = degf(dstrata_survey))[, 1]),
                         unname(confint(survey_grouped_results_mn, level=.75,
                                        df = degf(dstrata_survey))[, 2])),
                       c(srvyr_grouped_results[[3]], srvyr_grouped_results[[4]])))
test_that("Test that CIs are correct for .90",
          expect_equal(c(unname(confint(survey_grouped_results_mn, level=.9,
                                        df = degf(dstrata_survey))[, 1]),
                         unname(confint(survey_grouped_results_mn, level=.9,
                                        df = degf(dstrata_survey))[, 2])),
                       c(srvyr_grouped_results[[5]], srvyr_grouped_results[[6]])))
test_that("Test that CIs are correct for .95",
          expect_equal(c(unname(confint(survey_grouped_results_mn, level=.95,
                                        df = degf(dstrata_survey))[, 1]),
                         unname(confint(survey_grouped_results_mn, level=.95,
                                        df = degf(dstrata_survey))[, 2])),
                       c(srvyr_grouped_results[[7]], srvyr_grouped_results[[8]])))
test_that("Test that CIs are correct for .75",
          expect_equal(c(unname(confint(survey_grouped_results_tot, level=.75,
                                        df = degf(dstrata_survey))[, 1]),
                         unname(confint(survey_grouped_results_tot, level=.75,
                                        df = degf(dstrata_survey))[, 2])),
                       c(srvyr_grouped_results[[10]], srvyr_grouped_results[[11]])))
test_that("Test that CIs are correct for .90",
          expect_equal(c(unname(confint(survey_grouped_results_tot, level=.9,
                                        df = degf(dstrata_survey))[, 1]),
                         unname(confint(survey_grouped_results_tot, level=.9,
                                        df = degf(dstrata_survey))[, 2])),
                       c(srvyr_grouped_results[[12]], srvyr_grouped_results[[13]])))
test_that("Test that CIs are correct for .95",
          expect_equal(c(unname(confint(survey_grouped_results_tot, level=.95,
                                        df = degf(dstrata_survey))[, 1]),
                         unname(confint(survey_grouped_results_tot, level=.95,
                                        df = degf(dstrata_survey))[, 2])),
                       c(srvyr_grouped_results[[14]], srvyr_grouped_results[[15]])))