1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
|
#!/usr/bin/r -t
##
## Copyright (C) 2013 Romain Francois
## Copyright (C) 2014 - 2019 Christian Gunning and Dirk Eddelbuettel
## Copyright (C) 2019 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)),
big=list( n=5, size=1e6, prob=rep(1/1e3,1e3)),
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)
)
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))
})
})
|
