test_that("x used as basis of output (#3839)", { df1 <- tibble(x = 1:4, y = 1) df2 <- tibble(y = 1, x = c(4, 2)) expect_equal(duckplyr_intersect(df1, df2), tibble(x = c(2, 4), y = 1)) expect_equal(duckplyr_union(df1, df2), tibble(x = 1:4, y = 1)) expect_equal(duckplyr_union_all(df1, df2), tibble(x = c(1:4, 4, 2), y = 1)) expect_equal(duckplyr_setdiff(df1, df2), tibble(x = c(1, 3), y = 1)) expect_equal(duckplyr_symdiff(df1, df2), tibble(x = c(1, 3), y = 1)) }) test_that("set operations (apart from union_all) remove duplicates", { df1 <- tibble(x = c(1, 1, 2)) df2 <- tibble(x = 2) expect_equal(duckplyr_intersect(df1, df2), tibble(x = 2)) expect_equal(duckplyr_union(df1, df2), tibble(x = c(1, 2))) expect_equal(duckplyr_union_all(df1, df2), tibble(x = c(1, 1, 2, 2))) expect_equal(duckplyr_setdiff(df1, df2), tibble(x = 1)) expect_equal(duckplyr_symdiff(df1, df2), tibble(x = 1)) }) test_that("standard coercion rules are used (#799)", { df1 <- tibble(x = 1:2, y = c(1, 1)) df2 <- tibble(x = 1:2, y = 1:2) expect_equal(nrow(duckplyr_intersect(df1, df2)), 1) expect_equal(nrow(duckplyr_union(df1, df2)), 3) expect_equal(nrow(duckplyr_union_all(df1, df2)), 4) expect_equal(nrow(duckplyr_setdiff(df1, df2)), 1) expect_equal(nrow(duckplyr_symdiff(df1, df2)), 2) }) test_that("grouping metadata is reconstructed (#3587)", { df1 <- tibble(x = 1:4, g = rep(1:2, each = 2)) %>% group_by(g) df2 <- tibble(x = 3:6, g = rep(2:3, each = 2)) expect_equal(duckplyr_group_vars(duckplyr_intersect(df1, df2)), "g") expect_equal(duckplyr_group_vars(duckplyr_union(df1, df2)), "g") expect_equal(duckplyr_group_vars(duckplyr_union_all(df1, df2)), "g") expect_equal(duckplyr_group_vars(duckplyr_setdiff(df1, df2)), "g") expect_equal(duckplyr_group_vars(duckplyr_symdiff(df1, df2)), "g") }) test_that("also work with vectors", { expect_equal(duckplyr_intersect(1:3, 3:4), 3) expect_equal(duckplyr_union(1:3, 3:4), 1:4) expect_equal(duckplyr_union_all(1:3, 3:4), c(1:3, 3:4)) expect_equal(duckplyr_setdiff(1:3, 3:4), 1:2) expect_equal(duckplyr_symdiff(1:3, 3:4), c(1, 2, 4)) # removes duplicates expect_equal(duckplyr_symdiff(c(1, 1, 2), c(2, 2, 3)), c(1, 3)) }) test_that("extra arguments in ... error (#5891)", { df1 <- tibble(var = 1:3) df2 <- tibble(var = 2:4) expect_snapshot(error = TRUE, { duckplyr_intersect(df1, df2, z = 3) duckplyr_union(df1, df2, z = 3) duckplyr_union_all(df1, df2, z = 3) duckplyr_setdiff(df1, df2, z = 3) duckplyr_symdiff(df1, df2, z = 3) }) }) test_that("incompatible data frames error (#903)", { skip_if(Sys.getenv("DUCKPLYR_FORCE") == "TRUE") df1 <- tibble(x = 1) df2 <- tibble(x = 1, y = 1) expect_snapshot(error = TRUE, { duckplyr_intersect(df1, df2) duckplyr_union(df1, df2) duckplyr_union_all(df1, df2) duckplyr_setdiff(df1, df2) duckplyr_symdiff(df1, df2) }) }) test_that("is_compatible generates useful messages for different cases", { expect_snapshot({ cat(is_compatible(tibble(x = 1), 1)) cat(is_compatible(tibble(x = 1), tibble(x = 1, y = 2))) cat(is_compatible(tibble(x = 1, y = 1), tibble(y = 1, x = 1), ignore_col_order = FALSE)) cat(is_compatible(tibble(x = 1), tibble(y = 1))) cat(is_compatible(tibble(x = 1), tibble(x = 1L), convert = FALSE)) cat(is_compatible(tibble(x = 1), tibble(x = "a"))) }) }) # setequal ---------------------------------------------------------------- test_that("setequal ignores column and row order", { skip_if(Sys.getenv("DUCKPLYR_FORCE") == "TRUE") df1 <- tibble(x = 1:2, y = 3:4) df2 <- df1[2:1, 2:1] expect_true(duckplyr_setequal(df1, df2)) expect_true(duckplyr_setequal(df1, df2)) }) test_that("setequal ignores duplicated rows (#6057)", { skip_if(Sys.getenv("DUCKPLYR_FORCE") == "TRUE") df1 <- tibble(x = 1) df2 <- df1[c(1, 1, 1), ] expect_true(duckplyr_setequal(df1, df2)) expect_true(duckplyr_setequal(df2, df1)) }) test_that("setequal uses coercion rules (#6114)", { skip_if(Sys.getenv("DUCKPLYR_FORCE") == "TRUE") df1 <- tibble(x = 1) df2 <- tibble(x = 1L) expect_true(duckplyr_setequal(df1, df2)) expect_true(duckplyr_setequal(df2, df1)) }) test_that("setequal tibbles must have same rows and columns", { skip("TODO duckdb") # Different rows are the definition of not equal expect_false(duckplyr_setequal(tibble(x = 1:2), tibble(x = 2:3))) # Different or incompatible columns are an error, like the other set ops (#6786) expect_snapshot(error = TRUE, { duckplyr_setequal(tibble(x = 1:2), tibble(y = 1:2)) }) expect_snapshot(error = TRUE, { duckplyr_setequal(tibble(x = 1:2), tibble(x = c("a", "b"))) }) }) test_that("setequal checks y is a data frame", { expect_snapshot(duckplyr_setequal(mtcars, 1), error = TRUE) }) test_that("setequal checks for extra arguments", { expect_snapshot(duckplyr_setequal(mtcars, mtcars, z = 2), error = TRUE) })