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
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
|
test_that("errors for bad inputs", {
expect_error(vars_pull(letters, character()), "exactly one")
expect_error(vars_pull(letters, c("a", "b")), "exactly one")
expect_error(vars_pull(letters, !!c("a", "b")), "exactly one")
# FIXME
expect_error(
vars_pull(letters, aa),
"object 'aa' not found",
fixed = TRUE
)
expect_error(
vars_pull(letters, "foo"),
class = "vctrs_error_subscript_oob"
)
expect_error(
vars_pull(letters, 0),
class = "vctrs_error_subscript_type"
)
expect_error(
vars_pull(letters, 100),
class = "vctrs_error_subscript_oob"
)
expect_error(
vars_pull(letters, -100),
class = "vctrs_error_subscript_oob"
)
expect_error(
vars_pull(letters, -Inf),
class = "vctrs_error_subscript_type"
)
expect_error(
vars_pull(letters, TRUE),
class = "vctrs_error_subscript_type"
)
expect_error(
vars_pull(letters, NA),
class = "vctrs_error_subscript_type"
)
expect_error(
vars_pull(letters, na_int),
class = "vctrs_error_subscript_type"
)
expect_snapshot(error = TRUE, {
vars_pull(letters, character())
vars_pull(letters, c("a", "b"))
vars_pull(letters, !!c("a", "b"))
vars_pull(letters, aa)
vars_pull(letters, 0)
vars_pull(letters, 100)
vars_pull(letters, -100)
vars_pull(letters, -Inf)
vars_pull(letters, TRUE)
vars_pull(letters, NA)
vars_pull(letters, na_int)
vars_pull(letters, "foo")
})
})
test_that("gives informative error if quosure is missing", {
f <- function(arg) vars_pull(letters, {{ arg }})
expect_snapshot(f(), error = TRUE)
})
test_that("can pull variables with missing elements", {
expect_identical(vars_pull(c("a", ""), a), "a")
expect_identical(vars_pull(c("a", NA), a), "a")
})
test_that("missing values are detected in vars_pull() (#72)", {
lapply(list(NA_character_, NA_integer_, NA_real_, NA, NA_complex_), function(x) {
expect_error(vars_pull(c("a", "b"), !!x), class = "vctrs_error_subscript_type")
})
})
test_that("can pull with strings", {
expect_identical(vars_pull(letters, "b"), vars_pull(letters, b))
expect_error(vars_pull(letters, "foo"), class = "vctrs_error_subscript_oob")
# even if those strings are numbers (#200)
vars <- c("-1", "0", "1")
expect_equal(vars_pull(vars, "-1"), "-1")
expect_equal(vars_pull(vars, "0"), "0")
expect_equal(vars_pull(vars, "1"), "1")
})
test_that("can pull with all_of() without warning", {
expect_identical(vars_pull(letters, all_of("z")), "z")
})
test_that("can pull with negative values", {
expect_identical(vars_pull(letters, -1), "z")
expect_identical(vars_pull(letters, -3), "x")
})
test_that("vars_pull() has informative errors", {
expect_snapshot({
"# vars_pull() instruments base errors"
(expect_error(vars_pull(letters, foobar), ""))
})
})
test_that("vars_pull() errors mention correct calls", {
f <- function() stop("foo")
expect_snapshot((expect_error(vars_pull(letters, f()))))
})
test_that("vars_pull() produces correct backtraces", {
f <- function(base) g(base)
g <- function(base) h(base)
h <- function(base) if (base) stop("foo") else abort("foo")
local_options(
rlang_trace_trop_env = current_env(),
rlang_trace_format_srcrefs = FALSE
)
expect_snapshot({
print(expect_error(vars_pull(letters, f(base = TRUE))))
print(expect_error(vars_pull(letters, f(base = FALSE))))
})
})
|
