tests/testthat/test-dplyr-methods.R

In mikmart/flow: Flowchart numbers from dplyr workflows

context("test-dplyr-methods.R")

test_that("verbs preserve flow_df", {
  df <- flow(data.frame(x = c(1, 2, 2), y = 6:4))
  grouped <- dplyr::group_by(df, x)

  expect_is(dplyr::select(df, x), "flow_df")
  expect_is(dplyr::filter(df, x > 1), "flow_df")
  expect_is(dplyr::mutate(df, z = 7), "flow_df")
  expect_is(dplyr::rename(df, foo = x), "flow_df")
  expect_is(dplyr::arrange(df, y), "flow_df")
  expect_is(dplyr::group_by(df, x), "flow_df")
  expect_is(dplyr::transmute(df, z = 7), "flow_df")

  expect_is(dplyr::filter(grouped, mean(y) < 6), "flow_df")
  expect_is(dplyr::mutate(grouped, z = mean(y)), "flow_df")
})

test_that("add_count/tally preserve flow_df", {
  skip("waiting on generic methods")
  df <- flow(data.frame(x = c(1, 2, 2), y = 6:4))
  grouped <- dplyr::group_by(df, x)

  expect_is(dplyr::add_count(df, x), "flow_df")
  expect_is(dplyr::add_tally(df, y), "flow_df")

  expect_is(dplyr::add_count(grouped), "flow_df")
  expect_is(dplyr::add_tally(grouped, y), "flow_df")
})