tests/testthat/test-chained-transformation.R

In mivanic/outlookHelpeR:

test_that("Chained transformation", {
  data = rbind(
    as.data.frame(list(
      y = 4,
      x1 = 3,
      x2 = 5,
      p = '2019-01-01'
    )),
    as.data.frame(list(
      y = 2,
      x1 = 3,
      x2 = 4,
      p = '2018-12-01'
    )),
    as.data.frame(list(
      y = 4,
      x1 = 2,
      x2 = 2,
      p = '2018-11-01'
    )),
    as.data.frame(list(
      y = 6,
      x1 = 1,
      x2 = 1,
      p = '2018-10-01'
    )),
    as.data.frame(list(
      y = 7,
      x1 = 3,
      x2 = 2,
      p = '2018-09-01'
    )),
    as.data.frame(list(
      y = 2,
      x1 = 3,
      x2 = 4,
      p = '2018-08-01'
    )),
    as.data.frame(list(
      y = NA,
      x1 = 2,
      x2 = 2,
      p = '2018-07-01'
    )),
    as.data.frame(list(
      y = NA,
      x1 = 1,
      x2 = 1,
      p = '2018-06-01'
    )),
    as.data.frame(list(
      y = NA,
      x1 = NA,
      x2 = 2,
      p = '2018-05-01'
    )),
    as.data.frame(list(
      y = NA,
      x1 = NA,
      x2 = 7,
      p = '2018-04-01'
    ))
  )

  result = chainedTransfromations(data,
                                  function(data) {
                                    data$pp = 1
                                    return(data)
                                  },
                                  function(data) {
                                    data$x1b = data$x2 + data$y
                                    return(data)
                                  })

  expect(is.na(result[10, 'x1b']), 'Not NA')
  expect_equal(result[8, 'pp'], 1)
  expect_equal(result[6, 'x1b'], 6)
})