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
|
context("Test ggplot")
skip_if_not_installed("kernlab")
test_that("ggplot.train correctly orders factors", {
library(caret)
library(kernlab)
data(mtcars)
m <- train(
mpg ~ cyl + disp,
data = mtcars,
method = "svmRadial",
tuneGrid = expand.grid(C = 1:2, sigma = c(0.0001, 0.01, 1))
)
g <- ggplot(m, plotType = "level")
# Test plot data
obj_sigma <- as.numeric(levels(g$data$sigma))
obj_C <- as.numeric(levels(g$data$c))
expect_equal(obj_sigma, sort(obj_sigma))
expect_equal(obj_C, sort(obj_C))
# Test axes' labels on a built plot
build <- ggplot2::ggplot_build(g)
obj_x <- as.numeric(build$layout$panel_ranges[[1]]$x.labels)
obj_y <- as.numeric(build$layout$panel_ranges[[1]]$y.labels)
expect_equal(obj_x, sort(obj_x))
expect_equal(obj_y, sort(obj_y))
})
test_that("ggplot.train correctly orders facets' labels", {
library(caret)
library(kernlab)
data(mtcars)
m <- suppressWarnings(train(
mpg ~ cyl + disp,
data = mtcars,
method = "svmPoly",
tuneGrid = expand.grid(
degree = c(0.0001, 0.01, 1),
scale = c(0.0001, 0.01, 1),
C = c(0.0001, 0.01, 1)
)
))
g <- ggplot(m, plotType = "level", nameInStrip = TRUE)
# Test plot data
obj_C <- as.numeric(gsub(
'Cost: ',
'',
levels(g$data$C)
))
expect_equal(obj_C, sort(obj_C))
# Test axes' labels on a built plot
build <- ggplot2::ggplot_build(g)
obj_labels <- as.numeric(gsub(
'Cost: ',
'',
levels(build$layout$panel_layout$C)
))
expect_equal(obj_labels, sort(obj_labels))
})
|
