Merge branch 'baobao-geom_jitter' of https://github.com/ropensci/plotly into baobao-geom_jitter

Author: cpsievert

R/trace_generation.R

@@ -401,6 +401,14 @@ toBasic <- list(
     g$params$yend <- max(g$prestats.data$globymax)
     g
   },
+  jitter=function(g) {
+    if ("size" %in% names(g$data)) {
+      g$params$sizemin <- min(g$prestats.data$globsizemin)
+      g$params$sizemax <- max(g$prestats.data$globsizemax)
+    }
+    g$geom <- "point"
+    g
+  },
   point=function(g) {
     if ("size" %in% names(g$data)) {
       g$params$sizemin <- min(g$prestats.data$globsizemin)

tests/testthat/test-cookbook-scatterplots.R

@@ -71,3 +71,11 @@ g <- ggplot(dat, aes(x=xrnd, y=yrnd)) +
   geom_point(shape=1,      # Use hollow circles
              position=position_jitter(width=1,height=.5))
 save_outputs(g, "scatterplots-jitter")
+
+# Jitter the points using geom_jitter
+# Jitter range is 1 on the x-axis, .5 on the y-axis
+g <- ggplot(dat, aes(x = xrnd, y = yrnd)) +
+  geom_jitter(shape = 1,      # Use hollow circles
+             width = 1, height = 0.5)
+save_outputs(g, "scatterplots-geom_jitter")
+

tests/testthat/test-ggplot-jitter.R

@@ -0,0 +1,28 @@
+context("geom_jitter")
+
+# Expect trace function
+expect_traces <- function(gg, n_traces, name) {
+  stopifnot(is.ggplot(gg))
+  stopifnot(is.numeric(n_traces))
+  save_outputs(gg, paste0("jitter-", name))
+  L <- gg2list(gg)
+  all_traces <- L$data
+  no_data <- sapply(all_traces, function(tr) {
+    is.null(tr[["x"]]) && is.null(tr[["y"]])
+  })
+  has_data <- all_traces[!no_data]
+  expect_equal(length(has_data), n_traces)
+  list(traces = has_data, layout = L$layout)
+}
+
+set.seed(1001)
+p <- ggplot(mpg, aes(cyl, hwy)) + geom_jitter()
+
+test_that("geom_jitter is working", {
+  info <- expect_traces(p, 1, "basic")
+  tr <- info$traces[[1]]
+  expect_identical(tr$type, "scatter")
+  # default jitter is 40% of the resolution of the data.
+  diffs <- abs(mpg$cyl - tr$x)
+  expect_true(all(0 < diffs & diffs < 0.4))
+})