tests/testthat/test-icombinations.r

In ramhiser/itertools2: Iterators for efficient looping

context("icombinations iterator")

test_that("icombinations matches the output from utils::combn", {
  combn_out <- utils::combn(x=5, m=2, simplify=FALSE)
  it <- icombinations(seq_len(5), 2)

  expect_equal(as.list(it), combn_out)
})

test_that("icombinations matches the output from another utils::combn call", {
  combn_out <- utils::combn(x=3, m=3, simplify=FALSE)
  it <- icombinations(seq_len(3), 3)

  expect_equal(as.list(it), combn_out)
})

test_that("icombinations with replacement works correctly", {
  it <- icombinations(1:4, m=2, replacement=TRUE)

  expect_equal(iterators::nextElem(it), c(1, 1))
  expect_equal(iterators::nextElem(it), c(1, 2))
  expect_equal(iterators::nextElem(it), c(1, 3))
  expect_equal(iterators::nextElem(it), c(1, 4))
  expect_equal(iterators::nextElem(it), c(2, 2))
  expect_equal(iterators::nextElem(it), c(2, 3))
  expect_equal(iterators::nextElem(it), c(2, 4))
  expect_equal(iterators::nextElem(it), c(3, 3))
  expect_equal(iterators::nextElem(it), c(3, 4))
  expect_equal(iterators::nextElem(it), c(4, 4))
})