context("simData")

test_that("Exceptions are thrown", {
    stepPar <- c(1, 10, 1, 5, 0.2, 0.3)
    anglePar <- c(0, pi, 0.5, 2)

    expect_error(simData(1, 2, "gamma", "vm", stepPar, anglePar, nbCovs = 2, zeroInflation = TRUE),
                 NA)

    expect_error(simData(0, 2, "gamma", "vm", stepPar, anglePar, nbCovs = 2, zeroInflation = TRUE),
                 "nbAnimals should be at least 1.")
    expect_error(simData(1, 0, "gamma", "vm", stepPar, anglePar, nbCovs = 2, zeroInflation = TRUE),
                 "nbStates should be at least 1.")

    expect_error(simData(1, 2, "pois", "vm", stepPar, anglePar, nbCovs = 2, zeroInflation = TRUE))
    expect_error(simData(1, 2, "gamma", "norm", stepPar, anglePar, nbCovs = 2, zeroInflation = TRUE))

    stepPar2 <- c(stepPar, 1)
    expect_error(simData(1, 2, "gamma", "vm", stepPar2, anglePar, nbCovs = 2, zeroInflation = TRUE),
                 "Wrong number of parameters")
    anglePar2 <- c(anglePar, 1)
    expect_error(simData(1, 2, "gamma", "vm", stepPar, anglePar2, nbCovs = 2, zeroInflation = TRUE),
                 "Wrong number of parameters")

    stepPar <- c(-1, 10, 1, 5, 0.2, 0.3)
    expect_error(simData(1, 2, "gamma", "vm", stepPar, anglePar, nbCovs = 2, zeroInflation = TRUE))
    stepPar <- c(1, 10, 1, -0.5, 0.2, 0.3)
    expect_error(simData(1, 2, "gamma", "vm", stepPar, anglePar, nbCovs = 2, zeroInflation = TRUE))
    stepPar <- c(1, 10, 1, 5, 4, 0.3)
    expect_error(simData(1, 2, "gamma", "vm", stepPar, anglePar, nbCovs = 2, zeroInflation = TRUE))
})

test_that("The right slots are defined", {
    stepPar <- c(1, 10, 1, 5, 0.2, 0.3)
    anglePar <- c(0, pi, 0.5, 2)
    nbCovs <- 2
    data <- simData(1, 2, "gamma", "vm", stepPar, anglePar, nbCovs = nbCovs, zeroInflation = TRUE)

    expect_true(!is.null(data$ID))
    expect_true(!is.null(data$x))
    expect_true(!is.null(data$y))
    expect_true(!is.null(data$step))
    expect_true(!is.null(data$angle))

    expect_equal(length(which(names(data)!= "ID" & names(data)!= "x" & names(data)!= "y" &
                                  names(data)!= "step" & names(data)!= "angle")), nbCovs)
})

test_that("The returned object is of the correct class", {
    stepPar <- c(1, 10, 1, 5, 0.2, 0.3)
    anglePar <- c(0, pi, 0.5, 2)
    data <- simData(1, 2, "gamma", "vm", stepPar, anglePar, nbCovs = 2, zeroInflation = TRUE)

    expect_equal(class(data), c("moveData", "data.frame"))
})