tests/testthat/test-irep.r

In itertools2: itertools2: Functions creating iterators for efficient looping

context("irep iterator")

test_that("irep matches first example from base::rep", {
  it <- irep(1:4, 2)
  expect_equal(nextElem(it), 1)
  expect_equal(nextElem(it), 2)
  expect_equal(nextElem(it), 3)
  expect_equal(nextElem(it), 4)
  expect_equal(nextElem(it), 1)
  expect_equal(nextElem(it), 2)
  expect_equal(nextElem(it), 3)
  expect_equal(nextElem(it), 4)
  expect_error(nextElem(it), "StopIteration")
})

test_that("irep matches second example from base::rep", {
  it <- irep(1:4, each=2)
  expect_equal(nextElem(it), 1)
  expect_equal(nextElem(it), 1)
  expect_equal(nextElem(it), 2)
  expect_equal(nextElem(it), 2)
  expect_equal(nextElem(it), 3)
  expect_equal(nextElem(it), 3)
  expect_equal(nextElem(it), 4)
  expect_equal(nextElem(it), 4)
  expect_error(nextElem(it), "StopIteration")
})

test_that("irep matches fifth example from base::rep", {
  it <- irep(1:4, each=2, length.out=4)
  expect_equal(nextElem(it), 1)
  expect_equal(nextElem(it), 1)
  expect_equal(nextElem(it), 2)
  expect_equal(nextElem(it), 2)
  expect_error(nextElem(it), "StopIteration")
})

test_that("irep replicates a list and matches tenth example from base::rep", {
  # 8 integers plus two recycled 1's.
  fred <- list(happy=1:10, name="squash")
  it <- irep(fred, times=5)
  expect_equal(nextElem(it), 1:10)
  expect_equal(nextElem(it), "squash")
  expect_equal(nextElem(it), 1:10)
  expect_equal(nextElem(it), "squash")
  expect_equal(nextElem(it), 1:10)
  expect_equal(nextElem(it), "squash")
  expect_equal(nextElem(it), 1:10)
  expect_equal(nextElem(it), "squash")
  expect_equal(nextElem(it), 1:10)
  expect_equal(nextElem(it), "squash")
  expect_error(nextElem(it), "StopIteration")
})

test_that("irep replicates a factor and matches last example from base::rep", {
  # 8 integers plus two recycled 1's.
  x <- factor(LETTERS[1:4])
  it <- irep(x, 2)
  expect_equal(nextElem(it), x[1])
  expect_equal(nextElem(it), x[2])
  expect_equal(nextElem(it), x[3])
  expect_equal(nextElem(it), x[4])
  expect_equal(nextElem(it), x[1])
  expect_equal(nextElem(it), x[2])
  expect_equal(nextElem(it), x[3])
  expect_equal(nextElem(it), x[4])
  expect_error(nextElem(it), "StopIteration")
})

test_that("irep_len works on numeric vectors", {
  it <- irep_len(1:4, length.out=3)
  expect_equal(nextElem(it), 1)
  expect_equal(nextElem(it), 2)
  expect_equal(nextElem(it), 3)
  expect_error(nextElem(it), "StopIteration")
})

# Related to Issue #33
test_that("irep matches base::rep() when both times and each args are given", {
  it <- irep(1:4, times=2, each=3)
  expected_vector <- rep(1:4, times=2, each=3)
  expect_equal(unlist(as.list(it)), expected_vector)
})