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
96
97
98
99
100
101
102
103
104
105
|
get_pkg_path <- function(ctx) {
pkg_name <- package_name(ctx)
expect_type(pkg_name, "character")
pkg_path <- find.package(pkg_name)
pkg_path
}
utils::globalVariables("con")
utils::globalVariables("con2")
local_connection <- function(ctx, ..., .local_envir = parent.frame()) {
con <- connect(ctx, ...)
withr::local_db_connection(con, .local_envir = .local_envir)
}
local_closed_connection <- function(ctx, ...) {
con <- connect(ctx, ...)
dbDisconnect(con)
con
}
local_invalid_connection <- function(ctx, ...) {
con <- connect(ctx, ...)
dbDisconnect(con)
unserialize(serialize(con, NULL))
}
# Calls `dbClearResult()` on `query` after exiting `frame`.
local_result <- function(query, frame = caller_env()) {
res <- query
withr::defer(
{
dbClearResult(res)
},
envir = frame
)
res
}
# Calls `try_silent(dbRemoveTable())` after exiting `frame`.
local_remove_test_table <- function(con, name, frame = caller_env()) {
table_name <- dbQuoteIdentifier(con, name)
withr::defer(
try_silent(
dbRemoveTable(con, table_name)
),
envir = frame
)
}
get_penguins <- function(ctx) {
datasets_penguins <- unrowname(palmerpenguins::penguins[c(1, 153, 277), ])
# FIXME: better handling of DBI backends that do support factors
datasets_penguins$species <- as.character(datasets_penguins$species)
datasets_penguins$island <- as.character(datasets_penguins$island)
datasets_penguins$sex <- as.character(datasets_penguins$sex)
as.data.frame(datasets_penguins)
}
unrowname <- function(x) {
rownames(x) <- NULL
x
}
random_table_name <- function(n = 10) {
# FIXME: Use parallel-safe sequence of numbers
paste0("dbit", paste(sample(letters, n, replace = TRUE), collapse = ""))
}
try_silent <- function(code) {
tryCatch(
code,
error = function(e) NULL
)
}
check_df <- function(df) {
expect_s3_class(df, "data.frame")
if (length(df) >= 1L) {
lengths <- unname(lengths(df))
expect_equal(diff(lengths), rep(0L, length(lengths) - 1L))
expect_equal(nrow(df), lengths[[1]])
}
df_names <- names(df)
expect_true(all(df_names != ""))
expect_false(anyNA(df_names))
df
}
check_arrow <- function(stream, transform = identity) {
to <- function(schema, ptype) transform(ptype)
if (inherits(stream, "nanoarrow_array_stream")) {
on.exit(stream$release())
df <- nanoarrow::convert_array_stream(stream, to)
} else if (inherits(stream, "nanoarrow_array")) {
df <- nanoarrow::convert_array(stream, to)
} else {
stop("Unexpected conversion of type ", class(stream), ".", call. = FALSE)
}
check_df(df)
}
|
