#!/usr/bin/r -t
##
##  Copyright (C) 2013-2026  Romain Francois
##  Copyright (C) 2014-2026  Christian Gunning and Dirk Eddelbuettel
##  Copyright (C) 2019-2026  Dirk Eddelbuettel
##
##  This file is part of RcppArmadillo.
##
##  RcppArmadillo is free software: you can redistribute it and/or modify it
##  under the terms of the GNU General Public License as published by
##  the Free Software Foundation, either version 2 of the License, or
##  (at your option) any later version.
##
##  RcppArmadillo is distributed in the hope that it will be useful, but
##  WITHOUT ANY WARRANTY; without even the implied warranty of
##  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
##  GNU General Public License for more details.
##
##  You should have received a copy of the GNU General Public License
##  along with RcppArmadillo.  If not, see <http://www.gnu.org/licenses/>.

.runThisTest <- isTRUE(capabilities("long.double"))

if (!.runThisTest) exit_file("No long double support")

library(RcppArmadillo)

Rcpp::sourceCpp("cpp/rmultinom.cpp")

## Seed needs to be reset to compare R to C++
.seed=39

## test cases
## should be indentical between R and C++
tests <- list(vanilla=list( n=5, size=100, prob=rep(1/10,10)),
              fixup.prob=list( n=10, size=1e5, prob=1:10),
              n0=list( n=0, size=5, prob=1:10),
              size0=list( n=10, size=0, prob=1:10)
)
## In April 2026, R-devel improved rmultinom() vai a more precise calculation
## The routine in RcppArmadilloExtensions has been updated to but we can only
## compare with R-devel (i.e. R 4.7.0 or later)
if (getRversion() >= "4.7.0")
    tests[["big"]] <- list(n=5, size=1e6, prob=rep(1/1e3,1e3))


fail.tests <- list(na.prob=list( n=7, size=100, prob=c(1:10,NA)),
                   prob0=list( n=10, size=100, prob=0)
)

## these give errors
lapply(names(fail.tests), function(.name) {
    with(fail.tests[[.name]], {
        expect_error(rmultinom(n, size, prob)) #msg=sprintf("rmultinom.R.error.%s",.name)
    })
    with(fail.tests[[.name]], {
        expect_error(rmultinomC(n, size, prob)) #msg=sprintf("rmultinom.cpp.error.%s",.name)
    })
})

## for each test, check that results match
lapply(names(tests), function(.name) {
    with(tests[[.name]], {
        set.seed(.seed)
        r.multinom <- rmultinom(n, size, prob)
        set.seed(.seed)
        c.multinom <- rmultinomC(n, size, prob)
        expect_equal(r.multinom, c.multinom)# , msg=sprintf("rmultinom.%s",.name))
    })
})