-
Notifications
You must be signed in to change notification settings - Fork 2.1k
/
Copy pathtest-stat-density2d.R
104 lines (90 loc) · 4.16 KB
/
test-stat-density2d.R
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
skip_if_not_installed("MASS")
test_that("uses scale limits, not data limits", {
base <- ggplot(mtcars, aes(wt, mpg)) +
stat_density_2d() +
scale_x_continuous(limits = c(1, 6)) +
scale_y_continuous(limits = c(5, 40))
ret <- get_layer_data(base)
# Check that the contour data goes beyond data range.
# The specific values below are sort of arbitrary; but they go beyond the range
# of the data
expect_true(min(ret$x) < 1.2)
expect_true(max(ret$x) > 5.8)
expect_true(min(ret$y) < 8)
expect_true(max(ret$y) > 35)
})
test_that("stat_density2d can produce contour and raster data", {
p <- ggplot(faithful, aes(x = eruptions, y = waiting))
p_contour_lines <- p + stat_density_2d()
p_contour_bands <- p + stat_density_2d_filled()
p_raster <- p + stat_density_2d(contour = FALSE)
d_lines <- get_layer_data(p_contour_lines)
expect_true("level" %in% names(d_lines))
expect_false("level_low" %in% names(d_lines))
expect_true(is.numeric(d_lines$level))
d_bands <- get_layer_data(p_contour_bands)
expect_true("level" %in% names(d_bands))
expect_true("level_low" %in% names(d_bands))
expect_true(is.ordered(d_bands$level))
d_raster <- get_layer_data(p_raster)
expect_true("density" %in% names(d_raster))
expect_true("ndensity" %in% names(d_raster))
expect_true("count" %in% names(d_raster))
expect_true(unique(d_raster$level) == 1)
expect_true(unique(d_raster$piece) == 1)
# stat_density_2d() and stat_density_2d_filled() produce identical
# density output with `contour = FALSE`
# (`fill` and `colour` will differ due to different default aesthetic mappings)
d_raster2 <- get_layer_data(p + stat_density_2d_filled(contour = FALSE))
expect_identical(d_raster$x, d_raster2$x)
expect_identical(d_raster$y, d_raster2$y)
expect_identical(d_raster$density, d_raster2$density)
expect_identical(d_raster$ndensity, d_raster2$ndensity)
expect_identical(d_raster$count, d_raster2$count)
# stat_density_2d() with contouring is the same as stat_contour() on calculated density
p_lines2 <- ggplot(d_raster, aes(x, y, z = density)) + stat_contour()
d_lines2 <- get_layer_data(p_lines2)
expect_identical(d_lines$x, d_lines2$x)
expect_identical(d_lines$y, d_lines2$y)
expect_identical(d_lines$piece, d_lines2$piece)
expect_identical(d_lines$group, d_lines2$group)
expect_identical(d_lines$level, d_lines2$level)
# same for stat_density_2d_filled()
p_bands2 <- ggplot(d_raster, aes(x, y, z = density)) + stat_contour_filled()
d_bands2 <- get_layer_data(p_bands2)
expect_identical(d_bands$x, d_bands2$x)
expect_identical(d_bands$y, d_bands2$y)
expect_identical(d_bands$piece, d_bands2$piece)
expect_identical(d_bands$group, d_bands2$group)
expect_identical(d_bands$level, d_bands2$level)
expect_identical(d_bands$level_mid, d_bands2$level_mid)
# and for contour_var = "ndensity"
p_contour_lines <- p + stat_density_2d(contour_var = "ndensity")
d_lines <- get_layer_data(p_contour_lines)
p_lines2 <- ggplot(d_raster, aes(x, y, z = ndensity)) + stat_contour()
d_lines2 <- get_layer_data(p_lines2)
expect_identical(d_lines$x, d_lines2$x)
expect_identical(d_lines$y, d_lines2$y)
expect_identical(d_lines$piece, d_lines2$piece)
expect_identical(d_lines$group, d_lines2$group)
expect_identical(d_lines$level, d_lines2$level)
# and for contour_var = "count"
p_contour_bands <- p + stat_density_2d_filled(contour_var = "count")
d_bands <- get_layer_data(p_contour_bands)
p_bands2 <- ggplot(d_raster, aes(x, y, z = count)) + stat_contour_filled()
d_bands2 <- get_layer_data(p_bands2)
expect_identical(d_bands$x, d_bands2$x)
expect_identical(d_bands$y, d_bands2$y)
expect_identical(d_bands$piece, d_bands2$piece)
expect_identical(d_bands$group, d_bands2$group)
expect_identical(d_bands$level, d_bands2$level)
expect_identical(d_bands$level_mid, d_bands2$level_mid)
# error on incorrect contouring variable
expect_snapshot_error(ggplot_build(p + stat_density_2d(contour_var = "abcd")))
})
test_that("stat_density_2d handles faulty bandwidth", {
p <- ggplot(faithful, aes(eruptions, waiting)) +
stat_density_2d(h = c(0, NA))
expect_snapshot_warning(b <- ggplot_build(p))
expect_s3_class(layer_grob(b)[[1]], "null")
})