Fix typo and add test for multiple hlines

Author: mkcor

tests/testthat/test-ggplot-hline.R

@@ -1,19 +1,18 @@
 context("Hline")
-
 # Horizontal line
 
-test_that("Second trace be the hline", {
-  
-  x1 <- seq(from=0, to=3.5, by=0.5)
-  x2 <- x1 * 0.95
-  df <- data.frame("x1"=x1, "x2"=x2)
-  
-  gg <- ggplot(df) + geom_point(aes(x=x1, y=x2)) +
-    geom_hline(yintercept=1.1, colour="green", size=3)
+x1 <- seq(from=0, to=3.5, by=0.5)
+x2 <- x1 * 0.95
+df <- data.frame("x1"=x1, "x2"=x2)
+gg <- ggplot(df) + geom_point(aes(x=x1, y=x2))
+
+test_that("second trace be the hline", {
+  gg <- gg + geom_hline(yintercept=1.1, colour="green", size=3)
 
   L <- gg2list(gg)
 
   expect_equal(length(L), 3)
+  expect_equal(L[[2]]$y[1], 1.1)
   expect_true(L[[2]]$x[1] <= 0)
   expect_true(L[[2]]$x[2] >= 3.5)
   expect_identical(L[[2]]$mode, "lines")
@@ -22,4 +21,23 @@ test_that("Second trace be the hline", {
   expect_identical(L[[2]]$line$color, "rgb(0,255,0)")
 
   save_outputs(gg, "hline")
-})
+})
+
+test_that("vector yintercept results in multiple horizontal lines", {
+  gg <- gg + geom_hline(yintercept=1:3, colour="red", size=3)
+  
+  L <- gg2list(gg)
+  
+  expect_equal(length(L), 5)
+  expect_equal(L[[2]]y[1], 1)
+  expect_equal(L[[3]]y[1], 2)
+  expect_equal(L[[4]]y[1], 3)
+  expect_true(L[[4]]$x[1] <= 0)
+  expect_true(L[[4]]$x[2] >= 3.325)
+  expect_identical(L[[3]]$mode, "lines")
+  expect_identical(L[[3]]$line$shape, "linear")
+  expect_equal(L[[3]]$line$width, 3)
+  expect_identical(L[[3]]$line$color, "rgb(255,0,0)")
+  
+  save_outputs(gg, "hline-multiple")
+})

tests/testthat/test-ggplot-vline.R

@@ -12,7 +12,7 @@ test_that("second trace be the vline", {
   L <- gg2list(gg)
 
   expect_equal(length(L), 3)
-  expect_equal(L[[2]]x[1], 1.1)
+  expect_equal(L[[2]]$x[1], 1.1)
   expect_true(L[[2]]$y[1] <= 0)
   expect_true(L[[2]]$y[2] >= 3.325)
   expect_identical(L[[2]]$mode, "lines")
@@ -23,7 +23,7 @@ test_that("second trace be the vline", {
   save_outputs(gg, "vline")
 })
 
-test_that("vector yintercept results in multiple vertical lines", {
+test_that("vector xintercept results in multiple vertical lines", {
   gg <- gg + geom_vline(xintercept=1:2, colour="blue", size=3)
 
   L <- gg2list(gg)