context("test-summary.multiwave")

library(optimall)

MySurvey <- new_multiwave(phases = 2, waves = c(1, 2))
get_data(MySurvey, phase = 1, slot = "data") <-
  data.frame(a = c(1, 2, 3, 4), b = c(2, 3, 4, 5))
get_data(MySurvey, phase = 2, wave = 2, slot = "data") <-
  data.frame(a = c(1, 2, 3), b = c(2, 3, 4))
get_data(MySurvey, phase = 2, wave = 2, slot = "sampled_data") <-
  data.frame(a = c(1, 2, 3), b = c(2, 3, 4))

test_that("summary works", {
  expect_output(summary(MySurvey), "object with 2 phases containing 1, 2 wav")
  expect_output(summary(MySurvey), "4 obs. of 2 vars: a, b")
  expect_output(summary(MySurvey), "3 obs. of 2 vars: a, b")
})

MySurvey <- new_multiwave(phases = 2, waves = c(1, 2))
get_data(MySurvey, phase = 1, slot = "data") <-
  data.frame(a = c(1, 2, 3, 4), b = c(2, 3, 4, 5))
get_data(MySurvey, phase = 2, wave = 2, slot = "data") <-
  data.frame(a = c(1, 2, 3), b = c(2, 3, 4))
get_data(MySurvey, phase = 2, wave = 2, slot = "sampled_data") <-
  data.frame(a = c(1, 2, 3), b = c(2, 3, 4))
get_data(MySurvey, phase = 2, wave = 2, slot = "design") <-
  data.frame(a = c(1, 2, 3), b = c(2, 3, 4), c = c(6, 7, 8))
get_data(MySurvey, phase = 2, wave = 2, slot = "samples") <-
  c("4", "7", "11")

test_that("summary works with design and samples", {
  expect_output(summary(MySurvey), "object with 2 phases containing 1, 2 wav")
  expect_output(summary(MySurvey), "3 obs. of 3 vars: a, b")
  expect_output(summary(MySurvey), "vector of length 3")
})