1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
|
test_that("column order is matched", {
df1 <- memdb_frame(x = 1, y = 2)
df2 <- memdb_frame(y = 1, x = 2)
out <- collect(union(df1, df2))
expect_equal(out, tibble(x = c(1, 2), y = c(2, 1)))
})
test_that("missing columns filled with NULL", {
df1 <- memdb_frame(x = 1)
df2 <- memdb_frame(y = 2)
out <- collect(union_all(df1, df2))
expect_equal(out, tibble(x = c(1, NA), y = c(NA, 2)))
})
test_that("first edition works", {
con <- structure(list(), class = c("Test", "DBIConnection"))
lf <- lazy_frame(x = 1, con = con)
local_methods(
sql_escape_ident.Test = function(con, x) sql_quote(x, "`")
)
local_options(rlib_warning_verbosity = "quiet")
expect_error(union_all(lf, lf) %>% remote_query(), NA)
})
# SQL generation ----------------------------------------------------------
test_that("set ops generates correct sql", {
lf1 <- memdb_frame(x = 1)
lf2 <- memdb_frame(x = c(1, 2))
out <- lf1 %>%
union(lf2) %>%
collect()
expect_equal(out, tibble(x = c(1, 2)))
})
test_that("union and union all work for all backends", {
df <- tibble(x = 1:10, y = x %% 2)
tbls_full <- test_load(df)
tbls_filter <- test_load(filter(df, y == 0))
tbls_full %>%
purrr::map2(tbls_filter, union) %>%
expect_equal_tbls()
tbls_full %>%
purrr::map2(tbls_filter, union_all) %>%
expect_equal_tbls()
})
test_that("intersect and setdiff work for supported backends", {
df <- tibble(x = 1:10, y = x %% 2)
# MySQL doesn't support EXCEPT or INTERSECT
tbls_full <- test_load(df, ignore = c("mysql", "MariaDB"))
tbls_filter <- test_load(filter(df, y == 0), ignore = c("mysql", "MariaDB"))
tbls_full %>%
purrr::map2(tbls_filter, intersect) %>%
expect_equal_tbls()
tbls_full %>%
purrr::map2(tbls_filter, setdiff) %>%
expect_equal_tbls()
})
test_that("SQLite warns if set op attempted when tbl has LIMIT", {
mf <- memdb_frame(x = 1:2)
m1 <- head(mf, 1)
expect_error(union(mf, m1), "does not support")
expect_error(union(m1, mf), "does not support")
})
test_that("other backends can combine with a limit", {
df <- tibble(x = 1:2)
# sqlite only allows limit at top level
tbls_full <- test_load(df, ignore = "sqlite")
tbls_head <- lapply(test_load(df, ignore = "sqlite"), head, n = 1)
tbls_full %>%
purrr::map2(tbls_head, union) %>%
expect_equal_tbls()
tbls_full %>%
purrr::map2(tbls_head, union_all) %>%
expect_equal_tbls()
})
|
