tests/testthat/test-repDistribution.R

In alan-turing-institute/distr6: The Complete R6 Probability Distributions Interface

test_that("rep custom", {
  dist <- Distribution$new(pdf = function(x) 1 / 10, name = "Test",
                          type = Set$new(0:10, class = "integer"))
  rep_dist <- rep(dist, 3)
  expect_equal(getR6Class(rep_dist), "VectorDistribution")
  expect_equal(rep_dist$pdf(1), data.table(Test1 = 0.1, Test2 = 0.1, Test3 = 0.1))
})

test_that("rep dist/param", {
  bin <- Binomial$new(size = 2, prob = 0.1)
  mix <- MixtureDistribution$new(
    distribution = "Binomial",
    params = data.frame(size = rep(2, 10), prob = rep(0.1, 10))
  )
  repbin <- rep(bin, 10, "mix")
  expect_equal(repbin$pdf(1:5), mix$pdf(1:5))
})

test_that("rep kernel", {
  expect_silent(rep(Epanechnikov$new(), 5))
  expect_silent(rep(UniformKernel$new(), 2, "prod"))
})


test_that("rep keeps decorators", {
  r <- dstr("Exp", decorators = c("CoreStatistics", "ExoticStatistics"))
  expect_equal(rep(r, 2)$decorators, c("CoreStatistics", "ExoticStatistics"))
})