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
128
129
130
131
132
133
134
135
136
137
138
|
df <- head(mtcars)
df$character <- c("a", "b", "b", "c", "c", "a")
test_that("data_arrange works with one numeric column", {
skip_if_not_installed("poorman")
expect_identical(
poorman::arrange(df, carb),
data_arrange(df, "carb")
)
expect_identical(
poorman::arrange(df, -carb),
data_arrange(df, "-carb")
)
})
test_that("data_arrange works with one character column", {
skip_if_not_installed("poorman")
expect_identical(
poorman::arrange(df, character),
data_arrange(df, "character")
)
expect_identical(
poorman::arrange(df, desc(character)),
data_arrange(df, "-character")
)
})
test_that("data_arrange works with several columns", {
skip_if_not_installed("poorman")
expect_identical(
poorman::arrange(df, carb, gear),
data_arrange(df, c("carb", "gear"))
)
expect_identical(
poorman::arrange(df, -carb, gear),
data_arrange(df, c("-carb", "gear"))
)
expect_identical(
poorman::arrange(df, -carb, desc(character)),
data_arrange(df, c("-carb", "-character"))
)
})
test_that("data_arrange works without columns", {
expect_identical(data_arrange(df), df)
})
test_that("data_arrange ignores wrong names if safe = TRUE", {
expect_warning(
expect_identical(data_arrange(df, "foo"), df),
regexp = "don't exist"
)
expect_warning(
expect_identical(
data_arrange(df, c("gear", "foo")),
data_arrange(df, "gear")
),
regexp = "don't exist"
)
})
test_that("data_arrange errors if safe = FALSE", {
expect_error(data_arrange(df, "foo", safe = FALSE))
})
test_that("data_arrange errors if not coercable to data frame", {
expect_error(data_arrange(list(a = 1:5, b = letters[1:3]), select = "b"))
expect_equal(
data_arrange(list(a = 1:5, b = letters[5:1]), select = "b"),
structure(list(a = 5:1, b = c("a", "b", "c", "d", "e")), row.names = 5:1, class = "data.frame"),
ignore_attr = TRUE
)
})
test_that("data_arrange works with grouped df", {
set.seed(123)
x <- mtcars[sample(seq_len(nrow(mtcars)), 10, replace = TRUE), c("cyl", "mpg")]
g <- data_group(x, cyl)
expected <- data.frame(
cyl = c(4, 4, 4, 6, 6, 8, 8, 8, 8, 8),
mpg = c(22.8, 30.4, 32.4, 17.8, 19.2, 10.4, 15, 15.2, 15.5, 18.7)
)
class(expected) <- c("grouped_df", "data.frame")
rownames(expected) <- c(
"Datsun 710", "Honda Civic", "Fiat 128", "Merc 280C", "Merc 280",
"Cadillac Fleetwood", "Maserati Bora", "Merc 450SLC", "Dodge Challenger",
"Hornet Sportabout"
)
attributes(expected)$groups <- attributes(g)$groups
expect_identical(
data_arrange(g, "mpg"),
expected,
ignore_attr = TRUE
)
})
test_that("data_arrange works with NA", {
# without groups
tmp <- data.frame(
a = c(1, 2, 2, 8, 1, 3),
b = c(1, NA, 3, 3, NA, 5)
)
expect_identical(
data_arrange(tmp, "a"),
data.frame(
a = c(1, 1, 2, 2, 3, 8),
b = c(1, NA, NA, 3, 5, 3)
)
)
# with groups
g <- data_group(tmp, "b")
expected <- data.frame(
a = c(1, 2, 8, 3, 1, 2),
b = c(1, 3, 3, 5, NA, NA)
)
class(expected) <- c("grouped_df", "data.frame")
attributes(expected)$groups <- attributes(g)$groups
expect_identical(
data_arrange(g, "a"),
expected,
ignore_attr = TRUE
)
})
test_that("data_arrange works one-column data frames (and does not drop dimensions)", {
data(mtcars)
expect_s3_class(data_arrange(mtcars["gear"], select = "gear"), "data.frame")
expect_s3_class(data_arrange(mtcars[c("gear", "cyl")], select = "gear"), "data.frame")
})
|
