tests/testthat/test-iroundrobin.R

In iterors: Fast, Compact Iterators and Tools

test_that("i_roundrobin works with a single iterable object", {
  rr <- i_roundrobin(1:5)
  expect_equal(take(rr, 5), as.list(1:5))
  expect_equal(nextOr(rr, NA), NA)
})

test_that("i_roundrobin functions properly with two iterators of same type", {
  it <- iteror(1:5)
  it2 <- iteror(1:5)
  rr <- i_roundrobin(it, it2)
  expect_equal(take(rr, 10), as.list(rep(1:5, each=2)))
  expect_equal(nextOr(rr, NA), NA)
})

test_that("i_roundrobin functions properly with two iterators of different type", {
  it3 <- iteror(levels(iris$Species))
  it4 <- iteror(1:3)
  rr <- i_roundrobin(it3, it4)
  expect_equal(take(rr, 6), list("setosa", 1, "versicolor", 2, "virginica", 3))
  expect_equal(nextOr(rr, NA), NA)
})

test_that("i_roundrobin functions properly with two iterators of different type - part 2", {
  it5 <- iteror(letters[1:3])
  it6 <- iteror(c(TRUE, FALSE, FALSE))
  rr <- i_roundrobin(it5, it6)
  expect_equal(take(rr, 6), list("a", TRUE, "b", FALSE, "c", FALSE))
  expect_equal(nextOr(rr, NA), NA)
})

test_that("i_roundrobin functions properly with multiple iterators of same length", {
  it <- iteror(1:5)
  it2 <- iteror(1:5)
  it3 <- iteror(1:5)
  rr <- i_roundrobin(it, it2, it3)
  expect_equal(take(rr, 15), as.list(rep(1:5, each=3)))
  expect_equal(nextOr(rr, NA), NA)
})

test_that("i_roundrobin works functions properly with multiple iterators of unequal length", {
  it <- iteror(letters[1:3])
  it2 <- iteror(LETTERS[5])
  it3 <- iteror(letters[25:26])
  rr <- i_roundrobin(it, it2, it3)
  expect_equal(take(rr, 6), list("a", "E", "y", "b", "z", "c"))
  expect_equal(nextOr(rr, NA), NA)
})

test_that("i_roundrobin works functions properly with multiple vectors of unequal length", {
  rr <- i_roundrobin(1:3, 5, 25:26)
  expect_equal(take(rr, 6), list(1, 5, 25, 2, 26, 3))
  expect_equal(nextOr(rr, NA), NA)
})