tests/testthat/test-take.R

In dirkschumacher/transduceR: Functional programming in R

context("take")

test_that("take-nth takes every nth element.", {
  result <- transduce(take_nth(10), as_list, 1:20)
  expect_that(result, is_equivalent_to(list(10, 20)))
})

test_that("take-nth throws error if n < 1.", {
  expect_error(take_nth(0))
})

test_that("take-while works with basic example.", {
  result <- transduce(take_while(function(x) x < 5), as_sum, 1:20)
  expect_that(result, is_equivalent_to(sum(1:4)))
})