#-------------------------------------------------------------------------------
# Copyright (c) 2018-2022 University of Newcastle upon Tyne. All rights reserved.
#
# This program and the accompanying materials
# are made available under the terms of the GNU Public License v3.0.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.
#-------------------------------------------------------------------------------

#
# Set up
#

context("ds.glm::smk::gaussian::setup")

connect.studies.dataset.cnsim(list("LAB_TSC", "LAB_TRIG"))

test_that("setup", {
    ds_expect_variables(c("D"))
})

#
# Tests
#

context("ds.glm::smk::gaussian")
test_that("glm_gaussian", {
    res <- ds.glm('D$LAB_TSC~D$LAB_TRIG',family="gaussian",checks=FALSE)

    expect_length(res, 13)
    expect_equal(res$Nvalid, 7800)
    expect_equal(res$Nmissing, 1579)
    expect_equal(res$Ntotal, 9379)
    expect_length(res$disclosure.risk, 3)
    expect_equal(res$disclosure[1], 0)
    expect_equal(res$disclosure[3], 0)
    expect_equal(res$disclosure[2], 0)
    expect_length(res$errorMessage, 3)
    expect_equal(res$errorMessage[1], "No errors")
    expect_equal(res$errorMessage[2], "No errors")
    expect_equal(res$errorMessage[3], "No errors")
    expect_equal(res$nsubs, 7800)
    expect_equal(res$iter, 3)
    expect_true("family" %in% class(res$family))
    expect_equal(res$formula, "D$LAB_TSC ~ D$LAB_TRIG")
    expect_true("matrix" %in% class(res$coefficients))
    expect_equal(res$dev, 8936.32, tolerance=0.00001)
    expect_equal(res$df, 7798)
    expect_equal(res$output.information, "SEE TOP OF OUTPUT FOR INFORMATION ON MISSING DATA AND ERROR MESSAGES")
})

context("ds.glm::smk::gaussian, with check")
test_that("glm_gaussian, with check", {
    expect_warning(res <- ds.glm('D$LAB_TSC~D$LAB_TRIG', family="gaussian", checks=TRUE), "NAs introduced by coercion")

    expect_length(res, 13)
    expect_equal(res$Nvalid, 7800)
    expect_equal(res$Nmissing, 1579)
    expect_equal(res$Ntotal, 9379)
    expect_length(res$disclosure.risk, 3)
    expect_equal(res$disclosure[1], 0)
    expect_equal(res$disclosure[3], 0)
    expect_equal(res$disclosure[2], 0)
    expect_length(res$errorMessage, 3)
    expect_equal(res$errorMessage[1], "No errors")
    expect_equal(res$errorMessage[2], "No errors")
    expect_equal(res$errorMessage[3], "No errors")
    expect_equal(res$nsubs, 7800)
    expect_equal(res$iter, 3)
    expect_true("family" %in% class(res$family))
    expect_equal(res$formula, "D$LAB_TSC ~ D$LAB_TRIG")
    expect_true("matrix" %in% class(res$coefficients))
    expect_equal(res$dev, 8936.32, tolerance=0.00001)
    expect_equal(res$df, 7798)
    expect_equal(res$output.information, "SEE TOP OF OUTPUT FOR INFORMATION ON MISSING DATA AND ERROR MESSAGES")
})

#
# Done
#

context("ds.glm::smk::gaussian::shutdown")

test_that("shutdown", {
    ds_expect_variables(c("D", "LAB_TRIG", "LAB_TSC"))
})

disconnect.studies.dataset.cnsim()

context("ds.glm::smk::gaussian::done")