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
|
test_that("tbl_sql() works with string argument", {
name <- unclass(unique_table_name())
df <- memdb_frame(a = 1, .name = name)
expect_equal(collect(tbl_sql("sqlite", df$src, name)), collect(df))
})
test_that("same_src distinguishes srcs", {
con1 <- DBI::dbConnect(RSQLite::SQLite(), ":memory:", create = TRUE)
con2 <- DBI::dbConnect(RSQLite::SQLite(), ":memory:", create = TRUE)
on.exit({dbDisconnect(con1); dbDisconnect(con2)}, add = TRUE)
db1 <- copy_to(con1, iris[1:3, ], 'data1', temporary = FALSE)
db2 <- copy_to(con2, iris[1:3, ], 'data2', temporary = FALSE)
expect_true(same_src(db1, db1))
expect_false(same_src(db1, db2))
expect_false(same_src(db1, mtcars))
})
# tbl ---------------------------------------------------------------------
test_that("can generate sql tbls with raw sql", {
mf1 <- memdb_frame(x = 1:3, y = 3:1)
mf2 <- tbl(mf1$src, build_sql("SELECT * FROM ", mf1$lazy_query$x, con = simulate_dbi()))
expect_equal(collect(mf1), collect(mf2))
})
test_that("sql tbl can be printed", {
mf1 <- memdb_frame(x = 1:3, y = 3:1)
mf2 <- tbl(mf1$src, build_sql("SELECT * FROM ", mf1$lazy_query$x, con = simulate_dbi()))
expect_snapshot(mf2, transform = function(x) {
gsub("sqlite .* \\[:memory:\\]", "sqlite ?.?.? [:memory:]", x)
})
})
test_that("can refer to default schema explicitly", {
con <- sqlite_con_with_aux()
on.exit(DBI::dbDisconnect(con))
DBI::dbExecute(con, "CREATE TABLE t1 (x)")
expect_equal(as.character(tbl_vars(tbl(con, "t1"))), "x")
expect_equal(as.character(tbl_vars(tbl(con, in_schema("main", "t1")))), "x")
})
test_that("can distinguish 'schema.table' from 'schema'.'table'", {
con <- sqlite_con_with_aux()
on.exit(DBI::dbDisconnect(con))
DBI::dbExecute(con, "CREATE TABLE aux.t1 (x, y, z)")
DBI::dbExecute(con, "CREATE TABLE 'aux.t1' (a, b, c)")
expect_equal(as.character(tbl_vars(tbl(con, in_schema("aux", "t1")))), c("x", "y", "z"))
expect_equal(as.character(tbl_vars(tbl(con, ident("aux.t1")))), c("a", "b", "c"))
})
# n_groups ----------------------------------------------------------------
test_that("check basic group size implementation", {
db <- memdb_frame(x = rep(1:3, each = 10), y = rep(1:6, each = 5))
expect_equal(n_groups(db), 1L)
expect_equal(group_size(db), 30)
gb <- group_by(db, x)
expect_equal(n_groups(gb), 3L)
expect_equal(group_size(gb), rep(10, 3))
})
# tbl_sum -------------------------------------------------------------------
test_that("ungrouped output", {
mf <- copy_to_test("sqlite", tibble(x = 1:5, y = 1:5), name = "tbl_sum_test")
out1 <- tbl_sum(mf)
expect_named(out1, c("Source", "Database"))
expect_equal(out1[["Source"]], "table<tbl_sum_test> [?? x 2]")
expect_match(out1[["Database"]], "sqlite (.*) \\[:memory:\\]")
out2 <- tbl_sum(mf %>% group_by(x, y))
expect_named(out2, c("Source", "Database", "Groups"))
expect_equal(out2[["Groups"]], c("x, y"))
out3 <- tbl_sum(mf %>% arrange(x))
expect_named(out3, c("Source", "Database", "Ordered by"))
expect_equal(out3[["Ordered by"]], c("x"))
})
|
