inst/tinytest/test-sps_iterator.R

In sps: Sequential Poisson Sampling

# Streaming is the same as drawing.
local({
  set.seed(13026)
  x <- rlnorm(10)
  u <- runif(10)
  res <- integer(0)
  s <- sps_iterator(x, prn = u)
  for (i in 1:4) {
    res <- c(res, s())
    expect_true(setequal(sps(x, i, prn = u), res))
  }
  for (i in 6:10) {
    res <- c(res, s())
    expect_true(setequal(sps(x, i, prn = u), res))
  }

  expect_null(s())

  s <- sps_iterator(x, n = 4, prn = u)
  expect_true(setequal(c(3, 5), s()))

  res <- c(2, 3, 4, 5, 6)
  s <- sps_iterator(x, n = 5, prn = u)
  for (i in 6:10) {
    res <- c(res, s())
    expect_true(setequal(sps(x, i, prn = u), res))
  }
})

# Gives back order with same prn.
local({
  x <- c(1, 4, 5, 3, 2)
  prn <- rep(0.5, 5)
  res <- c(3, 2, 4, 5, 1)
  s <- sps_iterator(x, prn = prn)
  for (i in 1:5) {
    expect_equal(s(), res[i])
  }
})

# NULL cases work.
local({
  s <- sps_iterator(1:5, n = 5)
  expect_null(s())

  s <- sps_iterator(integer(0))
  expect_null(s())
})

# Errors work.
local({
  expect_error(sps_iterator(1:5, 7))
  expect_error(sps_iterator(1:5, prn = 1:6))
  expect_error(sps_iterator(0))
})

# Extra args work.
local({
  set.seed(1234)
  prn <- runif(5)
  s <- sps_iterator(1:5, n = 2, alpha = 0.5, prn = prn)
  expect_equal(s(), 3:4)

  expect_equal(as.integer(sps(1:5, 3, alpha = 0.5, prn = prn)), 3:5)
})