add test

Author: microly

tests/testthat/test-geom-sf.R

@@ -1,5 +1,46 @@
 context("geom-sf")
 
+test_that("geom_sf() determines the legend type automatically", {
+  skip_if_not_installed("sf")
+  if (packageVersion("sf") < "0.5.3") skip("Need sf 0.5.3")
+
+  mp <- sf::st_sf(
+    geometry = sf::st_sfc(sf::st_multipoint(rbind(c(1,1), c(2,2), c(3,3)))),
+    v = "a")
+
+  s1 <- rbind(c(0,3),c(0,4),c(1,5),c(2,5))
+  s2 <- rbind(c(0.2,3), c(0.2,4), c(1,4.8), c(2,4.8))
+  s3 <- rbind(c(0,4.4), c(0.6,5))
+
+  mls <- sf::st_sf(
+    geometry = sf::st_sfc(sf::st_multilinestring(list(s1,s2,s3))),
+    v = "a")
+
+  p1 <- rbind(c(0,0), c(1,0), c(3,2), c(2,4), c(1,4), c(0,0))
+  p2 <- rbind(c(1,1), c(1,2), c(2,2), c(1,1))
+  p3 <- rbind(c(3,0), c(4,0), c(4,1), c(3,1), c(3,0))
+  p4 <- rbind(c(3.3,0.3), c(3.8,0.3), c(3.8,0.8), c(3.3,0.8), c(3.3,0.3))[5:1,]
+  p5 <- rbind(c(3,3), c(4,2), c(4,3), c(3,3))
+
+  mpol <- sf::st_sf(
+    geometry = sf::st_sfc(sf::st_multipolygon(list(list(p1,p2), list(p3,p4), list(p5)))),
+    v = "a")
+
+  fun_geom_sf <- function(sf) {
+    p <- ggplot() + geom_sf(aes(colour = v), data = sf, show.legend = TRUE)
+    ggplot_build(p)
+  }
+
+  expect_identical(fun_geom_sf(mp)$plot$layers[[1]]$show.legend, TRUE)
+  expect_identical(fun_geom_sf(mp)$plot$layers[[1]]$geom_params$legend, "point")
+
+  expect_identical(fun_geom_sf(mls)$plot$layers[[1]]$show.legend, TRUE)
+  expect_identical(fun_geom_sf(mls)$plot$layers[[1]]$geom_params$legend, "line")
+
+  expect_identical(fun_geom_sf(mpol)$plot$layers[[1]]$show.legend, TRUE)
+  expect_identical(fun_geom_sf(mpol)$plot$layers[[1]]$geom_params$legend, "polygon")
+})
+
 test_that("geom_sf() removes rows containing missing aes", {
   skip_if_not_installed("sf")
   if (packageVersion("sf") < "0.5.3") skip("Need sf 0.5.3")