tests/testthat/test-lens-laws.R

In lenses: Elegant Data Manipulation with Lenses

context("test-lens-laws.R")

test_that("Index lens works", {
  v <- 1:10
  ix4 <- index_l(4)
  expect_equal(view(v, ix4), 4)
  expect_equal(set(v, ix4, 5)
           , c(1:3, 5, 5:10))
})

test_that("Lens composition works", {
  modified_frame <-
    setNames(seq_along(LETTERS), LETTERS) %>%
    as.list %>%
    as.data.frame %>%
    set(names_l %.%
         indexes_l(10:15) %.%
         index_l(3)
       , "hi")

  expect_equal(names(modified_frame)[12], "hi")
})