testthat::test_that("Bayes by Backprop", { # We will follow other packages such as diffeqr and skip # Julia related tests on CRAN testthat::skip_on_cran() # BayesFluxR_setup(installJulia = FALSE, env_path = ".", nthreads = 3, pkg_check = FALSE) test_setup(nthreads = 3, pkg_check = FALSE) y <- rnorm(100) x <- matrix(rnorm(100), nrow = 1) net <- Chain(Dense(1, 1)) prior <- prior.gaussian(net, 0.5) like <- likelihood.feedforward_normal(net, Gamma(2.0, 0.5)) init <- initialise.allsame(Normal(0, 0.5), like, prior) bnn <- BNN(x, y, like, prior, init) vi <- bayes_by_backprop(bnn, 10, 100) samples <- vi.get_samples(vi, n = 10) expect_equal(dim(samples)[1], BNN.totparams(bnn)) expect_equal(dim(samples)[2], 10) })