tests/testthat/test-take.r

In ramhiser/itertools2: Iterators for efficient looping

context("take")

test_that("take properly extracts n elements, n < length of the iterable", {
  it <- ichain(1:3, 4:5, 6)
  expect_equal(take(it, 4), as.list(1:4))
  expect_equal(iterators::nextElem(it), 5)
  expect_equal(iterators::nextElem(it), 6)
  expect_error(iterators::nextElem(it), "StopIteration")
})

test_that("take properly extracts n elements, n = length of the iterable", {
  it <- ichain(1:3, 4:5, 6)
  expect_equal(take(it, 6), as.list(1:6))
  expect_error(iterators::nextElem(it), "StopIteration")
})

test_that("take properly extracts n elements, n > length of the iterable", {
  it <- ichain(1:3, 4:5, 6)
  expect_equal(take(it, 7), as.list(1:6))
  expect_error(iterators::nextElem(it), "StopIteration")
})