tests/testthat/test-phop-vec.R

In DavisVaughan/slurrr: Sliding Window Functions

# ------------------------------------------------------------------------------
# phop_vec

test_that("phop_vec() works", {
  expect_identical(phop_vec(list(1L, 1L), 1, 1, ~.x + .y), 2L)
})

test_that("phop_vec() doesn't retains names of first input (#75)", {
  expect_named(phop_vec(list(c(x = 1L), c(y = 1L)), 1, 1, ~.x + .y), NULL)
})

test_that("phop_vec() can simplify automatically", {
  expect_identical(phop_vec(list(1, 2), 1, 1, ~.x + .y, .ptype = NULL), 3)
})

test_that("phop_vec() errors if it can't simplify", {
  fn <- function(x, y) {
    if (x == 1L) {
      1
    } else {
      "hi"
    }
  }
  expect_snapshot({
    (
      expect_error(
        phop_vec(list(1:2, 1:2), 1:2, 1:2, fn, .ptype = NULL),
        class = "vctrs_error_incompatible_type"
      )
    )
  })
})

# ------------------------------------------------------------------------------
# .ptype

test_that("`.ptype = NULL` validates that element lengths are 1", {
  expect_snapshot({
    (
      expect_error(
        phop_vec(
          list(1:2, 1:2),
          1:2,
          1:2,
          ~if (.x == 1L) {
            1:2
          } else {
            1
          },
          .ptype = NULL
        )
      )
    )
    (
      expect_error(
        phop_vec(
          list(1:2, 1:2),
          1:2,
          1:2,
          ~if (.x == 1L) {
            NULL
          } else {
            2
          },
          .ptype = NULL
        )
      )
    )
  })
})

test_that("`phop_vec()` falls back to `c()` method as required", {
  local_c_foobar()

  expect_identical(
    phop_vec(list(1:3, 1:3), 1:3, 1:3, ~foobar(.x), .ptype = foobar(integer())),
    foobar(1:3)
  )
  expect_condition(
    phop_vec(list(1:3, 1:3), 1:3, 1:3, ~foobar(.x), .ptype = foobar(integer())),
    class = "slider_c_foobar"
  )

  expect_identical(phop_vec(list(1:3, 1:3), 1:3, 1:3, ~foobar(.x)), foobar(1:3))
  expect_condition(
    phop_vec(list(1:3, 1:3), 1:3, 1:3, ~foobar(.x)),
    class = "slider_c_foobar"
  )
})