context("Model wrappers: glm")
library(finalfit)

test_that("glmuni gives glmlist", {
	expect_is(glmuni(colon_s, "mort_5yr", "age.factor"), "glmlist")
})

test_that("glmmulti gives glm", {
	expect_is(glmmulti(colon_s, "mort_5yr", "age.factor"), "glm")
})

test_that("glmmixed gives glmerMod", {
	expect_is(glmmixed(colon_s, "mort_5yr", "age.factor", "hospital"), "glmerMod")
})

test_that("glmmixed_boot gives list", {
	expect_is(glmmulti_boot(colon_s, "mort_5yr", "age.factor", R=50), "glmboot")
})


context("Model wrappers: lm")
library(finalfit)
test_that("lmuni gives lmlist", {
	expect_is(lmuni(colon_s, "nodes", "age.factor"), "lmlist")
})

test_that("lmmulti gives lm", {
	expect_is(lmmulti(colon_s, "nodes", "age.factor"), "lm")
})

test_that("lmmixed gives lmerMod", {
	expect_is(lmmixed(colon_s,  "nodes", "age.factor", "hospital"), "lmerMod")
})

context("Model wrappers: coxph")
library(finalfit)
test_that("coxphuni gives coxphlist", {
	expect_is(coxphuni(colon_s,  "Surv(time, status)", "age.factor"), "coxphlist")
})

test_that("coxphmulti gives coxph", {
	expect_is(coxphmulti(colon_s,  "Surv(time, status)", "age.factor"), "coxph")
})


#---------------


context("fit2df: glm")
library(finalfit)
test_that("fit2df(glm, metrics=TRUE) gives list", {
	expect_is(glm(mort_5yr ~  age.factor, data=colon_s, family="binomial") %>%
							fit2df(metrics=TRUE), "list")
})

test_that("fit2df(glmuni, metrics=TRUE) gives list", {
	expect_is(glmuni(colon_s, "mort_5yr", "age.factor") %>% fit2df(metrics=TRUE), "list")
})

test_that("fit2df(glmmulti, metrics=TRUE) gives list", {
	expect_is(glmmulti(colon_s, "mort_5yr", "age.factor") %>% fit2df(metrics=TRUE), "list")
})

test_that("fit2df(glmmixed, metrics=TRUE) gives list", {
	expect_is(glmmixed(colon_s, "mort_5yr", "age.factor", "hospital" )%>% fit2df(metrics=TRUE), "list")
})

test_that("fit2df(glmulti_boot) gives data.frame", {
	expect_is(glmmulti_boot(colon_s, "mort_5yr", "age.factor", R=50) %>% fit2df(), "data.frame")
})

context("fit2df: lm")
library(finalfit)
test_that("fit2df(lm, metrics=TRUE) gives list", {
	expect_is(lm(nodes ~  age.factor, data=colon_s) %>% fit2df(metrics=TRUE), "list")
})

test_that("fit2df(lmuni, metrics=TRUE) gives list", {
	expect_is(lmuni(colon_s, "nodes", "age.factor") %>% fit2df(metrics=TRUE), "list")
})

test_that("fit2df(lmmulti, metrics=TRUE) gives list", {
	expect_is(lmmulti(colon_s, "nodes", "age.factor") %>% fit2df(metrics=TRUE), "list")
})

test_that("fit2df(lmmixed, metrics=TRUE) gives list", {
	expect_is(lmmixed(colon_s, "nodes", "age.factor", random_effect="hospital") %>% fit2df(metrics=TRUE), "list")
})

context("fit2df: coxph")
library(finalfit)
test_that("fit2df(coxph) gives data.frame", {
	expect_is(coxph(Surv(time, status) ~ age.factor, data=colon_s) %>% fit2df(), "data.frame")
})

test_that("fit2df(coxphuni) gives data.frame", {
	expect_is(coxphuni(colon_s,  "Surv(time, status)", "age.factor") %>% fit2df(), "data.frame")
})

test_that("coxphmulti gives coxphlist", {
	expect_is(coxphmulti(colon_s,  "Surv(time, status)", "age.factor") %>% fit2df(), "data.frame")
})

test_that("fit2df(coxphuni) gives data.frame", {
	expect_is(crruni(colon_s,  "Surv(time, status)", "age.factor") %>% fit2df(), "data.frame")
})

test_that("coxphmulti gives coxphlist", {
	expect_is(crrmulti(colon_s,  "Surv(time, status)", "age.factor") %>% fit2df(), "data.frame")
})


#---------------


context("fit2df: svyglm")
library(finalfit)
library(survey)
data(api)
dstrat = svydesign(id=~1,strata=~stype, weights=~pw, data=apistrat, fpc=~fpc)

test_that("fit2df(svyglmuni) gives data.frame", {
	expect_is(svyglmuni(dstrat, "api00", "ell") %>% fit2df(), "data.frame")
})

test_that("fit2df(glmmulti) gives data.frame", {
	expect_is(svyglmmulti(dstrat, "api00", "ell") %>% fit2df(), "data.frame")
})