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
139
140
141
|
test_that("names of values used in manual scales", {
s1 <- scale_colour_manual(values = c("8" = "c", "4" = "a", "6" = "b"))
s1$train(c("4", "6", "8"))
expect_equal(s1$map(c("4", "6", "8")), c("a", "b", "c"))
s2 <- scale_colour_manual(values = c("8" = "c", "4" = "a", "6" = "b"), na.value = NA)
s2$train(c("4", "8"))
expect_equal(s2$map(c("4", "6", "8")), c("a", NA, "c"))
})
dat <- data_frame(g = c("B","A","A"))
p <- ggplot(dat, aes(g, fill = g)) + geom_bar()
col <- c("A" = "red", "B" = "green", "C" = "blue")
cols <- function(x) ggplot_build(x)$data[[1]][, "fill"]
test_that("named values work regardless of order", {
fill_scale <- function(order) scale_fill_manual(values = col[order],
na.value = "black")
# Order of value vector shouldn't matter
expect_equal(cols(p + fill_scale(1:3)), c("red", "green"))
expect_equal(cols(p + fill_scale(1:2)), c("red", "green"))
expect_equal(cols(p + fill_scale(2:1)), c("red", "green"))
expect_equal(cols(p + fill_scale(c(3, 2, 1))), c("red", "green"))
expect_equal(cols(p + fill_scale(c(3, 1, 2))), c("red", "green"))
expect_equal(cols(p + fill_scale(c(1, 3, 2))), c("red", "green"))
})
test_that("missing values are replaced with na.value", {
df <- data_frame(x = 1, y = 1:3, z = factor(c(1:2, NA), exclude = NULL))
p <- ggplot(df, aes(x, y, colour = z)) +
geom_point() +
scale_colour_manual(values = c("black", "black"), na.value = "red")
expect_equal(layer_data(p)$colour, c("black", "black", "red"))
})
test_that("insufficient values raise an error", {
df <- data_frame(x = 1, y = 1:3, z = factor(c(1:2, NA), exclude = NULL))
p <- ggplot(df, aes(x, y, colour = z)) + geom_point()
expect_error(ggplot_build(p + scale_colour_manual(values = "black")),
"Insufficient values")
# Should be sufficient
ggplot_build(p + scale_colour_manual(values = c("black", "black")))
})
test_that("values are matched when scale contains more unique values than are in the data", {
s <- scale_colour_manual(values = c("8" = "c", "4" = "a",
"22" = "d", "6" = "b"))
s$train(c("4", "6", "8"))
expect_equal(s$map(c("4", "6", "8")), c("a", "b", "c"))
})
test_that("generic scale can be used in place of aesthetic-specific scales", {
df <- data_frame(x = letters[1:3], y = LETTERS[1:3], z = factor(c(1, 2, 3)))
p1 <- ggplot(df, aes(z, z, shape = x, color = y, alpha = z)) +
scale_shape_manual(values = 1:3) +
scale_colour_manual(values = c("red", "green", "blue")) +
scale_alpha_manual(values = c(0.2, 0.4, 0.6))
p2 <- ggplot(df, aes(z, z, shape = x, color = y, alpha = z)) +
scale_discrete_manual(aesthetics = "shape", values = 1:3) +
scale_discrete_manual(aesthetics = "colour", values = c("red", "green", "blue")) +
scale_discrete_manual(aesthetics = "alpha", values = c(0.2, 0.4, 0.6))
expect_equal(layer_data(p1), layer_data(p2))
})
test_that("named values do not match with breaks in manual scales", {
s <- scale_fill_manual(
values = c("data_red" = "red", "data_black" = "black"),
breaks = c("data_black", "data_red")
)
s$train(c("data_black", "data_red"))
expect_equal(s$map(c("data_red", "data_black")), c("red", "black"))
})
test_that("unnamed values match breaks in manual scales", {
s <- scale_fill_manual(
values = c("red", "black"),
breaks = c("data_red", "data_black")
)
s$train(c("data_red", "data_black"))
expect_equal(s$map(c("data_red", "data_black")), c("red", "black"))
})
test_that("limits works (#3262)", {
# named character vector
s1 <- scale_colour_manual(values = c("8" = "c", "4" = "a", "6" = "b"), limits = c("4", "8"), na.value = NA)
s1$train(c("4", "6", "8"))
expect_equal(s1$map(c("4", "6", "8")), c("a", NA, "c"))
# unnamed character vector
s2 <- scale_colour_manual(values = c("c", "a", "b"), limits = c("4", "8"), na.value = NA)
s2$train(c("4", "6", "8"))
expect_equal(s2$map(c("4", "6", "8")), c("c", NA, "a"))
})
test_that("fewer values (#3451)", {
# named character vector
s1 <- scale_colour_manual(values = c("4" = "a", "8" = "c"), na.value = NA)
s1$train(c("4", "6", "8"))
expect_equal(s1$map(c("4", "6", "8")), c("a", NA, "c"))
# unnamed character vector
s2 <- scale_colour_manual(values = c("4", "8"), na.value = NA)
s2$train(c("4", "6", "8"))
expect_error(s2$map(c("4", "6", "8")), "Insufficient values")
})
test_that("limits and breaks (#4619)", {
# values don't change legend order
s1 <- scale_colour_manual(
values = c("8" = "c", "4" = "a", "6" = "b")
)
s1$train(c("8", "6", "4"))
expect_equal(s1$map(c("8", "6", "4")), c("c", "b", "a"))
expect_equal(s1$break_positions(), c("a", "b", "c"))
# limits change legend order
s2 <- scale_colour_manual(
values = c("8" = "c", "4" = "a", "6" = "b", "0" = "x"),
limits = c("0", "4", "6", "8")
)
s2$train(c("8", "6", "4"))
expect_equal(s2$map(c("4", "6", "8")), c("a", "b", "c"))
expect_equal(s2$break_positions(), c("x", "a", "b", "c"))
# breaks work
s3 <- scale_colour_manual(
values = c("8" = "c", "4" = "a", "6" = "b"),
breaks = c("4", "8")
)
s3$train(c("4", "6", "8"))
expect_equal(s3$map(c("4", "6", "8")), c("a", "b", "c"))
expect_equal(s3$break_positions(), c("a", "c"))
})
|
