#-------------------------------------------------------------------------------
# Copyright (c) 2019-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.levels::smk::setup")

connect.studies.dataset.cnsim(list("GENDER", "PM_BMI_CATEGORICAL"))

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

#
# Tests
#

context("ds.levels::smk")
test_that("simple levels", {
    ds.asFactor("D$GENDER", newobj.name = "gender")

    res <- ds.levels("gender")

    expect_length(res, 3)
    expect_length(res$sim1, 2)
    expect_length(res$sim1$ValidityMessage, 1)
    expect_equal(res$sim1$ValidityMessage, "VALID ANALYSIS")
    expect_length(res$sim1$Levels, 2)
    expect_equal(res$sim1$Levels[1], "0")
    expect_equal(res$sim1$Levels[2], "1")
    expect_length(res$sim2, 2)
    expect_length(res$sim2$ValidityMessage, 1)
    expect_equal(res$sim2$ValidityMessage, "VALID ANALYSIS")
    expect_length(res$sim2$Levels, 2)
    expect_equal(res$sim2$Levels[1], "0")
    expect_equal(res$sim2$Levels[2], "1")
    expect_length(res$sim3, 2)
    expect_length(res$sim3$ValidityMessage, 1)
    expect_equal(res$sim3$ValidityMessage, "VALID ANALYSIS")
    expect_length(res$sim3$Levels, 2)
    expect_equal(res$sim3$Levels[1], "0")
    expect_equal(res$sim3$Levels[2], "1")
})

context("ds.levels::smk")
test_that("simple levels", {
    ds.asFactor("D$PM_BMI_CATEGORICAL", newobj.name = "pm_bmi_categorical")

    res <- ds.levels("pm_bmi_categorical")

    expect_length(res, 3)
    expect_length(res$sim1, 2)
    expect_length(res$sim1$ValidityMessage, 1)
    expect_equal(res$sim1$ValidityMessage, "VALID ANALYSIS")
    expect_length(res$sim1$Levels, 3)
    expect_equal(res$sim1$Levels[1], "1")
    expect_equal(res$sim1$Levels[2], "2")
    expect_equal(res$sim1$Levels[3], "3")
    expect_length(res$sim2, 2)
    expect_length(res$sim2$ValidityMessage, 1)
    expect_equal(res$sim2$ValidityMessage, "VALID ANALYSIS")
    expect_length(res$sim2$Levels, 3)
    expect_equal(res$sim2$Levels[1], "1")
    expect_equal(res$sim2$Levels[2], "2")
    expect_equal(res$sim2$Levels[3], "3")
    expect_length(res$sim3, 2)
    expect_length(res$sim3$ValidityMessage, 1)
    expect_equal(res$sim3$ValidityMessage, "VALID ANALYSIS")
    expect_length(res$sim3$Levels, 3)
    expect_equal(res$sim3$Levels[1], "1")
    expect_equal(res$sim3$Levels[2], "2")
    expect_equal(res$sim3$Levels[3], "3")
})

#
# Done
#

context("ds.levels::smk::shutdown")

test_that("shutdown", {
    ds_expect_variables(c("D", "gender", "pm_bmi_categorical"))
})

disconnect.studies.dataset.cnsim()

context("ds.levels::smk::done")