tests/testthat/test-dplyr-dual.R

In tidyverse/multidplyr: A Multi-Process 'dplyr' Backend

test_that("joining data frame requires explicit copy", {
  pf <- partition(data.frame(x = 1:6), default_cluster())
  df <- data.frame(x = 1:3, y = 3:1)

  expect_error(left_join(pf, df), "same src")
  expect_error(left_join(pf, df, copy = TRUE), NA)
})

test_that("joins match local results", {
  pf1 <- partition(tibble(x = c(1, 2)), default_cluster())
  pf2 <- partition(tibble(x = c(1, 3)), default_cluster())

  # primarily testing that basic infrastructure works and that
  # I haven't accidentally typed the wrong verb name somewhere
  expect_equal(pf1 %>% inner_join(pf2, by = "x") %>% pull(x), 1)
  expect_equal(pf1 %>% left_join(pf2, by = "x") %>% pull(x), c(1, 2))
  expect_equal(pf1 %>% right_join(pf2, by = "x") %>% pull(x), c(1, 3))
  expect_equal(pf1 %>% full_join(pf2, by = "x") %>% pull(x), c(1, 2, 3))

  expect_equal(pf1 %>% semi_join(pf2, by = "x") %>% pull(x), 1)
  expect_equal(pf1 %>% anti_join(pf2, by = "x") %>% pull(x), 2)
})

test_that("set operations match local results", {
  pf1 <- partition(tibble(x = c(1, 2)), default_cluster())
  pf2 <- partition(tibble(x = c(1, 3)), default_cluster())

  expect_equal(pf1 %>% intersect(pf2) %>% pull(), 1)
  expect_setequal(pf1 %>% union(pf2) %>% pull(), c(1, 2, 3))
  expect_equal(pf1 %>% union_all(pf2) %>% pull() %>% sort(), c(1, 1, 2, 3))
  expect_equal(pf1 %>% setdiff(pf2) %>% pull(), 2)
})